path: root/src/main/kotlin/io/github/moulberry/notenoughupdates/miscfeatures/profileviewer/bestiary/BestiaryPage.kt

/*
 * Copyright (C) 2022-2023 NotEnoughUpdates contributors
 *
 * This file is part of NotEnoughUpdates.
 *
 * NotEnoughUpdates is free software: you can redistribute it
 * and/or modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation, either
 * version 3 of the License, or (at your option) any later version.
 *
 * NotEnoughUpdates is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with NotEnoughUpdates. If not, see <https://www.gnu.org/licenses/>.
 */
package io.github.moulberry.notenoughupdates.miscfeatures.profileviewer.bestiary

import io.github.moulberry.notenoughupdates.core.util.StringUtils
import io.github.moulberry.notenoughupdates.profileviewer.GuiProfileViewer
import io.github.moulberry.notenoughupdates.profileviewer.GuiProfileViewerPage
import io.github.moulberry.notenoughupdates.miscfeatures.profileviewer.bestiary.BestiaryData.calculateTotalBestiaryTiers
import io.github.moulberry.notenoughupdates.miscfeatures.profileviewer.bestiary.BestiaryData.hasMigrated
import io.github.moulberry.notenoughupdates.miscfeatures.profileviewer.bestiary.BestiaryData.parseBestiaryData
import io.github.moulberry.notenoughupdates.util.Constants
import io.github.moulberry.notenoughupdates.util.Utils
import net.minecraft.client.Minecraft
import net.minecraft.client.gui.ScaledResolution
import net.minecraft.client.renderer.GlStateManager
import net.minecraft.client.renderer.RenderHelper
import net.minecraft.item.ItemStack
import net.minecraft.util.EnumChatFormatting
import net.minecraft.util.ResourceLocation
import org.lwjgl.input.Mouse
import org.lwjgl.opengl.GL11
import java.awt.Color

/**
 * Individual mob entry in the Bestiary
 */
data class Mob(
    val name: String, val icon: ItemStack, val kills: Double, val deaths: Double, val mobLevelData: MobLevelData
)

/**
 * A Bestiary category as defined in `constants/bestiary.json`
 */
data class Category(
    val id: String,
    val name: String,
    val icon: ItemStack,
    val mobs: List<Mob>,
    val subCategories: List<Category>,
    val familyData: FamilyData
)

/**
 * Level data for one specific mob
 */
data class MobLevelData(
    val level: Int, val maxLevel: Boolean, val progress: Double, val totalProgress: Double, val mobKillData: MobKillData
)

/**
 * Kills data for one specific mob
 */
data class MobKillData(
    val tierKills: Double, val tierReq: Double, val cappedKills: Double, val cap: Double
)

/**
 * Family data for one specific category
 */
data class FamilyData(
    val found: Int, val completed: Int, val total: Int
)

class BestiaryPage(instance: GuiProfileViewer?) : GuiProfileViewerPage(instance) {
    private var selectedCategory = "dynamic"
    private var selectedSubCategory = ""
    private var tooltipToDisplay: MutableList<String> = mutableListOf()
    private var bestiaryLevel = 0.0
    private var computedCategories: MutableList<Category> = mutableListOf()
    private var lastProfileName = ""

    private val bestiaryTexture = ResourceLocation("notenoughupdates:pv_bestiary_tab.png")
    private val mobListXCount = 9
    private val mobListYCount = 5
    private val mobListXPadding = (240 - mobListXCount * 20) / (mobListXCount + 1).toFloat()
    private val mobListYPadding = (202 - mobListYCount * 20) / (mobListYCount + 1).toFloat()

    override fun drawPage(mouseX: Int, mouseY: Int, partialTicks: Float) {
        val guiLeft = GuiProfileViewer.getGuiLeft()
        val guiTop = GuiProfileViewer.getGuiTop()

        val selectedProfile = GuiProfileViewer.getSelectedProfile() ?: return
        val profileInfo = selectedProfile.profileJson
        val profileName = GuiProfileViewer.getProfileName()

        if (!hasMigrated(profileInfo) || Constants.BESTIARY == null) {
            Utils.drawStringCentered(
                "${EnumChatFormatting.RED}No valid bestiary data!",
                guiLeft + 431 / 2f,
                (guiTop + 101).toFloat(),
                true,
                0
            )
            lastProfileName = profileName
            return
        }
        // Do the initial parsing only once or on profile switch
        if (computedCategories.isEmpty() || lastProfileName != profileName) {
            computedCategories = parseBestiaryData(profileInfo)
            bestiaryLevel = calculateTotalBestiaryTiers(computedCategories).toDouble()
        }

        if (computedCategories.isEmpty() || Constants.BESTIARY == null) {
            Utils.drawStringCentered(
                "${EnumChatFormatting.RED}No valid bestiary data!",
                guiLeft + 431 / 2f,
                (guiTop + 101).toFloat(),
                true,
                0
            )
            lastProfileName = profileName
            return
        }
        lastProfileName = profileName

        val bestiarySize = computedCategories.size
        val bestiaryXSize = (350f / (bestiarySize - 1 + 0.0000001f)).toInt()


        // Render the category list
        for ((categoryXIndex, category) in computedCategories.withIndex()) {
            Minecraft.getMinecraft().textureManager.bindTexture(GuiProfileViewer.pv_elements)

            if (mouseX > guiLeft + 30 + bestiaryXSize * categoryXIndex &&
                mouseX < guiLeft + 30 + bestiaryXSize * categoryXIndex + 20
                && mouseY > guiTop + 10 && mouseY < guiTop + 10 + 20
            ) {
                tooltipToDisplay.add(EnumChatFormatting.GRAY.toString() + category.name)
                if (Mouse.getEventButtonState() && selectedCategory != category.id) {
                    selectedCategory = category.id
                    Utils.playPressSound()
                }
            }


            if (category.id == selectedCategory) {
                Utils.drawTexturedRect(
                    (guiLeft + 30 + bestiaryXSize * categoryXIndex).toFloat(),
                    (guiTop + 10).toFloat(),
                    20f,
                    20f,
                    20 / 256f,
                    0f,
                    20 / 256f,
                    0f,
                    GL11.GL_NEAREST
                )
            } else {
                Utils.drawTexturedRect(
                    (guiLeft + 30 + bestiaryXSize * categoryXIndex).toFloat(),
                    (guiTop + 10).toFloat(),
                    20f,
                    20f,
                    0f,
                    20 / 256f,
                    0f,
                    20 / 256f,
                    GL11.GL_NEAREST
                )
            }
            Utils.drawItemStack(category.icon, guiLeft + 32 + bestiaryXSize * categoryXIndex, guiTop + 12)
        }

        val scaledResolution = ScaledResolution(Minecraft.getMinecraft())
        val width = scaledResolution.scaledWidth
        val height = scaledResolution.scaledHeight

        Minecraft.getMinecraft().textureManager.bindTexture(bestiaryTexture)
        Utils.drawTexturedRect(guiLeft.toFloat(), guiTop.toFloat(), 431f, 202f, GL11.GL_NEAREST)
        GlStateManager.color(1f, 1f, 1f, 1f)
        val color = Color(128, 128, 128, 255)
        Utils.renderAlignedString(
            EnumChatFormatting.RED.toString() + "Milestone: ",
            "${EnumChatFormatting.GRAY}${(bestiaryLevel / 10)}",
            (guiLeft + 280).toFloat(),
            (guiTop + 50).toFloat(),
            110
        )

        // Render the subcategories in the bottom right corner, if applicable
        val selectedCategory = computedCategories.first { it.id == selectedCategory }
        if (selectedCategory.subCategories.isNotEmpty()) {
            if (selectedSubCategory == "") {
                selectedSubCategory = selectedCategory.subCategories.first().id
            }

            Utils.renderShadowedString(
                "${EnumChatFormatting.RED}Subcategories", (guiLeft + 317).toFloat(), (guiTop + 165).toFloat(), 1000
            )
            GlStateManager.color(1f, 1f, 1f, 1f)