1 files changed, 31 insertions, 0 deletions
diff --git a/src/main/kotlin/me/kyren223/trident/ToggleQuickMenuAction.kt b/src/main/kotlin/me/kyren223/trident/ToggleQuickMenuAction.kt
new file mode 100644
index 0000000..a8a78ab
--- /dev/null
+++ b/src/main/kotlin/me/kyren223/trident/ToggleQuickMenuAction.kt
@@ -0,0 +1,31 @@
+package me.kyren223.trident
+
+import com.intellij.openapi.actionSystem.AnAction
+import com.intellij.openapi.actionSystem.AnActionEvent
+import com.maddyhome.idea.vim.api.injector
+import com.maddyhome.idea.vim.command.MappingMode
+import com.maddyhome.idea.vim.key.MappingOwner
+
+class ToggleQuickMenuAction : AnAction() {
+    override fun actionPerformed(e: AnActionEvent) {
+        if (!remapped) remap()
+        if (e.project != null) {
+            TridentState.toggleQuickMenu(e.project!!)
+        }
+    }
+
+    private fun remap() {
+        if (!SettingsState.instance.enterToSelect) return
+        val keys = injector.parser.parseKeys(":action TridentQuickMenuSelect<cr>")
+        val keyGroup = injector.keyGroup
+        keyGroup.putKeyMapping(MappingMode.NVO,
+                injector.parser.parseKeys("<cr>"),
+                MappingOwner.Plugin.get("Trident"), keys, false
+        )
+        remapped = true
+    }
+
+    companion object {
+        var remapped = false
+    }
+}