feat(ext-terminal): refactor suggestCommand method to support 241 new UI context and add TerminalUtil class for message sending #135.

Author: phodal

exts/ext-terminal/src/222/main/kotlin/cc/unitmesh/terminal/TerminalUtil.kt

@@ -11,9 +11,9 @@ import org.jetbrains.plugins.terminal.TerminalView
 object TerminalUtil {
     fun sendMsg(project: Project, data: String, e: AnActionEvent) {
         val widget = getCurrentTerminalWidget(project) ?: return
-        suggestCommand(data, project) { string ->
+        suggestCommand(data, project, { string ->
             widget.terminalStarter?.sendString(string, true)
-        }
+        }, {})
     }
 
     fun getCurrentTerminalWidget(project: Project): JBTerminalWidget? {

exts/ext-terminal/src/233/main/kotlin/cc/unitmesh/terminal/TerminalUtil.kt

@@ -11,9 +11,9 @@ import org.jetbrains.plugins.terminal.TerminalToolWindowManager
 object TerminalUtil {
     fun sendMsg(project: Project, data: String, e: AnActionEvent) {
         val widget = getCurrentTerminalWidget(project) ?: return
-        suggestCommand(data, project) { string ->
+        suggestCommand(data, project, { string ->
             widget.terminalStarter?.sendString(string, true)
-        }
+        }, {})
     }
 
     fun getCurrentTerminalWidget(project: Project): JBTerminalWidget? {

exts/ext-terminal/src/241/main/kotlin/cc/unitmesh/terminal/TerminalUtil.kt

@@ -25,19 +25,19 @@ object TerminalUtil {
         }
 
         val sb = StringBuilder()
-        suggestCommand(data, project) { string ->
+        suggestCommand(data, project, { string ->
             sb.append(string)
             null
-        }
-
-        runInEdt {
-            CopyPasteManager.copyTextToClipboard(sb.toString())
-            controller.performPaste(e.dataContext)
-        }
+        }, {
+            runInEdt {
+                CopyPasteManager.copyTextToClipboard(sb.toString())
+                controller.performPaste(e.dataContext)
+            }
+        })
     }
 
     private fun tryGetBlockTerminalEditor(findWidgetByContent: TerminalWidget): TerminalPromptController? {
-        val terminalView = (findWidgetByContent.component as Wrapper).targetComponent
+        val terminalView = (findWidgetByContent.component as? Wrapper)?.targetComponent ?: return null
         if (terminalView is DataProvider) {
             val controller = terminalView.getData(TerminalPromptController.KEY.name)
             return (controller as? TerminalPromptController)
@@ -48,9 +48,9 @@ object TerminalUtil {
 
     private fun trySendMsgInOld(project: Project, data: String, content: Content): Boolean {
         val widget = TerminalToolWindowManager.getWidgetByContent(content) ?: return true
-        suggestCommand(data, project) { string ->
+        suggestCommand(data, project, { string ->
             widget.terminalStarter?.sendString(string, true)
-        }
+        }, {})
 
         return false
     }

exts/ext-terminal/src/main/kotlin/cc/unitmesh/terminal/ShellCommandSuggestAction.kt

@@ -125,7 +125,10 @@ class ShellCommandSuggestAction : DumbAwareAction() {
     }
 
     companion object {
-        fun suggestCommand(data: String, project: Project, function: (str: String) -> Unit?) {
+        fun suggestCommand(data: String, project: Project,
+                           chunk: (str: String) -> Unit?,
+                           done: ((str: String) -> Unit?)?
+        ) {
             val templateRender = TemplateRender(GENIUS_PRACTISES)
             val template = templateRender.getTemplate("shell-suggest.vm")
 
@@ -145,13 +148,17 @@ class ShellCommandSuggestAction : DumbAwareAction() {
                 AutoDevStatusService.notifyApplication(AutoDevStatus.InProgress)
 
                 try {
+                    val sb = StringBuilder()
                     stringFlow.collect {
+                        sb.append(it)
                         if (it.contains("\n")) {
                             throw Exception("Shell command suggestion failed")
                         }
 
-                        function(it)
+                        chunk(it)
                     }
+
+                    done?.invoke(sb.toString())
                 } finally {
                     AutoDevStatusService.notifyApplication(AutoDevStatus.Ready)
                 }