Merge branch 'sr-tream-pause_completion_on_battery' into preview

Author: sestinj

extensions/vscode/package-lock.json

@@ -80,6 +80,11 @@
           "default": true,
           "markdownDescription": "Enable Continue's tab autocomplete feature. Read our walkthrough to learn about configuration and how to share feedback: [continue.dev › Walkthrough: Tab Autocomplete](https://docs.continue.dev/walkthroughs/tab-autocomplete)"
         },
+        "continue.pauseTabAutocompleteOnBattery": {
+          "type": "boolean",
+          "default": false,
+          "markdownDescription": "Pause Continue's tab autocomplete feature when your battery is low."
+        },
         "continue.remoteConfigServerUrl": {
           "type": "string",
           "default": null,
@@ -553,6 +558,7 @@
     "request": "^2.88.2",
     "socket.io-client": "^4.7.2",
     "strip-ansi": "^7.1.0",
+    "systeminformation": "^5.22.10",
     "tailwindcss": "^3.3.2",
     "undici": "^6.2.0",
     "uuid": "^9.0.1",

extensions/vscode/package.json

@@ -10,7 +10,7 @@ import * as vscode from "vscode";
 import type { TabAutocompleteModel } from "../util/loadAutocompleteModel";
 import { getDefinitionsFromLsp } from "./lsp";
 import { RecentlyEditedTracker } from "./recentlyEdited";
-import { setupStatusBar, stopStatusBarLoading } from "./statusBar";
+import { StatusBarStatus, setupStatusBar, stopStatusBarLoading, getStatusBarStatus } from "./statusBar";
 
 interface VsCodeCompletionInput {
   document: vscode.TextDocument;
@@ -73,10 +73,7 @@ export class ContinueCompletionProvider
     token: vscode.CancellationToken,
     //@ts-ignore
   ): ProviderResult<InlineCompletionItem[] | InlineCompletionList> {
-    const enableTabAutocomplete =
-      vscode.workspace
-        .getConfiguration("continue")
-        .get<boolean>("enableTabAutocomplete") || false;
+    const enableTabAutocomplete = getStatusBarStatus() === StatusBarStatus.Enabled;
     if (token.isCancellationRequested || !enableTabAutocomplete) {
       return null;
     }
@@ -195,7 +192,7 @@ export class ContinueCompletionProvider
         injectDetails,
       };
 
-      setupStatusBar(true, true);
+      setupStatusBar(undefined, true);
       const outcome =
         await this.completionProvider.provideInlineCompletionItems(
           input,

extensions/vscode/src/autocomplete/completionProvider.ts

@@ -1,22 +1,75 @@
 import * as vscode from "vscode";
+import { Battery } from "../util/battery";
 
-const statusBarItemText = (enabled: boolean | undefined) =>
-  enabled ? "$(check) Continue" : "$(circle-slash) Continue";
+export enum StatusBarStatus {
+  Disabled,
+  Enabled,
+  Paused,
+}
+
+export const quickPickStatusText = (status: StatusBarStatus | undefined) => {
+  switch (status) {
+    case undefined:
+    case StatusBarStatus.Disabled:
+      return "$(circle-slash) Disable autocomplete";
+    case StatusBarStatus.Enabled:
+      return "$(check) Enable autocomplete";
+    case StatusBarStatus.Paused:
+      return "$(debug-pause) Pause autocomplete";
+  }
+};
+
+export const getStatusBarStatusFromQuickPickItemLabel = (
+  label: string,
+): StatusBarStatus | undefined => {
+  switch (label) {
+    case "$(circle-slash) Disable autocomplete":
+      return StatusBarStatus.Disabled;
+    case "$(check) Enable autocomplete":
+      return StatusBarStatus.Enabled;
+    case "$(debug-pause) Pause autocomplete":
+      return StatusBarStatus.Paused;
+    default:
+      return undefined;
+  }
+};
+
+const statusBarItemText = (status: StatusBarStatus | undefined) => {
+  switch (status) {
+    case undefined:
+    case StatusBarStatus.Disabled:
+      return "$(circle-slash) Continue";
+    case StatusBarStatus.Enabled:
+      return "$(check) Continue";
+    case StatusBarStatus.Paused:
+      return "$(debug-pause) Continue";
+  }
+};
 
-const statusBarItemTooltip = (enabled: boolean | undefined) =>
-  enabled ? "Tab autocomplete is enabled" : "Click to enable tab autocomplete";
+const statusBarItemTooltip = (status: StatusBarStatus | undefined) => {
+  switch (status) {
+    case undefined:
+    case StatusBarStatus.Disabled:
+      return "Click to enable tab autocomplete";
+    case StatusBarStatus.Enabled:
+      return "Tab autocomplete is enabled";
+    case StatusBarStatus.Paused:
+      return "Tab autocomplete is paused";
+  }
+};
 
+let statusBarStatus: StatusBarStatus | undefined = undefined;
 let statusBarItem: vscode.StatusBarItem | undefined = undefined;
 let statusBarFalseTimeout: NodeJS.Timeout | undefined = undefined;
 
 export function stopStatusBarLoading() {
   statusBarFalseTimeout = setTimeout(() => {
-    setupStatusBar(true, false);
+    setupStatusBar(StatusBarStatus.Enabled, false);
   }, 100);
 }
 
 export function setupStatusBar(
-  enabled: boolean | undefined,
+  status: StatusBarStatus | undefined,
   loading?: boolean,
 ) {
   if (loading !== false) {
@@ -33,17 +86,42 @@ export function setupStatusBar(
 
   statusBarItem.text = loading
     ? "$(loading~spin) Continue"
-    : statusBarItemText(enabled);
-  statusBarItem.tooltip = statusBarItemTooltip(enabled);
+    : statusBarItemText(status);
+  statusBarItem.tooltip = statusBarItemTooltip(status ?? statusBarStatus);
   statusBarItem.command = "continue.openTabAutocompleteConfigMenu";
 
   statusBarItem.show();
+  if (status !== undefined) statusBarStatus = status;
 
   vscode.workspace.onDidChangeConfiguration((event) => {
     if (event.affectsConfiguration("continue")) {
       const config = vscode.workspace.getConfiguration("continue");
       const enabled = config.get<boolean>("enableTabAutocomplete");
-      setupStatusBar(enabled);
+      if (enabled && statusBarStatus === StatusBarStatus.Paused) return;
+      setupStatusBar(
+        enabled ? StatusBarStatus.Enabled : StatusBarStatus.Disabled,
+      );
+    }
+  });
+}
+
+export function getStatusBarStatus(): StatusBarStatus | undefined {
+  return statusBarStatus;
+}
+
+export function monitorBatteryChanges(battery: Battery): vscode.Disposable {
+  return battery.onChangeAC((acConnected: boolean) => {
+    const config = vscode.workspace.getConfiguration("continue");
+    const enabled = config.get<boolean>("enableTabAutocomplete");
+    if (!!enabled) {
+      const pauseOnBattery = config.get<boolean>(
+        "pauseTabAutocompleteOnBattery",
+      );
+      setupStatusBar(
+        acConnected || !pauseOnBattery
+          ? StatusBarStatus.Enabled
+          : StatusBarStatus.Paused,
+      );
     }
   });
 }

extensions/vscode/src/autocomplete/statusBar.ts

@@ -11,9 +11,17 @@ import { fetchwithRequestOptions } from "core/util/fetchWithOptions";
 import { GlobalContext } from "core/util/GlobalContext";
 import { getConfigJsonPath, getDevDataFilePath } from "core/util/paths";
 import readLastLines from "read-last-lines";
+import {
+  StatusBarStatus,
+  getStatusBarStatus,
+  getStatusBarStatusFromQuickPickItemLabel,
+  quickPickStatusText,
+  setupStatusBar,
+} from "./autocomplete/statusBar";
 import { ContinueGUIWebviewViewProvider } from "./debugPanel";
 import { DiffManager } from "./diff/horizontal";
 import { VerticalPerLineDiffManager } from "./diff/verticalPerLine/manager";
+import { Battery } from "./util/battery";
 import { getPlatform } from "./util/util";
 import type { VsCodeWebviewProtocol } from "./webviewProtocol";
 
@@ -147,6 +155,7 @@ const commandsMap: (
   diffManager: DiffManager,
   verticalDiffManager: VerticalPerLineDiffManager,
   continueServerClientPromise: Promise<ContinueServerClient>,
+  battery: Battery,
 ) => { [command: string]: (...args: any) => any } = (
   ide,
   extensionContext,
@@ -155,6 +164,7 @@ const commandsMap: (
   diffManager,
   verticalDiffManager,
   continueServerClientPromise,
+  battery,
 ) => {
   async function streamInlineEdit(
     promptName: keyof ContextMenuConfig,
@@ -541,27 +551,72 @@ const commandsMap: (
     "continue.toggleTabAutocompleteEnabled": () => {
       const config = vscode.workspace.getConfiguration("continue");
       const enabled = config.get("enableTabAutocomplete");
-      config.update(
-        "enableTabAutocomplete",
-        !enabled,
-        vscode.ConfigurationTarget.Global,
+      const pauseOnBattery = config.get<boolean>(
+        "pauseTabAutocompleteOnBattery",
       );
+      if (!pauseOnBattery || battery.isACConnected()) {
+        config.update(
+          "enableTabAutocomplete",
+          !enabled,
+          vscode.ConfigurationTarget.Global,
+        );
+      } else {
+        if (enabled) {
+          const paused = getStatusBarStatus() === StatusBarStatus.Paused;
+          if (paused) {
+            setupStatusBar(StatusBarStatus.Enabled);
+          } else {
+            config.update(
+              "enableTabAutocomplete",
+              false,
+              vscode.ConfigurationTarget.Global,
+            );
+          }
+        } else {
+          setupStatusBar(StatusBarStatus.Paused);
+          config.update(
+            "enableTabAutocomplete",
+            true,
+            vscode.ConfigurationTarget.Global,
+          );
+        }
+      }
     },
     "continue.openTabAutocompleteConfigMenu": async () => {
       const config = vscode.workspace.getConfiguration("continue");
-      const enabled = config.get("enableTabAutocomplete");
       const quickPick = vscode.window.createQuickPick();
       const selected = new GlobalContext().get("selectedTabAutocompleteModel");
       const autocompleteModelTitles = ((
         await configHandler.loadConfig()
       ).tabAutocompleteModels
         ?.map((model) => model.title)
         .filter((t) => t !== undefined) || []) as string[];
+
+      // Toggle between Disabled, Paused, and Enabled
+      const pauseOnBattery =
+        config.get<boolean>("pauseTabAutocompleteOnBattery") &&
+        !battery.isACConnected();
+      const currentStatus = getStatusBarStatus();
+
+      let targetStatus: StatusBarStatus | undefined;
+      if (pauseOnBattery) {
+        // Cycle from Disabled -> Paused -> Enabled
+        targetStatus =
+          currentStatus === StatusBarStatus.Paused
+            ? StatusBarStatus.Enabled
+            : currentStatus === StatusBarStatus.Disabled
+              ? StatusBarStatus.Paused
+              : StatusBarStatus.Disabled;
+      } else {
+        // Toggle between Disabled and Enabled
+        targetStatus =
+          currentStatus === StatusBarStatus.Disabled
+            ? StatusBarStatus.Enabled
+            : StatusBarStatus.Disabled;
+      }
       quickPick.items = [
         {
-          label: enabled
-            ? "$(check) Disable autocomplete"
-            : "$(circle-slash) Enable autocomplete",
+          label: quickPickStatusText(targetStatus),
         },
         {
           label: "$(gear) Configure autocomplete options",
@@ -580,16 +635,14 @@ const commandsMap: (
       ];
       quickPick.onDidAccept(() => {
         const selectedOption = quickPick.selectedItems[0].label;
-        if (selectedOption === "$(circle-slash) Enable autocomplete") {
-          config.update(
-            "enableTabAutocomplete",
-            true,
-            vscode.ConfigurationTarget.Global,
-          );
-        } else if (selectedOption === "$(check) Disable autocomplete") {
+        const targetStatus =
+          getStatusBarStatusFromQuickPickItemLabel(selectedOption);
+
+        if (targetStatus !== undefined) {
+          setupStatusBar(targetStatus);
           config.update(
             "enableTabAutocomplete",
-            false,
+            targetStatus === StatusBarStatus.Enabled,
             vscode.ConfigurationTarget.Global,
           );
         } else if (
@@ -635,6 +688,7 @@ export function registerAllCommands(
   diffManager: DiffManager,
   verticalDiffManager: VerticalPerLineDiffManager,
   continueServerClientPromise: Promise<ContinueServerClient>,
+  battery: Battery,
 ) {
   for (const [command, callback] of Object.entries(
     commandsMap(
@@ -645,6 +699,7 @@ export function registerAllCommands(
       diffManager,
       verticalDiffManager,
       continueServerClientPromise,
+      battery,
     ),
   )) {
     context.subscriptions.push(