🐛 fix copy/paste/cut behavior in VS Code notebooks

extensions/vscode/src/debugPanel.ts

@@ -3,6 +3,7 @@ import { ConfigHandler } from "core/config/handler";
 import * as vscode from "vscode";
 import { VerticalPerLineDiffManager } from "./diff/verticalPerLine/manager";
 import { getTheme } from "./util/getTheme";
+import { getExtensionVersion } from "./util/util";
 import { getExtensionUri, getNonce, getUniqueId } from "./util/vscode";
 import { VsCodeWebviewProtocol } from "./webviewProtocol";
 
@@ -34,10 +35,10 @@ export class ContinueGUIWebviewViewProvider
   }
 
   public resetWebviewProtocolWebview(): void {
-    if (this._webview) { 
+    if (this._webview) {
       this.webviewProtocol.webview = this._webview;
-    } else{
-      console.warn("no webview found during reset")
+    } else {
+      console.warn("no webview found during reset");
     }
   }
 
@@ -148,6 +149,7 @@ export class ContinueGUIWebviewViewProvider
         <script type="module" nonce="${nonce}" src="${scriptUri}"></script>
 
         <script>localStorage.setItem("ide", "vscode")</script>
+        <script>localStorage.setItem("extensionVersion", '"${getExtensionVersion()}"')</script>
         <script>window.windowId = "${this.windowId}"</script>
         <script>window.vscMachineId = "${getUniqueId()}"</script>
         <script>window.vscMediaUrl = "${vscMediaUrl}"</script>

extensions/vscode/src/util/util.ts

@@ -100,7 +100,7 @@ export function getMetaKeyLabel() {
   }
 }
 
-export function getExtensionVersion() {
+export function getExtensionVersion(): string {
   const extension = vscode.extensions.getExtension("continue.continue");
-  return extension?.packageJSON.version || "";
+  return extension?.packageJSON.version || "0.1.0";
 }

gui/src/components/mainInput/TipTapEditor.tsx

@@ -37,7 +37,7 @@ import {
 } from "../../redux/slices/stateSlice";
 import { RootState } from "../../redux/store";
 import { isMetaEquivalentKeyPressed } from "../../util";
-import { isJetBrains, postToIde } from "../../util/ide";
+import { isJetBrains, isPrerelease, postToIde } from "../../util/ide";
 import CodeBlockExtension from "./CodeBlockExtension";
 import { SlashCommand } from "./CommandsExtension";
 import InputToolbar from "./InputToolbar";
@@ -405,6 +405,34 @@ function TipTapEditor(props: TipTapEditorProps) {
     },
   });
 
+  useEffect(() => {
+    if (isJetBrains() || !isPrerelease()) {
+      // This is only for VS Code .ipynb files
+      return;
+    }
+
+    const handleKeyDown = async (event: KeyboardEvent) => {
+      if (!editor) return;
+
+      if (event.metaKey && event.key === "x") {
+        document.execCommand("cut");
+        event.stopPropagation();
+      } else if (event.metaKey && event.key === "v") {
+        document.execCommand("paste");
+        event.stopPropagation();
+      } else if (event.metaKey && event.key === "c") {
+        document.execCommand("copy");
+        event.stopPropagation();
+      }
+    };
+
+    document.addEventListener("keydown", handleKeyDown);
+
+    return () => {
+      document.removeEventListener("keydown", handleKeyDown);
+    };
+  }, [editor]);
+
   useEffect(() => {
     if (mainEditorContent && editor) {
       editor.commands.setContent(mainEditorContent);

gui/src/util/ide.ts

@@ -6,6 +6,7 @@ import {
 } from "core/web/webviewProtocol";
 import { v4 as uuidv4 } from "uuid";
 import "vscode-webview";
+import { getLocalStorage } from "./localStorage";
 interface vscode {
   postMessage(message: any): vscode;
 }
@@ -190,3 +191,18 @@ export function appendText(text: string) {
 export function isJetBrains() {
   return localStorage.getItem("ide") === "jetbrains";
 }
+
+export function isPrerelease() {
+  const extensionVersion = getLocalStorage("extensionVersion");
+  if (!extensionVersion) {
+    console.warn(
+      `Could not find extension version in local storage, assuming it's a prerelease`,
+    );
+    return true;
+  }
+  const minor = parseInt(extensionVersion.split(".")[1], 10);
+  if (minor % 2 !== 0) {
+    return true;
+  }
+  return false;
+}

gui/src/util/localStorage.ts

@@ -5,6 +5,7 @@ type LocalStorageTypes = {
   mainTextEntryCounter: number;
   lastSessionId: string | undefined;
   inputHistory: JSONContent[];
+  extensionVersion: string;
 };
 
 export function getLocalStorage<T extends keyof LocalStorageTypes>(
@@ -14,7 +15,15 @@ export function getLocalStorage<T extends keyof LocalStorageTypes>(
   if (value === null) {
     return undefined;
   }
-  return JSON.parse(value);
+  try {
+    return JSON.parse(value);
+  } catch (error) {
+    console.error(
+      `Error parsing ${key} from local storage. Value was ${value}\n\n`,
+      error,
+    );
+    return undefined;
+  }
 }
 
 export function setLocalStorage<T extends keyof LocalStorageTypes>(