feat(custom): add basic variable

Author: phodal

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

@@ -4,9 +4,12 @@ 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.openapi.util.NlsSafe
+import com.intellij.psi.PsiElement
 import com.intellij.psi.PsiFile
 
 class CustomIntention(private val intentionConfig: CustomIntentionConfig) : AbstractChatIntention() {
+    val specConfig: Map<String, String> = CustomPromptConfig.load().spec
 
     override fun getText(): String = intentionConfig.title
 
@@ -15,7 +18,21 @@ class CustomIntention(private val intentionConfig: CustomIntentionConfig) : Abst
     override fun getActionType(): ChatActionType = super.getActionType()
 
     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 prompt: CustomIntentionPrompt = buildCustomPrompt(psiElement!!, selectedText, intentionConfig)
+
+    }
+
+    private fun buildCustomPrompt(
+        psiElement: PsiElement,
+        selectedText: @NlsSafe String,
+        config: CustomIntentionConfig,
+    ): CustomIntentionPrompt {
+        return CustomIntentionPrompt(selectedText, selectedText, listOf())
     }
 
     companion object {

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

@@ -0,0 +1,9 @@
+package cc.unitmesh.devti.custom
+
+import cc.unitmesh.devti.provider.context.ChatContextItem
+
+class CustomIntentionPrompt(
+    val displayPrompt: String,
+    val requestPrompt: String,
+    val contextItems: List<ChatContextItem> = listOf()
+)

src/main/kotlin/cc/unitmesh/devti/custom/variable/CustomIntentionVariableType.kt

@@ -0,0 +1,7 @@
+package cc.unitmesh.devti.custom.variable
+
+enum class CustomIntentionVariableType(@JvmField val description: String) {
+    SELECTION("Currently selected code fragment with language name"),
+    METHOD_INPUT_OUTPUT("Method input parameters's class as code snippets"),
+    ;
+}

src/main/kotlin/cc/unitmesh/devti/custom/variable/MethodInputOutputVariableResolver.kt

@@ -0,0 +1,29 @@
+package cc.unitmesh.devti.custom.variable
+
+import cc.unitmesh.devti.context.MethodContextProvider
+import com.intellij.psi.PsiElement
+
+class MethodInputOutputVariableResolver(val element: PsiElement) : VariableResolver {
+    override val type: CustomIntentionVariableType = CustomIntentionVariableType.METHOD_INPUT_OUTPUT
+
+    override fun resolve(): String {
+        var result = ""
+        val methodContext = MethodContextProvider(false, false).from(element)
+        if (methodContext.name == null) {
+            return ""
+        }
+
+        val input = methodContext.signature ?: ""
+        val output = methodContext.returnType ?: ""
+
+        // skip input if it is empty, skip output if it is void
+        if (input.isNotEmpty()) {
+            result += "Input: $input\n"
+        }
+        if (output.isNotEmpty() && output != "void") {
+            result += "Output: $output\n"
+        }
+
+        return result
+    }
+}

src/main/kotlin/cc/unitmesh/devti/custom/variable/SelectionVariableResolver.kt

@@ -0,0 +1,15 @@
+package cc.unitmesh.devti.custom.variable
+
+class SelectionVariableResolver(
+    private val languageName: String,
+    val code: String,
+) : VariableResolver {
+    override val type: CustomIntentionVariableType = CustomIntentionVariableType.SELECTION
+
+    override fun resolve(): String {
+        return """
+            |```$languageName
+            |$code
+            |```""".trimMargin()
+    }
+}

src/main/kotlin/cc/unitmesh/devti/custom/variable/VariableResolver.kt

@@ -0,0 +1,6 @@
+package cc.unitmesh.devti.custom.variable
+
+interface VariableResolver {
+    val type: CustomIntentionVariableType
+    fun resolve(): String
+}