Add support for WAILA (specifically with WTHIT).

1 files changed, 22 insertions, 0 deletions

diff --git a/src/main/java/lv/enes/mc/eris_alchemy/waila/EmcProvider.java b/src/main/java/lv/enes/mc/eris_alchemy/waila/EmcProvider.java
new file mode 100644
index 0000000..190ebbe
--- /dev/null
+++ b/src/main/java/lv/enes/mc/eris_alchemy/waila/EmcProvider.java
@@ -0,0 +1,22 @@
+package lv.enes.mc.eris_alchemy.waila;
+
+import lv.enes.mc.eris_alchemy.block.entity.EmcStorageEntity;
+import mcp.mobius.waila.api.IDataProvider;
+import mcp.mobius.waila.api.IDataWriter;
+import mcp.mobius.waila.api.IPluginConfig;
+import mcp.mobius.waila.api.IServerAccessor;
+import mcp.mobius.waila.api.data.EnergyData;
+import net.minecraft.world.level.block.entity.BlockEntity;
+
+public enum EmcProvider implements IDataProvider<BlockEntity> {
+        INSTANCE;
+
+        @Override
+        public void appendData(IDataWriter data, IServerAccessor<BlockEntity> accessor, IPluginConfig config) {
+                data.add(EnergyData.class, res -> {
+                        if (accessor.getTarget() instanceof EmcStorageEntity storage) {
+                                res.add(EnergyData.of(storage.getStoredEmc(), storage.getMaxEmc()));
+                        }
+                });
+        }
+}