From bedd5f31e7a1d7b915fbb2a47a92f58035c96415 Mon Sep 17 00:00:00 2001
From: Nate Sesti <sestinj@gmail.com>
Date: Tue, 27 Jun 2023 11:03:44 -0700
Subject: stop completion early, accept all improvements

---
 extension/src/lang-server/codeLens.ts | 4 ++++
 extension/src/suggestions.ts          | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

(limited to 'extension/src')

diff --git a/extension/src/lang-server/codeLens.ts b/extension/src/lang-server/codeLens.ts
index 3979b64d..629da0b8 100644
--- a/extension/src/lang-server/codeLens.ts
+++ b/extension/src/lang-server/codeLens.ts
@@ -28,6 +28,10 @@ class SuggestionsCodeLensProvider implements vscode.CodeLensProvider {
           title: "Reject ❌",
           command: "continue.rejectSuggestion",
           arguments: [suggestion],
+        }),
+        new vscode.CodeLens(range, {
+          title: "(⌘⇧↩/⌘⇧⌫ to accept/reject all)",
+          command: "",
         })
       );
     }
diff --git a/extension/src/suggestions.ts b/extension/src/suggestions.ts
index 4631c33e..209bf8b2 100644
--- a/extension/src/suggestions.ts
+++ b/extension/src/suggestions.ts
@@ -246,7 +246,7 @@ function handleAllSuggestions(accept: boolean) {
   if (!suggestions) return;
 
   while (suggestions.length > 0) {
-    selectSuggestion(accept ? "new" : "old");
+    selectSuggestion(accept ? "new" : "old", suggestions[0]);
   }
 }
 
-- 
cgit v1.2.3-70-g09d2