Basic Searching Works

1 files changed, 100 insertions, 0 deletions

diff --git a/src/main/java/me/shedaniel/rei/client/RecipeHelper.java b/src/main/java/me/shedaniel/rei/client/RecipeHelper.java
new file mode 100644
index 000000000..74780846c
--- /dev/null
+++ b/src/main/java/me/shedaniel/rei/client/RecipeHelper.java
@@ -0,0 +1,100 @@
+package me.shedaniel.rei.client;
+
+import com.google.common.collect.Lists;
+import com.google.common.collect.Maps;
+import me.shedaniel.rei.RoughlyEnoughItemsCore;
+import me.shedaniel.rei.api.IRecipeCategory;
+import me.shedaniel.rei.api.IRecipeDisplay;
+import me.shedaniel.rei.api.IRecipePlugin;
+import me.shedaniel.rei.listeners.RecipeSync;
+import net.minecraft.item.ItemStack;
+import net.minecraft.recipe.RecipeManager;
+import net.minecraft.util.Identifier;
+
+import java.util.*;
+import java.util.stream.Collectors;
+
+public class RecipeHelper implements RecipeSync {
+    
+    private static Map<Identifier, List<IRecipeDisplay>> recipeCategoryListMap;
+    private static List<IRecipeCategory> categories;
+    private static RecipeManager recipeManager;
+    
+    public RecipeHelper() {
+        this.recipeCategoryListMap = Maps.newHashMap();
+        this.categories = Lists.newArrayList();
+    }
+    
+    public static List<ItemStack> findCraftableByItems(List<ItemStack> inventoryItems) {
+        List<ItemStack> craftables = new ArrayList<>();
+        for(List<IRecipeDisplay> value : recipeCategoryListMap.values())
+            for(IRecipeDisplay recipeDisplay : value) {
+                int slotsCraftable = 0;
+                List<List<ItemStack>> requiredInput = (List<List<ItemStack>>) recipeDisplay.getRequiredItems();
+                for(List<ItemStack> slot : requiredInput) {
+                    if (slot.isEmpty()) {
+                        slotsCraftable++;
+                        continue;
+                    }
+                    boolean slotDone = false;
+                    for(ItemStack possibleType : inventoryItems) {
+                        for(ItemStack slotPossible : slot)
+                            if (ItemStack.areEqualIgnoreTags(slotPossible, possibleType)) {
+                                slotsCraftable++;
+                                slotDone = true;
+                                break;
+                            }
+                        if (slotDone)
+                            break;
+                    }
+                }
+                if (slotsCraftable == recipeDisplay.getRequiredItems().size())
+                    craftables.addAll((List<ItemStack>) recipeDisplay.getOutput());
+            }
+        return craftables.stream().distinct().collect(Collectors.toList());
+    }
+    
+    public static void registerCategory(IRecipeCategory category) {
+        categories.add(category);
+        recipeCategoryListMap.put(category.getIdentifier(), Lists.newArrayList());
+    }
+    
+    public static void registerRecipe(Identifier categoryIdentifier, IRecipeDisplay display) {
+        if (!recipeCategoryListMap.containsKey(categoryIdentifier))
+            return;
+        recipeCategoryListMap.get(categoryIdentifier).add(display);
+    }
+    
+    public static Map<IRecipeCategory, List<IRecipeDisplay>> getRecipesFor(ItemStack stack) {
+        Map<Identifier, List<IRecipeDisplay>> categoriesMap = new HashMap<>();
+        categories.forEach(f -> categoriesMap.put(f.getIdentifier(), new LinkedList<>()));
+        for(List<IRecipeDisplay> value : recipeCategoryListMap.values())
+            for(IRecipeDisplay recipeDisplay : value)
+                for(ItemStack outputStack : (List<ItemStack>) recipeDisplay.getOutput())
+                    if (ItemStack.areEqualIgnoreTags(stack, outputStack))
+                        categoriesMap.get(recipeDisplay.getRecipeCategory()).add(recipeDisplay);
+        categoriesMap.keySet().removeIf(f -> categoriesMap.get(f).isEmpty());
+        Map<IRecipeCategory, List<IRecipeDisplay>> recipeCategoryListMap = Maps.newHashMap();
+        categories.forEach(category -> {
+            if (categoriesMap.containsKey(category.getIdentifier()))
+                recipeCategoryListMap.put(category, categoriesMap.get(category.getIdentifier()));
+        });
+        return recipeCategoryListMap;
+    }
+    
+    public static RecipeManager getRecipeManager() {
+        return recipeManager;
+    }
+    
+    @Override
+    public void recipesLoaded(RecipeManager recipeManager) {
+        this.recipeManager = recipeManager;
+        this.recipeCategoryListMap.clear();
+        this.categories.clear();
+        RoughlyEnoughItemsCore.getListeners(IRecipePlugin.class).forEach(plugin -> {
+            plugin.registerPluginCategories();
+            plugin.registerRecipes();
+        });
+    }
+    
+}