123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251
import type { Command } from "../command.ts";import type { IArgument } from "../types.ts";import { FileType } from "../types/file.ts";
/** Generates bash completions script. */export class BashCompletionsGenerator {  /** Generates bash completions script for given command. */  public static generate(cmd: Command) {    return new BashCompletionsGenerator(cmd).generate();  }
  private constructor(protected cmd: Command) {}
  /** Generates bash completions code. */  private generate(): string {    const path = this.cmd.getPath();    const version: string | undefined = this.cmd.getVersion()      ? ` v${this.cmd.getVersion()}`      : "";
    return `#!/usr/bin/env bash# bash completion support for ${path}${version}
_${replaceSpecialChars(path)}() {  local word cur prev listFiles  local -a opts  COMPREPLY=()  cur="\${COMP_WORDS[COMP_CWORD]}"  prev="\${COMP_WORDS[COMP_CWORD-1]}"  cmd="_"  opts=()  listFiles=0
  _${replaceSpecialChars(this.cmd.getName())}_complete() {    local action="$1"; shift    mapfile -t values < <( ${this.cmd.getName()} completions complete "\${action}" "\${@}" )    for i in "\${values[@]}"; do      opts+=("$i")    done  }
  _${replaceSpecialChars(this.cmd.getName())}_expand() {    [ "$cur" != "\${cur%\\\\}" ] && cur="$cur\\\\"      # expand ~username type directory specifications    if [[ "$cur" == \\~*/* ]]; then      eval cur=$cur          elif [[ "$cur" == \\~* ]]; then      cur=\${cur#\\~}      COMPREPLY=( $( compgen -P '~' -u $cur ) )      return \${#COMPREPLY[@]}    fi  }
  _${replaceSpecialChars(this.cmd.getName())}_file_dir() {    listFiles=1    local IFS=$'\\t\\n' xspec #glob    _${replaceSpecialChars(this.cmd.getName())}_expand || return 0      if [ "\${1:-}" = -d ]; then      COMPREPLY=( \${COMPREPLY[@]:-} $( compgen -d -- $cur ) )      #eval "$glob"    # restore glob setting.      return 0    fi      xspec=\${1:+"!*.$1"}	# set only if glob passed in as $1    COMPREPLY=( \${COMPREPLY[@]:-} $( compgen -f -X "$xspec" -- "$cur" ) \          $( compgen -d -- "$cur" ) )  }
  ${this.generateCompletions(this.cmd).trim()}
  for word in "\${COMP_WORDS[@]}"; do    case "\${word}" in      -*) ;;      *)        cmd_tmp="\${cmd}_\${word//[^[:alnum:]]/_}"        if type "\${cmd_tmp}" &>/dev/null; then          cmd="\${cmd_tmp}"        fi    esac  done
  \${cmd}
  if [[ listFiles -eq 1 ]]; then    return 0  fi
  if [[ \${#opts[@]} -eq 0 ]]; then    # shellcheck disable=SC2207    COMPREPLY=($(compgen -f "\${cur}"))    return 0  fi
  local values  values="$( printf "\\n%s" "\${opts[@]}" )"  local IFS=$'\\n'  # shellcheck disable=SC2207  local result=($(compgen -W "\${values[@]}" -- "\${cur}"))  if [[ \${#result[@]} -eq 0 ]]; then    # shellcheck disable=SC2207    COMPREPLY=($(compgen -f "\${cur}"))  else    # shellcheck disable=SC2207    COMPREPLY=($(printf '%q\\n' "\${result[@]}"))  fi
  return 0}
complete -F _${replaceSpecialChars(path)} -o bashdefault -o default ${path}`;  }
  /** Generates bash completions method for given command and child commands. */  private generateCompletions(command: Command, path = "", index = 1): string {    path = (path ? path + " " : "") + command.getName();    const commandCompletions = this.generateCommandCompletions(      command,      path,      index,    );    const childCommandCompletions: string = command.getCommands(false)      .filter((subCommand: Command) => subCommand !== command)      .map((subCommand: Command) =>        this.generateCompletions(subCommand, path, index + 1)      )      .join("");
    return `${commandCompletions}
${childCommandCompletions}`;  }
  private generateCommandCompletions(    command: Command,    path: string,    index: number,  ): string {    const flags: string[] = this.getFlags(command);
    const childCommandNames: string[] = command.getCommands(false)      .map((childCommand: Command) => childCommand.getName());
    const completionsPath: string = ~path.indexOf(" ")      ? " " + path.split(" ").slice(1).join(" ")      : "";
    const optionArguments = this.generateOptionArguments(      command,      completionsPath,    );
    const completionsCmd: string = this.generateCommandCompletionsCommand(      command,      completionsPath,    );
    return `  __${replaceSpecialChars(path)}() {    opts=(${[...flags, ...childCommandNames].join(" ")})    ${completionsCmd}    if [[ \${cur} == -* || \${COMP_CWORD} -eq ${index} ]] ; then      return 0    fi    ${optionArguments}  }`;  }
  private getFlags(command: Command): string[] {    return command.getOptions(false)      .map((option) => option.flags)      .flat();  }
  private generateOptionArguments(    command: Command,    completionsPath: string,  ): string {    let opts = "";    const options = command.getOptions(false);    if (options.length) {      opts += 'case "${prev}" in';      for (const option of options) {        const flags: string = option.flags          .map((flag: string) => flag.trim())          .join("|");
        const completionsCmd: string = this.generateOptionCompletionsCommand(          command,          option.args,          completionsPath,          { standalone: option.standalone },        );
        opts += `\n      ${flags}) ${completionsCmd} ;;`;      }      opts += "\n    esac";    }
    return opts;  }
  private generateCommandCompletionsCommand(    command: Command,    path: string,  ) {    const args: IArgument[] = command.getArguments();    if (args.length) {      const type = command.getType(args[0].type);      if (type && type.handler instanceof FileType) {        return `_${replaceSpecialChars(this.cmd.getName())}_file_dir`;      }      // @TODO: add support for multiple arguments      return `_${replaceSpecialChars(this.cmd.getName())}_complete ${        args[0].action      }${path}`;    }
    return "";  }
  private generateOptionCompletionsCommand(    command: Command,    args: IArgument[],    path: string,    opts?: { standalone?: boolean },  ) {    if (args.length) {      const type = command.getType(args[0].type);      if (type && type.handler instanceof FileType) {        return `opts=(); _${replaceSpecialChars(this.cmd.getName())}_file_dir`;      }      // @TODO: add support for multiple arguments      return `opts=(); _${replaceSpecialChars(this.cmd.getName())}_complete ${        args[0].action      }${path}`;    }
    if (opts?.standalone) {      return "opts=()";    }
    return "";  }}
function replaceSpecialChars(str: string): string {  return str.replace(/[^a-zA-Z0-9]/g, "_");}