feat(mcp): add search ch functionality to MCP tools panel #371

Author: phodal

core/src/main/kotlin/cc/unitmesh/devti/mcp/editor/McpPreviewEditor.kt

@@ -22,6 +22,7 @@ import com.intellij.openapi.util.UserDataHolderBase
 import com.intellij.openapi.vfs.VirtualFile
 import com.intellij.psi.PsiManager
 import com.intellij.ui.JBColor
+import com.intellij.ui.SearchTextField
 import com.intellij.ui.components.JBLabel
 import com.intellij.ui.components.JBScrollPane
 import com.intellij.ui.components.JBTextField
@@ -41,6 +42,8 @@ import java.awt.FlowLayout
 import java.beans.PropertyChangeListener
 import javax.swing.*
 import javax.swing.border.CompoundBorder
+import javax.swing.event.DocumentEvent
+import javax.swing.event.DocumentListener
 
 open class McpPreviewEditor(
     val project: Project,
@@ -59,6 +62,7 @@ open class McpPreviewEditor(
     private lateinit var resultPanel: McpResultPanel
     private val config = McpLlmConfig()
     private val borderColor = JBColor(0xE5E7EB, 0x3C3F41) // Equivalent to Tailwind gray-200
+    private lateinit var searchField: SearchTextField
 
     init {
         createUI()
@@ -79,12 +83,24 @@ open class McpPreviewEditor(
     private fun createUI() {
         val headerPanel = panel {
             row {
-                val label = JBLabel("MCP Preview - Tools Panel").apply {
+                val label = JBLabel("MCP Tools").apply {
                     font = JBUI.Fonts.label(14.0f).asBold()
+                    border = JBUI.Borders.emptyLeft(8)
                     isOpaque = true
                 }
 
                 cell(label).align(Align.FILL).resizableColumn()
+                
+                searchField = SearchTextField().apply {
+                    textEditor.emptyText.text = "Search tools..."
+                    textEditor.document.addDocumentListener(object : DocumentListener {
+                        override fun insertUpdate(e: DocumentEvent) = filterTools()
+                        override fun removeUpdate(e: DocumentEvent) = filterTools()
+                        override fun changedUpdate(e: DocumentEvent) = filterTools()
+                    })
+                }
+                
+                cell(searchField).align(Align.FILL).resizableColumn()
             }
         }.apply {
             border = BorderFactory.createMatteBorder(0, 0, 1, 0, borderColor)
@@ -197,6 +213,11 @@ open class McpPreviewEditor(
         mainPanel.add(bottomPanel, BorderLayout.SOUTH)
     }
 
+    private fun filterTools() {
+        val searchText = searchField.text.trim()
+        toolListPanel.filterTools(searchText)
+    }
+
     private fun showConfigDialog() {
         val dialog = McpLlmConfigDialog(project, config, allTools)
 

core/src/main/kotlin/cc/unitmesh/devti/mcp/ui/McpToolListPanel.kt

@@ -23,6 +23,7 @@ import javax.swing.SwingUtilities
 class McpToolListPanel(private val project: Project) : JPanel() {
     private val mcpServerManager = CustomMcpServerManager.instance(project)
     private val allTools = mutableMapOf<String, List<Tool>>()
+    private var currentFilteredTools = mutableMapOf<String, List<Tool>>()
     private var loadingJob: Job? = null
     private val serverLoadingStatus = mutableMapOf<String, Boolean>()
     private val serverPanels = mutableMapOf<String, JPanel>()
@@ -41,6 +42,7 @@ class McpToolListPanel(private val project: Project) : JPanel() {
         serverLoadingStatus.clear()
         serverPanels.clear()
         allTools.clear()
+        currentFilteredTools.clear()
 
         SwingUtilities.invokeLater {
             removeAll()
@@ -69,6 +71,7 @@ class McpToolListPanel(private val project: Project) : JPanel() {
                 try {
                     val tools = mcpServerManager.collectServerInfo(serverName, serverConfig)
                     allTools[serverName] = tools
+                    currentFilteredTools[serverName] = tools
                     onToolsLoaded(allTools)
 
                     SwingUtilities.invokeLater {
@@ -85,6 +88,30 @@ class McpToolListPanel(private val project: Project) : JPanel() {
         }
     }
 
+    fun filterTools(searchText: String) {
+        if (searchText.isEmpty()) {
+            // If search text is empty, restore all tools
+            currentFilteredTools = allTools.toMutableMap()
+        } else {
+            // Filter tools based on search text
+            currentFilteredTools = allTools.mapValues { (_, tools) ->
+                tools.filter { tool ->
+                    tool.name.contains(searchText, ignoreCase = true) ||
+                    tool.description?.contains(searchText, ignoreCase = true) == true
+                }
+            }.toMutableMap()
+        }
+
+        // Update UI with filtered tools
+        SwingUtilities.invokeLater {
+            currentFilteredTools.forEach { (serverName, tools) ->
+                updateServerSection(serverName, tools)
+            }
+            revalidate()
+            repaint()
+        }
+    }
+
     private fun createServerSection(serverName: String) {
         val serverPanel = JPanel(BorderLayout()).apply {
             background = UIUtil.getPanelBackground()