validation.ts

import {argsert} from './argsert.js';
import {
  Dictionary,
  assertNotStrictEqual,
  Y18N,
  PlatformShim,
} from './typings/common-types.js';
import {levenshtein as distance} from './utils/levenshtein.js';
import {objFilter} from './utils/obj-filter.js';
import {UsageInstance} from './usage.js';
import {YargsInstance, Arguments} from './yargs-factory.js';
import {DetailedArguments} from './typings/yargs-parser-types.js';

const specialKeys = ['$0', '--', '_'];

// validation-type-stuff, missing params,
// bad implications, custom checks.
export function validation(
  yargs: YargsInstance,
  usage: UsageInstance,
  y18n: Y18N,
  shim: PlatformShim
) {
  const __ = y18n.__;
  const __n = y18n.__n;
  const self = {} as ValidationInstance;

  // validate appropriate # of non-option
  // arguments were provided, i.e., '_'.
  self.nonOptionCount = function nonOptionCount(argv) {
    const demandedCommands = yargs.getDemandedCommands();
    // don't count currently executing commands
    const positionalCount =
      argv._.length + (argv['--'] ? argv['--'].length : 0);
    const _s = positionalCount - yargs.getContext().commands.length;

    if (
      demandedCommands._ &&
      (_s < demandedCommands._.min || _s > demandedCommands._.max)
    ) {
      if (_s < demandedCommands._.min) {
        if (demandedCommands._.minMsg !== undefined) {
          usage.fail(
            // replace $0 with observed, $1 with expected.
            demandedCommands._.minMsg
              ? demandedCommands._.minMsg
                  .replace(/\$0/g, _s.toString())
                  .replace(/\$1/, demandedCommands._.min.toString())
              : null
          );
        } else {
          usage.fail(
            __n(
              'Not enough non-option arguments: got %s, need at least %s',
              'Not enough non-option arguments: got %s, need at least %s',
              _s,
              _s.toString(),
              demandedCommands._.min.toString()
            )
          );
        }
      } else if (_s > demandedCommands._.max) {
        if (demandedCommands._.maxMsg !== undefined) {
          usage.fail(
            // replace $0 with observed, $1 with expected.
            demandedCommands._.maxMsg
              ? demandedCommands._.maxMsg
                  .replace(/\$0/g, _s.toString())
                  .replace(/\$1/, demandedCommands._.max.toString())
              : null
          );
        } else {
          usage.fail(
            __n(
              'Too many non-option arguments: got %s, maximum of %s',
              'Too many non-option arguments: got %s, maximum of %s',
              _s,
              _s.toString(),
              demandedCommands._.max.toString()
            )
          );
        }
      }
    }
  };

  // validate the appropriate # of <required>
  // positional arguments were provided:
  self.positionalCount = function positionalCount(required, observed) {
    if (observed < required) {
      usage.fail(
        __n(
          'Not enough non-option arguments: got %s, need at least %s',
          'Not enough non-option arguments: got %s, need at least %s',
          observed,
          observed + '',
          required + ''
        )
      );
    }
  };

  // make sure all the required arguments are present.
  self.requiredArguments = function requiredArguments(
    argv,
    demandedOptions: Dictionary<string | undefined>
  ) {
    let missing: Dictionary<string | undefined> | null = null;

    for (const key of Object.keys(demandedOptions)) {
      if (
        !Object.prototype.hasOwnProperty.call(argv, key) ||
        typeof argv[key] === 'undefined'
      ) {
        missing = missing || {};
        missing[key] = demandedOptions[key];
      }
    }

    if (missing) {
      const customMsgs: string[] = [];
      for (const key of Object.keys(missing)) {
        const msg = missing[key];
        if (msg && customMsgs.indexOf(msg) < 0) {
          customMsgs.push(msg);
        }
      }

      const customMsg = customMsgs.length ? `\n${customMsgs.join('\n')}` : '';
      usage.fail(
        __n(
          'Missing required argument: %s',
          'Missing required arguments: %s',
          Object.keys(missing).length,
          Object.keys(missing).join(', ') + customMsg
        )
      );
    }
  };

  // check for unknown arguments (strict-mode).
  self.unknownArguments = function unknownArguments(
    argv,
    aliases,
    positionalMap,
    isDefaultCommand,
    checkPositionals = true
  ) {
    const commandKeys = yargs.getCommandInstance().getCommands();
    const unknown: string[] = [];
    const currentContext = yargs.getContext();

    Object.keys(argv).forEach(key => {
      if (
        specialKeys.indexOf(key) === -1 &&
        !Object.prototype.hasOwnProperty.call(positionalMap, key) &&
        !Object.prototype.hasOwnProperty.call(yargs._getParseContext(), key) &&
        !self.isValidAndSomeAliasIsNotNew(key, aliases)
      ) {
        unknown.push(key);
      }
    });

    if (
      checkPositionals &&
      (currentContext.commands.length > 0 ||
        commandKeys.length > 0 ||
        isDefaultCommand)
    ) {
      argv._.slice(currentContext.commands.length).forEach(key => {
        if (commandKeys.indexOf('' + key) === -1) {
          unknown.push('' + key);
        }
      });
    }

    if (unknown.length > 0) {
      usage.fail(
        __n(
          'Unknown argument: %s',
          'Unknown arguments: %s',
          unknown.length,
          unknown.join(', ')
        )
      );
    }
  };

  self.unknownCommands = function unknownCommands(argv) {
    const commandKeys = yargs.getCommandInstance().getCommands();
    const unknown: string[] = [];
    const currentContext = yargs.getContext();

    if (currentContext.commands.length > 0 || commandKeys.length > 0) {
      argv._.slice(currentContext.commands.length).forEach(key => {
        if (commandKeys.indexOf('' + key) === -1) {
          unknown.push('' + key);
        }
      });
    }

    if (unknown.length > 0) {
      usage.fail(
        __n(
          'Unknown command: %s',
          'Unknown commands: %s',
          unknown.length,
          unknown.join(', ')
        )
      );
      return true;
    } else {
      return false;
    }
  };

  // check for a key that is not an alias, or for which every alias is new,
  // implying that it was invented by the parser, e.g., during camelization
  self.isValidAndSomeAliasIsNotNew = function isValidAndSomeAliasIsNotNew(
    key,
    aliases
  ) {
    if (!Object.prototype.hasOwnProperty.call(aliases, key)) {
      return false;
    }
    const newAliases = (yargs.parsed as DetailedArguments).newAliases;
    for (const a of [key, ...aliases[key]]) {
      if (
        !Object.prototype.hasOwnProperty.call(newAliases, a) ||
        !newAliases[key]
      ) {
        return true;
      }
    }
    return false;
  };

  // validate arguments limited to enumerated choices
  self.limitedChoices = function limitedChoices(argv) {
    const options = yargs.getOptions();
    const invalid: Dictionary<any[]> = {};

    if (!Object.keys(options.choices).length) return;

    Object.keys(argv).forEach(key => {
      if (
        specialKeys.indexOf(key) === -1 &&
        Object.prototype.hasOwnProperty.call(options.choices, key)
      ) {
        [].concat(argv[key]).forEach(value => {
          // TODO case-insensitive configurability
          if (
            options.choices[key].indexOf(value) === -1 &&
            value !== undefined
          ) {
            invalid[key] = (invalid[key] || []).concat(value);
          }
        });
      }
    });

    const invalidKeys = Object.keys(invalid);

    if (!invalidKeys.length) return;

    let msg = __('Invalid values:');
    invalidKeys.forEach(key => {
      msg += `\n  ${__(
        'Argument: %s, Given: %s, Choices: %s',
        key,
        usage.stringifiedValues(invalid[key]),
        usage.stringifiedValues(options.choices[key])
      )}`;
    });
    usage.fail(msg);
  };

  // custom checks, added using the `check` option on yargs.
  let checks: CustomCheck[] = [];
  self.check = function check(f, global) {
    checks.push({
      func: f,
      global,
    });
  };

  self.customChecks = function customChecks(argv, aliases) {
    for (let i = 0, f; (f = checks[i]) !== undefined; i++) {
      const func = f.func;
      let result = null;
      try {
        result = func(argv, aliases);
      } catch (err) {
        usage.fail(err.message ? err.message : err, err);
        continue;
      }

      if (!result) {
        usage.fail(__('Argument check failed: %s', func.toString()));
      } else if (typeof result === 'string' || result instanceof Error) {
        usage.fail(result.toString(), result);
      }
    }
  };

  // check implications, argument foo implies => argument bar.
  let implied: Dictionary<KeyOrPos[]> = {};
  self.implies = function implies(key, value) {
    argsert(
      '<string|object> [array|number|string]',
      [key, value],
      arguments.length
    );

    if (typeof key === 'object') {
      Object.keys(key).forEach(k => {
        self.implies(k, key[k]);
      });
    } else {
      yargs.global(key);
      if (!implied[key]) {
        implied[key] = [];
      }
      if (Array.isArray(value)) {
        value.forEach(i => self.implies(key, i));
      } else {
        assertNotStrictEqual(value, undefined, shim);
        implied[key].push(value);
      }
    }
  };
  self.getImplied = function getImplied() {
    return implied;
  };

  function keyExists(argv: Arguments, val: any): any {
    // convert string '1' to number 1
    const num = Number(val);
    val = isNaN(num) ? val : num;

    if (typeof val === 'number') {
      // check length of argv._
      val = argv._.length >= val;
    } else if (val.match(/^--no-.+/)) {
      // check if key/value doesn't exist
      val = val.match(/^--no-(.+)/)[1];
      val = !argv[val];
    } else {
      // check if key/value exists
      val = argv[val];
    }
    return val;
  }

  self.implications = function implications(argv) {
    const implyFail: string[] = [];

    Object.keys(implied).forEach(key => {
      const origKey = key;
      (implied[key] || []).forEach(value => {
        let key = origKey;
        const origValue = value;
        key = keyExists(argv, key);
        value = keyExists(argv, value);

        if (key && !value) {
          implyFail.push(` ${origKey} -> ${origValue}`);
        }
      });
    });

    if (implyFail.length) {
      let msg = `${__('Implications failed:')}\n`;

      implyFail.forEach(value => {
        msg += value;
      });

      usage.fail(msg);
    }
  };

  let conflicting: Dictionary<(string | undefined)[]> = {};
  self.conflicts = function conflicts(key, value) {
    argsert('<string|object> [array|string]', [key, value], arguments.length);

    if (typeof key === 'object') {
      Object.keys(key).forEach(k => {
        self.conflicts(k, key[k]);
      });
    } else {
      yargs.global(key);
      if (!conflicting[key]) {
        conflicting[key] = [];
      }
      if (Array.isArray(value)) {
        value.forEach(i => self.conflicts(key, i));
      } else {
        conflicting[key].push(value);
      }
    }
  };
  self.getConflicting = () => conflicting;

  self.conflicting = function conflictingFn(argv) {
    Object.keys(argv).forEach(key => {
      if (conflicting[key]) {
        conflicting[key].forEach(value => {
          // we default keys to 'undefined' that have been configured, we should not
          // apply conflicting check unless they are a value other than 'undefined'.
          if (value && argv[key] !== undefined && argv[value] !== undefined) {
            usage.fail(
              __('Arguments %s and %s are mutually exclusive', key, value)
            );
          }
        });
      }
    });
  };

  self.recommendCommands = function recommendCommands(cmd, potentialCommands) {
    const threshold = 3; // if it takes more than three edits, let's move on.
    potentialCommands = potentialCommands.sort((a, b) => b.length - a.length);

    let recommended = null;
    let bestDistance = Infinity;
    for (
      let i = 0, candidate;
      (candidate = potentialCommands[i]) !== undefined;
      i++
    ) {
      const d = distance(cmd, candidate);
      if (d <= threshold && d < bestDistance) {
        bestDistance = d;
        recommended = candidate;
      }
    }
    if (recommended) usage.fail(__('Did you mean %s?', recommended));
  };

  self.reset = function reset(localLookup) {
    implied = objFilter(implied, k => !localLookup[k]);
    conflicting = objFilter(conflicting, k => !localLookup[k]);
    checks = checks.filter(c => c.global);
    return self;
  };

  const frozens: FrozenValidationInstance[] = [];
  self.freeze = function freeze() {
    frozens.push({
      implied,
      checks,
      conflicting,
    });
  };
  self.unfreeze = function unfreeze() {
    const frozen = frozens.pop();
    assertNotStrictEqual(frozen, undefined, shim);
    ({implied, checks, conflicting} = frozen);
  };

  return self;
}

/** Instance of the validation module. */
export interface ValidationInstance {
  check(f: CustomCheck['func'], global: boolean): void;
  conflicting(argv: Arguments): void;
  conflicts(
    key: string | Dictionary<string | string[]>,
    value?: string | string[]
  ): void;
  customChecks(argv: Arguments, aliases: DetailedArguments['aliases']): void;
  freeze(): void;
  getConflicting(): Dictionary<(string | undefined)[]>;
  getImplied(): Dictionary<KeyOrPos[]>;
  implications(argv: Arguments): void;
  implies(
    key: string | Dictionary<KeyOrPos | KeyOrPos[]>,
    value?: KeyOrPos | KeyOrPos[]
  ): void;
  isValidAndSomeAliasIsNotNew(
    key: string,
    aliases: DetailedArguments['aliases']
  ): boolean;
  limitedChoices(argv: Arguments): void;
  nonOptionCount(argv: Arguments): void;
  positionalCount(required: number, observed: number): void;
  recommendCommands(cmd: string, potentialCommands: string[]): void;
  requiredArguments(
    argv: Arguments,
    demandedOptions: Dictionary<string | undefined>
  ): void;
  reset(localLookup: Dictionary): ValidationInstance;
  unfreeze(): void;
  unknownArguments(
    argv: Arguments,
    aliases: DetailedArguments['aliases'],
    positionalMap: Dictionary,
    isDefaultCommand: boolean,
    checkPositionals?: boolean
  ): void;
  unknownCommands(argv: Arguments): boolean;
}

interface CustomCheck {
  func: (argv: Arguments, aliases: DetailedArguments['aliases']) => any;
  global: boolean;
}

interface FrozenValidationInstance {
  implied: Dictionary<KeyOrPos[]>;
  checks: CustomCheck[];
  conflicting: Dictionary<(string | undefined)[]>;
}

export type KeyOrPos = string | number;