feat(plan): sync plan state from agent to renderer

Integrate PlanStateService observation to update renderer with current plan. Add logging for plan updates and summary rendering.

Author: phodal

mpp-core/src/commonMain/kotlin/cc/unitmesh/agent/CodingAgent.kt

@@ -85,6 +85,14 @@ class CodingAgent(
     private val policyEngine = DefaultPolicyEngine()
     private val toolOrchestrator = ToolOrchestrator(toolRegistry, policyEngine, renderer, mcpConfigService = mcpToolConfigService)
 
+    /**
+     * Get the PlanStateService for observing plan state changes.
+     * Returns null if no plan tool is registered.
+     */
+    fun getPlanStateService(): cc.unitmesh.agent.plan.PlanStateService? {
+        return toolOrchestrator.getPlanStateService()
+    }
+
     private val errorRecoveryAgent = ErrorRecoveryAgent(projectPath, llmService)
     private val analysisAgent = AnalysisAgent(llmService, contentThreshold = 15000)
     private val mcpToolsInitializer = McpToolsInitializer()

mpp-idea/src/main/kotlin/cc/unitmesh/devins/idea/renderer/JewelRenderer.kt

@@ -14,11 +14,14 @@ import cc.unitmesh.agent.render.ToolCallInfo
 import cc.unitmesh.agent.tool.ToolType
 import cc.unitmesh.agent.tool.toToolType
 import cc.unitmesh.llm.compression.TokenInfo
+import com.intellij.openapi.diagnostic.Logger
 import kotlinx.coroutines.flow.MutableStateFlow
 import kotlinx.coroutines.flow.StateFlow
 import kotlinx.coroutines.flow.asStateFlow
 import kotlinx.coroutines.flow.update
 
+private val jewelRendererLogger = Logger.getInstance("JewelRenderer")
+
 /**
  * Jewel-compatible Renderer for IntelliJ IDEA plugin.
  *
@@ -84,6 +87,15 @@ class JewelRenderer : BaseRenderer() {
     private val _currentPlan = MutableStateFlow<AgentPlan?>(null)
     val currentPlan: StateFlow<AgentPlan?> = _currentPlan.asStateFlow()
 
+    /**
+     * Set the current plan directly.
+     * Used to sync with PlanStateService from CodingAgent.
+     */
+    fun setPlan(plan: AgentPlan?) {
+        jewelRendererLogger.info("setPlan: plan=${plan != null}, tasks=${plan?.tasks?.size ?: 0}")
+        _currentPlan.value = plan
+    }
+
     // BaseRenderer implementation
 
     override fun renderIterationHeader(current: Int, max: Int) {
@@ -137,17 +149,22 @@ class JewelRenderer : BaseRenderer() {
     }
 
     override fun renderToolCall(toolName: String, paramsStr: String) {
+        jewelRendererLogger.info("renderToolCall: toolName=$toolName, paramsStr length=${paramsStr.length}")
+
         val toolInfo = formatToolCallDisplay(toolName, paramsStr)
         val params = parseParamsString(paramsStr)
         val toolType = toolName.toToolType()
 
+        jewelRendererLogger.info("renderToolCall: parsed params keys=${params.keys}")
+
         // Handle task-boundary tool - update task list
         if (toolName == "task-boundary") {
             updateTaskFromToolCall(params)
         }
 
         // Handle plan management tool - update plan state
         if (toolName == "plan") {
+            jewelRendererLogger.info("renderToolCall: detected plan tool, calling updatePlanFromToolCall")
             updatePlanFromToolCall(params)
             // Skip rendering plan tool to timeline - it's shown in PlanSummaryBar
             return
@@ -302,10 +319,13 @@ class JewelRenderer : BaseRenderer() {
      * Internal method to update plan state
      */
     private fun updatePlanState(action: String, planMarkdown: String, taskIndex: Int?, stepIndex: Int?) {
+        jewelRendererLogger.info("updatePlanState: action=$action, planMarkdown length=${planMarkdown.length}")
         when (action) {
             "CREATE", "UPDATE" -> {
                 if (planMarkdown.isNotBlank()) {
-                    _currentPlan.value = MarkdownPlanParser.parseToPlan(planMarkdown)
+                    val plan = MarkdownPlanParser.parseToPlan(planMarkdown)
+                    jewelRendererLogger.info("Parsed plan: ${plan.tasks.size} tasks")
+                    _currentPlan.value = plan
                 }
             }
             "COMPLETE_STEP" -> {

mpp-idea/src/main/kotlin/cc/unitmesh/devins/idea/renderer/sketch/IdeaTerminalRenderer.kt

@@ -76,11 +76,10 @@ fun IdeaTerminalRenderer(
 
         // Command display
         CommandDisplay(command = command, isDangerous = isDangerous, dangerReason = dangerReason)
-        
         // Output (if available)
-        if (showOutput && executionResult != null) {
-            OutputDisplay(result = executionResult!!)
-        }
+//        if (showOutput && executionResult != null) {
+//            OutputDisplay(result = executionResult!!)
+//        }
     }
 }
 

mpp-idea/src/main/kotlin/cc/unitmesh/devins/idea/toolwindow/IdeaAgentViewModel.kt

@@ -262,10 +262,32 @@ class IdeaAgentViewModel(
                 enableLLMStreaming = true
             )
             agentInitialized = true
+
+            // Start observing PlanStateService and sync to renderer
+            startPlanStateObserver(codingAgent!!)
         }
         return codingAgent!!
     }
 
+    // Job for observing PlanStateService
+    private var planStateObserverJob: Job? = null
+
+    /**
+     * Start observing PlanStateService and sync plan state to renderer.
+     */
+    private fun startPlanStateObserver(agent: CodingAgent) {
+        // Cancel any existing observer
+        planStateObserverJob?.cancel()
+
+        val planStateService = agent.getPlanStateService() ?: return
+
+        planStateObserverJob = coroutineScope.launch {
+            planStateService.currentPlan.collect { plan ->
+                renderer.setPlan(plan)
+            }
+        }
+    }
+
     /**
      * Check if LLM service is configured.
      */

mpp-idea/src/main/kotlin/cc/unitmesh/devins/idea/toolwindow/changes/IdeaFileChangeSummary.kt

@@ -52,22 +52,25 @@ fun IdeaFileChangeSummary(
         return
     }
 
-    // Show diff dialog if a file is selected
-    selectedChange?.let { change ->
-        IdeaFileChangeDiffDialog(
-            project = project,
-            change = change,
-            onDismiss = { selectedChange = null },
-            onUndo = {
-                undoChange(project, change)
-                FileChangeTracker.removeChange(change)
-                selectedChange = null
-            },
-            onKeep = {
-                FileChangeTracker.removeChange(change)
-                selectedChange = null
+    // Handle diff dialog display when a file is selected
+    LaunchedEffect(selectedChange) {
+        selectedChange?.let { change ->
+            ApplicationManager.getApplication().invokeLater {
+                IdeaFileChangeDiffDialogWrapper.show(
+                    project = project,
+                    change = change,
+                    onUndo = {
+                        undoChange(project, change)
+                        FileChangeTracker.removeChange(change)
+                    },
+                    onKeep = {
+                        FileChangeTracker.removeChange(change)
+                    },
+                    onDismiss = {}
+                )
             }
-        )
+            selectedChange = null
+        }
     }
 
     Column(

mpp-idea/src/main/kotlin/cc/unitmesh/devins/idea/toolwindow/plan/IdeaPlanSummaryBar.kt

@@ -43,15 +43,21 @@ import org.jetbrains.jewel.ui.icons.AllIconsKeys
  * Displays a collapsible summary of the current plan above the input box.
  * Uses Jewel theming and components for native IntelliJ look and feel.
  */
+private val planSummaryLogger = com.intellij.openapi.diagnostic.Logger.getInstance("IdeaPlanSummaryBar")
+
 @Composable
 fun IdeaPlanSummaryBar(
     plan: AgentPlan?,
     modifier: Modifier = Modifier,
     onViewDetails: (() -> Unit)? = null,
     onDismiss: (() -> Unit)? = null
 ) {
+    // Log for debugging
+    planSummaryLogger.info("IdeaPlanSummaryBar called: plan=${plan != null}, tasks=${plan?.tasks?.size ?: 0}")
+
     // Don't render if no plan
     if (plan == null || plan.tasks.isEmpty()) {
+        planSummaryLogger.info("IdeaPlanSummaryBar: not rendering (plan is null or empty)")
         return
     }