tabby/clients/vscode/src/Commands.ts

import { ConfigurationTarget, workspace, window, commands } from "vscode";
import { ChoiceEvent, ApiError } from "./generated";
import { TabbyClient } from "./TabbyClient";

const target = ConfigurationTarget.Global;

type Command = {
  command: string;
  callback: (...args: any[]) => any;
  thisArg?: any;
};

const toogleEnabled: Command = {
  command: "tabby.toggleEnabled",
  callback: () => {
    const configuration = workspace.getConfiguration("tabby");
    const enabled = configuration.get("enabled", true);
    console.debug(`Toggle Enabled: ${enabled} -> ${!enabled}.`);
    configuration.update("enabled", !enabled, target, false);
  },
};

const setSuggestionDelay: Command = {
  command: "tabby.setSuggestionDelay",
  callback: () => {
    const configuration = workspace.getConfiguration("tabby");
    window
      .showInputBox({
        prompt: "Enter the suggestion delay in ms",
        value: configuration.get("suggestionDelay", "150"),
      })
      .then((delay) => {
        if (delay) {
          if (Number.parseInt(delay) !== null) {
            console.debug("Set suggestion delay: ", Number.parseInt(delay));
            configuration.update(
              "suggestionDelay",
              Number.parseInt(delay),
              target,
              false
            );
          } else {
            console.debug("Set suggestion delay error. Wrong input.");
          }
        }
      });
  },
};

const setServerUrl: Command = {
  command: "tabby.setServerUrl",
  callback: () => {
    const configuration = workspace.getConfiguration("tabby");
    window
      .showInputBox({
        prompt: "Enter the URL of your Tabby Server",
        value: configuration.get("serverUrl", ""),
      })
      .then((url) => {
        if (url) {
          console.debug("Set Tabby Server URL: ", url);
          configuration.update("serverUrl", url, target, false);
        }
      });
  },
};

const openSettings: Command = {
  command: "tabby.openSettings",
  callback: () => {
    commands.executeCommand("workbench.action.openSettings", "tabby");
  },
};

const tabbyClient = TabbyClient.getInstance();
const emitEvent: Command = {
  command: "tabby.emitEvent",
  callback: (event: ChoiceEvent) => {
    console.debug("Emit Event: ", event);
    tabbyClient.api.default.eventsV1EventsPost(event).then(() => {
      tabbyClient.changeStatus("ready");
    }).catch((err: ApiError) => {
      console.error(err);
      tabbyClient.changeStatus("disconnected");
    });
  },
};

export const tabbyCommands = [toogleEnabled, setServerUrl, setSuggestionDelay, openSettings, emitEvent].map((command) =>
  commands.registerCommand(command.command, command.callback, command.thisArg)
);