feat(custom): enable for custom by varible

Author: phodal

docs/autodev/custom-prompt-dsl.md

@@ -6,7 +6,7 @@
     "title": "转译",
     "autoInvoke": true,
     "matchRegex": ".*",
-    "template": "帮我重构下面的代码为 Kotlin 语言，方便于我理解: \n ${SPEC.XXX} ${SELECTION}"
+    "template": "帮我重构下面的代码为 Kotlin 语言，方便于我理解: \n ${SPEC_XXX} ${SELECTION}"
   }
 ]
 ```

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

@@ -29,10 +29,8 @@ class CustomIntention(private val intentionConfig: CustomIntentionConfig) : Abst
         val withRange = elementWithRange(editor, file, project) ?: return
         val selectedText = withRange.first
         val psiElement = withRange.second
-
         val prompt: CustomIntentionPrompt = buildCustomPrompt(psiElement!!, selectedText, intentionConfig)
 
-
         sendToChatPanel(project, getActionType(), object : ContextPrompter() {
             override fun displayPrompt(): String {
                 return prompt.displayPrompt
@@ -52,20 +50,19 @@ class CustomIntention(private val intentionConfig: CustomIntentionConfig) : Abst
         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()
 
+        val resolverMap = LinkedHashMap<String, VariableResolver>(10)
         for (resolver in variableResolvers) {
-            resolverMap[resolver.type] = resolver
+            resolverMap[resolver.variableName()] = resolver
         }
 
         resolverMap.forEach { (variableType, resolver) ->
             val value = resolver.resolve()
-            velocityContext.put(variableType.toString(), value)
+            velocityContext.put(variableType, value)
         }
 
         Velocity.evaluate(velocityContext, stringBuilderWriter, "", intentionConfig.template)

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

@@ -10,7 +10,7 @@ class SpecResolverService {
 
     fun createResolvers(): List<SpecVariableResolver> {
         return specs.map { (key, value) ->
-            SpecVariableResolver(key, value)
+            SpecVariableResolver("SPEC_$key", value)
         }
     }
 

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

@@ -6,4 +6,8 @@ class SpecVariableResolver(val key: String, val value: String) : VariableResolve
     override fun resolve(): String {
         return value
     }
+
+    override fun variableName(): String {
+        return key
+    }
 }

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

@@ -3,4 +3,6 @@ package cc.unitmesh.devti.custom.variable
 interface VariableResolver {
     val type: CustomIntentionVariableType
     fun resolve(): String
+
+    fun variableName() = type.name
 }