make tr energy compat last and use EnergyIo interface directly

badasintended · May 27, 2021 · 04ed08f · 04ed08f
1 parent 91bf615
commit 04ed08f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/src/fast-transfer-lib/java/badasintended/megane/impl/FastTransferLib.java b/src/fast-transfer-lib/java/badasintended/megane/impl/FastTransferLib.java
@@ -4,16 +4,14 @@
 import badasintended.megane.api.provider.EnergyProvider;
 import badasintended.megane.api.registry.MeganeRegistrar;
 import dev.technici4n.fasttransferlib.api.energy.EnergyIo;
-import net.minecraft.block.entity.BlockEntity;
 
 public class FastTransferLib implements MeganeModule {
 
     @Override
     public void register(MeganeRegistrar registrar) {
-        registrar.energy(1100, BlockEntity.class, EnergyProvider.of(
-            t -> t instanceof EnergyIo,
-            t -> ((EnergyIo) t).getEnergy(),
-            t -> ((EnergyIo) t).getEnergyCapacity()
+        registrar.energy(1100, EnergyIo.class, EnergyProvider.of(
+            EnergyIo::getEnergy,
+            EnergyIo::getEnergyCapacity
         ));
     }
 

diff --git a/src/team-reborn-energy/java/badasintended/megane/impl/TeamRebornEnergy.java b/src/team-reborn-energy/java/badasintended/megane/impl/TeamRebornEnergy.java
@@ -11,7 +11,7 @@ public class TeamRebornEnergy implements MeganeModule {
 
     @Override
     public void register(MeganeRegistrar registrar) {
-        registrar.energy(1100, BlockEntity.class, new EnergyProvider<BlockEntity>() {
+        registrar.energy(1200, BlockEntity.class, new EnergyProvider<BlockEntity>() {
             EnergyHandler handler;
 
             @Override