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()}"
            )
        }
    }
}