feat: update for cursor position

unit-mesh · Jul 13, 2023 · 12feb34 · 12feb34
1 parent df087a4
commit 12feb34
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/src/main/kotlin/cc/unitmesh/devti/actions/chat/CodeCompleteAction.kt b/src/main/kotlin/cc/unitmesh/devti/actions/chat/CodeCompleteAction.kt
@@ -17,16 +17,13 @@ class CodeCompleteAction : ChatBaseAction() {
 
         val primaryCaret = editor.caretModel.primaryCaret;
         val start = primaryCaret.selectionStart
-        var end = primaryCaret.selectionEnd
-
-        if (start == end) {
-            primaryCaret.setSelection(start, start + 1)
-            end = start + 1
-        }
+        val end = primaryCaret.selectionEnd
 
         return { response ->
             WriteCommandAction.runWriteCommandAction(project) {
-                document.replaceString(start, end, response)
+                document.insertString(start, response)
+                primaryCaret.removeSelection()
+                primaryCaret.moveToOffset(end + response.length)
             }
         }
     }