1 files changed, 28 insertions, 0 deletions
diff --git a/src/main/kotlin/repo/BetterRepoRecipeCache.kt b/src/main/kotlin/repo/BetterRepoRecipeCache.kt
new file mode 100644
index 0000000..91a6b50
--- /dev/null
+++ b/src/main/kotlin/repo/BetterRepoRecipeCache.kt
@@ -0,0 +1,28 @@
+
+package moe.nea.firmament.repo
+
+import io.github.moulberry.repo.IReloadable
+import io.github.moulberry.repo.NEURepository
+import io.github.moulberry.repo.data.NEURecipe
+import moe.nea.firmament.util.SkyblockId
+
+class BetterRepoRecipeCache(val essenceRecipeProvider: EssenceRecipeProvider) : IReloadable {
+    var usages: Map<SkyblockId, Set<NEURecipe>> = mapOf()
+    var recipes: Map<SkyblockId, Set<NEURecipe>> = mapOf()
+
+    override fun reload(repository: NEURepository) {
+        val usages = mutableMapOf<SkyblockId, MutableSet<NEURecipe>>()
+        val recipes = mutableMapOf<SkyblockId, MutableSet<NEURecipe>>()
+        val baseRecipes = repository.items.items.values
+            .asSequence()
+            .flatMap { it.recipes }
+        val extraRecipes = essenceRecipeProvider.recipes
+        (baseRecipes + extraRecipes)
+            .forEach { recipe ->
+                recipe.allInputs.forEach { usages.getOrPut(SkyblockId(it.itemId), ::mutableSetOf).add(recipe) }
+                recipe.allOutputs.forEach { recipes.getOrPut(SkyblockId(it.itemId), ::mutableSetOf).add(recipe) }
+            }
+        this.usages = usages
+        this.recipes = recipes
+    }
+}