argsbarg 1.4.3 → 1.5.0

package/src/builtins/completion-bash.ts

@@ -0,0 +1,240 @@
+import { CliCommand, CliOptionKind } from "../types.ts";
+import { collectScopes, type ScopeRec } from "./scopes.ts";
+import {
+  escShellSingleQuoted,
+  identToken,
+  kHelpLong,
+  kHelpShort,
+  kSchemaLong,
+  mainName,
+} from "./shell-helpers.ts";
+
+function emitConsumeLong(ident: string, scopes: ScopeRec[]): string {
+  let o = "_${ident}_nac_consume_long() {\n".replace("${ident}", ident);
+  o += "  local sid=\"$1\" w=\"$2\" nw=\"$3\"\n";
+  o += "  case $sid in\n";
+  for (const [i, sc] of scopes.entries()) {
+    o += "    " + i + ")\n";
+    o += "      case $w in\n";
+    o += "        " + kHelpLong + "|${kHelpLong}=*|${kHelpShort}) echo 1 ;;\n".replace(/\$\{kHelpLong\}/g, kHelpLong).replace(/\$\{kHelpShort\}/g, kHelpShort);
+    if (sc.path === "") {
+      o += "        " + kSchemaLong + ") echo 1 ;;\n";
+    }
+    for (const op of sc.opts) {
+      const base = "--" + op.name;
+      if (op.kind === "presence") {
+        o += "        " + base + "|${base}=*) echo 1 ;;\n".replace(/\$\{base\}/g, base);
+      } else {
+        o += "        " + base + "=*) echo 1 ;;\n";
+        o += "        " + base + ") echo 2 ;;\n";
+      }
+    }
+    o += "        *) echo 0 ;;\n";
+    o += "      esac\n";
+    o += "      ;;\n";
+  }
+  o += "    *) echo 0 ;;\n";
+  o += "  esac\n";
+  o += "}\n";
+  return o;
+}
+
+function emitConsumeShort(ident: string, scopes: ScopeRec[]): string {
+  let o = "_${ident}_nac_consume_short() {\n".replace("${ident}", ident);
+  o += "  local sid=\"$1\" w=\"$2\"\n";
+  o += "  case $sid in\n";
+  for (const [i, sc] of scopes.entries()) {
+    o += "    " + i + ")\n";
+    o += "      local rest=${w#-}\n";
+    o += "      local ch\n";
+    o += "      local saw=0\n";
+    o += "      while [[ -n $rest ]]; do\n";
+    o += "        ch=${rest:0:1}\n";
+    o += "        rest=${rest:1}\n";
+    o += "        case $ch in\n";
+    let boolChars = "";
+    for (const op of sc.opts) {
+      if (!op.shortName) continue;
+      if (op.kind === "presence") {
+        boolChars += op.shortName + "|";
+      } else {
+        o += "          " + op.shortName + ")\n";
+        o += "            if [[ $saw -ne 0 || -n $rest ]]; then echo 0; return; fi\n";
+        o += "            echo 2; return ;;\n";
+      }
+    }
+    if (boolChars.length > 0) {
+      boolChars = boolChars.slice(0, -1);
+      o += "          " + boolChars + ") ;;\n";
+    }
+    o += "          *) echo 0; return ;;\n";
+    o += "        esac\n";
+    o += "        saw=1\n";
+    o += "      done\n";
+    o += "      echo 1\n";
+    o += "      ;;\n";
+  }
+  o += "    *) echo 0 ;;\n";
+  o += "  esac\n";
+  o += "}\n";
+  return o;
+}
+
+function emitMatchChild(ident: string, scopes: ScopeRec[], pathIndex: Record<string, number>): string {
+  let o = "_${ident}_nac_match_child() {\n".replace("${ident}", ident);
+  o += "  local sid=\"$1\" w=\"$2\"\n";
+  o += "  case $sid in\n";
+  for (const [sid, sc] of scopes.entries()) {
+    if (sc.kids.length === 0) continue;
+    o += "    " + sid + ")\n";
+    o += "      case $w in\n";
+    for (const ch of sc.kids) {
+      const childPath = sc.path === "" ? ch.key : sc.path + "/" + ch.key;
+      const cid = pathIndex[childPath] ?? 0;
+      o += "        " + ch.key + ") echo " + cid + "; return 0 ;;\n";
+    }
+    o += "      esac\n";
+    o += "      ;;\n";
+  }
+  o += "  esac\n";
+  o += "  return 1\n";
+  o += "}\n";
+  return o;
+}
+
+function emitSimulate(ident: string): string {
+  let o = "_${ident}_nac_simulate() {\n".replace("${ident}", ident);
+  o += "  local i=1 sid=0 w steps next\n";
+  o += "  while (( i < COMP_CWORD )); do\n";
+  o += "    w=\"${COMP_WORDS[i]}\"\n";
+  o += "    if [[ $w == " + kHelpShort + " || $w == " + kHelpLong + " || $w == " + kSchemaLong + " ]]; then\n";
+  o += "      ((i++)); continue\n";
+  o += "    fi\n";
+  o += "    if [[ $w == --* ]]; then\n";
+  o += "      steps=$(_${ident}_nac_consume_long \"$sid\" \"$w\" \"${COMP_WORDS[i+1]}\")\n".replace("${ident}", ident);
+  o += "      case $steps in\n";
+  o += "        0) break ;;\n";
+  o += "        1) ((i++)) ;;\n";
+  o += "        2) ((i+=2)) ;;\n";
+  o += "        *) break ;;\n";
+  o += "      esac\n";
+  o += "      continue\n";
+  o += "    fi\n";
+  o += "    if [[ $w == -* ]]; then\n";
+  o += "      steps=$(_${ident}_nac_consume_short \"$sid\" \"$w\")\n".replace("${ident}", ident);
+  o += "      case $steps in\n";
+  o += "        0) break ;;\n";
+  o += "        1) ((i++)) ;;\n";
+  o += "        2) ((i++)); break ;;\n";
+  o += "        *) break ;;\n";
+  o += "      esac\n";
+  o += "      continue\n";
+  o += "    fi\n";
+  o += "    next=$(_${ident}_nac_match_child \"$sid\" \"$w\") || break\n".replace("${ident}", ident);
+  o += "    sid=$next\n";
+  o += "    ((i++))\n";
+  o += "  done\n";
+  o += "  REPLY_SID=$sid\n";
+  o += "}\n";
+  return o;
+}
+
+function emitEnumReplyBash(ident: string, scopes: ScopeRec[]): string {
+  let o = "_${ident}_nac_enum_reply() {\n".replace("${ident}", ident);
+  o += "  local sid=\"$1\" prev=\"$2\" cur=\"$3\"\n";
+  o += "  case $sid in\n";
+  for (const [i, sc] of scopes.entries()) {
+    const enumOpts = sc.opts.filter((op) => op.kind === CliOptionKind.Enum && (op.choices?.length ?? 0) > 0);
+    if (enumOpts.length === 0) continue;
+    o += "    " + i + ")\n";
+    o += "      case $prev in\n";
+    for (const op of enumOpts) {
+      const words = (op.choices ?? []).map((c) => escShellSingleQuoted(c)).join(" ");
+      o += "        --" + op.name + ") COMPREPLY=( $(compgen -W '" + words + "' -- \"$cur\") ); return 0 ;;\n";
+    }
+    o += "      esac\n";
+    o += "      ;;\n";
+  }
+  o += "  esac\n";
+  o += "  return 1\n";
+  o += "}\n";
+  return o;
+}
+
+function emitMainBodyBash(schema: CliCommand, ident: string): string {
+  const main = mainName(schema.key);
+  let o = "_${main}() {\n".replace("${main}", main);
+  o += "  local cur=\"${COMP_WORDS[COMP_CWORD]}\"\n";
+  o += "  local prev=\"${COMP_WORDS[COMP_CWORD-1]:-}\"\n";
+  o += "  _${ident}_nac_simulate\n".replace("${ident}", ident);
+  o += "  local sid=$REPLY_SID\n";
+  o += "  if _${ident}_nac_enum_reply \"$sid\" \"$prev\" \"$cur\"; then return; fi\n".replace("${ident}", ident);
+  o += "  if [[ $cur == -* ]]; then\n";
+  o += "    local oname=\"A_${ident}_${sid}_opts\"\n".replace("${ident}", ident);
+  o += "    local -a optsarr\n";
+  o += "    local -n optsref=\"$oname\"\n";
+  o += "    COMPREPLY=( $(compgen -W \"${optsref[*]}\" -- \"$cur\") )\n";
+  o += "  else\n";
+  o += "    local lname=\"A_${ident}_${sid}_leaf\"\n".replace("${ident}", ident);
+  o += "    local -n leafref=\"$lname\"\n";
+  o += "    if [[ $leafref -eq 0 ]]; then\n";
+  o += "      local cname=\"A_${ident}_${sid}_cmds\"\n".replace("${ident}", ident);
+  o += "      local -a cmdsarr\n";
+  o += "      local -n cmdsref=\"$cname\"\n";
+  o += "      COMPREPLY=( $(compgen -W \"${cmdsref[*]}\" -- \"$cur\") )\n";
+  o += "    else\n";
+  o += "      local pname=\"A_${ident}_${sid}_pos\"\n".replace("${ident}", ident);
+  o += "      local -n posref=\"$pname\"\n";
+  o += "      if [[ $posref -eq 1 ]]; then\n";
+  o += "        compopt -o filenames\n";
+  o += "      fi\n";
+  o += "    fi\n";
+  o += "  fi\n";
+  o += "}\n\n";
+  o += "complete -F _${main} ${schema.key}\n".replace("${main}", main).replace("${schema.key}", schema.key);
+  return o;
+}
+
+/** Returns a self-contained bash `complete` script for the given program schema. */
+export function completionBashScript(schema: CliCommand): string {
+  const ident = identToken(schema.key);
+  const scopes = collectScopes(schema);
+  const pathIndex: Record<string, number> = {};
+  for (const [i, s] of scopes.entries()) {
+    pathIndex[s.path] = i;
+  }
+
+  let out = "# Generated bash completion for " + schema.key + ".\n\n";
+
+  for (const [i, sc] of scopes.entries()) {
+    out += "A_" + ident + "_" + i + "_opts=()\n";
+    out += "A_" + ident + "_" + i + "_opts+=('" + kHelpLong + "' '" + kHelpShort + "')\n";
+    if (sc.path === "") {
+      out += "A_" + ident + "_" + i + "_opts+=('" + kSchemaLong + "')\n";
+    }
+    for (const o of sc.opts) {
+      out += "A_" + ident + "_" + i + "_opts+=('--" + o.name + "')\n";
+      if (o.shortName) {
+        out += "A_" + ident + "_" + i + "_opts+=('-" + o.shortName + "')\n";
+      }
+    }
+    out += "A_" + ident + "_" + i + "_leaf=" + (sc.kids.length === 0 ? "1" : "0") + "\n";
+    out += "A_" + ident + "_" + i + "_pos=" + (sc.wantsFiles ? "1" : "0") + "\n";
+    if (sc.kids.length > 0) {
+      out += "A_" + ident + "_" + i + "_cmds=(";
+      for (const ch of sc.kids) {
+        out += " '" + ch.key + "'";
+      }
+      out += ")\n";
+    }
+  }
+
+  out += emitConsumeLong(ident, scopes);
+  out += emitConsumeShort(ident, scopes);
+  out += emitMatchChild(ident, scopes, pathIndex);
+  out += emitSimulate(ident);
+  out += emitEnumReplyBash(ident, scopes);
+  out += emitMainBodyBash(schema, ident);
+
+  return out;
+}

package/src/builtins/completion-fish.ts

@@ -0,0 +1,73 @@
+import { CliCommand, CliOptionKind } from "../types.ts";
+import { collectScopes } from "./scopes.ts";
+import {
+  escFishSingleQuoted,
+  identToken,
+  kHelpLong,
+  kHelpShort,
+  kSchemaDesc,
+  kSchemaLong,
+} from "./shell-helpers.ts";
+
+function scopeCondition(ident: string, scopeIndex: number, path: string): string {
+  const fn = `__${ident}_scope_${scopeIndex}`;
+  let body = `function ${fn}\n`;
+  body += `    set -l tokens (commandline -opc)\n`;
+  if (path === "") {
+    body += `    test (count $tokens) -eq 0\n`;
+  } else {
+    const parts = path.split("/");
+    body += `    test (count $tokens) -eq ${parts.length}\n`;
+    for (let i = 0; i < parts.length; i++) {
+      body += `    and test $tokens[${i + 1}] = ${parts[i]}\n`;
+    }
+  }
+  body += `end\n\n`;
+  return body;
+}
+
+/** Returns a self-contained fish completion script for the given program schema. */
+export function completionFishScript(schema: CliCommand): string {
+  const ident = identToken(schema.key);
+  const app = schema.key;
+  const scopes = collectScopes(schema);
+
+  let out = "# Fish completion for " + app + "\n\n";
+
+  for (const [i, sc] of scopes.entries()) {
+    out += scopeCondition(ident, i, sc.path);
+    const cond = `__${ident}_scope_${i}`;
+
+    for (const ch of sc.kids) {
+      out += `complete -c ${app} -n '${cond}' -a '${escFishSingleQuoted(ch.key)}' -d '${escFishSingleQuoted(ch.description)}'\n`;
+    }
+
+    out += `complete -c ${app} -n '${cond}' -s h -l help -d '${escFishSingleQuoted("Show help for this command.")}'\n`;
+    if (sc.path === "") {
+      out += `complete -c ${app} -n '${cond}' -l schema -d '${escFishSingleQuoted(kSchemaDesc)}'\n`;
+    }
+
+    for (const op of sc.opts) {
+      if (op.kind === CliOptionKind.Presence) {
+        const shortPart = op.shortName ? `-s ${op.shortName} ` : "";
+        out += `complete -c ${app} -n '${cond}' ${shortPart}-l ${op.name} -d '${escFishSingleQuoted(op.description)}'\n`;
+      } else if (op.kind === CliOptionKind.Enum && (op.choices?.length ?? 0) > 0) {
+        const shortPart = op.shortName ? `-s ${op.shortName} ` : "";
+        out += `complete -c ${app} -n '${cond}' ${shortPart}-l ${op.name} -d '${escFishSingleQuoted(op.description)}'\n`;
+        const enumCond = `${cond}; and __fish_seen_argument -l ${op.name}`;
+        for (const choice of op.choices ?? []) {
+          out += `complete -c ${app} -n '${enumCond}' -a '${escFishSingleQuoted(choice)}'\n`;
+        }
+      } else {
+        const shortPart = op.shortName ? `-s ${op.shortName} ` : "";
+        out += `complete -c ${app} -n '${cond}' ${shortPart}-l ${op.name} -d '${escFishSingleQuoted(op.description)}' -r\n`;
+      }
+    }
+
+    if (sc.wantsFiles && sc.kids.length === 0) {
+      out += `complete -c ${app} -n '${cond}' -F\n`;
+    }
+  }
+
+  return out;
+}

package/src/builtins/completion-group.ts

@@ -0,0 +1,50 @@
+import { CliCommand } from "../types.ts";
+
+/**
+ * Builds the static `completion` / `bash` / `zsh` / `fish` command subtree (merged into the program root at runtime).
+ */
+export function cliBuiltinCompletionGroup(appName: string): CliCommand {
+  return {
+    key: "completion",
+    description: "Generate the autocompletion script for shells.",
+    commands: [
+      {
+        key: "bash",
+        description: "Print a bash tab-completion script.",
+        notes:
+          "Output is the whole script.\n" +
+          "Pipe it to a file, or feed it straight into your shell.\n\n" +
+          "To keep it across restarts, save it and source that file from ~/.bashrc.\n\n" +
+          "For example:\n\n" +
+          `echo 'eval \"$(${appName} completion bash)\"' >> ~/.bashrc\n` +
+          `\nor\n` +
+          `  ${appName} completion bash > ~/.bash_completion.d/${appName}\n` +
+          `  echo 'source ~/.bash_completion.d/${appName}' >> ~/.bashrc\n\n` +
+          "To try it only in this session (nothing written to disk):\n" +
+          `  source <(${appName} completion bash)`,
+        handler: () => {},
+      },
+      {
+        key: "zsh",
+        description: "Print a zsh tab-completion script.",
+        notes:
+          "Output is the whole script.\n\n" +
+          `fpath setup: ${appName} completion zsh > ~/.zsh/completions/_${appName}\n\n` +
+          `source setup: echo 'eval \"$(${appName} completion zsh)\"' >> ~/.zshrc\n\n` +
+          "To try it only in this session (nothing written to disk):\n" +
+          `  eval \"$(${appName} completion zsh)\"`,
+        handler: () => {},
+      },
+      {
+        key: "fish",
+        description: "Print a fish tab-completion script.",
+        notes:
+          "Output is the whole script.\n\n" +
+          "Install:\n" +
+          `  ${appName} completion fish > ~/.config/fish/completions/${appName}.fish\n\n` +
+          "Fish loads completions from that directory automatically.",
+        handler: () => {},
+      },
+    ],
+  };
+}

package/src/builtins/completion-zsh.ts

@@ -0,0 +1,244 @@
+import { CliCommand, CliOptionKind } from "../types.ts";
+import { collectScopes, type ScopeRec } from "./scopes.ts";
+import {
+  escShellSingleQuoted,
+  identToken,
+  kHelpLong,
+  kHelpShort,
+  kSchemaDesc,
+  kSchemaLong,
+  mainName,
+} from "./shell-helpers.ts";
+
+function emitScopeArraysZsh(ident: string, scopes: ScopeRec[]): string {
+  let out = "";
+  for (const [i, sc] of scopes.entries()) {
+    out += "typeset -g A_" + ident + "_" + i + "_opts\n";
+    out += "A_" + ident + "_" + i + "_opts=(";
+    out += "'" + escShellSingleQuoted(kHelpLong) + ":" + escShellSingleQuoted("Show help for this command.") + "' '" + escShellSingleQuoted(kHelpShort) + ":" + escShellSingleQuoted("Show help for this command.") + "'";
+    if (sc.path === "") {
+      out += " '" + escShellSingleQuoted(kSchemaLong) + ":" + escShellSingleQuoted(kSchemaDesc) + "'";
+    }
+    for (const o of sc.opts) {
+      out += " '" + escShellSingleQuoted("--" + o.name) + ":" + escShellSingleQuoted(o.description) + "'";
+      if (o.shortName) {
+        out += " '" + escShellSingleQuoted("-" + o.shortName) + ":" + escShellSingleQuoted(o.description) + "'";
+      }
+    }
+    out += ")\n";
+    out += "typeset -g A_" + ident + "_" + i + "_leaf=" + (sc.kids.length === 0 ? "1" : "0") + "\n";
+    out += "typeset -g A_" + ident + "_" + i + "_pos=" + (sc.wantsFiles ? "1" : "0") + "\n";
+    if (sc.kids.length > 0) {
+      out += "typeset -g A_" + ident + "_" + i + "_cmds=(";
+      for (const ch of sc.kids) {
+        out += " '" + escShellSingleQuoted(ch.key) + ":" + escShellSingleQuoted(ch.description) + "'";
+      }
+      out += ")\n";
+    }
+  }
+  return out;
+}
+
+function emitConsumeLongZsh(ident: string, scopes: ScopeRec[]): string {
+  let o = "_${ident}_nac_consume_long() {\n".replace("${ident}", ident);
+  o += "  local sid=\"$1\" w=\"$2\" nw=\"$3\"\n";
+  o += "  case $sid in\n";
+  for (const [i, sc] of scopes.entries()) {
+    o += "    " + i + ")\n";
+    o += "      case $w in\n";
+    o += "        " + kHelpLong + "|${kHelpLong}=*|${kHelpShort}) echo 1 ;;\n".replace(/\$\{kHelpLong\}/g, kHelpLong).replace(/\$\{kHelpShort\}/g, kHelpShort);
+    if (sc.path === "") {
+      o += "        " + kSchemaLong + ") echo 1 ;;\n";
+    }
+    for (const op of sc.opts) {
+      const base = "--" + op.name;
+      if (op.kind === "presence") {
+        o += "        " + base + "|${base}=*) echo 1 ;;\n".replace(/\$\{base\}/g, base);
+      } else {
+        o += "        " + base + "=*) echo 1 ;;\n";
+        o += "        " + base + ") echo 2 ;;\n";
+      }
+    }
+    o += "        *) echo 0 ;;\n";
+    o += "      esac\n";
+    o += "      ;;\n";
+  }
+  o += "    *) echo 0 ;;\n";
+  o += "  esac\n";
+  o += "}\n";
+  return o;
+}
+
+function emitConsumeShortZsh(ident: string, scopes: ScopeRec[]): string {
+  let o = "_${ident}_nac_consume_short() {\n".replace("${ident}", ident);
+  o += "  local sid=\"$1\" w=\"$2\"\n";
+  o += "  case $sid in\n";
+  for (const [i, sc] of scopes.entries()) {
+    o += "    " + i + ")\n";
+    o += "      local rest=${w#-}\n";
+    o += "      local ch\n";
+    o += "      local saw=0\n";
+    o += "      while [[ -n $rest ]]; do\n";
+    o += "        ch=${rest[1,1]}\n";
+    o += "        rest=${rest[2,-1]}\n";
+    o += "        case $ch in\n";
+    let boolChars = "";
+    for (const op of sc.opts) {
+      if (!op.shortName) continue;
+      if (op.kind === "presence") {
+        boolChars += op.shortName + "|";
+      } else {
+        o += "          " + op.shortName + ")\n";
+        o += "            if [[ $saw -ne 0 || -n $rest ]]; then echo 0; return; fi\n";
+        o += "            echo 2; return ;;\n";
+      }
+    }
+    if (boolChars.length > 0) {
+      boolChars = boolChars.slice(0, -1);
+      o += "          " + boolChars + ") ;;\n";
+    }
+    o += "          *) echo 0; return ;;\n";
+    o += "        esac\n";
+    o += "        saw=1\n";
+    o += "      done\n";
+    o += "      echo 1\n";
+    o += "      ;;\n";
+  }
+  o += "    *) echo 0 ;;\n";
+  o += "  esac\n";
+  o += "}\n";
+  return o;
+}
+
+function emitMatchChildZsh(ident: string, scopes: ScopeRec[], pathIndex: Record<string, number>): string {
+  let o = "_${ident}_nac_match_child() {\n".replace("${ident}", ident);
+  o += "  local sid=\"$1\" w=\"$2\"\n";
+  o += "  case $sid in\n";
+  for (const [sid, sc] of scopes.entries()) {
+    if (sc.kids.length === 0) continue;
+    o += "    " + sid + ")\n";
+    o += "      case $w in\n";
+    for (const ch of sc.kids) {
+      const childPath = sc.path === "" ? ch.key : sc.path + "/" + ch.key;
+      const cid = pathIndex[childPath] ?? 0;
+      o += "        " + ch.key + ") echo " + cid + "; return 0 ;;\n";
+    }
+    o += "      esac\n";
+    o += "      ;;\n";
+  }
+  o += "  esac\n";
+  o += "  return 1\n";
+  o += "}\n";
+  return o;
+}
+
+function emitSimulateZsh(ident: string): string {
+  let o = "_${ident}_nac_simulate() {\n".replace("${ident}", ident);
+  o += "  local i=2 sid=0 w steps next\n";
+  o += "  while (( i < CURRENT )); do\n";
+  o += "    w=$words[i]\n";
+  o += "    if [[ $w == " + kHelpShort + " || $w == " + kHelpLong + " || $w == " + kSchemaLong + " ]]; then\n";
+  o += "      ((i++)); continue\n";
+  o += "    fi\n";
+  o += "    if [[ $w == --* ]]; then\n";
+  o += "      steps=$(_${ident}_nac_consume_long \"$sid\" \"$w\" \"${words[i+1]}\")\n".replace("${ident}", ident);
+  o += "      case $steps in\n";
+  o += "        0) break ;;\n";
+  o += "        1) ((i++)) ;;\n";
+  o += "        2) ((i+=2)) ;;\n";
+  o += "        *) break ;;\n";
+  o += "      esac\n";
+  o += "      continue\n";
+  o += "    fi\n";
+  o += "    if [[ $w == -* ]]; then\n";
+  o += "      steps=$(_${ident}_nac_consume_short \"$sid\" \"$w\")\n".replace("${ident}", ident);
+  o += "      case $steps in\n";
+  o += "        0) break ;;\n";
+  o += "        1) ((i++)) ;;\n";
+  o += "        2) ((i++)); break ;;\n";
+  o += "        *) break ;;\n";
+  o += "      esac\n";
+  o += "      continue\n";
+  o += "    fi\n";
+  o += "    next=$(_${ident}_nac_match_child \"$sid\" \"$w\") || break\n".replace("${ident}", ident);
+  o += "    sid=$next\n";
+  o += "    ((i++))\n";
+  o += "  done\n";
+  o += "  REPLY_SID=$sid\n";
+  o += "}\n";
+  return o;
+}
+
+function emitEnumReplyZsh(ident: string, scopes: ScopeRec[]): string {
+  let o = "_${ident}_nac_enum_reply() {\n".replace("${ident}", ident);
+  o += "  local sid=$1 prev=$2\n";
+  o += "  case $sid in\n";
+  for (const [i, sc] of scopes.entries()) {
+    const enumOpts = sc.opts.filter((op) => op.kind === CliOptionKind.Enum && (op.choices?.length ?? 0) > 0);
+    if (enumOpts.length === 0) continue;
+    o += "    " + i + ")\n";
+    o += "      case $prev in\n";
+    for (const op of enumOpts) {
+      const vals = (op.choices ?? []).map((c) => escShellSingleQuoted(c)).join(" ");
+      o += "        --" + op.name + ") _values " + vals + "; return 0 ;;\n";
+    }
+    o += "      esac\n";
+    o += "      ;;\n";
+  }
+  o += "  esac\n";
+  o += "  return 1\n";
+  o += "}\n";
+  return o;
+}
+
+function emitMainBodyZsh(schema: CliCommand, ident: string): string {
+  const main = mainName(schema.key);
+  let o = "_${main}() {\n".replace("${main}", main);
+  o += "  local curcontext=\"$curcontext\" ret=1\n";
+  o += "  _${ident}_nac_simulate\n".replace("${ident}", ident);
+  o += "  local sid=$REPLY_SID\n";
+  o += "  if _${ident}_nac_enum_reply \"$sid\" \"$words[CURRENT-1]\"; then return 0; fi\n".replace("${ident}", ident);
+  o += "  if [[ $PREFIX == -* ]]; then\n";
+  o += "    local -a optsarr\n";
+  o += "    local oname=\"A_${ident}_${sid}_opts\"\n".replace("${ident}", ident);
+  o += "    optsarr=(${(@P)oname})\n";
+  o += "    _describe -t options 'option' optsarr && ret=0\n";
+  o += "  else\n";
+  o += "    local lname=\"A_${ident}_${sid}_leaf\"\n".replace("${ident}", ident);
+  o += "    if [[ ${(P)lname} -eq 0 ]]; then\n";
+  o += "      local -a cmdsarr\n";
+  o += "      local cname=\"A_${ident}_${sid}_cmds\"\n".replace("${ident}", ident);
+  o += "      cmdsarr=(${(@P)cname})\n";
+  o += "      _describe -t commands 'command' cmdsarr && ret=0\n";
+  o += "    else\n";
+  o += "      local pname=\"A_${ident}_${sid}_pos\"\n".replace("${ident}", ident);
+  o += "      if [[ ${(P)pname} -eq 1 ]]; then\n";
+  o += "        _files && ret=0\n";
+  o += "      fi\n";
+  o += "    fi\n";
+  o += "  fi\n";
+  o += "  return ret\n";
+  o += "}\n\n";
+  o += "compdef _${main} ${schema.key}\n".replace("${main}", main).replace("${schema.key}", schema.key);
+  return o;
+}
+
+/** Returns a self-contained zsh completion script for the given program schema. */
+export function completionZshScript(schema: CliCommand): string {
+  const ident = identToken(schema.key);
+  const scopes = collectScopes(schema);
+  const pathIndex: Record<string, number> = {};
+  for (const [i, s] of scopes.entries()) {
+    pathIndex[s.path] = i;
+  }
+
+  let out = "#compdef " + schema.key + "\n\n";
+  out += emitScopeArraysZsh(ident, scopes);
+  out += emitConsumeLongZsh(ident, scopes);
+  out += emitConsumeShortZsh(ident, scopes);
+  out += emitMatchChildZsh(ident, scopes, pathIndex);
+  out += emitSimulateZsh(ident);
+  out += emitEnumReplyZsh(ident, scopes);
+  out += emitMainBodyZsh(schema, ident);
+  return out;
+}