1 files changed, 66 insertions, 0 deletions
diff --git a/src/main/kotlin/features/inventory/CraftingOverlay.kt b/src/main/kotlin/features/inventory/CraftingOverlay.kt
new file mode 100644
index 0000000..031ef78
--- /dev/null
+++ b/src/main/kotlin/features/inventory/CraftingOverlay.kt
@@ -0,0 +1,66 @@
+
+
+package moe.nea.firmament.features.inventory
+
+import net.minecraft.client.gui.screen.ingame.GenericContainerScreen
+import net.minecraft.item.ItemStack
+import net.minecraft.util.Formatting
+import moe.nea.firmament.annotations.Subscribe
+import moe.nea.firmament.events.SlotRenderEvents
+import moe.nea.firmament.features.FirmamentFeature
+import moe.nea.firmament.rei.FirmamentReiPlugin.Companion.asItemEntry
+import moe.nea.firmament.rei.SBItemEntryDefinition
+import moe.nea.firmament.rei.recipes.SBCraftingRecipe
+import moe.nea.firmament.util.MC
+
+object CraftingOverlay : FirmamentFeature {
+
+    private var screen: GenericContainerScreen? = null
+    private var recipe: SBCraftingRecipe? = null
+    private val craftingOverlayIndices = listOf(
+        10, 11, 12,
+        19, 20, 21,
+        28, 29, 30,
+    )
+
+
+    fun setOverlay(screen: GenericContainerScreen, recipe: SBCraftingRecipe) {
+        this.screen = screen
+        this.recipe = recipe
+    }
+
+    override val identifier: String
+        get() = "crafting-overlay"
+
+    @Subscribe
+    fun onSlotRender(event: SlotRenderEvents.After) {
+        val slot = event.slot
+        val recipe = this.recipe ?: return
+        if (slot.inventory != screen?.screenHandler?.inventory) return
+        val recipeIndex = craftingOverlayIndices.indexOf(slot.index)
+        if (recipeIndex < 0) return
+        val expectedItem = recipe.neuRecipe.inputs[recipeIndex]
+        val actualStack = slot.stack ?: ItemStack.EMPTY!!
+        val actualEntry = SBItemEntryDefinition.getEntry(actualStack).value
+        if ((actualEntry.skyblockId.neuItem != expectedItem.itemId || actualEntry.getStackSize() < expectedItem.amount) && expectedItem.amount.toInt() != 0) {
+            event.context.fill(
+                event.slot.x,
+                event.slot.y,
+                event.slot.x + 16,
+                event.slot.y + 16,
+                0x80FF0000.toInt()
+            )
+        }
+        if (!slot.hasStack()) {
+            val itemStack = SBItemEntryDefinition.getEntry(expectedItem).asItemEntry().value
+            event.context.drawItem(itemStack, event.slot.x, event.slot.y)
+            event.context.drawItemInSlot(
+                MC.font,
+                itemStack,
+                event.slot.x,
+                event.slot.y,
+                "${Formatting.RED}${expectedItem.amount.toInt()}"
+            )
+        }
+    }
+}