377 lines
11 KiB
JavaScript
377 lines
11 KiB
JavaScript
import { argsert } from './argsert.js';
|
|
import { assertNotStrictEqual } from './typings/common-types.js';
|
|
import { levenshtein as distance } from './utils/levenshtein.js';
|
|
import { objFilter } from './utils/obj-filter.js';
|
|
const specialKeys = ['$0', '--', '_'];
|
|
export function validation(yargs, usage, y18n, shim) {
|
|
const __ = y18n.__;
|
|
const __n = y18n.__n;
|
|
const self = {};
|
|
self.nonOptionCount = function nonOptionCount(argv) {
|
|
const demandedCommands = yargs.getDemandedCommands();
|
|
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(
|
|
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(
|
|
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()
|
|
)
|
|
);
|
|
}
|
|
}
|
|
}
|
|
};
|
|
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 + ''
|
|
)
|
|
);
|
|
}
|
|
};
|
|
self.requiredArguments = function requiredArguments(argv) {
|
|
const demandedOptions = yargs.getDemandedOptions();
|
|
let missing = 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 = [];
|
|
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
|
|
)
|
|
);
|
|
}
|
|
};
|
|
self.unknownArguments = function unknownArguments(
|
|
argv,
|
|
aliases,
|
|
positionalMap,
|
|
isDefaultCommand,
|
|
checkPositionals = true
|
|
) {
|
|
const commandKeys = yargs.getCommandInstance().getCommands();
|
|
const unknown = [];
|
|
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 = [];
|
|
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;
|
|
}
|
|
};
|
|
self.isValidAndSomeAliasIsNotNew = function isValidAndSomeAliasIsNotNew(
|
|
key,
|
|
aliases
|
|
) {
|
|
if (!Object.prototype.hasOwnProperty.call(aliases, key)) {
|
|
return false;
|
|
}
|
|
const newAliases = yargs.parsed.newAliases;
|
|
for (const a of [key, ...aliases[key]]) {
|
|
if (
|
|
!Object.prototype.hasOwnProperty.call(newAliases, a) ||
|
|
!newAliases[key]
|
|
) {
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
};
|
|
self.limitedChoices = function limitedChoices(argv) {
|
|
const options = yargs.getOptions();
|
|
const invalid = {};
|
|
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) => {
|
|
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);
|
|
};
|
|
let checks = [];
|
|
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);
|
|
}
|
|
}
|
|
};
|
|
let implied = {};
|
|
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, val) {
|
|
const num = Number(val);
|
|
val = isNaN(num) ? val : num;
|
|
if (typeof val === 'number') {
|
|
val = argv._.length >= val;
|
|
} else if (val.match(/^--no-.+/)) {
|
|
val = val.match(/^--no-(.+)/)[1];
|
|
val = !argv[val];
|
|
} else {
|
|
val = argv[val];
|
|
}
|
|
return val;
|
|
}
|
|
self.implications = function implications(argv) {
|
|
const implyFail = [];
|
|
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 = {};
|
|
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) => {
|
|
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;
|
|
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 = [];
|
|
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;
|
|
}
|