Count RNG drops for Melon Dicer and Pumpkin Dicer

1 files changed, 130 insertions, 0 deletions

diff --git a/src/main/java/at/hannibal2/skyhanni/features/garden/DicerRngDropCounter.kt b/src/main/java/at/hannibal2/skyhanni/features/garden/DicerRngDropCounter.kt
new file mode 100644
index 000000000..1b170bc7f
--- /dev/null
+++ b/src/main/java/at/hannibal2/skyhanni/features/garden/DicerRngDropCounter.kt
@@ -0,0 +1,130 @@
+package at.hannibal2.skyhanni.features.garden
+
+import at.hannibal2.skyhanni.SkyHanniMod
+import at.hannibal2.skyhanni.events.ConfigLoadEvent
+import at.hannibal2.skyhanni.events.GardenToolChangeEvent
+import at.hannibal2.skyhanni.events.GuiRenderEvent
+import at.hannibal2.skyhanni.events.LorenzChatEvent
+import at.hannibal2.skyhanni.utils.ItemUtils.name
+import at.hannibal2.skyhanni.utils.LorenzUtils
+import at.hannibal2.skyhanni.utils.LorenzUtils.sortedDesc
+import at.hannibal2.skyhanni.utils.RenderUtils.renderStrings
+import net.minecraftforge.fml.common.eventhandler.SubscribeEvent
+
+class DicerRngDropCounter {
+    private val display = mutableListOf<String>()
+    private val drops = mutableMapOf<String, MutableMap<DropRarity, Int>>()
+    private val itemDrops = mutableListOf<ItemDrop>()
+
+    init {
+        drops["Melon"] = mutableMapOf()
+        drops["Pumpkin"] = mutableMapOf()
+
+        itemDrops.add(ItemDrop("Melon", DropRarity.UNCOMMON, "§a§lUNCOMMON DROP! §r§eDicer dropped §r§a1x §r§aEnchanted Melon§r§e!"))
+        itemDrops.add(ItemDrop("Melon", DropRarity.RARE, "§9§lRARE DROP! §r§eDicer dropped §r§a5x §r§aEnchanted Melon§r§e!"))
+        itemDrops.add(ItemDrop("Melon", DropRarity.CRAZY_RARE, "§d§lCRAZY RARE DROP! §r§eDicer dropped §r§a50x §r§aEnchanted Melon§r§e!"))
+        itemDrops.add(ItemDrop("Melon", DropRarity.PRAY_TO_RNGESUS, "§5§lPRAY TO RNGESUS DROP! §r§eDicer dropped §r§92x §r§9Enchanted Melon Block§r§e!"))
+
+        itemDrops.add(ItemDrop("Pumpkin", DropRarity.UNCOMMON, "§a§lUNCOMMON DROP! §r§eDicer dropped §r§f64x §r§fPumpkin§r§e!"))
+        itemDrops.add(ItemDrop("Pumpkin", DropRarity.RARE, "§9§lRARE DROP! §r§eDicer dropped §r§a1x §r§aEnchanted Pumpkin§r§e!"))
+        itemDrops.add(ItemDrop("Pumpkin", DropRarity.CRAZY_RARE, "§d§lCRAZY RARE DROP! §r§eDicer dropped §r§a10x §r§aEnchanted Pumpkin§r§e!"))
+        itemDrops.add(ItemDrop("Pumpkin", DropRarity.PRAY_TO_RNGESUS, "§5§lPRAY TO RNGESUS DROP! §r§eDicer dropped §r§a64x §r§aEnchanted Pumpkin§r§e!"))
+    }
+
+    enum class DropRarity(val displayName: String) {
+        UNCOMMON("§a§lUNCOMMON DROP"),
+        RARE("§9§lRARE DROP"),
+        CRAZY_RARE("§d§lCRAZY RARE DROP"),
+        PRAY_TO_RNGESUS("§5§lPRAY TO RNGESUS DROP"),
+    }
+
+    @SubscribeEvent
+    fun onChat(event: LorenzChatEvent) {
+        if (!isEnabled()) return
+
+        val message = event.message
+        for (drop in itemDrops) {
+            if (message == drop.message) {
+                addDrop(drop.itemName, drop.rarity)
+                saveConfig()
+                update()
+                return
+            }
+        }
+
+        if (message.contains("§r§eDicer dropped")) {
+            LorenzUtils.debug("Unknown dicer drop message!")
+            println("Unknown dicer drop message: '$message'")
+        }
+    }
+
+    private fun update() {
+        val newDisplay = drawDisplay()
+        display.clear()
+        display.addAll(newDisplay)
+    }
+
+    private fun drawDisplay(): List<String> {
+        val help = mutableListOf<String>()
+        val items = drops[itemInHand] ?: return help
+        help.add("§7RNG Drops for $toolname§7:")
+        for ((rarity, amount) in items.sortedDesc()) {
+            val displayName = rarity.displayName
+            help.add(" §7- §e${amount}x $displayName")
+        }
+
+        return help
+    }
+
+    private var itemInHand = ""
+    private var toolname = ""
+
+    @SubscribeEvent
+    fun onGardenToolChange(event: GardenToolChangeEvent) {
+        val crop = event.crop
+        itemInHand = if (crop == "Melon" || crop == "Pumpkin") crop else ""
+        if (itemInHand != "") {
+            toolname = event.heldItem!!.name!!
+        }
+        update()
+    }
+
+    private fun addDrop(item: String, rarity: DropRarity) {
+        val melon = drops[item]!!
+        val old = melon[rarity] ?: 0
+        melon[rarity] = old + 1
+    }
+
+    @SubscribeEvent
+    fun onRenderOverlay(event: GuiRenderEvent.GameOverlayRenderEvent) {
+        if (isEnabled()) {
+            SkyHanniMod.feature.garden.dicerCounterPos.renderStrings(display)
+        }
+    }
+
+    class ItemDrop(val itemName: String, val rarity: DropRarity, val message: String)
+
+    private fun saveConfig() {
+        val list = SkyHanniMod.feature.hidden.gardenDicerRngDrops
+        list.clear()
+        for (drop in drops) {
+            val itemName = drop.key
+            for ((rarity, amount) in drop.value) {
+                list[itemName + "." + rarity.name] = amount
+            }
+        }
+    }
+
+    @SubscribeEvent
+    fun onConfigLoad(event: ConfigLoadEvent) {
+        for ((internalName, amount) in SkyHanniMod.feature.hidden.gardenDicerRngDrops) {
+            val split = internalName.split(".")
+            val itemName = split[0]
+            val rarityName = split[1]
+            val rarity = DropRarity.valueOf(rarityName)
+            drops[itemName]!![rarity] = amount
+        }
+    }
+
+    fun isEnabled() = GardenAPI.inGarden() && SkyHanniMod.feature.garden.dicerCounterDisplay
+}
\ No newline at end of file