From 83cf83f1d23ab3d736e98665d2879cbe3cacb4da Mon Sep 17 00:00:00 2001
From: Phodal Huang <h@phodal.com>
Date: Thu, 18 Apr 2024 12:57:54 +0800
Subject: [PATCH] chore: init for runnerbar action

---
 .../main/resources/META-INF/autodev-core.xml  |  8 ++++
 .../devti/actions/run/RunPanelFixAction.kt    | 40 +++++++++++++++++++
 2 files changed, 48 insertions(+)
 create mode 100644 src/main/kotlin/cc/unitmesh/devti/actions/run/RunPanelFixAction.kt

diff --git a/src/233/main/resources/META-INF/autodev-core.xml b/src/233/main/resources/META-INF/autodev-core.xml
index 19c9c43585..e60ec8c718 100644
--- a/src/233/main/resources/META-INF/autodev-core.xml
+++ b/src/233/main/resources/META-INF/autodev-core.xml
@@ -294,6 +294,14 @@
             <add-to-group group-id="ConsoleEditorPopupMenu" anchor="first"/>
         </action>
 
+<!--        <action id="runnerBar.FixAction"-->
+<!--                class="cc.unitmesh.devti.actions.run.RunPanelFixAction"-->
+<!--                description="Ask AI fix error in tool bar"-->
+<!--                icon="cc.unitmesh.devti.AutoDevIcons.AI_COPILOT">-->
+
+<!--            <add-to-group group-id="RunTab.TopToolbar" anchor="first"/>-->
+<!--        </action>-->
+
         <action id="autodev.Vcs.CommitMessage"
                 class="cc.unitmesh.devti.actions.vcs.CommitMessageSuggestionAction"
                 icon="cc.unitmesh.devti.AutoDevIcons.AI_COPILOT"
diff --git a/src/main/kotlin/cc/unitmesh/devti/actions/run/RunPanelFixAction.kt b/src/main/kotlin/cc/unitmesh/devti/actions/run/RunPanelFixAction.kt
new file mode 100644
index 0000000000..a58fd492ea
--- /dev/null
+++ b/src/main/kotlin/cc/unitmesh/devti/actions/run/RunPanelFixAction.kt
@@ -0,0 +1,40 @@
+package cc.unitmesh.devti.actions.run
+
+import cc.unitmesh.devti.AutoDevNotifications
+import cc.unitmesh.devti.gui.chat.ChatActionType
+import cc.unitmesh.devti.gui.sendToChatWindow
+import cc.unitmesh.devti.provider.ContextPrompter
+import com.intellij.execution.ui.ConsoleView
+import com.intellij.execution.ui.RunContentDescriptor
+import com.intellij.execution.ui.RunContentManager
+import com.intellij.openapi.actionSystem.AnAction
+import com.intellij.openapi.actionSystem.AnActionEvent
+
+class RunPanelFixAction : AnAction() {
+    fun getActionType(): ChatActionType = ChatActionType.FIX_ISSUE
+
+    override fun actionPerformed(e: AnActionEvent) {
+        val project = e.project ?: return
+
+        val selectedContent: RunContentDescriptor = RunContentManager.getInstance(project).selectedContent ?: return
+        val console = selectedContent?.executionConsole as ConsoleView
+
+        val sb = StringBuilder()
+        console.addMessageFilter { line, _ ->
+            sb.append(line)
+            null
+        }
+
+        val content = sb.toString()
+        if (content.isNotEmpty()) {
+            sendToChatWindow(project, getActionType()) { panel, service ->
+                service.handlePromptAndResponse(panel, object : ContextPrompter() {
+                    override fun displayPrompt(): String = content
+                    override fun requestPrompt(): String = content
+                }, null, true)
+            }
+        } else {
+            AutoDevNotifications.error(project, "Cannot extract text from run panel.");
+        }
+    }
+}