aboutsummaryrefslogtreecommitdiff
path: root/src/main/java/at/hannibal2/skyhanni/features/misc/EnderNodeTracker.kt
blob: 133e12ab4689ece0ace34cab67e563e41d5acbde (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
package at.hannibal2.skyhanni.features.misc

import at.hannibal2.skyhanni.SkyHanniMod
import at.hannibal2.skyhanni.config.Storage
import at.hannibal2.skyhanni.data.IslandType
import at.hannibal2.skyhanni.data.ProfileStorageData
import at.hannibal2.skyhanni.events.*
import at.hannibal2.skyhanni.utils.ItemUtils.getInternalNameOrNull
import at.hannibal2.skyhanni.utils.LorenzUtils
import at.hannibal2.skyhanni.utils.LorenzUtils.addAsSingletonList
import at.hannibal2.skyhanni.utils.LorenzUtils.afterChange
import at.hannibal2.skyhanni.utils.LorenzUtils.editCopy
import at.hannibal2.skyhanni.utils.NEUItems.getNpcPriceOrNull
import at.hannibal2.skyhanni.utils.NEUItems.getPriceOrNull
import at.hannibal2.skyhanni.utils.NumberUtil.addSeparators
import at.hannibal2.skyhanni.utils.NumberUtil.format
import at.hannibal2.skyhanni.utils.RenderUtils.renderStringsAndItems
import io.github.moulberry.notenoughupdates.util.MinecraftExecutor
import net.minecraft.client.Minecraft
import net.minecraftforge.fml.common.eventhandler.SubscribeEvent

class EnderNodeTracker {
    private val config get() = SkyHanniMod.feature.misc.enderNodeTracker
    private val storage get() = ProfileStorageData.profileSpecific?.enderNodeTracker

    private var totalEnderArmor = 0
    private var miteGelInInventory = 0
    private var display = emptyList<List<Any>>()
    private var lootProfit = mapOf<EnderNode, Double>()

    private val enderNodeRegex = Regex("""ENDER NODE!.+You found (\d+x )?§r(.+)§r§f!""")
    private val endermanRegex = Regex("""(RARE|PET) DROP! §r(.+) §r§b\(""")

    @SubscribeEvent
    fun onChat(event: LorenzChatEvent) {
        if (!config.enabled) return
        if (!ProfileStorageData.loaded) return
        if (!isInTheEnd()) return

        // don't call removeColor because we want to distinguish enderman pet rarity
        val message = event.message.trim()
        var item: String? = null
        var amount = 1
        val storage = storage ?: return

        // check whether the loot is from an ender node or an enderman
        enderNodeRegex.find(message)?.let {
            storage.totalNodesMined++
            amount = it.groups[1]?.value?.substringBefore("x")?.toIntOrNull() ?: 1
            item = it.groups[2]?.value
        } ?: endermanRegex.find(message)?.let {
            amount = 1
            item = it.groups[2]?.value
        }

        when {
            item == null -> return
            isEnderArmor(item) -> totalEnderArmor++
            item == "§cEndermite Nest" -> {
                storage.totalEndermiteNests++
            }
        }

        // increment the count of the specific item found
        EnderNode.entries.find { it.displayName == item }?.let {
            val old = storage.lootCount[it] ?: 0
            storage.lootCount = storage.lootCount.editCopy {
                this[it] = old + amount
            }
        }
        update()
    }

    @SubscribeEvent
    fun onIslandChange(event: IslandChangeEvent) {
        if (!config.enabled) return
        if (event.newIsland != IslandType.THE_END) return
        miteGelInInventory = Minecraft.getMinecraft().thePlayer.inventory.mainInventory
            .filter { it?.getInternalNameOrNull() == EnderNode.MITE_GEL.internalName }
            .sumOf { it.stackSize }
    }

    @SubscribeEvent
    fun onSackChange(event: SackChangeEvent) {
        if (!config.enabled) return
        if (!ProfileStorageData.loaded) return
        if (!isInTheEnd()) return
        val storage = storage ?: return

        val change = event.sackChanges
            .firstOrNull { it.internalName == EnderNode.MITE_GEL.internalName && it.delta > 0 }
            ?: return
        val old = storage.lootCount[EnderNode.MITE_GEL] ?: 0
        storage.lootCount = storage.lootCount.editCopy {
            this[EnderNode.MITE_GEL] = old + change.delta
        }
        update()
    }

    @SubscribeEvent
    fun onInventoryUpdate(event: OwnInventoryItemUpdateEvent) {
        if (!config.enabled) return
        if (!isInTheEnd()) return
        if (!ProfileStorageData.loaded) return
        val storage = storage ?: return

        MinecraftExecutor.OnThread.execute {
            val newMiteGelInInventory = Minecraft.getMinecraft().thePlayer.inventory.mainInventory
                .filter { it?.getInternalNameOrNull() == EnderNode.MITE_GEL.internalName }
                .sumOf { it.stackSize }
            val change = newMiteGelInInventory - miteGelInInventory
            if (change > 0) {
                val old = storage.lootCount[EnderNode.MITE_GEL] ?: 0
                storage.lootCount = storage.lootCount.editCopy {
                    this[EnderNode.MITE_GEL] = old + change
                }
                update()
            }
            miteGelInInventory = newMiteGelInInventory
        }
    }

    @SubscribeEvent
    fun onRenderOverlay(event: GuiRenderEvent.GameOverlayRenderEvent) {
        if (!config.enabled) return
        if (!isInTheEnd()) return
        config.position.renderStringsAndItems(display, posLabel = "Ender Node Tracker")
    }

    @SubscribeEvent
    fun onConfigLoad(event: ConfigLoadEvent) {
        config.textFormat.afterChange {
            update()
        }
        val storage = storage ?: return

        totalEnderArmor = storage.lootCount.filter { isEnderArmor(it.key.displayName) }
            .map { it.value }
            .sum()
        update()
    }

    private fun calculateProfit(storage: Storage.ProfileSpecific.EnderNodeTracker): Map<EnderNode, Double> {
        if (!ProfileStorageData.loaded) return emptyMap()

        val newProfit = mutableMapOf<EnderNode, Double>()
        storage.lootCount.forEach { (item, amount) ->
            val price = if (isEnderArmor(item.displayName)) {
                10_000.0
            } else {
                (if (!LorenzUtils.noTradeMode) item.internalName.getPriceOrNull() else 0.0)
                    ?.coerceAtLeast(item.internalName.getNpcPriceOrNull() ?: 0.0)
                    ?.coerceAtLeast(georgePrice(item) ?: 0.0)
                    ?: 0.0
            }
            newProfit[item] = price * amount
        }
        return newProfit
    }

    private fun update() {
        val storage = storage ?: return
        lootProfit = calculateProfit(storage)
        display = formatDisplay(drawDisplay(storage))
    }

    private fun isInTheEnd() = LorenzUtils.skyBlockArea == "The End"

    private fun isEnderArmor(displayName: String?) = when (displayName) {
        EnderNode.END_HELMET.displayName,
        EnderNode.END_CHESTPLATE.displayName,
        EnderNode.END_LEGGINGS.displayName,
        EnderNode.END_BOOTS.displayName,
        EnderNode.ENDER_NECKLACE.displayName,
        EnderNode.ENDER_GAUNTLET.displayName -> true

        else -> false
    }

    private fun georgePrice(petRarity: EnderNode): Double? = when (petRarity) {
        EnderNode.COMMON_ENDERMAN_PET -> 100.0
        EnderNode.UNCOMMON_ENDERMAN_PET -> 500.0
        EnderNode.RARE_ENDERMAN_PET -> 2_000.0
        EnderNode.EPIC_ENDERMAN_PET -> 10_000.0
        EnderNode.LEGENDARY_ENDERMAN_PET -> 1_000_000.0
        else -> null
    }

    private fun drawDisplay(storage: Storage.ProfileSpecific.EnderNodeTracker) = buildList<List<Any>> {
        if (!ProfileStorageData.loaded) return emptyList<List<Any>>()

        addAsSingletonList("§5§lEnder Node Tracker")
        addAsSingletonList("§d${storage.totalNodesMined.addSeparators()} Ender Nodes mined")
        addAsSingletonList("§6${format(lootProfit.values.sum())} Coins made")
        addAsSingletonList(" ")
        addAsSingletonList("§b${storage.totalEndermiteNests.addSeparators()} §cEndermite Nest")

        for (item in EnderNode.entries.subList(0, 11)) {
            val count = (storage.lootCount[item] ?: 0).addSeparators()
            val profit = format(lootProfit[item] ?: 0.0)
            addAsSingletonList("§b$count ${item.displayName} §7(§6$profit§7)")
        }
        addAsSingletonList(" ")
        addAsSingletonList(
            "§b${totalEnderArmor.addSeparators()} §5Ender Armor " +
                    "§7(§6${format(totalEnderArmor * 10_000)}§7)"
        )
        for