feat(gui): enhance change management in planner results #352

- Add accept all and discard all functionality
- Improve change view with diff viewer and accept/discard options- Optimize performance with runWriteAction for file modifications
- Update UI to show/hide accept/discard buttons based on changes

Author: phodal

core/src/main/kotlin/cc/unitmesh/devti/gui/planner/PlannerResultSummary.kt

@@ -11,7 +11,9 @@ import com.intellij.icons.AllIcons
 import com.intellij.openapi.actionSystem.AnAction
 import com.intellij.openapi.actionSystem.AnActionEvent
 import com.intellij.openapi.application.runInEdt
+import com.intellij.openapi.application.runWriteAction
 import com.intellij.openapi.fileEditor.FileDocumentManager
+import com.intellij.openapi.fileEditor.FileEditorManager
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.ui.DialogWrapper
 import com.intellij.openapi.vcs.changes.Change
@@ -35,6 +37,26 @@ class PlannerResultSummary(
     private val changesPanel = JPanel(GridLayout(0, 1, 0, 1))
     private val statsLabel = JBLabel(AutoDevBundle.message("planner.stats.changes.empty"))
     private val rollbackWorker = RollbackWorker(project)
+    private val discardAllButton = HyperlinkLabel(AutoDevBundle.message("planner.action.discard.all")).apply {
+        icon = AllIcons.Actions.Cancel
+        addHyperlinkListener(object : HyperlinkListener {
+            override fun hyperlinkUpdate(e: HyperlinkEvent) {
+                if (e.eventType == HyperlinkEvent.EventType.ACTIVATED) {
+                    globalActionListener?.onDiscardAll()
+                }
+            }
+        })
+    }
+    private val acceptAllButton = HyperlinkLabel(AutoDevBundle.message("planner.action.accept.all")).apply {
+        icon = AllIcons.Actions.Commit
+        addHyperlinkListener(object : HyperlinkListener {
+            override fun hyperlinkUpdate(e: HyperlinkEvent) {
+                if (e.eventType == HyperlinkEvent.EventType.ACTIVATED) {
+                    globalActionListener?.onAcceptAll()
+                }
+            }
+        })
+    }
 
     interface ChangeActionListener {
         fun onView(change: Change)
@@ -62,41 +84,10 @@ class PlannerResultSummary(
 
     private var changeActionListener: ChangeActionListener = object : ChangeActionListener {
         override fun onView(change: Change) {
-            change.virtualFile?.also {
-                val diffFactory = DiffContentFactoryEx.getInstanceEx()
-                val oldCode = change.beforeRevision?.content ?: return
-                val newCode = change.afterRevision?.content ?: return
-                val currentDocContent = diffFactory.create(project, oldCode)
-                val newDocContent = diffFactory.create(newCode)
-
-                val diffRequest =
-                    SimpleDiffRequest("Diff", currentDocContent, newDocContent, "Original", "AI suggestion")
-
-                val diffViewer = SimpleDiffViewer(object : DiffContext() {
-                    override fun getProject() = this@PlannerResultSummary.project
-                    override fun isWindowFocused() = false
-                    override fun isFocusedInWindow() = false
-                    override fun requestFocusInWindow() = Unit
-                }, diffRequest)
-                diffViewer.init()
-
-                val dialog = object : DialogWrapper(project) {
-                    init {
-                        init()
-                        title = "Diff Viewer"
-                    }
-
-                    override fun createCenterPanel(): JComponent = diffViewer.component
-                    override fun doOKAction() {
-                        super.doOKAction()
-                        changeActionListener.onAccept(change)
-                    }
-                    override fun doCancelAction() {
-                        super.doCancelAction()
-                    }
+            runWriteAction {
+                change.virtualFile?.also {
+                    showDiffView(change)
                 }
-
-                dialog.show()
             }
         }
 
@@ -111,7 +102,7 @@ class PlannerResultSummary(
             val file = change.virtualFile ?: return
             val content = change.afterRevision?.content ?: change.beforeRevision?.content
 
-            runInEdt {
+            runWriteAction {
                 if (content != null) {
                     val document = FileDocumentManager.getInstance().getDocument(file)
                     document?.setText(content)
@@ -120,6 +111,45 @@ class PlannerResultSummary(
         }
     }
 
+    private fun showDiffView(change: Change) {
+        val diffFactory = DiffContentFactoryEx.getInstanceEx()
+        val oldCode = change.beforeRevision?.content ?: return
+        val newCode = change.afterRevision?.content ?: return
+        val currentDocContent = diffFactory.create(project, oldCode)
+        val newDocContent = diffFactory.create(newCode)
+
+        val diffRequest =
+            SimpleDiffRequest("Diff", currentDocContent, newDocContent, "Original", "AI suggestion")
+
+        val diffViewer = SimpleDiffViewer(object : DiffContext() {
+            override fun getProject() = this@PlannerResultSummary.project
+            override fun isWindowFocused() = false
+            override fun isFocusedInWindow() = false
+            override fun requestFocusInWindow() = Unit
+        }, diffRequest)
+        diffViewer.init()
+
+        val dialog = object : DialogWrapper(project) {
+            init {
+                init()
+                title = "Diff Viewer"
+                setOKButtonText("Apply")
+            }
+
+            override fun createCenterPanel(): JComponent = diffViewer.component
+            override fun doOKAction() {
+                super.doOKAction()
+                changeActionListener.onAccept(change)
+            }
+
+            override fun doCancelAction() {
+                super.doCancelAction()
+            }
+        }
+
+        dialog.show()
+    }
+
     init {
         border = JBUI.Borders.customLine(UIUtil.getBoundsColor(), 1, 0, 0, 0)
 
@@ -138,28 +168,6 @@ class PlannerResultSummary(
             }
 
             val actionsPanel = JPanel(FlowLayout(FlowLayout.RIGHT, 4, 0)).apply {
-                val discardAllButton = HyperlinkLabel(AutoDevBundle.message("planner.action.discard.all")).apply {
-                    icon = AllIcons.Actions.Cancel
-                    addHyperlinkListener(object : HyperlinkListener {
-                        override fun hyperlinkUpdate(e: HyperlinkEvent) {
-                            if (e.eventType == HyperlinkEvent.EventType.ACTIVATED) {
-                                globalActionListener?.onDiscardAll()
-                            }
-                        }
-                    })
-                }
-
-                val acceptAllButton = HyperlinkLabel(AutoDevBundle.message("planner.action.accept.all")).apply {
-                    icon = AllIcons.Actions.Commit
-                    addHyperlinkListener(object : HyperlinkListener {
-                        override fun hyperlinkUpdate(e: HyperlinkEvent) {
-                            if (e.eventType == HyperlinkEvent.EventType.ACTIVATED) {
-                                globalActionListener?.onAcceptAll()
-                            }
-                        }
-                    })
-                }
-
                 add(discardAllButton)
                 add(acceptAllButton)
             }
@@ -193,6 +201,9 @@ class PlannerResultSummary(
                 foreground = UIUtil.getLabelDisabledForeground()
                 border = JBUI.Borders.empty(8)
             })
+
+            discardAllButton.isVisible = false
+            acceptAllButton.isVisible = false
         } else {
             statsLabel.text = " - " + AutoDevBundle.message("planner.stats.changes.count", changes.size)
             changes.forEach { change ->
@@ -202,6 +213,9 @@ class PlannerResultSummary(
                 val changePanel = createChangeItemPanel(change, fileName, filePath)
                 changesPanel.add(changePanel)
             }
+
+            discardAllButton.isVisible = true
+            acceptAllButton.isVisible = true
         }
 
         changesPanel.revalidate()
@@ -232,7 +246,9 @@ class PlannerResultSummary(
                 addHyperlinkListener(object : HyperlinkListener {
                     override fun hyperlinkUpdate(e: HyperlinkEvent) {
                         if (e.eventType == HyperlinkEvent.EventType.ACTIVATED) {
-                            changeActionListener.onView(change)
+                            change.virtualFile?.also {
+                                FileEditorManager.getInstance(project).openFile(it, true)
+                            }
                         }
                     }
                 })
@@ -293,4 +309,3 @@ class PlannerResultSummary(
         return KeyboardAccessibleActionButton(anAction)
     }
 }
-