refactor(patch): move diff stream logic to SingleFileDiffView

- Relocate diff stream handling from DiffLangSketch to SingleFileDiffView.
- Add repair and run stream buttons with improved UI and functionality.
- Clean up unused code and streamline patch handling.

Author: phodal

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

@@ -9,6 +9,7 @@ import cc.unitmesh.devti.llms.tokenizer.Tokenizer
 import cc.unitmesh.devti.llms.tokenizer.TokenizerFactory
 import cc.unitmesh.devti.provider.RelatedClassesProvider
 import cc.unitmesh.devti.settings.AutoDevSettingsState
+import cc.unitmesh.devti.util.isInProject
 import com.intellij.codeInsight.lookup.LookupManagerListener
 import com.intellij.icons.AllIcons
 import com.intellij.ide.IdeTooltip
@@ -385,10 +386,12 @@ class AutoDevInputSection(private val project: Project, val disposable: Disposab
         }
 
     val focusableComponent: JComponent get() = input
-}
 
-private fun DefaultListModel<ModelWrapper>.addIfAbsent(vfile: VirtualFile) {
-    if (elements().asIterator().asSequence().none { it.virtualFile == vfile }) {
-        addElement(ModelWrapper(vfile))
+    private fun DefaultListModel<ModelWrapper>.addIfAbsent(vfile: VirtualFile) {
+        if (!isInProject(vfile, project)) return
+        if (elements().asIterator().asSequence().none { it.virtualFile == vfile }) {
+            addElement(ModelWrapper(vfile))
+        }
     }
+
 }

core/src/main/kotlin/cc/unitmesh/devti/sketch/ui/patch/DiffLangSketch.kt

@@ -53,9 +53,8 @@ class DiffLangSketch(private val myProject: Project, private var patchContent: S
     private val filePatches: MutableList<TextFilePatch> = myReader.textPatches
 
     init {
-        val header = createHeaderAction()
-
-        myHeaderPanel.add(header, BorderLayout.EAST)
+//        val header = createHeaderAction()
+//        myHeaderPanel.add(header, BorderLayout.EAST)
         mainPanel.add(myHeaderPanel)
         mainPanel.border = JBUI.Borders.compound(JBUI.Borders.empty(0, 10))
 
@@ -112,67 +111,15 @@ class DiffLangSketch(private val myProject: Project, private var patchContent: S
             }
         }
 
-        val runStreamButton = JButton(AllIcons.Actions.RunAll).apply {
-            this.toolTipText = AutoDevBundle.message("sketch.patch.action.runDiff.tooltip")
-            addActionListener {
-                showStreamDiff()
-            }
-        }
-
-        val repairButton = JButton(AllIcons.Actions.Redo).apply {
-            this.toolTipText = AutoDevBundle.message("sketch.patch.action.repairDiff.tooltip")
-            addActionListener {
-                AutoDevNotifications.error(myProject, "Not implemented yet")
-            }
-        }
-
         val panel = JPanel()
         panel.layout = BoxLayout(panel, BoxLayout.X_AXIS)
         panel.add(acceptButton)
         panel.add(rejectButton)
         panel.add(viewDiffButton)
-        panel.add(runStreamButton)
-        panel.add(repairButton)
 
         return panel
     }
 
-    private fun showStreamDiff() {
-        val matchedPatches =
-            MatchPatchPaths(myProject).execute(filePatches, true)
-
-        val patchGroups = MultiMap<VirtualFile, AbstractFilePatchInProgress<*>>()
-        for (patchInProgress in matchedPatches) {
-            patchGroups.putValue(patchInProgress.base, patchInProgress)
-        }
-
-        val parsedPatch = filePatches.single()
-
-        val findFile = myProject.findFile(parsedPatch.beforeFileName!!)
-        if (findFile == null) {
-            AutoDevNotifications.error(myProject, "PatchProcessor: no before file found")
-            return
-        }
-
-        FileEditorManager.getInstance(myProject).openFile(findFile, true)
-        val editor =  FileEditorManager.getInstance(myProject).selectedTextEditor ?: return
-        val oldCode = findFile.readText()
-        val newText = GenericPatchApplier.apply(oldCode, parsedPatch.hunks)!!.patchedText
-
-        val diffStreamHandler = DiffStreamHandler(
-            myProject,
-            editor = editor,
-            0,
-            oldCode.lines().size,
-            onClose = {
-            },
-            onFinish = {
-
-            })
-
-        diffStreamHandler.normalDiff(oldCode, newText)
-    }
-
     private fun handleAcceptAction() {
         PsiDocumentManager.getInstance(myProject).commitAllDocuments();
         val commandProcessor: CommandProcessor = CommandProcessor.getInstance()

core/src/main/kotlin/cc/unitmesh/devti/sketch/ui/patch/SingleFileDiffView.kt

@@ -1,6 +1,8 @@
 package cc.unitmesh.devti.sketch.ui.patch
 
 import cc.unitmesh.devti.AutoDevBundle
+import cc.unitmesh.devti.AutoDevNotifications
+import cc.unitmesh.devti.diff.DiffStreamHandler
 import com.intellij.diff.DiffContentFactoryEx
 import com.intellij.diff.chains.SimpleDiffRequestChain
 import com.intellij.diff.chains.SimpleDiffRequestProducer
@@ -26,8 +28,17 @@ import com.intellij.ui.dsl.builder.AlignX
 import com.intellij.ui.dsl.builder.RightGap
 import com.intellij.ui.dsl.builder.panel
 import cc.unitmesh.devti.sketch.ui.LangSketch
+import cc.unitmesh.devti.util.findFile
+import com.intellij.openapi.command.CommandProcessor
+import com.intellij.openapi.command.UndoConfirmationPolicy
+import com.intellij.openapi.diff.impl.patch.ApplyPatchStatus
+import com.intellij.openapi.vcs.changes.patch.AbstractFilePatchInProgress
+import com.intellij.openapi.vcs.changes.patch.MatchPatchPaths
 import com.intellij.openapi.vfs.VfsUtilCore
+import com.intellij.psi.PsiDocumentManager
+import com.intellij.util.containers.MultiMap
 import java.awt.BorderLayout
+import java.awt.Dimension
 import java.awt.event.MouseAdapter
 import java.awt.event.MouseEvent
 import javax.swing.BorderFactory
@@ -45,6 +56,8 @@ class SingleFileDiffView(
     private val myHeaderPanel: JPanel = JPanel(BorderLayout())
     private var filePanel: DialogPanel? = null
     var diffFile: ChainDiffVirtualFile? = null
+    private val appliedPatch = GenericPatchApplier.apply(virtualFile.readText(), patch.hunks)
+    private val oldCode = virtualFile.readText()
 
     init {
         val contentPanel = JPanel(BorderLayout())
@@ -142,11 +155,9 @@ class SingleFileDiffView(
         val fileEditor = FileEditorManager.getInstance(myProject).getSelectedEditor(virtualFile)
 
         val rollback = JButton(AllIcons.Actions.Rollback).apply {
+            preferredSize = Dimension(32, 32)
             toolTipText = AutoDevBundle.message("sketch.patch.action.rollback.tooltip")
             isEnabled = undoManager.isUndoAvailable(fileEditor)
-            border = null
-            isFocusPainted = false
-            isContentAreaFilled = false
 
             addMouseListener(object : MouseAdapter() {
                 override fun mouseClicked(e: MouseEvent?) {
@@ -157,9 +168,70 @@ class SingleFileDiffView(
             })
         }
 
-        return listOf(rollback)
+        val runStreamButton = JButton(AllIcons.Actions.RunAll).apply {
+            preferredSize = Dimension(32, 32)
+            toolTipText = AutoDevBundle.message("sketch.patch.action.runDiff.tooltip")
+            isEnabled = appliedPatch?.status == ApplyPatchStatus.SUCCESS
+
+            addActionListener {
+                val document = FileDocumentManager.getInstance().getDocument(virtualFile) ?: return@addActionListener
+                PsiDocumentManager.getInstance(myProject).commitDocument(document)
+                CommandProcessor.getInstance().executeCommand(myProject, {
+                    showStreamDiff()
+                }, "RunStream", null, UndoConfirmationPolicy.REQUEST_CONFIRMATION, false)
+            }
+        }
+
+        val repairButton = JButton(AllIcons.Toolwindows.ToolWindowBuild).apply {
+            preferredSize = Dimension(32, 32)
+            toolTipText = AutoDevBundle.message("sketch.patch.action.repairDiff.tooltip")
+            isEnabled = appliedPatch?.status != ApplyPatchStatus.SUCCESS
+
+            addActionListener {
+                FileEditorManager.getInstance(myProject).openFile(virtualFile, true)
+                val editor = FileEditorManager.getInstance(myProject).selectedTextEditor ?: return@addActionListener
+
+                val diffStreamHandler = DiffStreamHandler(
+                    myProject,
+                    editor = editor,
+                    0,
+                    oldCode.lines().size,
+                    onClose = {
+                    },
+                    onFinish = {
+
+                    })
+
+                diffStreamHandler.streamDiffLinesToEditor(
+                    oldCode, "Please repair the diff, return all repaird code. \n" +
+                            "Here is the original code: \n\n$oldCode" + "\n\nHere is the patched code: \n\n${patch.singleHunkPatchText}"
+                )
+            }
+        }
+
+        return listOf(rollback, runStreamButton, repairButton)
     }
 
+    private fun showStreamDiff() {
+        FileEditorManager.getInstance(myProject).openFile(virtualFile, true)
+        val editor = FileEditorManager.getInstance(myProject).selectedTextEditor ?: return
+        val newText = appliedPatch!!.patchedText
+
+        val diffStreamHandler = DiffStreamHandler(
+            myProject,
+            editor = editor,
+            0,
+            oldCode.lines().size,
+            onClose = {
+            },
+            onFinish = {
+
+            })
+
+        diffStreamHandler.normalDiff(oldCode, newText)
+    }
+
+
     override fun getViewText(): String = virtualFile.readText()
 
     override fun updateViewText(text: String) {}