path: root/src/main/java/at/hannibal2/skyhanni/features/mining/powdertracker/PowderTracker.kt

package at.hannibal2.skyhanni.features.mining.powdertracker

import at.hannibal2.skyhanni.SkyHanniMod
import at.hannibal2.skyhanni.config.ConfigUpdaterMigrator
import at.hannibal2.skyhanni.config.features.mining.PowderTrackerConfig.PowderDisplayEntry
import at.hannibal2.skyhanni.data.BossbarData
import at.hannibal2.skyhanni.data.IslandType
import at.hannibal2.skyhanni.data.model.TabWidget
import at.hannibal2.skyhanni.events.ConfigLoadEvent
import at.hannibal2.skyhanni.events.GuiRenderEvent
import at.hannibal2.skyhanni.events.IslandChangeEvent
import at.hannibal2.skyhanni.events.LorenzChatEvent
import at.hannibal2.skyhanni.events.LorenzWorldChangeEvent
import at.hannibal2.skyhanni.events.SecondPassedEvent
import at.hannibal2.skyhanni.skyhannimodule.SkyHanniModule
import at.hannibal2.skyhanni.utils.CollectionUtils.addSearchString
import at.hannibal2.skyhanni.utils.ConditionalUtils.afterChange
import at.hannibal2.skyhanni.utils.ConfigUtils
import at.hannibal2.skyhanni.utils.LorenzUtils.isInIsland
import at.hannibal2.skyhanni.utils.NumberUtil.addSeparators
import at.hannibal2.skyhanni.utils.NumberUtil.formatLong
import at.hannibal2.skyhanni.utils.RegexUtils.groupOrNull
import at.hannibal2.skyhanni.utils.RegexUtils.matchMatcher
import at.hannibal2.skyhanni.utils.SimpleTimeMark
import at.hannibal2.skyhanni.utils.TimeUtils
import at.hannibal2.skyhanni.utils.renderables.Searchable
import at.hannibal2.skyhanni.utils.repopatterns.RepoPattern
import at.hannibal2.skyhanni.utils.tracker.SkyHanniTracker
import at.hannibal2.skyhanni.utils.tracker.TrackerData
import com.google.gson.JsonArray
import com.google.gson.JsonNull
import com.google.gson.annotations.Expose
import net.minecraftforge.fml.common.eventhandler.SubscribeEvent
import kotlin.time.Duration
import kotlin.time.Duration.Companion.minutes

@SkyHanniModule
object PowderTracker {

    private val config get() = SkyHanniMod.feature.mining.powderTracker

    private val patternGroup = RepoPattern.group("mining.powder.tracker")
    private val pickedPattern by patternGroup.pattern(
        "picked",
        "  §r§6§lCHEST LOCKPICKED ",
    )
    private val powderStartedPattern by patternGroup.pattern(
        "powder.started",
        ".*§r§b§l2X POWDER STARTED!.*",
    )
    private val powderEndedPattern by patternGroup.pattern(
        "powder.ended",
        ".*§r§b§l2X POWDER ENDED!.*",
    )
    private val powderBossBarPattern by patternGroup.pattern(
        "powder.bossbar",
        "§e§lPASSIVE EVENT §b§l2X POWDER §e§lRUNNING FOR §a§l(?<time>.*)§r",
    )

    /**
     * REGEX-TEST: Ends in: §r§b5m 27s
     */
    private val tablistEventDuration by patternGroup.pattern(
        "powder.duration",
        "Ends in: §r§b(?<duration>.*)",
    )

    /**
     * REGEX-TEST: §b§lCOMPACT! §r§fYou found an §r§aEnchanted Hard Stone§r§f!
     */
    private val compactedPattern by patternGroup.pattern(
        "compacted",
        "§b§lCOMPACT! §r§fYou found an §r§aEnchanted Hard Stone§r§f!",
    )

    private var lastChestPicked = SimpleTimeMark.farPast()
    private var isGrinding = false
    private val gemstoneInfo = ResourceInfo(0L, 0L, 0, 0.0, mutableListOf())
    private val mithrilInfo = ResourceInfo(0L, 0L, 0, 0.0, mutableListOf())
    private val diamondEssenceInfo = ResourceInfo(0L, 0L, 0, 0.0, mutableListOf())
    private val goldEssenceInfo = ResourceInfo(0L, 0L, 0, 0.0, mutableListOf())
    private val chestInfo = ResourceInfo(0L, 0L, 0, 0.0, mutableListOf())
    private val hardStoneInfo = ResourceInfo(0L, 0L, 0, 0.0, mutableListOf())
    private var doublePowder = false
    private var powderTimer = Duration.ZERO
    private var eventEnded = false
    private val gemstones = listOf(
        "Ruby" to "§c",
        "Sapphire" to "§b",
        "Amber" to "§6",
        "Amethyst" to "§5",
        "Jade" to "§a",
        "Topaz" to "§e",
    )

    init {
        PowderChestReward.entries.forEach { it.chatPattern }
    }

    @SubscribeEvent
    fun onSecondPassed(event: SecondPassedEvent) {
        if (!isEnabled()) return
        calculateResourceHour(gemstoneInfo)
        calculateResourceHour(mithrilInfo)
        calculateResourceHour(diamondEssenceInfo)
        calculateResourceHour(goldEssenceInfo)
        calculateResourceHour(chestInfo)
        calculateResourceHour(hardStoneInfo)

        if (TabWidget.EVENT.isActive) {
            for ((index, line) in TabWidget.EVENT.lines.withIndex()) {
                if (line.contains("2x Powder")) {
                    if (eventEnded) return
                    doublePowder = true
                    val durationLine = TabWidget.EVENT.lines.getOrNull(index + 1) ?: return
                    tablistEventDuration.matchMatcher(durationLine) {
                        val duration = group("duration")
                        powderTimer = TimeUtils.getDuration(duration)
                        tracker.update()
                    }
                }
            }
        } else {
            powderBossBarPattern.matchMatcher(BossbarData.getBossbar()) {
                val duration = group("time")
                powderTimer = TimeUtils.getDuration(duration)

                doublePowder = powderTimer > Duration.ZERO
                tracker.update()
            }
        }

        if (lastChestPicked.passedSince() > 1.minutes) {
            isGrinding = false
        }
    }

    private val tracker = SkyHanniTracker("Powder Tracker", { Data() }, { it.powderTracker }) { formatDisplay(drawDisplay(it)) }

    class Data : TrackerData() {

        override fun reset() {
            rewards.clear()
            totalChestPicked = 0
            totalHardStoneCompacted = 0
        }

        @Expose
        var totalChestPicked = 0

        @Expose
        var totalHardStoneCompacted = 0

        @Expose
        var rewards: MutableMap<PowderChestReward, Long> = mutableMapOf()
    }

    @SubscribeEvent
    fun onRenderOverlay(event: GuiRenderEvent) {
        if (!isEnabled()) return

        if (config.onlyWhenPowderGrinding && !isGrinding) return

        tracker.renderDisplay(config.position)
    }

    @SubscribeEvent
    fun onChat(event: LorenzChatEvent) {
        if (!isEnabled()) return
        val msg = event.message

        pickedPattern.matchMatcher(msg) {
            tracker.modify {
                it.totalChestPicked += 1
            }
            isGrinding = true
            lastChestPicked = SimpleTimeMark.now()
        }

        powderStartedPattern.matchMatcher(msg) {
            doublePowder = true
            eventEnded = false
        }
        powderEndedPattern.matchMatcher(msg) {
            doublePowder = false
            eventEnded = true
        }

        compactedPattern.matchMatcher(msg) {
            tracker.modify {
                it.totalHardStoneCompacted += 1
            }
        }

        for (reward in PowderChestReward.entries) {
            reward.chatPattern.matchMatcher(msg) {
                tracker.modify {
                    val count = it.rewards[reward] ?: 0
                    var amount = groupOrNull("amount")?.formatLong() ?: 1
                    if ((reward == PowderChestReward.MITHRIL_POWDER || reward == PowderChestReward.GEMSTONE_POWDER) && doublePowder) {
                        amount *= 2
                    }
                    it.rewards[reward] = count + amount
                }
            }
        }
        tracker.update()
    }

    @SubscribeEvent
    fun onConfigLoad(event: ConfigLoadEvent) {
        config.textFormat.afterChange {
            tracker.update()
        }
    }

    @SubscribeEvent
    fun onWorldChange(event: LorenzWorldChangeEvent) {
        if (!isEnabled()) return
        gemstoneInfo.perHour = 0.0
        gemstoneInfo.stoppedChecks = 0
        gemstoneInfo.perMin.clear()
        mithrilInfo.perHour = 0.0
        mithrilInfo.stoppedChecks = 0
        mithrilInfo.perMin.clear()
        diamondEssenceInfo.perHour = 0.0
        diamondEssenceInfo.stoppedChecks = 0
        diamondEssenceInfo.perMin.clear()
        goldEssenceInfo.perHour = 0.0
        goldEssenceInfo.stoppedChecks = 0
        goldEssenceInfo.perMin.clear()
        chestInfo.perHour = 0.0
        chestInfo.stoppedChecks = 0
        chestInfo.perMin.clear()
        hardStoneInfo