123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426
import { bold, red } from "https://deno.land/std@v0.37.1/fmt/colors.ts";import { existsSync, walkSync } from "https://deno.land/std@v0.37.1/fs/mod.ts";import * as path from "https://deno.land/std@v0.37.1/path/mod.ts";
export class DrakeError extends Error {  constructor(message?: string) {    super(message);    this.name = "DrakeError";  }}
type Env = { [name: string]: any; "--tasks": string[] };
/**  * The Drake `env` object contains the command-line options, tasks an variables:  *  * Options are keyed by their long option name e.g.  `env["--dry-run"]`. Unspecified flag options  * are undefined; unspecified value options are assigned their default value.  *  * Tasks names are stored in the `env["--tasks"]` string array. A default task can be specified by  * setting `env["--default-task"]` to the task name.  *  * Variable values are keyed by name. For example `vers=1.0.1` on the command-line is available as  * `env["vers"]` and `env.vers`.  */export const env: Env = { "--tasks": [] };
export function parseEnv(args: string[], env: Env): void {  let arg: string | undefined;  while (!!(arg = args.shift())) {    const match = arg.match(/^([a-zA-Z]\w*)=(.*)$/);    if (match) {      env[match[1]] = match[2];      continue;    }    switch (arg) {      case "-a":      case "--always-make":        env["-a"] = true;        env["--always-make"] = true;        break;      case "-d":      case "--directory":        arg = args.shift();        if (arg === undefined) {          abort("missing --directory option value");        }        env["--directory"] = arg;        break;      case "-f":      case "--drakefile":        arg = args.shift();        if (arg === undefined) {          abort("missing --drakefile option value");        }        env["--drakefile"] = arg;        break;      case "-h":      case "--help":        env["--help"] = true;        break;      case "-l":      case "--list-tasks":        env["--list-tasks"] = true;        break;      case "-L":        env["--list-all"] = true;        break;      case "-n":      case "--dry-run":        env["--dry-run"] = true;        break;      case "-q":      case "--quiet":        env["--quiet"] = true;        break;      case "--version":        env["--version"] = true;        break;      default:        if (arg.startsWith("-")) {          abort(`illegal option: ${arg}`);        }        env["--tasks"].push(arg);        break;    }  }}
/** Print error message to to `stderr` and terminate execution. */export function abort(message: string): never {  if (env["--abort-exits"]) {    console.error(`${red(bold("drake error:"))} ${message}`);    Deno.exit(1);  } else {    throw new DrakeError(message);  }}
/** * Log a message to the console. Do not log the message if the `--quiet` command-line option is set. */export function log(message: string): void {  if (!env["--quiet"]) {    console.log(message);  }}
/** * Quote string array values with double-quotes then join them with a separator. * Double-quote characters are escaped with a backspace. * The separator defaults to a space character. */export function quote(values: string[], sep: string = " "): string {  values = values.map(value => `"${value.replace(/"/g, '\\"')}"`);  return values.join(sep);}
/** Read the entire contents of a file synchronously to a UTF-8 string. */export function readFile(filename: string): string {  return new TextDecoder("utf-8").decode(Deno.readFileSync(filename));}
/* Write text to a file synchronously. If the file exists it will be overwritten. */export function writeFile(filename: string, text: string): void {  Deno.writeFileSync(filename, new TextEncoder().encode(text));}
/** Find and replace in text file synchronously. */export function updateFile(  filename: string,  find: RegExp,  replace: string): void {  writeFile(filename, readFile(filename).replace(find, replace));}
/** * Update the modification time of each file to the current time. * If a file does not exist then create a zero length file. * Missing parent directory paths are also created. */export function touch(...files: string[]): void {  for (const file of files) {    const dir = path.dirname(file);    if (!existsSync(dir)) {      Deno.mkdirSync(dir, { recursive: true });    }    Deno.openSync(file, "w").close();  }}
/** * Return `true` if either the target file does not exist or its modification time is older then one * or more prerequisite files. Otherwise return `false`. */export function outOfDate(target: string, prereqs: string[]): boolean {  if (!existsSync(target)) {    return true;  }  const targetStat = Deno.statSync(target);  for (const prereq of prereqs) {    const prereqStat = Deno.statSync(prereq);    if (!targetStat.modified || !prereqStat.modified) {      continue;    }    if (targetStat.modified < prereqStat.modified) {      return true;    }  }  return false;}
/** * Return true if `name` is a normal task name. Normal task names contain one or more alphanumeric, * underscore and hyphen characters and cannot start with a hyphen. * *     isNormalTask("hello-world")    // true *     isNormalTask("io.ts")          // false *     isNormalTask("./hello-world")  // false * */export function isNormalTask(name: string): boolean {  return /^\w[\w-]*$/.test(name);}
/** * Return true if `name` is a file task name. File task names are valid file paths. *  *     isFileTask("io.ts")          // true *     isFileTask("hello-world")    // false *     isFileTask("./hello-world")  // true *  */export function isFileTask(name: string): boolean {  return !isNormalTask(name);}
/** * The path name is normalized and, if necessary, prefixed with a period and path separator to * distinguish it from non-file task name. * *     normalizePath("hello-world")   // "./hello-world" *     normalizePath("./lib/io.ts")   // "lib/io.ts" */export function normalizePath(name: string): string {  name = path.normalize(name);  if (isNormalTask(name)) {    name = "." + path.sep + name;  }  return name;}
/** Normalise Drake task name. Throw an error if the name is blank or it contains wildcard * characters. */export function normalizeTaskName(name: string): string {  name = name.trim();  if (name === "") {    abort("blank task name");  }  if (path.isGlob(name)) {    abort(`wildcard task name not allowed: ${name}`);  }  if (isFileTask(name)) {    name = normalizePath(name);  }  return name;}
/** * Return a list prerequisite task names. * Globs are expanded and path names are normalized. */export function normalizePrereqs(prereqs: string[]): string[] {  const result: string[] = [];  for (let prereq of prereqs) {    prereq = prereq.trim();    if (prereq === "") {      abort("blank prerequisite name");    }    if (!isFileTask(prereq)) {      result.push(prereq);    } else if (path.isGlob(prereq)) {      result.push(...glob(prereq));    } else {      result.push(normalizePath(prereq));    }  }  return result;}
/** * Return a  sorted array of normalized file names matching the wildcard patterns. * Wildcard patterns can include the `**` (globstar) pattern. * e.g. `glob("tmp/*.ts", "lib/*.ts", "mod.ts");` */export function glob(...patterns: string[]): string[] {  function glob1(pattern: string): string[] {    const globOptions = { extended: true, globstar: true };    pattern = path.normalizeGlob(pattern, globOptions);    let root = path.dirname(pattern);    while (root !== "." && path.isGlob(root)) {      root = path.dirname(root);    }    const regexp = path.globToRegExp(pattern, globOptions);    const iter = walkSync(root, { match: [regexp], includeDirs: false });    return Array.from(iter, info => info.filename);  }  let result: string[] = [];  for (const pattern of patterns) {    result = [...result, ...glob1(pattern)];  }  // Drop dups, normalise and sort paths.  return [...new Set(result)].map(p => normalizePath(p)).sort();}
function shArgs(command: string): [string[], string | undefined] {  let cmdArgs: string[];  let cmdFile: string | undefined;  if (Deno.build.os === "win") {    cmdFile = Deno.makeTempFileSync(      { prefix: "drake_", suffix: ".cmd" }    );    writeFile(cmdFile, `@echo off\n${command}`);    cmdArgs = [cmdFile];  } else {    const shellExe = Deno.env("SHELL")!;    if (!shellExe) {      abort(`cannot locate shell: missing SHELL environment variable`);    }    cmdArgs = [shellExe, "-c", command];  }  return [cmdArgs, cmdFile];}
export interface ShOpts {  cwd?: string;  env?: { [key: string]: string };  stdout?: Deno.ProcessStdio;  stderr?: Deno.ProcessStdio;}
/** * Execute commands in the command shell. *  * - If `commands` is a string execute it. * - If `commands` is an array of commands execute them asynchronously. * - If any command fails throw an error. * - If `opts.stdout` or `opts.stderr` is set to `"null"` then the respective outputs are ignored. * - `opts.cwd` sets the shell current working directory (defaults to the parent process working directory). * - The `opts.env` mapping passes additional environment variables to the shell. *  * Examples: *  *     await sh("echo Hello World"); *     await sh(["echo Hello 1", "echo Hello 2", "echo Hello 3"]); *     await sh("echo Hello World", { stdout: "null" }); */export async function sh(commands: string | string[], opts: ShOpts = {}) {  if (typeof commands === "string") {    commands = [commands];  }  const tempFiles: string[] = [];  const processes: Deno.Process[] = [];  const results: Deno.ProcessStatus[] = [];  try {    for (const cmd of commands) {      let cmdArgs: string[];      let cmdFile: string | undefined;      [cmdArgs, cmdFile] = shArgs(cmd);      if (cmdFile) tempFiles.push(cmdFile);      const p = Deno.run({        cmd: cmdArgs,        cwd: opts.cwd,        env: opts.env,        stdout: opts.stdout ?? "inherit",        stderr: opts.stderr ?? "inherit"      });      processes.push(p);    }    results.push(...await Promise.all(processes.map(p => p.status())));  } finally {    for (const p of processes) {      p.close();    }  }  for (const f of tempFiles) {    Deno.removeSync(f);  }  for (const i in results) {    const cmd = commands[i];    const code = results[i].code;    if (code === undefined) {      abort(`sh: ${cmd}: undefined exit code`);    }    if (code !== 0) {      abort(`sh: ${cmd}: error code: ${code}`);    }  }}
export type ShOutput = {  code: number | undefined;  output: string;  error: string;};
export interface ShCaptureOpts extends ShOpts {  input?: string;}
/** * Execute `command` in the command shell and return a promise for the exit `code`, `output` (the * stdout output) and `error` (the stderr output). * * - If the `opts.input` string has been assigned then it is piped to the shell `stdin`. * - `opts.cwd` sets the shell current working directory (defaults to the parent process working *   directory). * - The `opts.env` mapping passes additional environment variables to the shell. * - `opts.stdout` and `opts.stderr` have `Deno.ProcessStdio` semantics and default to `"piped"`. * * Examples: * *     const { code, stdout, stderr } = await shCapture("echo Hello");  */export async function shCapture(  command: string,  opts: ShCaptureOpts = {}): Promise<ShOutput> {  let cmdArgs: string[];  let cmdFile: string | undefined;  [cmdArgs, cmdFile] = shArgs(command);  const p = Deno.run({    cmd: cmdArgs,    cwd: opts.cwd,    env: opts.env,    stdin: opts.input !== undefined ? "piped" : undefined,    stdout: opts.stdout ?? "piped",    stderr: opts.stderr ?? "piped"  });  let status: Deno.ProcessStatus;  let outputBytes, errorBytes: Uint8Array;  try {    if (p.stdin) {      await p.stdin.write(new TextEncoder().encode(opts.input));      p.stdin.close();    }    [status, outputBytes, errorBytes] = await Promise.all(      [        p.status(),        p.stdout ? p.output() : Promise.resolve(new Uint8Array()),        p.stderr ? p.stderrOutput() : Promise.resolve(new Uint8Array())      ]    );  } finally {    p.close();  }  if (cmdFile) Deno.removeSync(cmdFile);  return {    code: status.code,    output: new TextDecoder().decode(outputBytes),    error: new TextDecoder().decode(errorBytes)  };}