1 files changed, 13 insertions, 3 deletions
diff --git a/extension/src/continueIdeClient.ts b/extension/src/continueIdeClient.ts
index 48000b76..3b46d90a 100644
--- a/extension/src/continueIdeClient.ts
+++ b/extension/src/continueIdeClient.ts
@@ -294,7 +294,7 @@ class IdeProtocolClient {
         break;
       case "getTerminalContents":
         messenger.send("getTerminalContents", {
-          contents: await this.getTerminalContents(),
+          contents: await this.getTerminalContents(data.commands),
         });
         break;
       case "listDirectoryContents":
@@ -631,9 +631,19 @@ class IdeProtocolClient {
     return contents;
   }
 
-  async getTerminalContents(): Promise<string> {
+  async getTerminalContents(commands: number = -1): Promise<string> {
     const tempCopyBuffer = await vscode.env.clipboard.readText();
-    await vscode.commands.executeCommand("workbench.action.terminal.selectAll");
+    if (commands < 0) {
+      await vscode.commands.executeCommand(
+        "workbench.action.terminal.selectAll"
+      );
+    } else {
+      for (let i = 0; i < commands; i++) {
+        await vscode.commands.executeCommand(
+          "workbench.action.terminal.selectToPreviousCommand"
+        );
+      }
+    }
     await vscode.commands.executeCommand(
       "workbench.action.terminal.copySelection"
     );