package moe.nea.firmament.features.chat

import com.mojang.brigadier.arguments.StringArgumentType.string
import moe.nea.firmament.annotations.Subscribe
import moe.nea.firmament.commands.get
import moe.nea.firmament.commands.suggestsList
import moe.nea.firmament.commands.thenArgument
import moe.nea.firmament.commands.thenExecute
import moe.nea.firmament.events.CommandEvent
import moe.nea.firmament.events.MaskCommands
import moe.nea.firmament.features.FirmamentFeature
import moe.nea.firmament.gui.config.ManagedConfig
import moe.nea.firmament.repo.RepoManager
import moe.nea.firmament.util.MC

object AutoCompletions : FirmamentFeature {

    object TConfig : ManagedConfig(identifier) {
        val provideWarpTabCompletion by toggle("warp-complete") { true }
        val replaceWarpIsByWarpIsland by toggle("warp-is") { true }
    }

    override val config: ManagedConfig?
        get() = TConfig
    override val identifier: String
        get() = "auto-completions"

    @Subscribe
    fun onMaskCommands(event: MaskCommands) {
        if (TConfig.provideWarpTabCompletion) {
            event.mask("warp")
        }
    }

    @Subscribe
    fun onCommandEvent(event: CommandEvent) {
        if (!TConfig.provideWarpTabCompletion) return
        event.deleteCommand("warp")
        event.register("warp") {
            thenArgument("to", string()) { toArg ->
                suggestsList {
                    RepoManager.neuRepo.constants?.islands?.warps?.flatMap { listOf(it.warp) + it.aliases } ?: listOf()
                }
                thenExecute {
                    val warpName = get(toArg)
                    if (warpName == "is" && TConfig.replaceWarpIsByWarpIsland) {
                        MC.sendServerCommand("warp island")
                    } else {
                        MC.sendServerCommand("warp $warpName")
                    }
                }
            }
        }
    }
}