lock suggestions until done streaming

1 files changed, 9 insertions, 0 deletions

diff --git a/extension/src/continueIdeClient.ts b/extension/src/continueIdeClient.ts
index 2e641054..1ccc070c 100644
--- a/extension/src/continueIdeClient.ts
+++ b/extension/src/continueIdeClient.ts
@@ -1,5 +1,6 @@
 // import { ShowSuggestionRequest } from "../schema/ShowSuggestionRequest";
 import {
+  editorSuggestionsLocked,
   showSuggestion as showSuggestionInEditor,
   SuggestionRanges,
 } from "./suggestions";
@@ -119,6 +120,9 @@ class IdeProtocolClient {
         this.openFile(data.filepath);
         // TODO: Close file if False
         break;
+      case "setSuggestionsLocked":
+        this.setSuggestionsLocked(data.filepath, data.locked);
+        break;
       case "showSuggestion":
         this.showSuggestion(data.edit);
         break;
@@ -204,6 +208,11 @@ class IdeProtocolClient {
     openEditorAndRevealRange(filepath, undefined, vscode.ViewColumn.One);
   }
 
+  setSuggestionsLocked(filepath: string, locked: boolean) {
+    editorSuggestionsLocked.set(filepath, locked);
+    // TODO: Rerender?
+  }
+
   async getUserSecret(key: string) {
     // Check if secret already exists in VS Code settings (global)
     let secret = vscode.workspace.getConfiguration("continue").get(key);