feat(custom): make custom prompt works

Author: phodal

java/src/main/kotlin/cc/unitmesh/idea/promting/JavaContextPrompter.kt

@@ -175,6 +175,8 @@ open class JavaContextPrompter : ContextPrompter() {
 
             ChatActionType.EXPLAIN_BUSINESS -> {
             }
+
+            ChatActionType.CUSTOM_ACTION -> TODO()
         }
 
         return prompt

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

@@ -1,12 +1,19 @@
 package cc.unitmesh.devti.custom
 
+import cc.unitmesh.devti.custom.variable.*
 import cc.unitmesh.devti.gui.chat.ChatActionType
+import cc.unitmesh.devti.gui.sendToChatPanel
 import cc.unitmesh.devti.intentions.AbstractChatIntention
+import cc.unitmesh.devti.provider.ContextPrompter
+import cc.unitmesh.devti.provider.builtin.DefaultContextPrompter
 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
+import org.apache.velocity.VelocityContext
+import org.apache.velocity.app.Velocity
+import java.io.StringWriter
 
 class CustomIntention(private val intentionConfig: CustomIntentionConfig) : AbstractChatIntention() {
     val specConfig: Map<String, String> = CustomPromptConfig.load().spec
@@ -15,7 +22,7 @@ class CustomIntention(private val intentionConfig: CustomIntentionConfig) : Abst
 
     override fun getFamilyName(): String = "Custom Intention"
 
-    override fun getActionType(): ChatActionType = super.getActionType()
+    override fun getActionType(): ChatActionType = ChatActionType.CUSTOM_ACTION
 
     override fun invoke(project: Project, editor: Editor?, file: PsiFile?) {
         if (editor == null || file == null) return
@@ -25,14 +32,46 @@ class CustomIntention(private val intentionConfig: CustomIntentionConfig) : Abst
 
         val prompt: CustomIntentionPrompt = buildCustomPrompt(psiElement!!, selectedText, intentionConfig)
 
+
+        sendToChatPanel(project, getActionType(), object : ContextPrompter() {
+            override fun displayPrompt(): String {
+                return prompt.displayPrompt
+            }
+
+            override fun requestPrompt(): String {
+                return prompt.requestPrompt
+            }
+        })
     }
 
     private fun buildCustomPrompt(
         psiElement: PsiElement,
         selectedText: @NlsSafe String,
         config: CustomIntentionConfig,
     ): CustomIntentionPrompt {
-        return CustomIntentionPrompt(selectedText, selectedText, listOf())
+        val stringBuilderWriter = StringWriter()
+        val velocityContext = VelocityContext()
+
+        val resolverMap = LinkedHashMap<CustomIntentionVariableType, VariableResolver>(10)
+
+        val variableResolvers = arrayOf(
+            MethodInputOutputVariableResolver(psiElement),
+            SelectionVariableResolver(psiElement.language.displayName ?: "", selectedText),
+        ) + SpecResolverService.getInstance().createResolvers()
+
+        for (resolver in variableResolvers) {
+            resolverMap[resolver.type] = resolver
+        }
+
+        resolverMap.forEach { (variableType, resolver) ->
+            val value = resolver.resolve()
+            velocityContext.put(variableType.toString(), value)
+        }
+
+        Velocity.evaluate(velocityContext, stringBuilderWriter, "", intentionConfig.template)
+        val output = stringBuilderWriter.toString()
+
+        return CustomIntentionPrompt(output, output, listOf())
     }
 
     companion object {

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

@@ -0,0 +1,13 @@
+package cc.unitmesh.devti.custom.variable
+
+import com.intellij.openapi.components.Service
+
+@Service(Service.Level.APP)
+class AllVariablesResolverService  {
+
+    fun buildResolver() {
+
+    }
+
+
+}

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

@@ -3,5 +3,6 @@ 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"),
+    SPEC_VARIABLE("Load from spec config, and config to items"),
     ;
 }

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

@@ -1,7 +1,7 @@
 package cc.unitmesh.devti.custom.variable
 
 class SelectionVariableResolver(
-    private val languageName: String,
+    val languageName: String,
     val code: String,
 ) : VariableResolver {
     override val type: CustomIntentionVariableType = CustomIntentionVariableType.SELECTION

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

@@ -0,0 +1,22 @@
+package cc.unitmesh.devti.custom.variable
+
+import cc.unitmesh.devti.custom.CustomPromptConfig
+import com.intellij.openapi.application.ApplicationManager
+import com.intellij.openapi.components.Service
+
+@Service(Service.Level.APP)
+class SpecResolverService {
+    private val specs = CustomPromptConfig.load().spec
+
+    fun createResolvers(): List<SpecVariableResolver> {
+        return specs.map { (key, value) ->
+            SpecVariableResolver(key, value)
+        }
+    }
+
+    companion object {
+        fun getInstance(): SpecResolverService {
+            return ApplicationManager.getApplication().getService(SpecResolverService::class.java)
+        }
+    }
+}

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

@@ -0,0 +1,9 @@
+package cc.unitmesh.devti.custom.variable
+
+class SpecVariableResolver(val key: String, val value: String) : VariableResolver {
+    override val type: CustomIntentionVariableType get() = CustomIntentionVariableType.SPEC_VARIABLE
+
+    override fun resolve(): String {
+        return value
+    }
+}