Add support for WAILA (specifically with WTHIT).

2 files changed, 16 insertions, 6 deletions

diff --git a/src/main/java/lv/enes/mc/eris_alchemy/block/entity/EmcStorageEntity.java b/src/main/java/lv/enes/mc/eris_alchemy/block/entity/EmcStorageEntity.java
new file mode 100644
index 0000000..465e743
--- /dev/null
+++ b/src/main/java/lv/enes/mc/eris_alchemy/block/entity/EmcStorageEntity.java
@@ -0,0 +1,6 @@
+package lv.enes.mc.eris_alchemy.block.entity;
+
+public interface EmcStorageEntity {
+        double getMaxEmc();
+        double getStoredEmc();
+}
diff --git a/src/main/java/lv/enes/mc/eris_alchemy/block/entity/EnergyCondenserEntity.java b/src/main/java/lv/enes/mc/eris_alchemy/block/entity/EnergyCondenserEntity.java
index 33a6250..105e9e3 100644
--- a/src/main/java/lv/enes/mc/eris_alchemy/block/entity/EnergyCondenserEntity.java
+++ b/src/main/java/lv/enes/mc/eris_alchemy/block/entity/EnergyCondenserEntity.java
@@ -25,7 +25,7 @@ import org.quiltmc.loader.api.minecraft.ClientOnly;
 
 import java.util.stream.IntStream;
 
-public class EnergyCondenserEntity extends ChestLikeEntity implements ExtendedScreenHandlerFactory {
+public class EnergyCondenserEntity extends ChestLikeEntity implements EmcStorageEntity, ExtendedScreenHandlerFactory {
         private final static int WIDTH = 13;
         private final static int HEIGHT = 7;
 
@@ -49,6 +49,11 @@ public class EnergyCondenserEntity extends ChestLikeEntity implements ExtendedSc
                 return items;
         }
 
+        @Override
+        public double getMaxEmc() {
+                return Emc.get(items.get(0)).orElse(Double.POSITIVE_INFINITY);
+        }
+
         @ClientOnly
         @Nonnull
         public Material getMaterial() {
@@ -61,6 +66,7 @@ public class EnergyCondenserEntity extends ChestLikeEntity implements ExtendedSc
                 return Blocks.ENERGY_CONDENSER;
         }
 
+        @Override
         public double getStoredEmc() {
                 return storedEmc.getValue();
         }
@@ -89,11 +95,9 @@ public class EnergyCondenserEntity extends ChestLikeEntity implements ExtendedSc
         @Override
         public void tick(Level world, BlockPos pos, BlockState state) {
                 super.tick(world, pos, state);
-                Emc.get(items.get(0)).ifPresent(cost -> {
-                        tryConsumeEmc(cost);
-                        tryCloneTemplate(cost);
-                });
-
+                var cost = getMaxEmc();
+                tryConsumeEmc(cost);
+                tryCloneTemplate(cost);
                 this.storedEmc.syncIfChanged(ContainerOpenersCounterUtil.getOpeners(openersCounter));
         }