Support Chipped fully.

1 files changed, 46 insertions, 4 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 d278bcc..cb782e5 100644
--- a/src/main/java/lv/enes/mc/eris_alchemy/Emc.java
+++ b/src/main/java/lv/enes/mc/eris_alchemy/Emc.java
@@ -64,10 +64,9 @@ public final class Emc {
                 return get(itemId)
                                 .stream()
                                 .map(value -> {
-                                        if (item instanceof BlockItem blockItem) {
-                                                if (blockItem.getBlock() instanceof EmcStorageBlock block) {
-                                                        return value + block.getStoredEmc(stack);
-                                                }
+                                        if (item instanceof BlockItem blockItem
+                                                        && blockItem.getBlock() instanceof EmcStorageBlock block) {
+                                                return value + block.getStoredEmc(stack);
                                         }
 
                                         return value;
@@ -220,11 +219,54 @@ public final class Emc {
                 sortedItems.stream()
                                 .filter(id -> !VALUES.containsKey(id))
                                 .forEach(id -> calcEmc(id, recipes).ifPresent(v -> VALUES.put(id, OptionalDouble.of(v))));
+
+                if (ForeignUtils.isClassAvailable("earth.terrarium.chipped.common.recipes.ChippedRecipe")) {
+                        reinitForChipped(sortedItems, recipes);
+                }
+
                 ErisAlchemy.LOGGER.info("Done calculating EMC values...");
 
                 sync();
         }
 
+        private static void reinitForChipped(
+                        Set<ResourceLocation> items,
+                        HashMap<ResourceLocation, List<SimplifiedRecipe>> recipes
+        ) {
+                items.stream()
+                                .filter(id -> !VALUES.containsKey(id))
+                                .forEach(item -> {
+                                        var myRecipes = recipes.getOrDefault(item, List.of())
+                                                        .stream()
+                                                        .filter(SimplifiedRecipe::fromChipped)
+                                                        .toList();
+
+                                        if (myRecipes.size() != 1) {
+                                                if (!myRecipes.isEmpty()) {
+                                                        ErisAlchemy.LOGGER.warn("Item {} has multiple chipped recipes, skipping...", item);
+                                                }
+                                                return;
+                                        }
+
+                                        var recipe = myRecipes.get(0);
+                                        if (recipe.input().size() != 1) {
+                                                ErisAlchemy.LOGGER.warn("Chipped recipe for {} has multiple inputs, skipping...", item);
+                                                return;
+                                        }
+
+                                        var stacks = recipe.input().get(0).get().getItems();
+                                        if (stacks.length != 1) {
+                                                ErisAlchemy.LOGGER.warn("Chipped recipe for {} has multiple stack inputs, skipping...", item);
+                                                return;
+                                        }
+
+                                        var inputId = ItemUtils.getId(stacks[0]);
+                                        if (VALUES.containsKey(inputId)) {
+                                                VALUES.put(item, VALUES.get(inputId));
+                                        }
+                                });
+        }
+
         private static void sortDps(
                         Set<ResourceLocation> permSorted,
                         ConsList<ResourceLocation> tmpSorted,