feat: add include file in context to editor context menu (#475)

* feat: add include file in context to editor context menu * fix: custom title for IncludeFilesInContextAction in editor context menu
carlrobertoh · Apr 18, 2024 · 9666590 · 9666590
1 parent 29b36c5
commit 9666590
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 5 deletions.
diff --git a/src/main/java/ee/carlrobert/codegpt/actions/IncludeFilesInContextAction.java b/src/main/java/ee/carlrobert/codegpt/actions/IncludeFilesInContextAction.java
@@ -52,6 +52,10 @@ public IncludeFilesInContextAction() {
     super(CodeGPTBundle.get("action.includeFilesInContext.title"));
   }
 
+  public IncludeFilesInContextAction(String customTitleKey) {
+    super(CodeGPTBundle.get(customTitleKey));
+  }
+
   @Override
   public void actionPerformed(@NotNull AnActionEvent e) {
     var project = e.getProject();
@@ -93,11 +97,6 @@ public void nodeStateChanged(@NotNull CheckedTreeNode node) {
   }
 
   private @Nullable FileCheckboxTree getCheckboxTree(DataContext dataContext) {
-    var psiElement = CommonDataKeys.PSI_ELEMENT.getData(dataContext);
-    if (psiElement != null) {
-      return new PsiElementCheckboxTree(psiElement);
-    }
-
     var selectedVirtualFiles = VIRTUAL_FILE_ARRAY.getData(dataContext);
     if (selectedVirtualFiles != null) {
       return new VirtualFileCheckboxTree(selectedVirtualFiles);

diff --git a/src/main/java/ee/carlrobert/codegpt/actions/editor/EditorActionsUtil.java b/src/main/java/ee/carlrobert/codegpt/actions/editor/EditorActionsUtil.java
@@ -11,6 +11,7 @@
 import com.intellij.openapi.project.Project;
 import ee.carlrobert.codegpt.CodeGPTKeys;
 import ee.carlrobert.codegpt.ReferencedFile;
+import ee.carlrobert.codegpt.actions.IncludeFilesInContextAction;
 import ee.carlrobert.codegpt.conversations.message.Message;
 import ee.carlrobert.codegpt.settings.configuration.ConfigurationSettings;
 import ee.carlrobert.codegpt.toolwindow.chat.ChatToolWindowContentManager;
@@ -75,6 +76,8 @@ protected void actionPerformed(Project project, Editor editor, String selectedTe
         };
         group.add(action);
       });
+      group.addSeparator();
+      group.add(new IncludeFilesInContextAction("action.includeFileInContext.title"));
     }
   }
 

diff --git a/src/main/resources/messages/codegpt.properties b/src/main/resources/messages/codegpt.properties
@@ -5,6 +5,7 @@ action.generateCommitMessage.description=Generate commit message
 action.generateCommitMessage.serviceWarning=Messages can only be generated with OpenAI or Azure service
 action.generateCommitMessage.missingCredentials=Credentials not provided
 action.includeFilesInContext.title=Include In Context...
+action.includeFileInContext.title=Include File In Context...
 action.includeFilesInContext.dialog.title=Include In Context
 action.includeFilesInContext.dialog.description=Choose the files that you wish to include in the final prompt
 action.includeFilesInContext.dialog.repeatableContext.label=Repeatable context: