Add support for WAILA (specifically with WTHIT).

1 files changed, 5 insertions, 9 deletions

diff --git a/src/main/java/lv/enes/mc/eris_alchemy/Emc.java b/src/main/java/lv/enes/mc/eris_alchemy/Emc.java
index 576fdc2..d278bcc 100644
--- a/src/main/java/lv/enes/mc/eris_alchemy/Emc.java
+++ b/src/main/java/lv/enes/mc/eris_alchemy/Emc.java
@@ -2,6 +2,7 @@ package lv.enes.mc.eris_alchemy;
 
 import jakarta.annotation.Nullable;
 import lv.enes.mc.eris_alchemy.ErisAlchemyRegistry.NetworkingConstants;
+import lv.enes.mc.eris_alchemy.block.EmcStorageBlock;
 import lv.enes.mc.eris_alchemy.recipe.BannedRecipe;
 import lv.enes.mc.eris_alchemy.recipe.SimplifiedRecipe;
 import lv.enes.mc.eris_alchemy.utils.*;
@@ -63,17 +64,12 @@ public final class Emc {
                 return get(itemId)
                                 .stream()
                                 .map(value -> {
-                                        EmcStorage storage = null;
-                                        if (item instanceof EmcStorage emcStorage) {
-                                                storage = emcStorage;
-                                        } else if (item instanceof BlockItem blockItem) {
-                                                if (blockItem.getBlock() instanceof EmcStorage emcStorage) {
-                                                        storage = emcStorage;
+                                        if (item instanceof BlockItem blockItem) {
+                                                if (blockItem.getBlock() instanceof EmcStorageBlock block) {
+                                                        return value + block.getStoredEmc(stack);
                                                 }
                                         }
-                                        if (storage != null) {
-                                                return value + storage.getStoredEmc(stack);
-                                        }
+
                                         return value;
                                 })
                                 .findFirst();