forked from hero/www_hero
196 lines
4.4 KiB
JavaScript
196 lines
4.4 KiB
JavaScript
|
const flagSymbol = Symbol('arg flag');
|
||
|
|
||
|
class ArgError extends Error {
|
||
|
constructor(msg, code) {
|
||
|
super(msg);
|
||
|
this.name = 'ArgError';
|
||
|
this.code = code;
|
||
|
|
||
|
Object.setPrototypeOf(this, ArgError.prototype);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
function arg(
|
||
|
opts,
|
||
|
{
|
||
|
argv = process.argv.slice(2),
|
||
|
permissive = false,
|
||
|
stopAtPositional = false
|
||
|
} = {}
|
||
|
) {
|
||
|
if (!opts) {
|
||
|
throw new ArgError(
|
||
|
'argument specification object is required',
|
||
|
'ARG_CONFIG_NO_SPEC'
|
||
|
);
|
||
|
}
|
||
|
|
||
|
const result = { _: [] };
|
||
|
|
||
|
const aliases = {};
|
||
|
const handlers = {};
|
||
|
|
||
|
for (const key of Object.keys(opts)) {
|
||
|
if (!key) {
|
||
|
throw new ArgError(
|
||
|
'argument key cannot be an empty string',
|
||
|
'ARG_CONFIG_EMPTY_KEY'
|
||
|
);
|
||
|
}
|
||
|
|
||
|
if (key[0] !== '-') {
|
||
|
throw new ArgError(
|
||
|
`argument key must start with '-' but found: '${key}'`,
|
||
|
'ARG_CONFIG_NONOPT_KEY'
|
||
|
);
|
||
|
}
|
||
|
|
||
|
if (key.length === 1) {
|
||
|
throw new ArgError(
|
||
|
`argument key must have a name; singular '-' keys are not allowed: ${key}`,
|
||
|
'ARG_CONFIG_NONAME_KEY'
|
||
|
);
|
||
|
}
|
||
|
|
||
|
if (typeof opts[key] === 'string') {
|
||
|
aliases[key] = opts[key];
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
let type = opts[key];
|
||
|
let isFlag = false;
|
||
|
|
||
|
if (
|
||
|
Array.isArray(type) &&
|
||
|
type.length === 1 &&
|
||
|
typeof type[0] === 'function'
|
||
|
) {
|
||
|
const [fn] = type;
|
||
|
type = (value, name, prev = []) => {
|
||
|
prev.push(fn(value, name, prev[prev.length - 1]));
|
||
|
return prev;
|
||
|
};
|
||
|
isFlag = fn === Boolean || fn[flagSymbol] === true;
|
||
|
} else if (typeof type === 'function') {
|
||
|
isFlag = type === Boolean || type[flagSymbol] === true;
|
||
|
} else {
|
||
|
throw new ArgError(
|
||
|
`type missing or not a function or valid array type: ${key}`,
|
||
|
'ARG_CONFIG_VAD_TYPE'
|
||
|
);
|
||
|
}
|
||
|
|
||
|
if (key[1] !== '-' && key.length > 2) {
|
||
|
throw new ArgError(
|
||
|
`short argument keys (with a single hyphen) must have only one character: ${key}`,
|
||
|
'ARG_CONFIG_SHORTOPT_TOOLONG'
|
||
|
);
|
||
|
}
|
||
|
|
||
|
handlers[key] = [type, isFlag];
|
||
|
}
|
||
|
|
||
|
for (let i = 0, len = argv.length; i < len; i++) {
|
||
|
const wholeArg = argv[i];
|
||
|
|
||
|
if (stopAtPositional && result._.length > 0) {
|
||
|
result._ = result._.concat(argv.slice(i));
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
if (wholeArg === '--') {
|
||
|
result._ = result._.concat(argv.slice(i + 1));
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
if (wholeArg.length > 1 && wholeArg[0] === '-') {
|
||
|
/* eslint-disable operator-linebreak */
|
||
|
const separatedArguments =
|
||
|
wholeArg[1] === '-' || wholeArg.length === 2
|
||
|
? [wholeArg]
|
||
|
: wholeArg
|
||
|
.slice(1)
|
||
|
.split('')
|
||
|
.map((a) => `-${a}`);
|
||
|
/* eslint-enable operator-linebreak */
|
||
|
|
||
|
for (let j = 0; j < separatedArguments.length; j++) {
|
||
|
const arg = separatedArguments[j];
|
||
|
const [originalArgName, argStr] =
|
||
|
arg[1] === '-' ? arg.split(/=(.*)/, 2) : [arg, undefined];
|
||
|
|
||
|
let argName = originalArgName;
|
||
|
while (argName in aliases) {
|
||
|
argName = aliases[argName];
|
||
|
}
|
||
|
|
||
|
if (!(argName in handlers)) {
|
||
|
if (permissive) {
|
||
|
result._.push(arg);
|
||
|
continue;
|
||
|
} else {
|
||
|
throw new ArgError(
|
||
|
`unknown or unexpected option: ${originalArgName}`,
|
||
|
'ARG_UNKNOWN_OPTION'
|
||
|
);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
const [type, isFlag] = handlers[argName];
|
||
|
|
||
|
if (!isFlag && j + 1 < separatedArguments.length) {
|
||
|
throw new ArgError(
|
||
|
`option requires argument (but was followed by another short argument): ${originalArgName}`,
|
||
|
'ARG_MISSING_REQUIRED_SHORTARG'
|
||
|
);
|
||
|
}
|
||
|
|
||
|
if (isFlag) {
|
||
|
result[argName] = type(true, argName, result[argName]);
|
||
|
} else if (argStr === undefined) {
|
||
|
if (
|
||
|
argv.length < i + 2 ||
|
||
|
(argv[i + 1].length > 1 &&
|
||
|
argv[i + 1][0] === '-' &&
|
||
|
!(
|
||
|
argv[i + 1].match(/^-?\d*(\.(?=\d))?\d*$/) &&
|
||
|
(type === Number ||
|
||
|
// eslint-disable-next-line no-undef
|
||
|
(typeof BigInt !== 'undefined' && type === BigInt))
|
||
|
))
|
||
|
) {
|
||
|
const extended =
|
||
|
originalArgName === argName ? '' : ` (alias for ${argName})`;
|
||
|
throw new ArgError(
|
||
|
`option requires argument: ${originalArgName}${extended}`,
|
||
|
'ARG_MISSING_REQUIRED_LONGARG'
|
||
|
);
|
||
|
}
|
||
|
|
||
|
result[argName] = type(argv[i + 1], argName, result[argName]);
|
||
|
++i;
|
||
|
} else {
|
||
|
result[argName] = type(argStr, argName, result[argName]);
|
||
|
}
|
||
|
}
|
||
|
} else {
|
||
|
result._.push(wholeArg);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
arg.flag = (fn) => {
|
||
|
fn[flagSymbol] = true;
|
||
|
return fn;
|
||
|
};
|
||
|
|
||
|
// Utility types
|
||
|
arg.COUNT = arg.flag((v, name, existingCount) => (existingCount || 0) + 1);
|
||
|
|
||
|
// Expose error class
|
||
|
arg.ArgError = ArgError;
|
||
|
|
||
|
module.exports = arg;
|