fix: fix paste override copyed lines

Author: shezhangzhang

.vscode/launch.json

@@ -10,8 +10,8 @@
       "type": "extensionHost",
       "request": "launch",
       "args": [
-        "--extensionDevelopmentPath=${workspaceFolder}",
-        "--disable-extensions"
+        "--extensionDevelopmentPath=${workspaceFolder}"
+        // "--disable-extensions"
       ],
       "outFiles": ["${workspaceFolder}/dist/**/*.js"],
       "preLaunchTask": "${defaultBuildTask}"

assets/decorations.gif

@@ -27,7 +27,9 @@ export default class DecorationManager {
     isEdit: boolean = false,
     diffLine: number = 0,
     startLine?: number,
-    endLine?: number
+    endLine?: number,
+    originalStartLine?: number,
+    originalEndLine?: number
   ) {
     /**
      * Active editor changed event
@@ -46,14 +48,32 @@ export default class DecorationManager {
 
     if (throttle) {
       this.timeout = setTimeout(() => {
-        this.setupDecorations(diffLine, startLine, endLine);
+        this.setupDecorations(
+          diffLine,
+          startLine,
+          endLine,
+          originalStartLine,
+          originalEndLine
+        );
       }, 500);
     } else {
-      this.setupDecorations(diffLine, startLine, endLine);
+      this.setupDecorations(
+        diffLine,
+        startLine,
+        endLine,
+        originalStartLine,
+        originalEndLine
+      );
     }
   }
 
-  setupDecorations(diffLine: number, startLine?: number, endLine?: number) {
+  setupDecorations(
+    diffLine: number,
+    startLine?: number,
+    endLine?: number,
+    originalStartLine?: number,
+    originalEndLine?: number
+  ) {
     if (startLine && endLine) {
       const lines = this.getLines(startLine, endLine);
 
@@ -68,6 +88,17 @@ export default class DecorationManager {
       }
 
       if (diffLine > 0) {
+        /**
+         * paste override copyed lines
+         */
+        if (originalStartLine && originalEndLine) {
+          const originalLines = this.getLines(
+            originalStartLine,
+            originalEndLine
+          );
+          this.clearCurrentFileDecoration(originalLines);
+        }
+
         this.updateFileDecorationMap(diffLine, startLine);
         this.setDecorations(lines);
       }

src/decoration-manager.ts

@@ -26,18 +26,18 @@ export default function setupEventListener(
         diffLine = activeEditor.document.lineCount - fileLineCount;
         fileLineCount = activeEditor.document.lineCount;
 
-        const [startLine, endLine] = getStartEndLine(
-          event.document,
-          event.contentChanges[0]
-        );
+        const [startLine, endLine, originalStartLine, originalEndLine] =
+          getStartEndLine(event.document, event.contentChanges[0]);
 
         decorationManager.setActiveEditor(activeEditor);
         decorationManager.triggerUpdateDecorations(
-          diffLine === 0,
+          event.reason ? false : diffLine === 0,
           true,
           diffLine,
           startLine,
-          endLine
+          endLine,
+          originalStartLine,
+          originalEndLine
         );
       }
     },
@@ -65,6 +65,8 @@ export default function setupEventListener(
     const newLines = change.text.split("\n").length;
     let startLine = change.range.start.line;
     let endLine = change.range.end.line;
+    const originalStartLine = startLine;
+    const originalEndLine = endLine;
     const startLinePos = document.lineAt(startLine);
 
     const isStartLineEmpty = startLinePos.isEmptyOrWhitespace;
@@ -95,7 +97,7 @@ export default function setupEventListener(
     /**
      * enter
      */
-    if (!isStartLineEmpty && diffLine === 1) {
+    if (!isStartLineEmpty && diffLine === 1 && startLine === endLine) {
       startLine += 1;
       endLine += 1;
     }
@@ -107,6 +109,13 @@ export default function setupEventListener(
       startLine += 1;
     }
 
+    /**
+     * paste override copyed lines
+     */
+    if (!isStartLineLastPos) {
+      return [startLine, endLine, originalStartLine, originalEndLine];
+    }
+
     return [startLine, endLine];
   }
 }