package moe.nea.firmament.compat.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 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.compat.rei.SBItemEntryDefinition
import moe.nea.firmament.compat.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 = SBItemEntryDefinition.getPassthrough(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)
				add(Widgets.createBurningFire(ingredientsCenter).animationDurationTicks(25.0))
				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))
				)
			}
		}
	}

}
