package moe.nea.firmament.features.diana

import kotlin.time.Duration.Companion.seconds
import net.minecraft.particle.ParticleTypes
import net.minecraft.sound.SoundEvents
import net.minecraft.util.math.Vec3d
import moe.nea.firmament.annotations.Subscribe
import moe.nea.firmament.events.ParticleSpawnEvent
import moe.nea.firmament.events.SoundReceiveEvent
import moe.nea.firmament.events.WorldKeyboardEvent
import moe.nea.firmament.events.WorldReadyEvent
import moe.nea.firmament.events.WorldRenderLastEvent
import moe.nea.firmament.events.subscription.SubscriptionOwner
import moe.nea.firmament.features.FirmamentFeature
import moe.nea.firmament.util.MC
import moe.nea.firmament.util.SBData
import moe.nea.firmament.util.SkyBlockIsland
import moe.nea.firmament.util.SkyblockId
import moe.nea.firmament.util.TimeMark
import moe.nea.firmament.util.WarpUtil
import moe.nea.firmament.util.render.RenderInWorldContext
import moe.nea.firmament.util.skyBlockId

object AncestralSpadeSolver : SubscriptionOwner {
    var lastDing = TimeMark.farPast()
        private set
    private val pitches = mutableListOf<Float>()
    val particlePositions = mutableListOf<Vec3d>()
    var nextGuess: Vec3d? = null
        private set

    val ancestralSpadeId = SkyblockId("ANCESTRAL_SPADE")
    private var lastTeleportAttempt = TimeMark.farPast()

    fun isEnabled() =
        DianaWaypoints.TConfig.ancestralSpadeSolver
            && SBData.skyblockLocation == SkyBlockIsland.HUB
            && MC.player?.inventory?.containsAny { it.skyBlockId == ancestralSpadeId } == true // TODO: add a reactive property here

    @Subscribe
    fun onKeyBind(event: WorldKeyboardEvent) {
        if (!isEnabled()) return
        if (!event.matches(DianaWaypoints.TConfig.ancestralSpadeTeleport)) return

        if (lastTeleportAttempt.passedTime() < 3.seconds) return
        WarpUtil.teleportToNearestWarp(SkyBlockIsland.HUB, nextGuess ?: return)
        lastTeleportAttempt = TimeMark.now()
    }

    @Subscribe
    fun onParticleSpawn(event: ParticleSpawnEvent) {
        if (!isEnabled()) return
        if (event.particleEffect != ParticleTypes.DRIPPING_LAVA) return
        if (event.offset.x != 0.0F || event.offset.y != 0F || event.offset.z != 0F)
            return
        particlePositions.add(event.position)
        if (particlePositions.size > 20) {
            particlePositions.removeFirst()
        }
    }

    @Subscribe
    fun onPlaySound(event: SoundReceiveEvent) {
        if (!isEnabled()) return
        if (!SoundEvents.BLOCK_NOTE_BLOCK_HARP.matchesId(event.sound.value().id)) return

        if (lastDing.passedTime() > 1.seconds) {
            particlePositions.clear()
            pitches.clear()
        }
        lastDing = TimeMark.now()

        pitches.add(event.pitch)
        if (pitches.size > 20) {
            pitches.removeFirst()
        }

        if (particlePositions.size < 3) {
            return
        }

        val averagePitchDelta =
            if (pitches.isEmpty()) return
            else pitches
                .zipWithNext { a, b -> b - a }
                .average()

        val soundDistanceEstimate = (Math.E / averagePitchDelta) - particlePositions.first().distanceTo(event.position)

        if (soundDistanceEstimate > 1000) {
            return
        }

        val lastParticleDirection = particlePositions
            .takeLast(3)
            .let { (a, _, b) -> b.subtract(a) }
            .normalize()

        nextGuess = event.position.add(lastParticleDirection.multiply(soundDistanceEstimate))
    }

    @Subscribe
    fun onWorldRender(event: WorldRenderLastEvent) {
        if (!isEnabled()) return
        RenderInWorldContext.renderInWorld(event) {
            nextGuess?.let {
                color(1f, 1f, 0f, 0.5f)
                tinyBlock(it, 1f)
                color(1f, 1f, 0f, 1f)
                tracer(it, lineWidth = 3f)
            }
            if (particlePositions.size > 2 && lastDing.passedTime() < 10.seconds && nextGuess != null) {
                color(0f, 1f, 0f, 0.7f)
                line(particlePositions)
            }
        }
    }

    @Subscribe
    fun onSwapWorld(event: WorldReadyEvent) {
        nextGuess = null
        particlePositions.clear()
        pitches.clear()
        lastDing = TimeMark.farPast()
    }

    override val delegateFeature: FirmamentFeature
        get() = DianaWaypoints

}