From d47c4cafb856ffe2efd7d08ceddc8ceda475a8c6 Mon Sep 17 00:00:00 2001
From: Nate Sesti <sestinj@gmail.com>
Date: Thu, 29 Jun 2023 22:46:56 -0700
Subject: lock suggestions until done streaming

---
 extension/src/lang-server/codeLens.ts | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

(limited to 'extension/src/lang-server/codeLens.ts')

diff --git a/extension/src/lang-server/codeLens.ts b/extension/src/lang-server/codeLens.ts
index 5b55589c..03a9a0a7 100644
--- a/extension/src/lang-server/codeLens.ts
+++ b/extension/src/lang-server/codeLens.ts
@@ -1,5 +1,5 @@
 import * as vscode from "vscode";
-import { editorToSuggestions } from "../suggestions";
+import { editorToSuggestions, editorSuggestionsLocked } from "../suggestions";
 
 class SuggestionsCodeLensProvider implements vscode.CodeLensProvider {
   public provideCodeLenses(
@@ -10,6 +10,7 @@ class SuggestionsCodeLensProvider implements vscode.CodeLensProvider {
     if (!suggestions) {
       return [];
     }
+    const locked = editorSuggestionsLocked.get(document.uri.fsPath.toString());
 
     const codeLenses: vscode.CodeLens[] = [];
     for (const suggestion of suggestions) {
@@ -20,12 +21,12 @@ class SuggestionsCodeLensProvider implements vscode.CodeLensProvider {
       codeLenses.push(
         new vscode.CodeLens(range, {
           title: "Accept ✅",
-          command: "continue.acceptSuggestion",
+          command: locked ? "" : "continue.acceptSuggestion",
           arguments: [suggestion],
         }),
         new vscode.CodeLens(range, {
           title: "Reject ❌",
-          command: "continue.rejectSuggestion",
+          command: locked ? "" : "continue.rejectSuggestion",
           arguments: [suggestion],
         })
       );
-- 
cgit v1.2.3-70-g09d2