fig.ts

const SETTINGS_PATH = "~/.fig/tools/all-settings.json";

interface Settings {
  settingName: string;
  description: string;
  type: "boolean" | "text" | "single_select" | "multiselect";
  options?: Array<string | { name: string; description: string }>;
  default?: string;
}

const devCompletionsFolderGenerator: Fig.Generator = {
  script: 'ls -d -1 "$PWD/"**/',
  postProcess: (out) =>
    out.split("\n").map((folder) => {
      const paths = folder.split("/");
      paths.pop();

      return {
        name: paths.pop(),
        insertValue: folder,
        icon: `fig://${folder}`,
      };
    }),
};

const disableForCommandsGenerator: Fig.Generator = {
  script: "fig settings autocomplete.disableForCommands",
  postProcess: (out) => {
    const existing = out.split("\n").filter((item) => item.length > 0);

    const append: Fig.Suggestion = {
      name: "Disable...",
      icon: "fig://icon?type=box",
      insertValue: JSON.stringify(existing.concat(["{cursor}"])),
    };

    const enabledAll: Fig.Suggestion = {
      name: "Enable all commands",
      icon: "fig://icon?type=box",
      insertValue: "[]",
    };

    return [append, enabledAll].concat(
      existing.map((disabledCommand) => {
        return {
          name: `Enable ${disabledCommand}`,
          icon: "fig://icon?type=box",
          insertValue: JSON.stringify(
            existing.filter((cmd) => cmd != disabledCommand)
          ),
        };
      })
    );
  },
};

const themesGenerator: Fig.Generator = {
  script: "ls -1 ~/.fig/themes",
  postProcess: (output) => {
    const builtinThemes = [
      {
        name: "light",
        icon: "fig://template?color=ffffff&badge=☀️",
        priority: 51,
      },
      {
        name: "dark",
        icon: "fig://template?color=000000&badge=🌙",
        priority: 51,
      },
    ];
    return output
      .split("\n")
      .map((theme) => ({ name: theme.replace(".json", "") }))
      .concat(builtinThemes);
  },
};

const SETTINGS_GENERATOR: Record<string, Fig.Generator> = {
  "autocomplete.devCompletionsFolder": devCompletionsFolderGenerator,
  "autocomplete.disableForCommands": disableForCommandsGenerator,
  "autocomplete.theme": themesGenerator,
};

const completionSpec: Fig.Spec = {
  name: "fig",
  description: "Autocomplete for your terminal",
  subcommands: [
    {
      name: "source",
      description: "(Re)connect fig to the current shell session",
    },
    { name: "update", description: "Update completion specs and app" },
    {
      name: "theme",
      description: "Set the Theme of fig",
      args: {
        name: "theme",
        generators: themesGenerator,
      },
    },
    {
      name: "settings",
      description: "Update preferences",
      generateSpec: async (_, executeShellCommand) => {
        const settings: Settings[] = JSON.parse(
          await executeShellCommand(`cat ${SETTINGS_PATH}`)
        );

        return {
          name: "settings",
          subcommands: settings.map(
            ({
              settingName: name,
              description,
              type,
              options,
              default: defaultValue,
            }) => {
              const suggestions =
                type === "boolean"
                  ? ["true", "false"]
                  : options?.map((option) => ({
                      name: option["name"] || option,
                      description: option["description"] || "",
                    }));
              const insertValue =
                type === "multiselect" ? `${name} '{cursor}'` : undefined;

              const generators = SETTINGS_GENERATOR[name];

              return {
                name,
                description,
                icon: "fig://icon?type=commandkey",
                insertValue,
                args: {
                  name: type,
                  default: defaultValue,
                  suggestions: generators ? [] : suggestions,
                  generators,
                },
              };
            }
          ),
        };
      },
    },
    {
      name: "remove",
      description: "Remove a completion spec",
      args: {
        name: "spec",
        description: "The CLI completion spec to remove",
        generators: {
          script: "ls -1Ap ~/.fig/autocomplete",
          postProcess: (data) => {
            console.log(data);
            const out = data.split("\n").reduce((acc, curr) => {
              if (
                [
                  ".gitignore",
                  "README.md",
                  "package.json",
                  "package-lock.json",
                ].includes(curr)
              )
                return acc;
              else {
                acc.push({
                  name: curr.trim().split(".")[0],
                  icon: "fig://icon?type=box",
                });
                return acc;
              }
            }, []);
            return out;
          },
        },
      },
    },
    {
      name: "invite",
      description: "Share Fig with a teammate ⭐",
      icon: "fig://icon?type=invite",
    },
    { name: "report", description: "Report an issue" },
    {
      name: "tweet",
      description: "Tweet about Fig",
      icon: "fig://icon?type=twitter",
    },
    { name: "docs", description: "View docs in browser" },
    { name: "list", description: "List all available completion specs" },
    { name: "onboarding", description: "Re-run Fig's onboarding" },
    { name: "diagnostic", description: "Display diagnostic information" },
    {
      name: "issue",
      description: "Create a new Github issue in withfig/fig",
      icon: "fig://icon?type=github",
    },

    { name: "quit", description: "Quit the Fig application" },

    {
      name: "team:upload",
      description: "Share an completion spec with your team",
      args: {
        name: "spec",
        template: "filepaths",
      },
    },
    { name: "team:download", description: "Download your team's spec" },
    {
      name: "integrations:iterm",
      description: "Install the iTerm tab integration",
    },
    {
      name: "set:path",
      description: "Sync your $PATH variable with Fig",
    },
  ],

  options: [
    {
      name: ["-h", "--help"],
      description: "Overview of Fig CLI",
    },
    {
      name: "--version",
      description: "The current Fig version",
    },
  ],
};

export default completionSpec;