1 files changed, 84 insertions, 0 deletions
diff --git a/src/main/kotlin/me/kyren223/trident/ui/TridentMappingsMenu.kt b/src/main/kotlin/me/kyren223/trident/ui/TridentMappingsMenu.kt
new file mode 100644
index 0000000..a84b561
--- /dev/null
+++ b/src/main/kotlin/me/kyren223/trident/ui/TridentMappingsMenu.kt
@@ -0,0 +1,84 @@
+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 me.kyren223.trident.utils.setEditorMode
+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
+        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
+                setEditorMode(editor, 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() {
+        val map = editor.text.split("\n")
+            .map { entry -> entry.split("=") }
+            .filter { entry -> entry.size == 2 }
+            .associate { it[0] to it[1] }
+        TridentMappings.set(editor.project, map)
+    }
+
+    override fun createSouthPanel(): JComponent? {
+        return null
+    }
+
+    companion object {
+        fun open(project: Project) {
+            val content = TridentMappings.get(project).entries
+                .joinToString("\n") { "${it.key}=${it.value}" }
+            val menu = TridentMappingsMenu(content)
+            menu.show()
+        }
+    }
+}