123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131
#!/usr/bin/env deno --allow-run// Copyright 2019 Yoshiya Hinosawa. All rights reserved. MIT license.
import writeFile = Deno.writeFile;
const { exit, args, readFile } = Deno;import { parse } from "https://deno.land/std@v0.3.1/flags/mod.ts";import { red, green, blue } from "https://deno.land/std@v0.3.1/colors/mod.ts";import { globrex } from "https://deno.land/std@v0.3.1/fs/globrex.ts";import { encode } from "https://deno.land/std@v0.3.1/strings/strings.ts";
import { xrun, decode } from "./util.ts";
function match(filename, glob) {  return globrex(glob, { globstar: true }).regex.test(filename);}
async function readConfig(config = ".licenserc.json") {  let data;  let configObj;  try {    data = await readFile(config);  } catch (e) {    console.log(`Error: config file "${config}" not found.`);    exit(1);  }
  try {    configObj = JSON.parse(decode(data));  } catch (e) {    console.log(`Error: Failed to parse "${config}" as JSON.`);    console.log(e);    exit(1);  }
  const ignore: string[] = configObj.ignore || [];  delete configObj.ignore;
  const entries: Array<[string, string | string[]]> = Object.entries(configObj);
  return { ignore, config: entries };}
const checkFile = async (  filename: string,  copyright: string | string[],  quiet: boolean,  inject: boolean) => {  const sourceCode = decode(await readFile(filename));  const copyrightLines: string[] = Array.isArray(copyright)    ? copyright    : [String(copyright)];  if (copyrightLines.every(line => sourceCode.includes(line))) {    if (!quiet) {      console.log(filename, "...", green("ok"));    }    return true;  }
  if (inject) {    console.log(`${filename} ${blue("missing copyright. injecting ... done")}`);    await writeFile(      filename,      encode(copyrightLines.join("\n") + "\n" + sourceCode)    );    return true;  } else {    console.log(filename, red("missing copyright!"));  }
  return false;};
const main = async opts => {  if (opts.help) {    console.log(`Usage: license_checker.ts [options]
Options:  -H, --help               Show this help message and exit.  -V, --version            Show the version number and exit.  -q, --quiet              Don't print messages except errors.  -i, --inject             Inject license into head if missing.  -c, --config <filename>  Specify config filename. Default is '.licenserc.json'.`);    exit(0);  }
  if (opts.version) {    console.log("1.5.0");    exit(0);  }
  const { config, ignore } = await readConfig();  const filenames = (await xrun(["git", "ls-files"])).trim().split("\n");
  const tasks = [];
  for (const filename of filenames) {    for (const [glob, copyright] of config) {      if (ignore.some(pattern => filename.includes(pattern))) {        continue;      }      if (match(filename, glob)) {        tasks.push(checkFile(filename, copyright, opts.quiet, opts.inject));      }    }  }
  const results = await Promise.all(tasks);
  if (results.includes(false)) {    exit(1);    return;  }  exit(0);};
main(  parse(args.slice(1), {    boolean: ["quiet", "help", "version", "inject"],    alias: {      q: "quiet",      i: "inject",      H: "help",      V: "version"    }  }));