package at.hannibal2.skyhanni.utils

import net.minecraft.init.Items
import net.minecraft.item.ItemStack
import net.minecraft.nbt.NBTTagCompound
import net.minecraft.nbt.NBTTagList
import net.minecraft.nbt.NBTTagString
import net.minecraftforge.common.util.Constants
import java.util.*

object ItemUtil {
    private val PET_PATTERN = "§7\\[Lvl \\d+] (?<color>§[0-9a-fk-or]).+".toRegex()
    const val NBT_INTEGER = 3
    private const val NBT_STRING = 8
    private const val NBT_LIST = 9
    private const val NBT_COMPOUND = 10

    /**
     * Returns the display name of a given item
     * @author Mojang
     * @param item the Item to get the display name of
     * @return the display name of the item
     */
    @JvmStatic
    fun getDisplayName(item: ItemStack): String {
        var s = item.item.getItemStackDisplayName(item)
        if (item.tagCompound != null && item.tagCompound.hasKey("display", 10)) {
            val nbtTagCompound = item.tagCompound.getCompoundTag("display")
            if (nbtTagCompound.hasKey("Name", 8)) {
                s = nbtTagCompound.getString("Name")
            }
        }
        return s
    }

    /**
     * Returns the Skyblock Item ID of a given Skyblock item
     *
     * @author BiscuitDevelopment
     * @param item the Skyblock item to check
     * @return the Skyblock Item ID of this item or `null` if this isn't a valid Skyblock item
     */
    @JvmStatic
    fun getSkyBlockItemID(item: ItemStack?): String? {
        if (item == null) {
            return null
        }
        val extraAttributes = getExtraAttributes(item) ?: return null
        return if (!extraAttributes.hasKey("id", NBT_STRING)) {
            null
        } else extraAttributes.getString("id")
    }

    /**
     * Returns the `ExtraAttributes` compound tag from the item's NBT data.
     *
     * @author BiscuitDevelopment
     * @param item the item to get the tag from
     * @return the item's `ExtraAttributes` compound tag or `null` if the item doesn't have one
     */
    @JvmStatic
    fun getExtraAttributes(item: ItemStack?): NBTTagCompound? {
        return if (item == null || !item.hasTagCompound()) {
            null
        } else item.getSubCompound("ExtraAttributes", false)
    }

    /**
     * Returns the Skyblock Item ID of a given Skyblock Extra Attributes NBT Compound
     *
     * @author BiscuitDevelopment
     * @param extraAttributes the NBT to check
     * @return the Skyblock Item ID of this item or `null` if this isn't a valid Skyblock NBT
     */
    @JvmStatic
    fun getSkyBlockItemID(extraAttributes: NBTTagCompound?): String? {
        if (extraAttributes != null) {
            val itemId = extraAttributes.getString("id")
            if (itemId.isNotEmpty()) {
                return itemId
            }
        }
        return null
    }

    /**
     * Returns a string list containing the nbt lore of an ItemStack, or
     * an empty list if this item doesn't have a lore. The returned lore
     * list is unmodifiable since it has been converted from an NBTTagList.
     *
     * @author BiscuitDevelopment
     * @param itemStack the ItemStack to get the lore from
     * @return the lore of an ItemStack as a string list
     */
    @JvmStatic
    fun getItemLore(itemStack: ItemStack): List<String> {
        if (itemStack.hasTagCompound() && itemStack.tagCompound.hasKey("display", NBT_COMPOUND)) {
            val display = itemStack.tagCompound.getCompoundTag("display")
            if (display.hasKey("Lore", NBT_LIST)) {
                val lore = display.getTagList("Lore", NBT_STRING)
                val loreAsList = ArrayList<String>(lore.tagCount())
                for (lineNumber in 0 until lore.tagCount()) {
                    loreAsList.add(lore.getStringTagAt(lineNumber))
                }
                return Collections.unmodifiableList(loreAsList)
            }
        }
        return emptyList()
    }

//            @JvmStatic
//            fun hasRightClickAbility(itemStack: ItemStack): Boolean {
//                for (line in getItemLore(itemStack)) {
//                    val stripped = line.stripControlCodes()
//                    if (stripped.startsWith("Item Ability:") && stripped.endsWith("RIGHT CLICK")) return true
//                }
//                return false
//            }

//            /**
//             * Returns the rarity of a given Skyblock item
//             * Modified
//             * @author BiscuitDevelopment
//             * @param item the Skyblock item to check
//             * @return the rarity of the item if a valid rarity is found, `null` if no rarity is found, `null` if item is `null`
//             */
//            fun getRarity(item: ItemStack?): ItemRarity {
//                if (item == null || !item.hasTagCompound()) {
//                    return ItemRarity.NONE
//                }
//                val display = item.getSubCompound("display", false)
//                if (display == null || !display.hasKey("Lore")) {
//                    return ItemRarity.NONE
//                }
//                val lore = display.getTagList("Lore", Constants.NBT.TAG_STRING)
//                val name = display.getString("Name")
//
//                // Determine the item's rarity
//                for (i in (lore.tagCount() - 1) downTo 0) {
//                    val currentLine = lore.getStringTagAt(i)
//                    val rarityMatcher = RARITY_PATTERN.find(currentLine)
//                    if (rarityMatcher != null) {
//                        val rarity = rarityMatcher.groups["rarity"]?.value ?: continue
//                        ItemRarity.values().find {
//                            it.rarityName == rarity.stripControlCodes().substringAfter("SHINY ")
//                        }?.let {
//                            return it
//                        }
//                    }
//                }
//                val petRarityMatcher = PET_PATTERN.find(name)
//                if (petRarityMatcher != null) {
//                    val color = petRarityMatcher.groupValues.getOrNull(1) ?: return ItemRarity.NONE
//                    return ItemRarity.byBaseColor(color) ?: ItemRarity.NONE
//                }
//
//                // If the item doesn't have a valid rarity, return null
//                return ItemRarity.NONE
//            }

    fun isPet(item: ItemStack?): Boolean {
        if (item == null || !item.hasTagCompound()) {
            return false
        }
        val display = item.getSubCompound("display", false)
        if (display == null || !display.hasKey("Lore")) {
            return false
        }
        val name = display.getString("Name")

        return PET_PATTERN.matches(name)
    }

    fun setSkullTexture(item: ItemStack, texture: String, SkullOwner: String): ItemStack {
        val textureTagCompound = NBTTagCompound()
        textureTagCompound.setString("Value", texture)

        val textures = NBTTagList()
        textures.appendTag(textureTagCompound)

        val properties = NBTTagCompound()
        properties.setTag("textures", textures)

        val skullOwner = NBTTagCompound()
        skullOwner.setString("Id", SkullOwner)
        skullOwner.setTag("Properties", properties)

        val nbtTag = NBTTagCompound()
        nbtTag.setTag("SkullOwner", skullOwner)

        item.tagCompound = nbtTag
        return item
    }

    fun getSkullTexture(item: ItemStack): String? {
        if (item.item != Items.skull) return null
        val nbt = item.tagCompound
        if (!nbt.hasKey("SkullOwner")) return null
        return nbt.getCompoundTag("SkullOwner").getCompoundTag("Properties")
            .getTagList("textures", Constants.NBT.TAG_COMPOUND).getCompoundTagAt(0).getString("Value")
    }

    fun ItemStack.setLore(lines: List<String>): ItemStack {
        setTagInfo("display", getSubCompound("display", true).apply {
            setTag("Lore", NBTTagList().apply {
                for (line in lines) appendTag(NBTTagString(line))
            })
        })
        return this
    }

    fun NBTTagList.asStringSet() = (0..tagCount()).mapTo(hashSetOf()) { getStringTagAt(it) }
}