Add moonx autocomplete support

package.json

@@ -13,6 +13,7 @@
   "type": "module",
   "main": "src/index.ts",
   "bin": {
+    "mx": "bin/moonx",
     "moonx": "bin/moonx"
   },
   "files": [
@@ -39,6 +40,7 @@
     "@trivago/prettier-plugin-sort-imports": "^4.2.1",
     "@types/nunjucks": "^3.2.6",
     "@types/semver": "^7.5.5",
+    "@withfig/autocomplete-types": "^1.29.0",
     "bun-types": "^1.0.11",
     "prettier": "^3.0.3",
     "semver": "^7.5.4",

scripts/fig.ts

@@ -0,0 +1,31 @@
+const generator: Fig.Generator = {
+  script(args) {
+    console.log(args);
+    return [
+      "moonx",
+      "_moonx_list",
+      ...args.filter(
+        (arg) => !arg.startsWith("-") && arg.length && "moonx" !== arg,
+      ),
+    ];
+  },
+  trigger: "moonx",
+  postProcess(out) {
+    console.log("here");
+    const lines = out.split("\n");
+    return lines.map((line) => {
+      return {
+        name: line,
+      };
+    });
+  },
+};
+
+const completionSpec: Fig.Spec = {
+  name: "moonx",
+  args: {
+    isVariadic: true,
+    generators: generator,
+  },
+};
+export default completionSpec;

src/cli/exec.ts

@@ -0,0 +1,16 @@
+export function exec(
+  command: string,
+  wss: Array<string>,
+  commands: Map<string, string[]>,
+) {
+  const wokspaces = commands.get(command);
+  if (!wokspaces) {
+    throw new Error(`task ${command} does not exist`);
+  }
+  if (!wss.length) {
+    return [`:${command}`];
+  }
+  return wss
+    .filter((ws) => wokspaces.includes(ws))
+    .map((ws) => `${ws}:${command}`);
+}

src/cli/list.ts

@@ -0,0 +1,17 @@
+export function list(
+  [command, ...wss]: Array<string>,
+  commands: Map<string, string[]>,
+) {
+  if (!commands.has(command) && wss.length) {
+    console.log(`task ${command} does not exist`);
+    return [];
+  }
+  if (!commands.has(command)) {
+    return Array.from(commands.keys());
+  }
+  const workspaces = commands.get(command)!;
+  if (!wss.length) {
+    return workspaces;
+  }
+  return wss.filter((ws) => workspaces.includes(ws));
+}

src/index.ts

@@ -1,6 +1,8 @@
 import { cac } from "cac";
 import { spawnSync } from "child_process";
 
+import { exec } from "./cli/exec.js";
+import { list } from "./cli/list.js";
 import { help } from "./utils/help.js";
 import { logger } from "./utils/logger.js";
 import { scan } from "./utils/scan-moon.js";
@@ -18,57 +20,23 @@ const commands = await scan();
 const cli = cac("moonx");
 
 cli
-  .command("_moonx_list [...items]", "List all available tasks")
-  .action(([command, ...wss]) => {
-    const workspaces = commands.get(command);
-
-    if (!command) {
-      return console.log(Array.from(commands.keys()).join("\n"));
-    }
-    if (!workspaces) {
-      return;
-    }
-    if (wss.length > 0) {
-      return console.log(
-        workspaces.filter((ws) => !wss.includes(ws)).join("\n"),
-      );
-    }
-    console.log(workspaces.join("\n"));
+  .command("_moonx_list [...params]", "List all available tasks")
+  .action((arg) => {
+    const result = list(arg, commands);
+    const stdout = Bun.stdout.writer();
+    stdout.write(result.join("\n"));
   });
 
-for (const [name, workspaces] of commands) {
+for (const name in commands) {
   cli
     .command(`${name} [...workspaces]`, "", { allowUnknownOptions: true })
-    .usage(`${name} [...workspaces] [options]`)
+    .usage(`${name} [...workspaces] [MOONX_OPTIONS] -- [OPTIONS]`)
     .action(async (wss: Array<string>, options) => {
-      const args = ["moon", "run"];
+      console.log(wss);
+      const workspaces = exec(name, wss, commands);
       const rest = ["--", ...options["--"]];
-      if (wss.length === 0) {
-        return Bun.spawnSync({
-          cmd: [args, `:${name}`, rest].flat(),
-          stdout: "inherit",
-          stderr: "inherit",
-          stdin: "inherit",
-          onExit(_, exitCode) {
-            if (exitCode) {
-              logger.error(`task ${name} failed`);
-              process.exit(exitCode);
-            }
-          },
-        });
-      }
-      wss = wss.filter((ws) => {
-        if (!workspaces.includes(ws)) {
-          logger.warn(`task ${name} does not exist on workspace ${ws}`);
-          return false;
-        }
-        return true;
-      });
-      if (wss.length === 0) {
-        return;
-      }
       return Bun.spawnSync({
-        cmd: [args, wss.map((ws) => `${name}:${ws}`), rest].flat(),
+        cmd: ["moon", "run", workspaces, rest].flat(),
         stdout: "inherit",
         stderr: "inherit",
         stdin: "inherit",

tsconfig.json

@@ -4,7 +4,7 @@
     "target": "ESNext",
     "module": "NodeNext",
     "skipLibCheck": true,
-    "types": ["bun-types"],
+    "types": ["bun-types", "@withfig/autocomplete-types"],
     "resolveJsonModule": true,
     "moduleResolution": "NodeNext"
   },