feat(refactor): add post-action support for refactoring and code completion #129

This commit introduces a new `postAction` field to the `ChatContext` class, which allows for custom actions to be performed after a refactoring or code completion suggestion is applied. The `ChatCodingPanel` now includes a `SuggestionPanel` to display the suggested code, and the `ChatCodingService` has been updated to invoke the `postAction` callback instead of the deprecated `replaceSelectedText` callback. Additionally, the `RefactorThisAction` now includes a more detailed static analysis result and uses a random key from the `refactorIntentionsKeys` to generate a suggestion message.

Author: phodal

docs/features/chat-with-code.md

@@ -15,6 +15,8 @@ permalink: /features/chat-width-code
 
 ## Refactor this
 
+In [#129](https://github.com/unit-mesh/auto-dev/issues/129), we provide more information for you to refactor code.
+
 ## Chat with this
 
 ## Write test for this

src/main/kotlin/cc/unitmesh/devti/actions/chat/RefactorThisAction.kt

@@ -4,6 +4,7 @@ import cc.unitmesh.devti.AutoDevBundle
 import cc.unitmesh.devti.actions.chat.base.ChatBaseAction
 import cc.unitmesh.devti.gui.chat.ChatActionType
 import cc.unitmesh.devti.gui.chat.ChatCodingPanel
+import cc.unitmesh.devti.util.parser.Code
 import com.intellij.codeInsight.daemon.impl.DaemonCodeAnalyzerEx
 import com.intellij.lang.LanguageCommenters
 import com.intellij.openapi.actionSystem.AnActionEvent
@@ -13,7 +14,6 @@ import com.intellij.openapi.editor.Editor
 import com.intellij.openapi.project.Project
 import com.intellij.psi.PsiElement
 
-
 class RefactorThisAction : ChatBaseAction() {
     init{
         val presentation = getTemplatePresentation()
@@ -42,7 +42,7 @@ class RefactorThisAction : ChatBaseAction() {
         val commentSymbol = commentPrefix(element)
 
         return collectProblems(project, editor, element)?.let {
-            "\n\n$commentSymbol relative static analysis result: $it"
+            "\n\n$commentSymbol relative static analysis result: \n\n$it"
         } ?: ""
     }
 
@@ -101,9 +101,16 @@ class RefactorThisAction : ChatBaseAction() {
         val start = primaryCaret.selectionStart;
         val end = primaryCaret.selectionEnd
 
+        // get random key from refactorIntentionsKeys
+        val key = refactorIntentionsKeys.random()
+        val msg = AutoDevBundle.message(key)
+
         return { response ->
+            panel.showSuggestion(msg)
+
+            val code = Code.parse(response).text
             WriteCommandAction.runWriteCommandAction(project) {
-                document.replaceString(start, end, response)
+                document.replaceString(start, end, code)
             }
         }
     }

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

@@ -36,6 +36,8 @@ import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.delay
 import kotlinx.coroutines.flow.*
 import kotlinx.coroutines.withContext
+import org.jetbrains.annotations.Nls
+import java.awt.BorderLayout
 import java.awt.event.ActionListener
 import java.awt.event.MouseAdapter
 import java.awt.event.MouseEvent
@@ -56,6 +58,8 @@ class ChatCodingPanel(private val chatCodingService: ChatCodingService, val disp
     private val delaySeconds: String
         get() = AutoDevSettingsState.getInstance().delaySeconds
 
+    private var suggestionPanel: JPanel = JPanel(BorderLayout())
+
     init {
         focusMouseListener = object : MouseAdapter() {
             override fun mouseClicked(e: MouseEvent?) {
@@ -115,6 +119,7 @@ class ChatCodingPanel(private val chatCodingService: ChatCodingService, val disp
 
         panelContent = panel {
             row { cell(myScrollPane).fullWidth().fullHeight() }.resizableRow()
+            row { cell(suggestionPanel).fullWidth() }
             row { cell(progressBar).fullWidth() }
             row { cell(actionLink).alignRight() }
             row {
@@ -301,5 +306,26 @@ class ChatCodingPanel(private val chatCodingService: ChatCodingService, val disp
     fun moveCursorToStart() {
         inputSection.moveCursorToStart()
     }
+
+    fun showSuggestion(msg: @Nls String) {
+        val label = panel {
+            row {
+                cell(JBLabel(msg)).fullWidth().also {
+                    it.component.addMouseListener(object : MouseAdapter() {
+                        override fun mouseClicked(e: MouseEvent?) {
+                            inputSection.text = msg
+                            inputSection.requestFocus()
+
+                            suggestionPanel.removeAll()
+                            updateUI()
+                        }
+                    })
+                }
+            }
+        }
+
+        suggestionPanel.add(label)
+        updateUI()
+    }
 }
 

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

@@ -62,11 +62,11 @@ class ChatCodingService(var actionType: ChatActionType, val project: Project) {
             LLMCoroutineScope.scope(project).launch {
                 when {
                     actionType === ChatActionType.REFACTOR -> ui.updateReplaceableContent(response) {
-                        context?.replaceSelectedText?.invoke(getCodeSection(it, context.prefixText, context.suffixText))
+                        context?.postAction?.invoke(it)
                     }
 
                     actionType === ChatActionType.CODE_COMPLETE -> ui.updateReplaceableContent(response) {
-                        context?.replaceSelectedText?.invoke(getCodeSection(it, context.prefixText, context.suffixText))
+                        context?.postAction?.invoke(getCodeSection(it, context.prefixText, context.suffixText))
                     }
 
                     else -> ui.updateMessage(response)

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

@@ -1,7 +1,7 @@
 package cc.unitmesh.devti.gui.chat
 
 data class ChatContext(
-    val replaceSelectedText: ((response: String) -> Unit)? = null,
+    val postAction: ((response: String) -> Unit)? = null,
     val prefixText: String,
     val suffixText: String
 )