package moe.nea.firmament.features.inventory

import java.awt.Color
import net.minecraft.client.gui.DrawContext
import net.minecraft.item.ItemStack
import net.minecraft.util.Formatting
import net.minecraft.util.Identifier
import moe.nea.firmament.annotations.Subscribe
import moe.nea.firmament.events.HotbarItemRenderEvent
import moe.nea.firmament.events.SlotRenderEvents
import moe.nea.firmament.features.FirmamentFeature
import moe.nea.firmament.gui.config.ManagedConfig
import moe.nea.firmament.util.MC
import moe.nea.firmament.util.item.loreAccordingToNbt
import moe.nea.firmament.util.collections.lastNotNullOfOrNull
import moe.nea.firmament.util.collections.memoizeIdentity
import moe.nea.firmament.util.unformattedString

object ItemRarityCosmetics : FirmamentFeature {
    override val identifier: String
        get() = "item-rarity-cosmetics"

    object TConfig : ManagedConfig(identifier) {
        val showItemRarityBackground by toggle("background") { false }
        val showItemRarityInHotbar by toggle("background-hotbar") { false }
    }

    override val config: ManagedConfig
        get() = TConfig

    private val rarityToColor = mapOf(
        "UNCOMMON" to Formatting.GREEN,
        "COMMON" to Formatting.WHITE,
        "RARE" to Formatting.DARK_BLUE,
        "EPIC" to Formatting.DARK_PURPLE,
        "LEGENDARY" to Formatting.GOLD,
        "LEGENJERRY" to Formatting.GOLD,
        "MYTHIC" to Formatting.LIGHT_PURPLE,
        "DIVINE" to Formatting.BLUE,
        "SPECIAL" to Formatting.DARK_RED,
        "SUPREME" to Formatting.DARK_RED,
    ).mapValues {
        val c = Color(it.value.colorValue!!)
        Triple(c.red / 255F, c.green / 255F, c.blue / 255F)
    }

    private fun getSkyblockRarity0(itemStack: ItemStack): Triple<Float, Float, Float>? {
        return itemStack.loreAccordingToNbt.lastNotNullOfOrNull {
            val entry = it.unformattedString
            rarityToColor.entries.find { (k, v) -> k in entry }?.value
        }
    }

    val getSkyblockRarity = ::getSkyblockRarity0.memoizeIdentity(100)


    fun drawItemStackRarity(drawContext: DrawContext, x: Int, y: Int, item: ItemStack) {
        val (r, g, b) = getSkyblockRarity(item) ?: return
        drawContext.drawSprite(
            x, y,
            0,
            16, 16,
            MC.guiAtlasManager.getSprite(Identifier.of("firmament:item_rarity_background")),
            r, g, b, 1F
        )
    }


    @Subscribe
    fun onRenderSlot(it: SlotRenderEvents.Before) {
        if (!TConfig.showItemRarityBackground) return
        val stack = it.slot.stack ?: return
        drawItemStackRarity(it.context, it.slot.x, it.slot.y, stack)
    }

    @Subscribe
    fun onRenderHotbarItem(it: HotbarItemRenderEvent) {
        if (!TConfig.showItemRarityInHotbar) return
        val stack = it.item
        drawItemStackRarity(it.context, it.x, it.y, stack)
    }
}