package moe.nea.firmament.rei.recipes

import io.github.moulberry.repo.data.NEUForgeRecipe
import me.shedaniel.math.Point
import me.shedaniel.math.Rectangle
import me.shedaniel.rei.api.client.gui.Renderer
import me.shedaniel.rei.api.client.gui.widgets.Widget
import me.shedaniel.rei.api.client.gui.widgets.Widgets
import me.shedaniel.rei.api.client.registry.display.DisplayCategory
import me.shedaniel.rei.api.common.category.CategoryIdentifier
import me.shedaniel.rei.api.common.util.EntryStacks
import kotlin.math.cos
import kotlin.math.sin
import kotlin.time.Duration.Companion.seconds
import net.minecraft.block.Blocks
import net.minecraft.text.Text
import moe.nea.firmament.Firmament
import moe.nea.firmament.rei.SBItemEntryDefinition
import moe.nea.firmament.rei.plus

class SBForgeRecipe(override val neuRecipe: NEUForgeRecipe) : SBRecipe() {
    override fun getCategoryIdentifier(): CategoryIdentifier<*> = Category.categoryIdentifier

    object Category : DisplayCategory<SBForgeRecipe> {
        override fun getCategoryIdentifier(): CategoryIdentifier<SBForgeRecipe> =
            CategoryIdentifier.of(Firmament.MOD_ID, "forge_recipe")

        override fun getTitle(): Text = Text.literal("Forge Recipes")
        override fun getDisplayHeight(): Int {
            return 104
        }

        override fun getIcon(): Renderer = EntryStacks.of(Blocks.ANVIL)
        override fun setupDisplay(display: SBForgeRecipe, bounds: Rectangle): List<Widget> {
            return buildList {
                add(Widgets.createRecipeBase(bounds))
                add(Widgets.createResultSlotBackground(Point(bounds.minX + 124, bounds.minY + 46)))
                val arrow = Widgets.createArrow(Point(bounds.minX + 90, bounds.minY + 54 - 18 / 2))
                add(arrow)
                add(Widgets.createTooltip(arrow.bounds, Text.stringifiedTranslatable("firmament.recipe.forge.time", display.neuRecipe.duration.seconds)))
                val ingredientsCenter = Point(bounds.minX + 49 - 8, bounds.minY + 54 - 8)
                val count = display.neuRecipe.inputs.size
                if (count == 1) {
                    add(
                        Widgets.createSlot(Point(ingredientsCenter.x, ingredientsCenter.y)).markInput()
                            .entry(SBItemEntryDefinition.getEntry(display.neuRecipe.inputs.single()))
                    )
                } else {
                    display.neuRecipe.inputs.forEachIndexed { idx, ingredient ->
                        val rad = Math.PI * 2 * idx / count
                        add(
                            Widgets.createSlot(
                                Point(
                                    cos(rad) * 30,
                                    sin(rad) * 30,
                                ) + ingredientsCenter
                            ).markInput().entry(SBItemEntryDefinition.getEntry(ingredient))
                        )
                    }
                }
                add(
                    Widgets.createSlot(Point(bounds.minX + 124, bounds.minY + 46)).markOutput().disableBackground()
                        .entry(SBItemEntryDefinition.getEntry(display.neuRecipe.outputStack))
                )
            }
        }
    }

}