pref: 优化panle输出, 可以在输出时选中, 并且不会一直刷新processbar (#250)

Author: c913168497

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

@@ -220,25 +220,29 @@ class ChatCodingPanel(private val chatCodingService: ChatCodingService, val disp
     private suspend fun updateMessageInUi(content: Flow<String>): String {
         val messageView = MessageView("", ChatRole.Assistant, "")
         myList.add(messageView)
-        val startTime = System.currentTimeMillis() // 记录代码开始执行的时间
-
+        val startTime = System.currentTimeMillis()
         var text = ""
-        content.onCompletion {
-            logger.info("onCompletion ${it?.message}")
-            hiddenProgressBar()
-        }.catch {
-            it.printStackTrace()
-        }.collect {
-            text += it
-
+        val batchSize = 5
+        val buffer = mutableListOf<String>()
+
+        content
+            .buffer(capacity = 10)
+            .collect { newText ->
+                buffer.add(newText)
+                if (buffer.size >= batchSize) {
+                    text += buffer.joinToString("")
+                    buffer.clear()
+                    messageView.updateContent(text)
+                }
+            }
+        // 处理剩余的缓冲内容
+        if (buffer.isNotEmpty()) {
+            text += buffer.joinToString("")
             messageView.updateContent(text)
-            messageView.scrollToBottom()
         }
 
         if (delaySeconds.isNotEmpty()) {
             val elapsedTime = System.currentTimeMillis() - startTime
-            // waiting for the last message to be rendered, like sleep 5 ms?
-            // 此处的 20s 出自 openAI 免费账户访问 3/min
             withContext(Dispatchers.IO) {
                 val delaySec = delaySeconds.toLong()
                 val remainingTime = maxOf(delaySec * 1000 - elapsedTime, 0)
@@ -247,7 +251,6 @@ class ChatCodingPanel(private val chatCodingService: ChatCodingService, val disp
         }
 
         messageView.reRenderAssistantOutput()
-
         return text
     }
 

core/src/main/kotlin/cc/unitmesh/devti/gui/component/DisplayComponent.kt

@@ -26,11 +26,23 @@ class DisplayComponent(question: String) : JEditorPane() {
     }
 
     fun updateMessage(content: String) {
+        // 记录当前的选中状态
+        val selectionStart = this.selectionStart
+        val selectionEnd = this.selectionEnd
         this.text = content
+        if (selectionStart != selectionEnd) {
+            SwingUtilities.invokeLater {
+                this.setSelectionStart(selectionStart)
+                this.setSelectionEnd(selectionEnd)
+            }
+        }
     }
 
+
     private fun stripHtmlAndUnescapeXmlEntities(input: String): String {
+        // 使用 Jsoup 去除 HTML 标签
         val text = Jsoup.parse(input).text()
+        // 使用 Apache Commons Text 解码 XML 实体
         return StringEscapeUtils.unescapeXml(text)
     }
-}
+}