feat(gui): add support for custom variables in chat input #51

This commit introduces the ability to use custom variables in the chat input by providing a list of available variables and a popup to select them. The custom variables are now rendered in the chat input section, and a popup is created to display the list of variables when the user types a dollar sign. The popup is dismissed when the user clicks outside of it or presses a key.

Author: phodal

src/main/kotlin/cc/unitmesh/devti/gui/chat/AutoDevInputSection.kt

@@ -5,6 +5,7 @@ import cc.unitmesh.devti.AutoDevIcons
 import cc.unitmesh.devti.counit.configurable.customAgentSetting
 import cc.unitmesh.devti.counit.model.CustomAgentConfig
 import cc.unitmesh.devti.counit.model.CustomAgentState
+import cc.unitmesh.devti.custom.variable.CustomVariable
 import cc.unitmesh.devti.llms.tokenizer.Tokenizer
 import cc.unitmesh.devti.llms.tokenizer.TokenizerImpl
 import cc.unitmesh.devti.settings.AutoDevSettingsState
@@ -108,6 +109,10 @@ class AutoDevInputSection(private val project: Project, val disposable: Disposab
 
                 // check new input == $
                 if (event.newFragment.contentEquals("$") || event.newFragment.contentEquals("¥")) {
+                    if (popup == null) {
+                        popup = createPopup()
+                    }
+
                     if (popup?.isVisible == true) {
                         popup?.cancel()
                     }
@@ -122,8 +127,6 @@ class AutoDevInputSection(private val project: Project, val disposable: Disposab
             }
         }
 
-        popup = createPopup()
-
         input.addDocumentListener(documentListener)
         input.recreateDocument()
 
@@ -175,14 +178,22 @@ class AutoDevInputSection(private val project: Project, val disposable: Disposab
         tokenizer = TokenizerImpl.INSTANCE
     }
 
-    private fun createPopup() = JBPopupFactory.getInstance()
-        .createComponentPopupBuilder(AutoDevVariableList(listOf(
-            AutoDevVariableListComponent(),
-        ) ,null), null )
-        .setRequestFocus(false)
-        .setMinSize(
-            Dimension(this@AutoDevInputSection.width, 0)
-        ).createPopup()
+    private fun createPopup(): JBPopup {
+        val devVariableList = AutoDevVariableList.from(CustomVariable.all()) { item ->
+            input.text += item.customVariable.variable
+            this.popup?.cancel()
+        }
+
+        devVariableList.selectedIndex = 0
+
+        val popups = JBPopupFactory.getInstance()
+            .createComponentPopupBuilder(devVariableList, null)
+            .setRequestFocus(false)
+            .setMinSize(Dimension(this@AutoDevInputSection.width, 0))
+            .createPopup()
+
+        return popups
+    }
 
 
     private fun loadRagApps(): List<CustomAgentConfig> {

src/main/kotlin/cc/unitmesh/devti/gui/chat/AutoDevVariableList.kt

@@ -1,38 +1,47 @@
 package cc.unitmesh.devti.gui.chat
 
+import cc.unitmesh.devti.custom.variable.CustomVariable
 import com.intellij.ui.Gray
 import com.intellij.ui.JBColor
 import com.intellij.ui.components.JBList
 import java.awt.BorderLayout
 import java.awt.Component
 import java.awt.event.MouseEvent
-import java.awt.event.MouseListener
 import javax.swing.*
 import javax.swing.border.Border
 import javax.swing.event.MouseInputAdapter
 
 class AutoDevVariableList(
-    val list: List<AutoDevVariableListComponent>,
-    val callback: ((AutoDevVariableListComponent) -> Unit?)?,
-) : JBList<AutoDevVariableListComponent>() {
+    val list: List<AutoDevVariableListItemComponent>,
+    val callback: ((AutoDevVariableListItemComponent) -> Unit),
+) : JBList<AutoDevVariableListItemComponent>(list) {
     init {
         border = BorderFactory.createEmptyBorder(0, 4, 0, 4)
         setCellRenderer(VariableListCellRenderer())
         addMouseListener(object : MouseInputAdapter() {
             override fun mouseClicked(event: MouseEvent?) {
                 val item = selectedValue ?: return
-                callback?.invoke(item)
+                callback.invoke(item)
             }
         })
     }
+
+    companion object {
+        fun from(all: List<CustomVariable>, function: (AutoDevVariableListItemComponent) -> Unit): AutoDevVariableList {
+            val list = all.map {
+                AutoDevVariableListItemComponent(it)
+            }
+            return AutoDevVariableList(list, function)
+        }
+    }
 }
 
-class VariableListCellRenderer : ListCellRenderer<AutoDevVariableListComponent> {
+class VariableListCellRenderer : ListCellRenderer<AutoDevVariableListItemComponent> {
     private var emptyBorder: Border = BorderFactory.createEmptyBorder(1, 1, 1, 1)
 
     override fun getListCellRendererComponent(
-        jList: JList<out AutoDevVariableListComponent>,
-        value: AutoDevVariableListComponent?,
+        jList: JList<out AutoDevVariableListItemComponent>,
+        value: AutoDevVariableListItemComponent?,
         index: Int,
         isSelected: Boolean,
         cellHasFocus: Boolean,
@@ -62,15 +71,14 @@ class VariableListCellRenderer : ListCellRenderer<AutoDevVariableListComponent>
         value.border = border
         return value
     }
-
 }
 
-class AutoDevVariableListComponent : JPanel() {
+class AutoDevVariableListItemComponent(val customVariable: CustomVariable) : JPanel() {
     init {
-        add(JLabel("$" + "selection"), BorderLayout.WEST)
-        val label = JLabel("Used to get the currently selected text")
+        add(JLabel("$${customVariable.variable}"), BorderLayout.WEST)
+        val label = JLabel(customVariable.description)
         label.border = BorderFactory.createEmptyBorder(0, 8, 0, 0)
         label.foreground = JBColor.namedColor("Component.infoForeground", JBColor(Gray.x99, Gray.x78))
         add(label, BorderLayout.EAST)
     }
-}
+}