refactor(custom): extract for range

Author: phodal

src/main/kotlin/cc/unitmesh/devti/custom/CustomIntention.kt

@@ -1,13 +1,23 @@
 package cc.unitmesh.devti.custom
 
+import cc.unitmesh.devti.gui.chat.ChatActionType
 import cc.unitmesh.devti.intentions.AbstractChatIntention
+import com.intellij.openapi.editor.Editor
+import com.intellij.openapi.project.Project
+import com.intellij.psi.PsiFile
 
-class CustomIntention(val intentionConfig: CustomIntentionConfig) : AbstractChatIntention() {
+class CustomIntention(private val intentionConfig: CustomIntentionConfig) : AbstractChatIntention() {
 
     override fun getText(): String = intentionConfig.title
 
     override fun getFamilyName(): String = "Custom Intention"
 
+    override fun getActionType(): ChatActionType = super.getActionType()
+
+    override fun invoke(project: Project, editor: Editor?, file: PsiFile?) {
+
+    }
+
     companion object {
         fun create(intentionConfig: CustomIntentionConfig): CustomIntention {
             return CustomIntention(intentionConfig)

src/main/kotlin/cc/unitmesh/devti/gui/chat/ChatActionType.kt

@@ -17,7 +17,9 @@ enum class ChatActionType {
     FIX_ISSUE,
     CREATE_DDL,
     CREATE_CHANGELOG,
-    CUSTOM_COMPLETE;
+    CUSTOM_COMPLETE,
+    CUSTOM_ACTION
+    ;
 
     override fun toString(): String {
         return instruction()
@@ -83,6 +85,7 @@ $diff
             CREATE_CHANGELOG -> "generate release note"
             CHAT -> ""
             CUSTOM_COMPLETE -> ""
+            CUSTOM_ACTION -> ""
             EXPLAIN_BUSINESS -> "Recover the original business scene and functionality of the provided code by describing it in a User Story format. Ensure that your explanation avoids any technical jargon or technology-related terms."
         }
     }

src/main/kotlin/cc/unitmesh/devti/intentions/AbstractChatIntention.kt

@@ -10,6 +10,7 @@ import com.intellij.codeInsight.intention.IntentionAction
 import com.intellij.lang.injection.InjectedLanguageManager
 import com.intellij.openapi.editor.Editor
 import com.intellij.openapi.project.Project
+import com.intellij.openapi.util.NlsSafe
 import com.intellij.openapi.util.TextRange
 import com.intellij.psi.ElementDescriptionUtil
 import com.intellij.psi.PsiElement
@@ -36,27 +37,39 @@ abstract class AbstractChatIntention : IntentionAction {
      */
     override fun invoke(project: Project, editor: Editor?, file: PsiFile?) {
         if (editor == null || file == null) return
+        val withRange = elementWithRange(editor, file, project) ?: return
 
+        val selectedText = withRange.first
+        val psiElement = withRange.second
+
+        val actionType = getActionType()
+
+        val prompter = ContextPrompter.prompter(file.language.displayName)
+        prompter.initContext(actionType, selectedText, file, project, editor.caretModel.offset, psiElement)
+        sendToChatPanel(project, actionType, prompter)
+    }
+
+    fun elementWithRange(
+        editor: Editor,
+        file: PsiFile,
+        project: Project,
+    ): Pair<@NlsSafe String, PsiElement?>? {
         var selectedText = editor.selectionModel.selectedText
-        val elementToExplain = getElementToAction(project, editor)
+        val psiElement = getElementToAction(project, editor)
 
         if (selectedText == null) {
-            if (elementToExplain == null) {
-                return
+            if (psiElement == null) {
+                return null
             }
-            selectElement(elementToExplain, editor)
+            selectElement(psiElement, editor)
             selectedText = editor.selectionModel.selectedText
         }
 
         if (selectedText == null) {
-            return
+            return null
         }
 
-        val actionType = getActionType()
-
-        val prompter = ContextPrompter.prompter(file.language.displayName)
-        prompter.initContext(actionType, selectedText, file, project, editor.caretModel.offset, elementToExplain)
-        sendToChatPanel(project, actionType, prompter)
+        return Pair(selectedText, psiElement)
     }
 
     protected fun selectElement(elementToExplain: PsiElement, editor: Editor) {