fix(sketch): improve run button for DevIns code execution

Replace AutoDevInsertCodeAction with direct DumbAwareAction to handle
text insertion. Update UI elements including borders, spacing, and
the "Fewer lines" label to "Hidden". Fix presentation handling for
the run button and simplify EditorFragment implementation.

Author: phodal

core/src/main/kotlin/cc/unitmesh/devti/gui/chat/view/MessageView.kt

@@ -16,6 +16,7 @@ import com.intellij.openapi.actionSystem.AnActionEvent
 import com.intellij.openapi.actionSystem.DefaultActionGroup
 import com.intellij.openapi.application.ApplicationManager
 import com.intellij.openapi.application.runInEdt
+import com.intellij.openapi.editor.colors.EditorColorsManager
 import com.intellij.openapi.fileTypes.PlainTextLanguage
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.ui.DialogPanel

core/src/main/kotlin/cc/unitmesh/devti/sketch/ui/code/CodeHighlightSketch.kt

@@ -4,19 +4,19 @@ import cc.unitmesh.devti.AutoDevBundle
 import cc.unitmesh.devti.AutoDevIcons
 import cc.unitmesh.devti.AutoDevNotifications
 import cc.unitmesh.devti.command.dataprovider.BuiltinCommand
-import cc.unitmesh.devti.gui.snippet.AutoDevInsertCodeAction
-import cc.unitmesh.devti.gui.snippet.AutoDevRunAction
+import cc.unitmesh.devti.gui.chat.ui.AutoInputService
 import cc.unitmesh.devti.provider.BuildSystemProvider
 import cc.unitmesh.devti.provider.RunService
+import cc.unitmesh.devti.sketch.AutoSketchMode
 import cc.unitmesh.devti.sketch.ui.LangSketch
 import cc.unitmesh.devti.util.parser.CodeFence
 import com.intellij.icons.AllIcons
 import com.intellij.ide.scratch.ScratchRootType
 import com.intellij.lang.Language
 import com.intellij.openapi.Disposable
-import com.intellij.openapi.actionSystem.ActionManager
 import com.intellij.openapi.actionSystem.ActionToolbar
 import com.intellij.openapi.actionSystem.DataProvider
+import com.intellij.openapi.actionSystem.Presentation
 import com.intellij.openapi.actionSystem.impl.ActionButton
 import com.intellij.openapi.application.ReadAction
 import com.intellij.openapi.command.WriteCommandAction
@@ -27,6 +27,7 @@ import com.intellij.openapi.editor.ex.EditorEx
 import com.intellij.openapi.fileEditor.FileDocumentManager
 import com.intellij.openapi.fileEditor.FileEditor
 import com.intellij.openapi.fileTypes.PlainTextLanguage
+import com.intellij.openapi.project.DumbAwareAction
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.util.Disposer
 import com.intellij.openapi.util.text.StringUtil
@@ -88,7 +89,7 @@ open class CodeHighlightSketch(
     fun initEditor(text: String, fileName: String? = null) {
         if (hasSetupAction) return
         hasSetupAction = true
-        
+
         if (isUser) {
             setupSimpleEditor(text, fileName)
             return
@@ -117,53 +118,55 @@ open class CodeHighlightSketch(
 
     private fun setupSimpleEditor(text: String, fileName: String?) {
         val editor = EditorUtil.createCodeViewerEditor(project, text, ideaLanguage, fileName, this)
-        
+
         border = if (withLeftRightBorder) {
             JBEmptyBorder(4, 4, 4, 4)
         } else {
             JBEmptyBorder(4, 0, 0, 0)
         }
-        
+
         editor.component.isOpaque = true
         editorFragment = EditorFragment(editor, editorLineThreshold, previewEditor)
         add(editorFragment!!.getContent())
-        
+
         setupToolbarAndStyling(fileName, editor)
     }
-    
+
     private fun setupRegularEditor(editor: EditorEx) {
         editorFragment = EditorFragment(editor, editorLineThreshold, previewEditor)
         add(editorFragment!!.getContent())
     }
-    
+
     private fun setupToolbarAndStyling(fileName: String?, editor: EditorEx) {
         val isDeclarePackageFile = BuildSystemProvider.isDeclarePackageFile(fileName)
         val lowercase = textLanguage?.lowercase()
-        
+
         if (textLanguage != null && lowercase != "markdown" && lowercase != "plain text") {
             if (showToolbar && lowercase != "devin") {
-                toolbar = setupActionBar(project, editor, isDeclarePackageFile, showBottomBorder = devInsCollapsedPanel != null)
+                toolbar = setupActionBar(
+                    project,
+                    editor,
+                    isDeclarePackageFile,
+                    showBottomBorder = devInsCollapsedPanel != null
+                )
             }
         } else {
             editorFragment?.editor?.backgroundColor = JBColor.PanelBackground
         }
 
         when (lowercase) {
             "devin" -> editorFragment?.editor?.setBorder(JBEmptyBorder(1, 1, 0, 1))
-            "markdown" -> { /* no border changes needed */ }
+            "markdown" -> { /* no border changes needed */
+            }
+
             else -> editorFragment?.editor?.setBorder(JBEmptyBorder(1, 0, 0, 0))
         }
     }
 
     private fun setupDevInsView(text: String) {
         devInsCollapsedPanel = JPanel(BorderLayout()).apply {
             border = JBUI.Borders.empty(2)
-
-            val runAction =
-                ActionManager.getInstance().getAction("AutoDev.ToolWindow.Snippet.InsertCode") as? AutoDevInsertCodeAction
-                    ?: AutoDevInsertCodeAction()
-
-            runButton = createRunButton(runAction)
+            runButton = createRunButton(text)
 
             val firstLine = text.lines().firstOrNull() ?: ""
             val previewLabel = JBLabel(firstLine).apply {
@@ -210,10 +213,19 @@ open class CodeHighlightSketch(
         updateRunButtonIcon()
     }
 
-    private fun createRunButton(runAction: AutoDevInsertCodeAction): ActionButton {
+    var devinRunButtonPresentation = Presentation()
+    private fun createRunButton(newText: String): ActionButton {
+        devinRunButtonPresentation.icon = AutoDevIcons.RUN
         return ActionButton(
-            runAction,
-            runAction.templatePresentation.clone(),
+            DumbAwareAction.create {
+                val sketchService = project.getService(AutoSketchMode::class.java)
+                if (sketchService.listener == null) {
+                    AutoInputService.getInstance(project).putText(newText)
+                } else {
+                    sketchService.send(newText)
+                }
+            },
+            devinRunButtonPresentation,
             "AutoDevToolbar",
             JBUI.size(24, 24)
         )
@@ -222,7 +234,7 @@ open class CodeHighlightSketch(
     private fun updateRunButtonIcon() {
         runButton?.let { button: ActionButton ->
             val icon = if (isComplete) AutoDevIcons.RUN else AutoDevIcons.LOADING
-            button.setIcon(icon)
+            devinRunButtonPresentation.setIcon(icon)
             button.repaint()
         }
     }
@@ -243,7 +255,7 @@ open class CodeHighlightSketch(
     }
 
     private fun createFewerLinesLabel(): JBLabel {
-        return JBLabel("Fewer lines", AllIcons.General.ChevronUp, JBLabel.LEFT).apply {
+        return JBLabel("Hidden", AllIcons.General.ChevronUp, JBLabel.LEFT).apply {
             border = JBUI.Borders.empty(4, 8)
             cursor = Cursor.getPredefinedCursor(Cursor.HAND_CURSOR)
             isOpaque = true
@@ -314,7 +326,7 @@ open class CodeHighlightSketch(
             if (lineCount > editorLineThreshold) {
                 editorFragment?.updateExpandCollapseLabel()
             }
-            
+
             // Auto-collapse DevIns view when complete
             if (complete && isDevIns && !isCollapsed) {
                 toggleEditorVisibility()

core/src/main/kotlin/cc/unitmesh/devti/sketch/ui/code/EditorFragment.kt

@@ -1,38 +1,17 @@
 package cc.unitmesh.devti.sketch.ui.code
 
 import com.intellij.icons.AllIcons
-import com.intellij.openapi.editor.Editor
-import com.intellij.openapi.editor.event.CaretEvent
-import com.intellij.openapi.editor.event.CaretListener
 import com.intellij.openapi.editor.ex.EditorEx
 import com.intellij.openapi.fileEditor.FileEditor
-import com.intellij.ui.JBColor
 import com.intellij.ui.components.JBLabel
 import com.intellij.util.ui.JBUI
 import com.intellij.util.ui.components.BorderLayoutPanel
-import java.awt.Color
 import java.awt.Dimension
 import java.awt.Insets
 import java.awt.event.MouseAdapter
 import java.awt.event.MouseEvent
-import javax.swing.Box
 import javax.swing.JComponent
 
-class EditorPadding(private val editor: Editor, pad: Int) :
-    Box.Filler(Dimension(pad, pad), Dimension(pad, pad), Dimension(pad, pad)) {
-    init {
-        setOpaque(true)
-        editor.caretModel.addCaretListener(object : CaretListener {
-            override fun caretPositionChanged(event: CaretEvent) {
-                this@EditorPadding.repaint()
-            }
-        })
-    }
-
-    override fun getBackground(): Color = editor.contentComponent.getBackground()
-}
-
-
 class EditorFragment(
     var editor: EditorEx,
     private val editorLineThreshold: Int = EDITOR_LINE_THRESHOLD,
@@ -41,22 +20,36 @@ class EditorFragment(
     private val expandCollapseTextLabel: JBLabel = JBLabel("", 0).apply {
         isOpaque = true
         isVisible = false
-        border = JBUI.Borders.customLine(JBColor.border(), 0, 1, 1, 1)
+        border = JBUI.Borders.empty()
     }
 
-    private val content: BorderLayoutPanel = createContentPanel().also {
-        it.border = JBUI.Borders.empty()
+    private val content: BorderLayoutPanel = createContentPanel().apply {
+        this.border = JBUI.Borders.empty()
     }
 
     private var collapsed = false
 
+    init {
+        expandCollapseTextLabel.addMouseListener(object : MouseAdapter() {
+            override fun mouseClicked(e: MouseEvent?) {
+                toggleCollapsedState()
+            }
+        })
+
+        editor.component.border = JBUI.Borders.empty()
+        editor.contentComponent.border = JBUI.Borders.empty()
+        editor.setBorder(JBUI.Borders.empty())
+        editor.headerComponent?.border = JBUI.Borders.empty()
+    }
+
     private fun createContentPanel(): BorderLayoutPanel {
         return object : BorderLayoutPanel() {
             override fun getPreferredSize(): Dimension {
                 val preferredSize = super.getPreferredSize()
                 if (editor.document.lineCount > editorLineThreshold && collapsed) {
                     return calculatePreferredSize(preferredSize, this@EditorFragment.editorLineThreshold)
                 }
+
                 return preferredSize
             }
         }.apply {
@@ -84,14 +77,6 @@ class EditorFragment(
         return height + headerHeight + labelHeight + insets.height
     }
 
-    init {
-        expandCollapseTextLabel.addMouseListener(object : MouseAdapter() {
-            override fun mouseClicked(e: MouseEvent?) {
-                toggleCollapsedState()
-            }
-        })
-    }
-
     fun getContent(): JComponent = content
 
     fun isCollapsed(): Boolean = collapsed