refactor(gui): improve chat coding panel UI and add message rendering logic #51

The chat coding panel has been refactored to improve the user interface and add a new message rendering logic. The changes include:

Author: phodal

src/main/kotlin/cc/unitmesh/devti/counit/CustomAgentChatProcessor.kt

@@ -32,8 +32,12 @@ class CustomAgentChatProcessor(val project: Project) {
         selectedAgent.state = CustomAgentState.FINISHED
         when (selectedAgent.responseAction) {
             ResponseAction.Direct -> {
-                ui.addMessage(response, false, response)
+                ui.addMessage(response, false, response).also {
+                    it.updateContent(response)
+                    it.reRenderAssistantOutput()
+                }
                 ui.hiddenProgressBar()
+                ui.updateUI()
             }
 
             ResponseAction.TextChunk -> {
@@ -42,8 +46,12 @@ class CustomAgentChatProcessor(val project: Project) {
             }
 
             ResponseAction.Flow -> {
-                ui.addMessage(response, false, response)
+                ui.addMessage(response, false, response).also {
+                    it.updateContent(response)
+                    it.reRenderAssistantOutput()
+                }
                 ui.hiddenProgressBar()
+                ui.updateUI()
             }
 
             ResponseAction.WebView -> {

src/main/kotlin/cc/unitmesh/devti/gui/chat/ChatCodingPanel.kt

@@ -118,6 +118,9 @@ class ChatCodingPanel(private val chatCodingService: ChatCodingService, val disp
         }
     }
 
+    /**
+     * Add a message to the chat panel and update ui
+     */
     fun addMessage(message: String, isMe: Boolean = false, displayPrompt: String = ""): MessageView {
         val role = if (isMe) ChatRole.User else ChatRole.Assistant
         val displayText = displayPrompt.ifEmpty { message }
@@ -129,7 +132,6 @@ class ChatCodingPanel(private val chatCodingService: ChatCodingService, val disp
         scrollToBottom()
         progressBar.isIndeterminate = true
         updateUI()
-
         return messageView
     }
 
@@ -200,10 +202,7 @@ class ChatCodingPanel(private val chatCodingService: ChatCodingService, val disp
         }.collect {
             text += it
 
-            // 以下两个 API 设计不合理，如果必须要同时调用，那就只提供一个就好了
-            messageView.updateSourceContent(text)
             messageView.updateContent(text)
-
             messageView.scrollToBottom()
         }
 

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

@@ -103,16 +103,13 @@ class MessageView(private val message: String, val role: ChatRole, private val d
         }
     }
 
+    private var answer: String = ""
+
     fun updateContent(content: String) {
+        this.answer = content
         MessageWorker(content).execute()
     }
 
-
-    private var answer: String = ""
-    fun updateSourceContent(source: String) {
-        answer = source
-    }
-
     fun scrollToBottom() {
         SwingUtilities.invokeLater {
             val bounds: Rectangle = bounds
@@ -122,7 +119,6 @@ class MessageView(private val message: String, val role: ChatRole, private val d
 
     fun reRenderAssistantOutput() {
         ApplicationManager.getApplication().invokeLater {
-
             centerPanel.remove(component)
             centerPanel.updateUI()
 

src/main/kotlin/com/intellij/temporary/gui/block/CodeBlockView.kt

@@ -8,7 +8,6 @@ import com.intellij.openapi.Disposable
 import com.intellij.openapi.actionSystem.ActionPlaces
 import com.intellij.openapi.actionSystem.ex.ActionUtil
 import com.intellij.openapi.actionSystem.impl.ActionToolbarImpl
-import com.intellij.openapi.application.ApplicationManager
 import com.intellij.openapi.application.ReadAction
 import com.intellij.openapi.editor.Document
 import com.intellij.openapi.editor.Editor
@@ -124,7 +123,7 @@ class CodeBlockView(
             val markupModel: MarkupModelEx = editor.markupModel
             (markupModel as EditorMarkupModel).isErrorStripeVisible = false
 
-            val settings = editor.getSettings().also {
+            val settings = editor.settings.also {
                 it.isDndEnabled = false
                 it.isLineNumbersShown = false
                 it.additionalLinesCount = 0
@@ -160,14 +159,12 @@ class CodeBlockView(
             message: CompletableMessage
         ): CodePartEditorInfo {
             val forceFoldEditorByDefault = message.getRole() === ChatRole.User
-            val content = graphProperty.get()
-
-            val createCodeViewerFile: VirtualFile = createCodeViewerFile(language, content)
+            val createCodeViewerFile = createCodeViewerFile(language, graphProperty.get())
             val document: Document =
                 createCodeViewerFile.findDocument() ?: throw IllegalStateException("Document not found")
 
             val editor: EditorEx =
-                createCodeViewerEditor(project, createCodeViewerFile as LightVirtualFile, document, disposable)
+                createCodeViewerEditor(project, createCodeViewerFile, document, disposable)
 
             val toolbarActionGroup = ActionUtil.getActionGroup("AutoDev.ToolWindow.Snippet.Toolbar")!!
             toolbarActionGroup.let {