1 files changed, 33 insertions, 4 deletions
diff --git a/extension/src/commands.ts b/extension/src/commands.ts
index 479e8db0..4e2f4571 100644
--- a/extension/src/commands.ts
+++ b/extension/src/commands.ts
@@ -9,6 +9,36 @@ import { ideProtocolClient } from "./activation/activate";
 
 let focusedOnContinueInput = false;
 
+function addHighlightedCodeToContext(edit: boolean) {
+  focusedOnContinueInput = !focusedOnContinueInput;
+  const editor = vscode.window.activeTextEditor;
+  if (editor) {
+    const selection = editor.selection;
+    if (selection.isEmpty) return;
+    const range = new vscode.Range(selection.start, selection.end);
+    const contents = editor.document.getText(range);
+    ideProtocolClient?.sendHighlightedCode(
+      [
+        {
+          filepath: editor.document.uri.fsPath,
+          contents,
+          range: {
+            start: {
+              line: selection.start.line,
+              character: selection.start.character,
+            },
+            end: {
+              line: selection.end.line,
+              character: selection.end.character,
+            },
+          },
+        },
+      ],
+      edit
+    );
+  }
+}
+
 export const setFocusedOnContinueInput = (value: boolean) => {
   focusedOnContinueInput = value;
 };
@@ -32,11 +62,11 @@ const commandsMap: { [command: string]: (...args: any) => any } = {
     debugPanelWebview?.postMessage({
       type: "focusContinueInput",
     });
-
-    focusedOnContinueInput = !focusedOnContinueInput;
+    addHighlightedCodeToContext(false);
   },
   "continue.focusContinueInputWithEdit": async () => {
     vscode.commands.executeCommand("continue.continueGUIView.focus");
+    addHighlightedCodeToContext(true);
     debugPanelWebview?.postMessage({
       type: "focusContinueInputWithEdit",
     });
@@ -47,8 +77,7 @@ const commandsMap: { [command: string]: (...args: any) => any } = {
   },
   "continue.quickTextEntry": async () => {
     const text = await vscode.window.showInputBox({
-      placeHolder:
-        "Ask a question or enter a slash command",
+      placeHolder: "Ask a question or enter a slash command",
       title: "Continue Quick Input",
     });
     if (text) {