refactor: try to add custom agent json

Author: phodal

src/233/main/resources/META-INF/autodev-core.xml

@@ -29,7 +29,8 @@
 
         <applicationService serviceImplementation="cc.unitmesh.devti.settings.AutoDevSettingsState"/>
 
-        <statusBarWidgetFactory id="AIAssistant" implementation="cc.unitmesh.devti.statusbar.AutoDevStatusBarWidgetFactory"/>
+        <statusBarWidgetFactory id="AIAssistant"
+                                implementation="cc.unitmesh.devti.statusbar.AutoDevStatusBarWidgetFactory"/>
 
         <runConfigurationProducer
                 implementation="cc.unitmesh.devti.runconfig.command.AutoDevFeatureConfigurationProducer"/>
@@ -49,13 +50,13 @@
                     icon="cc.unitmesh.devti.AutoDevIcons.AI_COPILOT"
                     factoryClass="cc.unitmesh.devti.gui.AutoDevToolWindowFactory"/>
 
-<!--        <toolWindow id="AutoDev Pair"-->
-<!--                    doNotActivateOnStart="true"-->
-<!--                    anchor="left"-->
-<!--                    secondary="true"-->
-<!--                    canCloseContents="false"-->
-<!--                    icon="cc.unitmesh.devti.AutoDevIcons.AI_PAIR"-->
-<!--                    factoryClass="cc.unitmesh.devti.gui.AutoDevPairToolWindowFactory"/>-->
+        <!--        <toolWindow id="AutoDev Pair"-->
+        <!--                    doNotActivateOnStart="true"-->
+        <!--                    anchor="left"-->
+        <!--                    secondary="true"-->
+        <!--                    canCloseContents="false"-->
+        <!--                    icon="cc.unitmesh.devti.AutoDevIcons.AI_PAIR"-->
+        <!--                    factoryClass="cc.unitmesh.devti.gui.AutoDevPairToolWindowFactory"/>-->
 
         <notificationGroup id="AI notification group" displayType="STICKY_BALLOON" bundle="messages.AutoDevBundle"
                            key="name"/>
@@ -72,6 +73,16 @@
         <daemon.intentionActionFilter
                 implementation="cc.unitmesh.devti.gui.snippet.error.CodeBlockIntentionActionFilter"/>
 
+<!--        <completion.contributor-->
+<!--                language="JSON"-->
+<!--                id="CustomAgentJsonCompletionContributor"-->
+<!--                implementationClass="cc.unitmesh.devti.custom.completion.CustomAgentJsonCompletionContributor"-->
+<!--                order="first, before JsonCompletionContributor"/>-->
+    </extensions>
+
+    <extensions defaultExtensionNs="JavaScript.JsonSchema">
+        <ProviderFactory implementation="cc.unitmesh.devti.custom.schema.AutoDevJsonSchemaProviderFactory"/>
+        <ProviderFactory implementation="cc.unitmesh.devti.custom.schema.AutoDevPromptJsonSchemaProviderFactory"/>
     </extensions>
 
     <extensionPoints>
@@ -136,7 +147,7 @@
 
         <extensionPoint qualifiedName="cc.unitmesh.layeredArchProvider"
                         interface="cc.unitmesh.devti.provider.architecture.LayeredArchProvider"
-                        dynamic="true"  />
+                        dynamic="true"/>
 
         <!-- TODO: find better way to share context -->
         <extensionPoint qualifiedName="cc.unitmesh.contextPrompter"
@@ -186,11 +197,6 @@
         <chatContextProvider implementation="cc.unitmesh.devti.provider.builtin.LanguageContextProvider"/>
     </extensions>
 
-    <extensions defaultExtensionNs="JavaScript.JsonSchema">
-        <ProviderFactory implementation="cc.unitmesh.devti.custom.schema.AutoDevJsonSchemaProviderFactory"/>
-        <ProviderFactory implementation="cc.unitmesh.devti.custom.schema.AutoDevPromptJsonSchemaProviderFactory"/>
-    </extensions>
-
     <actions>
         <group id="AutoDevIntentionsActionGroup" class="cc.unitmesh.devti.intentions.IntentionsActionGroup"
                icon="cc.unitmesh.devti.AutoDevIcons.AI_COPILOT" searchable="false">

src/main/kotlin/cc/unitmesh/devti/custom/completion/CustomAgentJsonCompletionContributor.kt

@@ -0,0 +1,49 @@
+package cc.unitmesh.devti.custom.completion
+
+import cc.unitmesh.devti.agent.model.CustomAgentConfig
+import cc.unitmesh.devti.custom.schema.CUSTOM_AGENT_FILE_NAME
+import com.intellij.codeInsight.completion.*
+import com.intellij.codeInsight.lookup.LookupElementBuilder
+import com.intellij.json.JsonElementTypes
+import com.intellij.json.psi.JsonStringLiteral
+import com.intellij.patterns.PlatformPatterns
+import com.intellij.patterns.PsiElementPattern
+import com.intellij.psi.PsiElement
+import com.intellij.util.ProcessingContext
+
+class CustomAgentJsonCompletionContributor : CompletionContributor() {
+    internal fun jsonPropertyName() = PlatformPatterns.psiElement(JsonElementTypes.IDENTIFIER)
+
+    internal fun jsonStringValue() =
+        PlatformPatterns.psiElement(JsonElementTypes.SINGLE_QUOTED_STRING).withParent<JsonStringLiteral>()
+
+    internal inline fun <reified T : PsiElement> PsiElementPattern<*, *>.withParent() = this.withParent(T::class.java)
+
+    init {
+        extend(
+            CompletionType.BASIC,
+            jsonPropertyName(),
+            CustomAgentFieldsProvider()
+        )
+    }
+
+    override fun fillCompletionVariants(parameters: CompletionParameters, result: CompletionResultSet) {
+        if (parameters.position.containingFile.name != CUSTOM_AGENT_FILE_NAME) {
+            return
+        }
+
+        super.fillCompletionVariants(parameters, result)
+    }
+}
+
+class CustomAgentFieldsProvider : CompletionProvider<CompletionParameters>() {
+    override fun addCompletions(
+        parameters: CompletionParameters,
+        context: ProcessingContext,
+        result: CompletionResultSet,
+    ) {
+        CustomAgentConfig::class.java.declaredFields.forEach {
+            result.addElement(LookupElementBuilder.create("\"" + it.name + "\""))
+        }
+    }
+}