package at.hannibal2.skyhanni.features.dungeon

import at.hannibal2.skyhanni.SkyHanniMod
import at.hannibal2.skyhanni.data.IslandType
import at.hannibal2.skyhanni.data.mob.Mob
import at.hannibal2.skyhanni.data.mob.MobData
import at.hannibal2.skyhanni.events.ConfigLoadEvent
import at.hannibal2.skyhanni.events.LorenzRenderWorldEvent
import at.hannibal2.skyhanni.events.LorenzTickEvent
import at.hannibal2.skyhanni.events.MobEvent
import at.hannibal2.skyhanni.skyhannimodule.SkyHanniModule
import at.hannibal2.skyhanni.utils.ColorUtils.toChromaColor
import at.hannibal2.skyhanni.utils.ConditionalUtils.onToggle
import at.hannibal2.skyhanni.utils.LorenzUtils.isInIsland
import at.hannibal2.skyhanni.utils.RenderUtils.draw3DLine
import at.hannibal2.skyhanni.utils.RenderUtils.drawWaypointFilled
import at.hannibal2.skyhanni.utils.RenderUtils.exactPlayerEyeLocation
import at.hannibal2.skyhanni.utils.getLorenzVec
import net.minecraftforge.fml.common.eventhandler.SubscribeEvent
import java.awt.Color

@SkyHanniModule
object DungeonMobManager {

    private val config get() = SkyHanniMod.feature.dungeon.objectHighlighter
    private val starredConfig get() = config.starred
    private val fel get() = config.fel

    private val staredInvisible = mutableSetOf<Mob>()
    private val felOnTheGround = mutableSetOf<Mob>()

    @SubscribeEvent
    fun onConfigLoad(event: ConfigLoadEvent) {
        onToggle(
            starredConfig.highlight,
            starredConfig.colour,
        ) {
            val color = if (starredConfig.highlight.get()) getStarColor() else null
            MobData.skyblockMobs.filter { it.hasStar }.forEach {
                handleStar0(it, color)
            }
            if (!starredConfig.highlight.get()) {
                staredInvisible.clear()
            }
        }
        onToggle(
            fel.highlight,
            fel.colour,
        ) {
            if (fel.highlight.get()) {
                if (felOnTheGround.isEmpty()) {
                    MobData.skyblockMobs.forEach(::handleFel)
                }
            } else {
                felOnTheGround.clear()
            }
        }
    }

    @SubscribeEvent
    fun onMobSpawn(event: MobEvent.Spawn.SkyblockMob) {
        if (event.mob.mobType != Mob.Type.DUNGEON) return
        handleStar(event.mob)
        handleFel(event.mob)
    }

    @SubscribeEvent
    fun onMobDeSpawn(event: MobEvent.DeSpawn.SkyblockMob) {
        if (event.mob.mobType != Mob.Type.DUNGEON) return
        if (starredConfig.highlight.get()) {
            staredInvisible.remove(event.mob)
        }
        handleFelDespawn(event.mob)
    }

    @SubscribeEvent
    fun onLorenzTick(event: LorenzTickEvent) {
        if (!IslandType.CATACOMBS.isInIsland()) return
        handleInvisibleStar()
    }

    @SubscribeEvent
    fun onRenderWorld(event: LorenzRenderWorldEvent) {
        if (!fel.highlight.get()) return
        if (fel.line) {
            felOnTheGround.filter { it.canBeSeen(30) }.forEach {
                event.draw3DLine(
                    it.baseEntity.getLorenzVec().add(y = 0.15),
                    event.exactPlayerEyeLocation(),
                    fel.colour.get().toChromaColor(),
                    3,
                    true,
                )
            }
        }

        felOnTheGround.removeIf { mob ->
            event.drawWaypointFilled(
                mob.baseEntity.getLorenzVec().add(-0.5, -0.23, -0.5),
                fel.colour.get().toChromaColor(),
                seeThroughBlocks = false,
                beacon = false,
                extraSize = -0.2,
                minimumAlpha = 0.8f,
                inverseAlphaScale = true,
            )
            !mob.isInvisible()
        }
    }

    private fun handleStar(mob: Mob) {
        if (!starredConfig.highlight.get()) return
        if (!mob.hasStar) return
        handleStar0(mob, getStarColor())
    }

    private fun handleInvisibleStar() {
        if (!starredConfig.highlight.get()) return
        staredInvisible.removeIf {
            val visible = !it.isInvisible()
            if (visible) {
                it.highlight(getStarColor())
            }
            visible
        }
    }

    private fun getStarColor(): Color = starredConfig.colour.get().toChromaColor()

    private fun handleStar0(mob: Mob, colour: Color?) {
        if (mob.isInvisible()) {
            staredInvisible.add(mob)
            return
        }
        mob.highlight(colour)
    }

    private fun handleFel(mob: Mob) {
        if (!fel.highlight.get()) return
        if (mob.name != "Fels") return
        if (!mob.isInvisible()) return
        felOnTheGround.add(mob)
    }

    private fun handleFelDespawn(mob: Mob) {
        if (!fel.highlight.get()) return
        felOnTheGround.remove(mob)
    }
}