feat(mcp): add dynamic tool config listener to sketch

Subscribe to tool configuration changes and automatically update system prompts when selected tools change. Add message bus connection management and proper cleanup in dispose method.

Author: phodal

core/src/main/kotlin/cc/unitmesh/devti/sketch/SketchInputListener.kt

@@ -7,6 +7,7 @@ import cc.unitmesh.devti.gui.chat.ui.AutoDevInputSection
 import cc.unitmesh.devti.gui.chat.ui.AutoDevInputTrigger
 import cc.unitmesh.devti.llm2.model.LlmConfig
 import cc.unitmesh.devti.llms.cancelHandler
+import cc.unitmesh.devti.mcp.ui.SketchConfigListener
 import cc.unitmesh.devti.observer.agent.AgentStateService
 import cc.unitmesh.devti.prompting.SimpleDevinPrompter
 import cc.unitmesh.devti.provider.devins.LanguageProcessor
@@ -17,8 +18,10 @@ import com.intellij.openapi.Disposable
 import com.intellij.openapi.application.ApplicationManager
 import com.intellij.openapi.diagnostic.logger
 import com.intellij.openapi.project.Project
+import com.intellij.util.messages.MessageBusConnection
 import kotlinx.coroutines.flow.cancellable
 import kotlinx.coroutines.launch
+import io.modelcontextprotocol.kotlin.sdk.Tool
 
 open class SketchInputListener(
     private val project: Project,
@@ -32,12 +35,38 @@ open class SketchInputListener(
     open var systemPrompt = ""
     open var planPrompt = ""
     val planTemplate = templateRender.getTemplate("plan.vm")
+    private var messageBusConnection: MessageBusConnection? = null
 
     open suspend fun setup() {
         val customContext = SketchRunContext.create(project, null, "")
         systemPrompt = templateRender.renderTemplate(template, customContext)
         planPrompt = templateRender.renderTemplate(planTemplate, customContext)
         toolWindow.addSystemPrompt(systemPrompt)
+
+        // Subscribe to tool configuration changes
+        setupToolConfigListener()
+    }
+
+    private fun setupToolConfigListener() {
+        messageBusConnection = ApplicationManager.getApplication().messageBus.connect()
+        messageBusConnection?.subscribe(SketchConfigListener.TOPIC, object : SketchConfigListener {
+            override fun onSelectedToolsChanged(tools: Map<String, Set<Tool>>) {
+                AutoDevCoroutineScope.scope(project).launch {
+                    updateSystemPrompt()
+                }
+            }
+        })
+    }
+
+    private suspend fun updateSystemPrompt() {
+        val customContext = SketchRunContext.create(project, null, "")
+        val newSystemPrompt = templateRender.renderTemplate(template, customContext)
+        val newPlanPrompt = templateRender.renderTemplate(planTemplate, customContext)
+
+        systemPrompt = newSystemPrompt
+        planPrompt = newPlanPrompt
+
+        toolWindow.updateSystemPrompt(newSystemPrompt)
     }
 
     override fun onStop(component: AutoDevInputSection) {
@@ -114,6 +143,8 @@ open class SketchInputListener(
 
     override fun dispose() {
         connection.disconnect()
+        messageBusConnection?.disconnect()
+        messageBusConnection = null
     }
 
     fun stop() {

core/src/main/kotlin/cc/unitmesh/devti/sketch/SketchToolWindow.kt

@@ -274,11 +274,16 @@ open class SketchToolWindow(
         }
     }
 
-//    fun clearSystemPromptPanel() {
-//        systemPromptPanel.removeAll()
-//        systemPromptPanel.revalidate()
-//        systemPromptPanel.repaint()
-//    }
+    fun updateSystemPrompt(text: String) {
+        runInEdt {
+            systemPromptPanel.removeAll()
+            systemPromptPanel.add(createSingleTextView(text, language = "VTL"))
+            systemPromptPanel.revalidate()
+            systemPromptPanel.repaint()
+            this.revalidate()
+            this.repaint()
+        }
+    }
 
     fun updateHistoryPanel() {
         runInEdt {