3 files changed, 10 insertions, 9 deletions
diff --git a/extension/src/commands.ts b/extension/src/commands.ts
index 4ceac25d..7d190634 100644
--- a/extension/src/commands.ts
+++ b/extension/src/commands.ts
@@ -28,14 +28,11 @@ const commandsMap: { [command: string]: (...args: any) => any } = {
     }
   },
   "continue.focusContinueInput": async () => {
-    if (focusedOnContinueInput) {
-      vscode.commands.executeCommand("workbench.action.focusActiveEditorGroup");
-    } else {
-      vscode.commands.executeCommand("continue.continueGUIView.focus");
-      debugPanelWebview?.postMessage({
-        type: "focusContinueInput",
-      });
-    }
+    vscode.commands.executeCommand("continue.continueGUIView.focus");
+    debugPanelWebview?.postMessage({
+      type: "focusContinueInput",
+    });
+
     focusedOnContinueInput = !focusedOnContinueInput;
   },
   "continue.focusContinueInputWithEdit": async () => {
diff --git a/extension/src/diffs.ts b/extension/src/diffs.ts
index 98b8753a..b7acd109 100644
--- a/extension/src/diffs.ts
+++ b/extension/src/diffs.ts
@@ -62,7 +62,7 @@ class DiffManager {
   }
 
   private escapeFilepath(filepath: string): string {
-    return filepath.replace(/\\/g, "_").replace(/\//g, "_");
+    return encodeURIComponent(filepath);
   }
 
   private remoteTmpDir: string = "/tmp/continue";
diff --git a/extension/src/lang-server/codeLens.ts b/extension/src/lang-server/codeLens.ts
index ba80e557..ec03f73e 100644
--- a/extension/src/lang-server/codeLens.ts
+++ b/extension/src/lang-server/codeLens.ts
@@ -69,6 +69,10 @@ class DiffViewerCodeLensProvider implements vscode.CodeLensProvider {
           title: `Reject All ❌ (${getMetaKeyLabel()}⇧⌫)`,
           command: "continue.rejectDiff",
           arguments: [document.uri.fsPath],
+        }),
+        new vscode.CodeLens(range, {
+          title: `Further Edit ✏️ (${getMetaKeyLabel()}⇧M)`,
+          command: "continue.focusContinueInputWithEdit",
         })
       );
       return codeLenses;