From 9081089c2e47a20e4108ee322204e8c8dbde4169 Mon Sep 17 00:00:00 2001
From: Uko Kokņevičs
Date: Sat, 27 Jan 2024 04:57:45 +0200
Subject: Add support for WAILA (specifically with WTHIT).

---
 src/main/java/lv/enes/mc/eris_alchemy/Emc.java | 14 +++++---------
 1 file changed, 5 insertions(+), 9 deletions(-)

(limited to 'src/main/java/lv/enes/mc/eris_alchemy/Emc.java')

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();
-- 
cgit v1.2.3