working version

1 files changed, 281 insertions, 0 deletions

diff --git a/src/main/kotlin/io/github/moulberry/notenoughupdates/miscfeatures/MuseumCheapestItemOverlay.kt b/src/main/kotlin/io/github/moulberry/notenoughupdates/miscfeatures/MuseumCheapestItemOverlay.kt
new file mode 100644
index 00000000..1526513e
--- /dev/null
+++ b/src/main/kotlin/io/github/moulberry/notenoughupdates/miscfeatures/MuseumCheapestItemOverlay.kt
@@ -0,0 +1,281 @@
+/*
+ * Copyright (C) 2022 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
+
+import io.github.moulberry.notenoughupdates.NEUManager
+import io.github.moulberry.notenoughupdates.NotEnoughUpdates
+import io.github.moulberry.notenoughupdates.core.util.ArrowPagesUtils
+import io.github.moulberry.notenoughupdates.mixins.AccessorGuiContainer
+import io.github.moulberry.notenoughupdates.util.Utils
+import io.github.moulberry.notenoughupdates.util.stripControlCodes
+import net.minecraft.client.Minecraft
+import net.minecraft.client.gui.GuiScreen
+import net.minecraft.client.gui.inventory.GuiChest
+import net.minecraft.client.renderer.GlStateManager
+import net.minecraft.init.Items
+import net.minecraft.inventory.Slot
+import net.minecraft.item.ItemDye
+import net.minecraft.util.EnumChatFormatting
+import net.minecraft.util.ResourceLocation
+import net.minecraftforge.client.event.GuiScreenEvent
+import net.minecraftforge.client.event.GuiScreenEvent.BackgroundDrawnEvent
+import net.minecraftforge.fml.common.eventhandler.SubscribeEvent
+import org.lwjgl.input.Mouse
+import org.lwjgl.opengl.GL11
+
+
+object MuseumCheapestItemOverlay {
+    data class MuseumItem(var name: String, var value: Double, var priceRefreshedAt: Long)
+
+    private val backgroundResource: ResourceLocation by lazy {
+        ResourceLocation("notenoughupdates:dungeon_chest_worth.png")
+    }
+
+    private const val ITEMS_PER_PAGE = 8
+    private var topLeft = intArrayOf(237, 110)
+    private var currentPage: Int = 0
+    private var previousSlots: List<Slot> = emptyList()
+    private var itemsToDonate: MutableList<MuseumItem> = emptyList<MuseumItem>().toMutableList()
+
+    //category -> was the highest page visited?
+    private var checkedPages: HashMap<String, Boolean> = hashMapOf(
+        //this page only shows items when you have already donated them -> there is no useful information to gather
+        "Special Items" to true,
+        "Weapons" to false,
+        "Armor Sets" to false,
+        "Rarities" to false
+    )
+
+    @SubscribeEvent
+    fun onDrawBackground(event: BackgroundDrawnEvent) {
+        if (!shouldRender(event.gui)) return
+        val chest = event.gui as GuiChest
+
+        val slots = chest.inventorySlots.inventorySlots
+        if (!slots.equals(previousSlots)) {
+            parseItems(slots)
+            checkIfHighestPageWasVisited(slots)
+            sortByPrice()
+        }
+        previousSlots = slots
+
+        val xSize = (event.gui as AccessorGuiContainer).xSize
+        val guiLeft = (event.gui as AccessorGuiContainer).guiLeft
+        val guiTop = (event.gui as AccessorGuiContainer).guiTop
+
+        drawBackground(guiLeft, xSize, guiTop)
+        drawLines(guiLeft, xSize, guiTop)
+    }
+
+    @SubscribeEvent
+    fun onMouseClick(event: GuiScreenEvent.MouseInputEvent.Pre) {
+        if (!shouldRender(event.gui)) return
+        if (!Mouse.getEventButtonState()) return
+        val guiLeft = (event.gui as AccessorGuiContainer).guiLeft
+        val guiTop = (event.gui as AccessorGuiContainer).guiTop
+        ArrowPagesUtils.onPageSwitchMouse(
+            guiLeft, guiTop, topLeft, currentPage, totalPages()
+        ) { pageChange: Int -> currentPage = pageChange }
+    }
+
+
+    private fun sortByPrice() {
+        itemsToDonate.sortBy { it.value }
+    }
+
+
+    private fun drawLines(guiLeft: Int, xSize: Int, guiTop: Int) {
+        //render
+        val lines = buildLines()
+        lines.forEachIndexed { index, line ->
+            if (index == ITEMS_PER_PAGE && !visitedAllPages()) {
+                Minecraft.getMinecraft().fontRendererObj.drawStringWithShadow(
+                    "${EnumChatFormatting.RED}Visit all pages for accurate info!",
+                    (guiLeft + 185).toFloat(),
+                    (guiTop + 85).toFloat(),
+                    0
+                )
+            } else {
+                Utils.renderAlignedString(
+                    "${EnumChatFormatting.BLUE}${line.name}",
+                    "${EnumChatFormatting.BLUE}${
+                        if (line.value == Double.MAX_VALUE) "${EnumChatFormatting.RED}Unknown" else "${EnumChatFormatting.AQUA}${
+                            Utils.shortNumberFormat(
+                                line.value,
+                                0
+                            )
+                        }"
+                    }",
+                    (guiLeft + 187).toFloat(),
+                    (guiTop + 5 + (index * 10)).toFloat(),
+                    160
+                )
+            }
+        }
+
+        ArrowPagesUtils.onDraw(guiLeft, guiTop, topLeft, currentPage, totalPages())
+        return
+    }
+
+    private fun buildLines(): List<MuseumItem> {
+        val list = emptyList<MuseumItem>().toMutableList()
+        for (i in ITEMS_PER_PAGE * currentPage..ITEMS_PER_PAGE * currentPage + ITEMS_PER_PAGE) {
+            if (i >= itemsToDonate.size) {
+                break
+            }
+            list.add(itemsToDonate[i])
+        }
+        return list
+    }
+
+    private fun parseItems(slots: List<Slot>) {
+        //iterate upper chest with 56 slots
+        val time = System.currentTimeMillis()
+        for (i in 0..53) {
+            val stack = slots[i].stack ?: continue
+            //check for gray dye which indicates that the item has not been donated
+            if (stack.item is ItemDye && stack.itemDamage == 8) {
+                val name = stack.displayName.stripControlCodes()
+                val internalNames = guessInternalNames(name, Utils.getOpenChestName().endsWith("Armor Sets"))
+//                println("$name resolves to ${internalNames.toString()}")
+                var totalValue = 0.0
+                internalNames.forEach {
+                    val itemValue = NotEnoughUpdates.INSTANCE.manager.auctionManager.getBazaarOrBin(it, false)
+                    if (itemValue == -1.0 || itemValue == 0.0) {
+                        totalValue = Double.MAX_VALUE
+                        return@forEach
+                    } else {
+                        totalValue += itemValue
+                    }
+                }
+                if (totalValue == 0.0) {
+                    totalValue = Double.MAX_VALUE
+                }
+                itemsToDonate.add(MuseumItem(name, totalValue, time))
+            }
+        }
+    }
+
+    private fun checkIfHighestPageWasVisited(slots: List<Slot>) {
+        val category = getCategory()
+        val nextPageSlot = slots[53]
+        //if the "Next Page" arrow is missing, we are at the highest page
+        if ((nextPageSlot.stack ?: return).item != Items.arrow) {
+            checkedPages[category] = true
+        }
+    }
+
+    private fun guessInternalNames(itemName: String, armor: Boolean): List<String> {
+        return if (armor) {
+            //TODO: convert this to use titlewordmap as well
+            val suffixes = listOf(
+                "HELMET",
+                "CHESTPLATE",
+                "LEGGINGS",
+                "BOOTS",
+                "ARMOR_HELMET",
+                "ARMOR_CHESTPLATE",
+                "ARMOR_LEGGINGS",
+                "ARMOR_BOOTS"
+            )
+
+            val prefixes = listOf(
+                "HELMET",
+                "LEGGINGS",
+                "CHESTPLATE",
+                "BOOTS"
+            )
+            val validNames = mutableListOf<String>()
+            val name = itemName
+                .replace(" Armor", "")
+                .replace("Armor ", "")
+                .replace("'s", "")
+                .replace(" ", "_")
+                .uppercase()
+
+            for (suffix in suffixes) {
+                val possibleName = name + "_" + suffix
+                if (NotEnoughUpdates.INSTANCE.manager.isValidInternalName(possibleName)) {
+                    validNames.add(possibleName)
+                } else {
+                    //catch for example: ARMOR_OF_YOG_LEGGINGS, where "Armor" does not need to be replaced to get a valid internalname
+                    val unchangedName = itemName.replace(" ", "_").uppercase() + "_" + suffix
+                    if (NotEnoughUpdates.INSTANCE.manager.isValidInternalName(unchangedName)) {
+                        validNames.add(unchangedName)
+                    }
+                }
+            }
+
+            for (prefix in prefixes) {
+                val possibleName = prefix + "_" + name
+                if (NotEnoughUpdates.INSTANCE.manager.isValidInternalName(possibleName)) {
+                    validNames.add(possibleName)
+                }
+
+            }
+
+            validNames
+        } else {
+            val monochromeName = NEUManager.cleanForTitleMapSearch(itemName)
+            val internalName = monochromeName.split(" ")
+                .asSequence()
+                .mapNotNull { NotEnoughUpdates.INSTANCE.manager.titleWordMap[it]?.keys }
+                .flatten()
+                .filter {
+                    val item = NotEnoughUpdates.INSTANCE.manager.createItem(it) ?: return@filter false
+                    item.displayName != null && item.displayName.isNotEmpty() && NEUManager.cleanForTitleMapSearch(item.displayName) in monochromeName
+                }
+                .firstOrNull() ?: return emptyList()
+
+            listOf(internalName)
+        }
+    }
+
+
+    private fun drawBackground(guiLeft: Int, xSize: Int, guiTop: Int) {
+        Minecraft.getMinecraft().textureManager.bindTexture(backgroundResource)
+        GL11.glColor4f(1F, 1F, 1F, 1F)
+        GlStateManager.disableLighting()
+        Utils.drawTexturedRect(
+            guiLeft.toFloat() + xSize + 4,
+            guiTop.toFloat(),
+            180F,
+            101F,
+            0F,
+            180 / 256F,
+            0F,
+            101 / 256F,
+            GL11.GL_NEAREST
+        )
+    }
+
+    private fun shouldRender(gui: GuiScreen): Boolean =
+        NotEnoughUpdates.INSTANCE.config.misc.museumCheapestItemOverlay && gui is GuiChest && Utils.getOpenChestName()
+            .startsWith("Museum ➜")
+
+    private fun getCategory(): String = Utils.getOpenChestName().substring(9, Utils.getOpenChestName().length)
+
+    private fun visitedAllPages(): Boolean = !checkedPages.containsValue(false)
+
+    private fun totalPages(): Int = when (itemsToDonate.size % ITEMS_PER_PAGE) {
+        0 -> itemsToDonate.size / ITEMS_PER_PAGE
+        else -> (itemsToDonate.size / ITEMS_PER_PAGE) + 1
+    }
+}