feat(code): support multiple code blocks in CodeHighlightSketch #331

- Add example structure to plan.devin files for clarity
- Refactor CodeHighlightSketch to handle multiple code blocks using VerticalLayout
- Remove outdated TODO comment in AgentStateService

Author: phodal

core/src/main/kotlin/cc/unitmesh/devti/observer/agent/AgentStateService.kt

@@ -33,7 +33,6 @@ class AgentStateService {
      */
     fun processMessages(messages: List<Message>): List<Message> {
         state.messages = messages
-        /// todo compress message in here
         return messages
     }
 

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

@@ -34,8 +34,10 @@ import com.intellij.openapi.util.text.StringUtil
 import com.intellij.openapi.vfs.VirtualFile
 import com.intellij.psi.PsiManager
 import cc.unitmesh.devti.util.whenDisposed
+import com.intellij.openapi.ui.VerticalFlowLayout
 import com.intellij.testFramework.LightVirtualFile
 import com.intellij.ui.components.JBPanel
+import com.intellij.ui.components.panels.VerticalLayout
 import com.intellij.util.concurrency.annotations.RequiresReadLock
 import com.intellij.util.ui.JBEmptyBorder
 import com.intellij.util.ui.JBUI
@@ -162,32 +164,38 @@ open class CodeHighlightSketch(
             }
         }
 
-        val parse = CodeFence.parse(editorFragment!!.editor.document.text)
-        var panel: JComponent? = null
-        when (parse.originLanguage) {
-            "diff", "patch" -> {
-                val langSketch = LanguageSketchProvider.provide("patch")?.create(project, parse.text) ?: return
-                panel = langSketch.getComponent()
-                langSketch.onDoneStream(allText)
-            }
+        val codes = CodeFence.parseAll(editorFragment!!.editor.document.text)
+        val blockedPanel = JPanel(VerticalLayout(JBUI.scale(0)))
 
-            "html" -> {
-                val langSketch = LanguageSketchProvider.provide("html")?.create(project, parse.text) ?: return
-                panel = langSketch.getComponent()
-                langSketch.onDoneStream(allText)
-            }
+        codes.forEach { code ->
+            var panel: JComponent? = null
+            when (code.originLanguage) {
+                "diff", "patch" -> {
+                    val langSketch = LanguageSketchProvider.provide("patch")?.create(project, code.text) ?: return
+                    panel = langSketch.getComponent()
+                    langSketch.onDoneStream(allText)
+                }
+
+                "html" -> {
+                    val langSketch = LanguageSketchProvider.provide("html")?.create(project, code.text) ?: return
+                    panel = langSketch.getComponent()
+                    langSketch.onDoneStream(allText)
+                }
 
-            "bash", "shell" -> {
-                val langSketch = LanguageSketchProvider.provide("shell")?.create(project, parse.text) ?: return
-                panel = langSketch.getComponent()
-                langSketch.onDoneStream(allText)
+                "bash", "shell" -> {
+                    val langSketch = LanguageSketchProvider.provide("shell")?.create(project, code.text) ?: return
+                    panel = langSketch.getComponent()
+                    langSketch.onDoneStream(allText)
+                }
             }
-        }
 
-        if (panel == null) return
+            if (panel == null) return@forEach
+
+            panel.border = JBEmptyBorder(4)
+            blockedPanel.add(panel)
+        }
 
-        panel.border = JBEmptyBorder(4)
-        add(panel, BorderLayout.SOUTH)
+        add(blockedPanel, BorderLayout.SOUTH)
 
         editorFragment?.updateExpandCollapseLabel()
 

core/src/main/resources/genius/en/code/plan.devin

@@ -58,6 +58,14 @@ Here is the rule you should follow:
 If `<user.question>` directly contradicts any of these steps, follow the instructions from `<user.question>`
 first. Be thorough in your thinking process, so it's okay if it is lengthy.
 
+For example:
+
+<THOUGHT>
+<PLAN>
+Some plan
+</PLAN>
+</THOUGHT>
+
 Here is user.question:
 
 <user.question>user.question</user.question>

core/src/main/resources/genius/zh/code/plan.devin

@@ -58,6 +58,14 @@ Here is the rule you should follow:
 If `<user.question>` directly contradicts any of these steps, follow the instructions from `<user.question>`
 first. Be thorough in your thinking process, so it's okay if it is lengthy.
 
+For example:
+
+<THOUGHT>
+<PLAN>
+Some plan
+</PLAN>
+</THOUGHT>
+
 Here is user.question:
 
 <user.question>user.question</user.question>