From db3f4ac98c0125183042b4c7b0829eadd833eee4 Mon Sep 17 00:00:00 2001
From: Carl-Robert Linnupuu <carlrobertoh@gmail.com>
Date: Sun, 30 Jun 2024 01:15:00 +0300
Subject: [PATCH] feat: add default keymap for edit code action

---
 .../codegpt/actions/editor/EditCodeAction.kt         | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/main/kotlin/ee/carlrobert/codegpt/actions/editor/EditCodeAction.kt b/src/main/kotlin/ee/carlrobert/codegpt/actions/editor/EditCodeAction.kt
index 803c565e..4c9220e7 100644
--- a/src/main/kotlin/ee/carlrobert/codegpt/actions/editor/EditCodeAction.kt
+++ b/src/main/kotlin/ee/carlrobert/codegpt/actions/editor/EditCodeAction.kt
@@ -1,11 +1,15 @@
 package ee.carlrobert.codegpt.actions.editor
 
+import com.intellij.openapi.actionSystem.CustomShortcutSet
 import com.intellij.openapi.application.runInEdt
 import com.intellij.openapi.editor.Editor
 import com.intellij.openapi.project.Project
 import ee.carlrobert.codegpt.Icons
 import ee.carlrobert.codegpt.ui.EditCodePopover
+import java.awt.event.InputEvent
+import java.awt.event.KeyEvent
 import javax.swing.Icon
+import javax.swing.KeyStroke
 
 class EditCodeAction : BaseEditorAction {
 
@@ -16,6 +20,14 @@ class EditCodeAction : BaseEditorAction {
         "Allow LLM to edit code directly in your editor",
         icon
     ) {
+        registerCustomShortcutSet(
+            CustomShortcutSet(
+                KeyStroke.getKeyStroke(
+                    KeyEvent.VK_E,
+                    InputEvent.SHIFT_DOWN_MASK or InputEvent.META_DOWN_MASK
+                )
+            ), null
+        )
         EditorActionsUtil.registerAction(this)
     }