refactor(devins-lang): init agent library to use variable in agent #100

Author: phodal

exts/devins-lang/src/main/kotlin/cc/unitmesh/devti/language/completion/AgentToolLibraryCompletion.kt

@@ -0,0 +1,35 @@
+package cc.unitmesh.devti.language.completion
+
+import cc.unitmesh.devti.language.DevInIcons
+import com.intellij.codeInsight.completion.CompletionParameters
+import com.intellij.codeInsight.completion.CompletionProvider
+import com.intellij.codeInsight.completion.CompletionResultSet
+import com.intellij.codeInsight.completion.PrioritizedLookupElement
+import com.intellij.codeInsight.lookup.LookupElementBuilder
+import com.intellij.util.ProcessingContext
+
+class AgentToolLibraryCompletion : CompletionProvider<CompletionParameters>() {
+    override fun addCompletions(
+        parameters: CompletionParameters,
+        context: ProcessingContext,
+        result: CompletionResultSet
+    ) {
+        val command = PrioritizedLookupElement.withPriority(
+            LookupElementBuilder.create("commands")
+                .withIcon(DevInIcons.DEFAULT)
+                .withTypeText("DevIns all commands for AI Agent to call", true),
+            0.0
+        )
+
+        val agent = PrioritizedLookupElement.withPriority(
+            LookupElementBuilder.create("agents")
+                .withIcon(DevInIcons.DEFAULT)
+                .withTypeText("DevIns all agent for AI Agent to call", true),
+            0.0
+        )
+
+        result.addElement(command)
+        result.addElement(agent)
+    }
+
+}

exts/devins-lang/src/main/kotlin/cc/unitmesh/devti/language/completion/DevInCompletionContributor.kt

@@ -15,7 +15,10 @@ import com.intellij.psi.PsiElement
 class DevInCompletionContributor : CompletionContributor() {
     init {
         extend(CompletionType.BASIC, PlatformPatterns.psiElement(DevInTypes.LANGUAGE_ID), CodeFenceLanguageCompletion())
-        extend(CompletionType.BASIC, PlatformPatterns.psiElement(DevInTypes.VARIABLE_ID), CustomVariableCompletion())
+
+        extend(CompletionType.BASIC, PlatformPatterns.psiElement(DevInTypes.VARIABLE_ID), VariableCompletionProvider())
+        extend(CompletionType.BASIC, PlatformPatterns.psiElement(DevInTypes.VARIABLE_ID), AgentToolLibraryCompletion())
+
         extend(CompletionType.BASIC, PlatformPatterns.psiElement(DevInTypes.COMMAND_ID), BuiltinCommandCompletion())
         extend(CompletionType.BASIC, PlatformPatterns.psiElement(DevInTypes.AGENT_ID), CustomAgentCompletion())
 

exts/devins-lang/src/main/kotlin/cc/unitmesh/devti/language/completion/provider/CustomVariableCompletion.kt renamed to exts/devins-lang/src/main/kotlin/cc/unitmesh/devti/language/completion/provider/VariableCompletionProvider.kt

@@ -4,17 +4,22 @@ import cc.unitmesh.devti.custom.compile.CustomVariable
 import com.intellij.codeInsight.completion.CompletionParameters
 import com.intellij.codeInsight.completion.CompletionProvider
 import com.intellij.codeInsight.completion.CompletionResultSet
+import com.intellij.codeInsight.completion.PrioritizedLookupElement
 import com.intellij.codeInsight.lookup.LookupElementBuilder
 import com.intellij.util.ProcessingContext
 
-class CustomVariableCompletion: CompletionProvider<CompletionParameters>() {
+class VariableCompletionProvider : CompletionProvider<CompletionParameters>() {
     override fun addCompletions(
         parameters: CompletionParameters,
         context: ProcessingContext,
         result: CompletionResultSet,
     ) {
         CustomVariable.all().forEach {
-            val withTypeText = LookupElementBuilder.create(it.variable).withTypeText(it.description, true)
+            val withTypeText =
+                PrioritizedLookupElement.withPriority(
+                    LookupElementBuilder.create(it.variable).withTypeText(it.description, true),
+                    99.0
+                )
             result.addElement(withTypeText)
         }
     }