feat(completion): add support for text replacement in code completion

- Add `isReplacement` parameter to `CodeCompletionRequest` class.
- Modify `CodeCompletionRequest.create()` method to include `isReplacement` parameter.
- Add `Replace` interaction type in `TeamPromptExecTask.kt`.
- Implement text replacement logic in `BaseCompletionTask.kt`.
- Create `replaceText()` function in `InsertUtil.kt` to perform text replacement.

This commit adds support for text replacement in code completion. It introduces a new `isReplacement` parameter to the `CodeCompletionRequest` class, which allows specifying whether the code completion is for replacement purposes. The `CodeCompletionRequest.create()` method is modified to include this new parameter. In `TeamPromptExecTask.kt`, a new interaction type called `Replace` is added, which triggers the code completion process with text replacement enabled. The text replacement logic is implemented in the `BaseCompletionTask.kt` file, which replaces the selected text with the suggested completion. Finally, a new function called `replaceText()` is created in `InsertUtil.kt` to perform the actual text replacement.

Author: phodal

src/main/kotlin/cc/unitmesh/devti/custom/team/CustomActionPrompt.kt

@@ -125,5 +125,6 @@ enum class InteractionType {
     AppendCursor,
     AppendCursorStream,
     OutputFile,
+    Replace
     ;
 }

src/main/kotlin/cc/unitmesh/devti/custom/team/TeamPromptExecTask.kt

@@ -7,7 +7,9 @@ import cc.unitmesh.devti.gui.chat.ChatActionType
 import cc.unitmesh.devti.gui.sendToChatWindow
 import cc.unitmesh.devti.intentions.action.task.BaseCompletionTask
 import cc.unitmesh.devti.intentions.action.task.CodeCompletionRequest
+import com.intellij.openapi.application.invokeLater
 import com.intellij.openapi.application.runReadAction
+import com.intellij.openapi.application.runWriteAction
 import com.intellij.openapi.editor.Editor
 import com.intellij.openapi.progress.ProgressIndicator
 import com.intellij.openapi.progress.ProgressManager
@@ -72,7 +74,21 @@ class TeamPromptExecTask(
                 ProgressManager.getInstance()
                     .runProcessWithProgressAsynchronously(task, BackgroundableProcessIndicator(task))
             }
+
+            InteractionType.Replace -> {
+                val msgString = systemPrompt + "\n" + userPrompt
+                val request = runReadAction {
+                    CodeCompletionRequest.create(editor, offset, element, null, msgString, isReplacement = true)
+                } ?: return
+
+                val task = object : BaseCompletionTask(request) {
+                    override fun keepHistory(): Boolean = false
+                    override fun promptText(): String = msgString
+                }
+
+                ProgressManager.getInstance()
+                    .runProcessWithProgressAsynchronously(task, BackgroundableProcessIndicator(task))
+            }
         }
     }
-
 }

src/main/kotlin/cc/unitmesh/devti/intentions/action/task/BaseCompletionTask.kt

@@ -53,7 +53,7 @@ abstract class BaseCompletionTask(private val request: CodeCompletionRequest) :
         var currentOffset = request.offset
 
         indicator.isIndeterminate = true
-        indicator.fraction = 0.8
+        indicator.fraction = 0.5
         indicator.text = AutoDevBundle.message("intentions.request.background.process.title")
 
         LLMCoroutineScope.scope(request.project).launch {
@@ -67,13 +67,18 @@ abstract class BaseCompletionTask(private val request: CodeCompletionRequest) :
 
                 suggestion.append(char as String)
                 invokeLater {
-                    if (!isCanceled) {
+                    if (!isCanceled && !request.isReplacement) {
                         InsertUtil.insertStreamingToDoc(project, char, editor, currentOffset)
                         currentOffset += char.length
                     }
                 }
             }
 
+            if (request.isReplacement) {
+                InsertUtil.replaceText(project, editor, request.element, suggestion.toString())
+            }
+
+            indicator.fraction = 0.8
             AutoDevStatusService.notifyApplication(AutoDevStatus.Done)
             logger.info("Suggestion: $suggestion")
         }

src/main/kotlin/cc/unitmesh/devti/intentions/action/task/CodeCompletionRequest.kt

@@ -19,10 +19,18 @@ class CodeCompletionRequest(
     val documentVersion: Long,
     val element: PsiElement?,
     val editor: Editor,
-    val suffixText: String
+    val suffixText: String,
+    val isReplacement: Boolean = false,
 ) : Disposable {
     companion object {
-        fun create(editor: Editor, offset: Int, element: PsiElement?, prefix: String?, suffix: String?): CodeCompletionRequest? {
+        fun create(
+            editor: Editor,
+            offset: Int,
+            element: PsiElement?,
+            prefix: String?,
+            suffix: String?,
+            isReplacement: Boolean = false
+        ): CodeCompletionRequest? {
             val project = editor.project ?: return null
             val document = editor.document
             val file = PsiDocumentManager.getInstance(project).getPsiFile(document) ?: return null
@@ -47,7 +55,8 @@ class CodeCompletionRequest(
                 documentVersion,
                 element,
                 editor,
-                suffixText
+                suffixText,
+                isReplacement
             )
 
         }

src/main/kotlin/cc/unitmesh/devti/util/InsertUtil.kt

@@ -1,13 +1,15 @@
 package cc.unitmesh.devti.util
 
 import cc.unitmesh.devti.AutoDevBundle
+import com.intellij.openapi.application.runReadAction
 import com.intellij.openapi.command.WriteCommandAction
 import com.intellij.openapi.editor.Document
 import com.intellij.openapi.editor.Editor
 import com.intellij.openapi.editor.ScrollType
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.util.TextRange
 import com.intellij.psi.PsiDocumentManager
+import com.intellij.psi.PsiElement
 import com.intellij.psi.codeStyle.CodeStyleManager
 
 object InsertUtil {
@@ -42,4 +44,15 @@ object InsertUtil {
         editor.caretModel.moveToOffset(currentOffset + char.length)
         editor.scrollingModel.scrollToCaret(ScrollType.MAKE_VISIBLE)
     }
+
+    fun replaceText(project: Project, editor: Editor, element: PsiElement?, output: String) {
+        val primaryCaret = editor.caretModel.primaryCaret;
+        val start = runReadAction { primaryCaret.selectionStart }
+        val end = runReadAction { primaryCaret.selectionEnd }
+        val document = runReadAction { editor.document }
+
+        WriteCommandAction.runWriteCommandAction(project) {
+            document.replaceString(start, end, output)
+        }
+    }
 }