package me.kyren223.trident.ui

import com.intellij.openapi.editor.EditorSettings
import com.intellij.openapi.editor.LogicalPosition
import com.intellij.openapi.project.Project
import com.intellij.openapi.ui.DialogWrapper
import com.intellij.ui.EditorTextField
import com.maddyhome.idea.vim.newapi.vim
import com.maddyhome.idea.vim.state.mode.Mode
import me.kyren223.trident.data.SettingsState
import me.kyren223.trident.utils.TridentMappings
import java.awt.event.FocusEvent
import java.awt.event.FocusListener
import javax.swing.JComponent

class TridentMappingsMenu(private val content: String) : DialogWrapper(true) {
    private lateinit var editor: EditorTextField

    init {
        val settings = SettingsState.instance.state
        this.title = "Trident Mappings"
        setSize(settings.width, settings.height)
        super.init()
    }

    override fun createCenterPanel(): JComponent {
        editor = EditorTextField(this.content)
        editor.setOneLineMode(false)
        editor.addSettingsProvider {
            it.setFontSize(SettingsState.instance.fontSize)
            it.isInsertMode = false
            it.caretModel.moveToLogicalPosition(LogicalPosition(0, 0))
            it.settings.isLineNumbersShown = true
            it.settings.lineNumerationType = EditorSettings.LineNumerationType.ABSOLUTE
        }

        editor.addFocusListener(object : FocusListener {
            override fun focusGained(e: FocusEvent?) {
                val editor = editor.editor!!.vim
                editor.mode = Mode.NORMAL()
            }

            override fun focusLost(e: FocusEvent?) { /* Do nothing */ }
        })

        return editor
    }

    override fun getPreferredFocusedComponent(): JComponent {
        return this.editor
    }

    override fun doOKAction() {
        save()
        super.doOKAction()
    }

    override fun doCancelAction() {
        save()
        super.doCancelAction()
    }

    private fun save() {
        map = editor.text.split("\n")
            .map { entry -> entry.split("=") }
            .filter { entry -> entry.size == 2 }
            .associate { it[0] to it[1] }
    }

    override fun createSouthPanel(): JComponent? {
        return null
    }

    companion object {
        var map: Map<String, String>? = null
        fun open(project: Project) {
            val content = TridentMappings.get(project).entries
                .joinToString("\n") { "${it.key}=${it.value}" }
            val menu = TridentMappingsMenu(content)
            menu.show()
            map?.let { TridentMappings.set(project, it) }
            map = null
        }
    }
}