Add AskIssueIntention

devchat-ai · Jul 19, 2024 · a5fb4cc · a5fb4cc
1 parent b29e221
commit a5fb4cc
Show file tree

Hide file tree

Showing 5 changed files with 50 additions and 2 deletions.
diff --git a/src/main/kotlin/ai/devchat/plugin/actions/AskIssueIntention.kt b/src/main/kotlin/ai/devchat/plugin/actions/AskIssueIntention.kt
@@ -0,0 +1,39 @@
+package ai.devchat.plugin.actions
+
+import ai.devchat.core.DevChatActions
+import ai.devchat.core.handlers.SendUserMessageHandler
+import ai.devchat.plugin.DevChatToolWindow
+import com.alibaba.fastjson.JSONObject
+import com.intellij.codeInsight.intention.IntentionAction
+import com.intellij.codeInsight.intention.PriorityAction
+import com.intellij.openapi.editor.Editor
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.wm.ToolWindowManager
+import com.intellij.psi.PsiFile
+
+class AskIssueIntention : IntentionAction, PriorityAction {
+    override fun getText(): String = "Ask DevChat"
+    override fun getFamilyName(): String = "DevChat"
+    override fun getPriority(): PriorityAction.Priority = PriorityAction.Priority.HIGH
+    override fun isAvailable(project: Project, editor: Editor?, file: PsiFile?): Boolean {
+        return true
+    }
+    override fun invoke(project: Project, editor: Editor?, file: PsiFile?) {
+        editor?.let{
+            val line = it.document.getLineNumber(it.caretModel.offset)
+            val lineStartOffset: Int = it.document.getLineStartOffset(line)
+            val lineEndOffset: Int = it.document.getLineEndOffset(line)
+            it.selectionModel.setSelection(lineStartOffset, lineEndOffset)
+            val payload = JSONObject(mapOf("message" to "/ask_issue"))
+
+            ToolWindowManager.getInstance(editor.project!!).getToolWindow("DevChat")?.show {
+                if (DevChatToolWindow.loaded) {
+                    SendUserMessageHandler(DevChatActions.SEND_USER_MESSAGE_REQUEST,null, payload).executeAction()
+                } else {
+                    SendUserMessageHandler.cache = payload
+                }
+            }
+        }
+    }
+    override fun startInWriteAction(): Boolean = false
+}
diff --git a/...in/ai/devchat/plugin/actions/Intention.kt → ...vchat/plugin/actions/FixIssueIntention.kt b/...in/ai/devchat/plugin/actions/Intention.kt → ...vchat/plugin/actions/FixIssueIntention.kt
@@ -11,7 +11,7 @@ import com.intellij.openapi.project.Project
 import com.intellij.openapi.wm.ToolWindowManager
 import com.intellij.psi.PsiFile
 
-class Intention : IntentionAction, PriorityAction {
+class FixIssueIntention : IntentionAction, PriorityAction {
     override fun getText(): String = "Fix using DevChat"
     override fun getFamilyName(): String = "DevChat"
     override fun getPriority(): PriorityAction.Priority = PriorityAction.Priority.HIGH

diff --git a/src/main/resources/META-INF/plugin.xml b/src/main/resources/META-INF/plugin.xml
@@ -51,7 +51,11 @@
         <editorFactoryListener implementation="ai.devchat.plugin.completion.editor.EditorListener" />
         <actionPromoter order="last" implementation="ai.devchat.plugin.completion.editor.EditorActionPromoter"/>
         <intentionAction>
-            <className>ai.devchat.plugin.actions.Intention</className>
+            <className>ai.devchat.plugin.actions.FixIssueIntention</className>
+            <category>DevChat</category>
+        </intentionAction>
+        <intentionAction>
+            <className>ai.devchat.plugin.actions.AskIssueIntention</className>
             <category>DevChat</category>
         </intentionAction>
     </extensions>

diff --git a/src/main/resources/intentionDescriptions/AskIssueIntention/description.html b/src/main/resources/intentionDescriptions/AskIssueIntention/description.html
@@ -0,0 +1,5 @@
+<html>
+<body>
+Ask about this issue with DevChat.
+</body>
+</html>
diff --git a/...onDescriptions/Intention/description.html → ...ptions/FixIssueIntention/description.html b/...onDescriptions/Intention/description.html → ...ptions/FixIssueIntention/description.html