forked from zhurui/management
1056 lines
34 KiB
JavaScript
1056 lines
34 KiB
JavaScript
/**
|
|
* dashdash - A light, featureful and explicit option parsing library for
|
|
* node.js.
|
|
*/
|
|
// vim: set ts=4 sts=4 sw=4 et:
|
|
|
|
var assert = require('assert-plus');
|
|
var format = require('util').format;
|
|
var fs = require('fs');
|
|
var path = require('path');
|
|
|
|
|
|
var DEBUG = true;
|
|
if (DEBUG) {
|
|
var debug = console.warn;
|
|
} else {
|
|
var debug = function () {};
|
|
}
|
|
|
|
|
|
|
|
// ---- internal support stuff
|
|
|
|
// Replace {{variable}} in `s` with the template data in `d`.
|
|
function renderTemplate(s, d) {
|
|
return s.replace(/{{([a-zA-Z]+)}}/g, function (match, key) {
|
|
return d.hasOwnProperty(key) ? d[key] : match;
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Return a shallow copy of the given object;
|
|
*/
|
|
function shallowCopy(obj) {
|
|
if (!obj) {
|
|
return (obj);
|
|
}
|
|
var copy = {};
|
|
Object.keys(obj).forEach(function (k) {
|
|
copy[k] = obj[k];
|
|
});
|
|
return (copy);
|
|
}
|
|
|
|
|
|
function space(n) {
|
|
var s = '';
|
|
for (var i = 0; i < n; i++) {
|
|
s += ' ';
|
|
}
|
|
return s;
|
|
}
|
|
|
|
|
|
function makeIndent(arg, deflen, name) {
|
|
if (arg === null || arg === undefined)
|
|
return space(deflen);
|
|
else if (typeof (arg) === 'number')
|
|
return space(arg);
|
|
else if (typeof (arg) === 'string')
|
|
return arg;
|
|
else
|
|
assert.fail('invalid "' + name + '": not a string or number: ' + arg);
|
|
}
|
|
|
|
|
|
/**
|
|
* Return an array of lines wrapping the given text to the given width.
|
|
* This splits on whitespace. Single tokens longer than `width` are not
|
|
* broken up.
|
|
*/
|
|
function textwrap(s, width) {
|
|
var words = s.trim().split(/\s+/);
|
|
var lines = [];
|
|
var line = '';
|
|
words.forEach(function (w) {
|
|
var newLength = line.length + w.length;
|
|
if (line.length > 0)
|
|
newLength += 1;
|
|
if (newLength > width) {
|
|
lines.push(line);
|
|
line = '';
|
|
}
|
|
if (line.length > 0)
|
|
line += ' ';
|
|
line += w;
|
|
});
|
|
lines.push(line);
|
|
return lines;
|
|
}
|
|
|
|
|
|
/**
|
|
* Transform an option name to a "key" that is used as the field
|
|
* on the `opts` object returned from `<parser>.parse()`.
|
|
*
|
|
* Transformations:
|
|
* - '-' -> '_': This allow one to use hyphen in option names (common)
|
|
* but not have to do silly things like `opt["dry-run"]` to access the
|
|
* parsed results.
|
|
*/
|
|
function optionKeyFromName(name) {
|
|
return name.replace(/-/g, '_');
|
|
}
|
|
|
|
|
|
|
|
// ---- Option types
|
|
|
|
function parseBool(option, optstr, arg) {
|
|
return Boolean(arg);
|
|
}
|
|
|
|
function parseString(option, optstr, arg) {
|
|
assert.string(arg, 'arg');
|
|
return arg;
|
|
}
|
|
|
|
function parseNumber(option, optstr, arg) {
|
|
assert.string(arg, 'arg');
|
|
var num = Number(arg);
|
|
if (isNaN(num)) {
|
|
throw new Error(format('arg for "%s" is not a number: "%s"',
|
|
optstr, arg));
|
|
}
|
|
return num;
|
|
}
|
|
|
|
function parseInteger(option, optstr, arg) {
|
|
assert.string(arg, 'arg');
|
|
var num = Number(arg);
|
|
if (!/^[0-9-]+$/.test(arg) || isNaN(num)) {
|
|
throw new Error(format('arg for "%s" is not an integer: "%s"',
|
|
optstr, arg));
|
|
}
|
|
return num;
|
|
}
|
|
|
|
function parsePositiveInteger(option, optstr, arg) {
|
|
assert.string(arg, 'arg');
|
|
var num = Number(arg);
|
|
if (!/^[0-9]+$/.test(arg) || isNaN(num) || num === 0) {
|
|
throw new Error(format('arg for "%s" is not a positive integer: "%s"',
|
|
optstr, arg));
|
|
}
|
|
return num;
|
|
}
|
|
|
|
/**
|
|
* Supported date args:
|
|
* - epoch second times (e.g. 1396031701)
|
|
* - ISO 8601 format: YYYY-MM-DD[THH:MM:SS[.sss][Z]]
|
|
* 2014-03-28T18:35:01.489Z
|
|
* 2014-03-28T18:35:01.489
|
|
* 2014-03-28T18:35:01Z
|
|
* 2014-03-28T18:35:01
|
|
* 2014-03-28
|
|
*/
|
|
function parseDate(option, optstr, arg) {
|
|
assert.string(arg, 'arg');
|
|
var date;
|
|
if (/^\d+$/.test(arg)) {
|
|
// epoch seconds
|
|
date = new Date(Number(arg) * 1000);
|
|
/* JSSTYLED */
|
|
} else if (/^\d{4}-\d{2}-\d{2}(T\d{2}:\d{2}:\d{2}(\.\d+)?Z?)?$/i.test(arg)) {
|
|
// ISO 8601 format
|
|
date = new Date(arg);
|
|
} else {
|
|
throw new Error(format('arg for "%s" is not a valid date format: "%s"',
|
|
optstr, arg));
|
|
}
|
|
if (date.toString() === 'Invalid Date') {
|
|
throw new Error(format('arg for "%s" is an invalid date: "%s"',
|
|
optstr, arg));
|
|
}
|
|
return date;
|
|
}
|
|
|
|
var optionTypes = {
|
|
bool: {
|
|
takesArg: false,
|
|
parseArg: parseBool
|
|
},
|
|
string: {
|
|
takesArg: true,
|
|
helpArg: 'ARG',
|
|
parseArg: parseString
|
|
},
|
|
number: {
|
|
takesArg: true,
|
|
helpArg: 'NUM',
|
|
parseArg: parseNumber
|
|
},
|
|
integer: {
|
|
takesArg: true,
|
|
helpArg: 'INT',
|
|
parseArg: parseInteger
|
|
},
|
|
positiveInteger: {
|
|
takesArg: true,
|
|
helpArg: 'INT',
|
|
parseArg: parsePositiveInteger
|
|
},
|
|
date: {
|
|
takesArg: true,
|
|
helpArg: 'DATE',
|
|
parseArg: parseDate
|
|
},
|
|
arrayOfBool: {
|
|
takesArg: false,
|
|
array: true,
|
|
parseArg: parseBool
|
|
},
|
|
arrayOfString: {
|
|
takesArg: true,
|
|
helpArg: 'ARG',
|
|
array: true,
|
|
parseArg: parseString
|
|
},
|
|
arrayOfNumber: {
|
|
takesArg: true,
|
|
helpArg: 'NUM',
|
|
array: true,
|
|
parseArg: parseNumber
|
|
},
|
|
arrayOfInteger: {
|
|
takesArg: true,
|
|
helpArg: 'INT',
|
|
array: true,
|
|
parseArg: parseInteger
|
|
},
|
|
arrayOfPositiveInteger: {
|
|
takesArg: true,
|
|
helpArg: 'INT',
|
|
array: true,
|
|
parseArg: parsePositiveInteger
|
|
},
|
|
arrayOfDate: {
|
|
takesArg: true,
|
|
helpArg: 'INT',
|
|
array: true,
|
|
parseArg: parseDate
|
|
},
|
|
};
|
|
|
|
|
|
|
|
// ---- Parser
|
|
|
|
/**
|
|
* Parser constructor.
|
|
*
|
|
* @param config {Object} The parser configuration
|
|
* - options {Array} Array of option specs. See the README for how to
|
|
* specify each option spec.
|
|
* - allowUnknown {Boolean} Default false. Whether to throw on unknown
|
|
* options. If false, then unknown args are included in the _args array.
|
|
* - interspersed {Boolean} Default true. Whether to allow interspersed
|
|
* arguments (non-options) and options. E.g.:
|
|
* node tool.js arg1 arg2 -v
|
|
* '-v' is after some args here. If `interspersed: false` then '-v'
|
|
* would not be parsed out. Note that regardless of `interspersed`
|
|
* the presence of '--' will stop option parsing, as all good
|
|
* option parsers should.
|
|
*/
|
|
function Parser(config) {
|
|
assert.object(config, 'config');
|
|
assert.arrayOfObject(config.options, 'config.options');
|
|
assert.optionalBool(config.interspersed, 'config.interspersed');
|
|
var self = this;
|
|
|
|
// Allow interspersed arguments (true by default).
|
|
this.interspersed = (config.interspersed !== undefined
|
|
? config.interspersed : true);
|
|
|
|
// Don't allow unknown flags (true by default).
|
|
this.allowUnknown = (config.allowUnknown !== undefined
|
|
? config.allowUnknown : false);
|
|
|
|
this.options = config.options.map(function (o) { return shallowCopy(o); });
|
|
this.optionFromName = {};
|
|
this.optionFromEnv = {};
|
|
for (var i = 0; i < this.options.length; i++) {
|
|
var o = this.options[i];
|
|
if (o.group !== undefined && o.group !== null) {
|
|
assert.optionalString(o.group,
|
|
format('config.options.%d.group', i));
|
|
continue;
|
|
}
|
|
assert.ok(optionTypes[o.type],
|
|
format('invalid config.options.%d.type: "%s" in %j',
|
|
i, o.type, o));
|
|
assert.optionalString(o.name, format('config.options.%d.name', i));
|
|
assert.optionalArrayOfString(o.names,
|
|
format('config.options.%d.names', i));
|
|
assert.ok((o.name || o.names) && !(o.name && o.names),
|
|
format('exactly one of "name" or "names" required: %j', o));
|
|
assert.optionalString(o.help, format('config.options.%d.help', i));
|
|
var env = o.env || [];
|
|
if (typeof (env) === 'string') {
|
|
env = [env];
|
|
}
|
|
assert.optionalArrayOfString(env, format('config.options.%d.env', i));
|
|
assert.optionalString(o.helpGroup,
|
|
format('config.options.%d.helpGroup', i));
|
|
assert.optionalBool(o.helpWrap,
|
|
format('config.options.%d.helpWrap', i));
|
|
assert.optionalBool(o.hidden, format('config.options.%d.hidden', i));
|
|
|
|
if (o.name) {
|
|
o.names = [o.name];
|
|
} else {
|
|
assert.string(o.names[0],
|
|
format('config.options.%d.names is empty', i));
|
|
}
|
|
o.key = optionKeyFromName(o.names[0]);
|
|
o.names.forEach(function (n) {
|
|
if (self.optionFromName[n]) {
|
|
throw new Error(format(
|
|
'option name collision: "%s" used in %j and %j',
|
|
n, self.optionFromName[n], o));
|
|
}
|
|
self.optionFromName[n] = o;
|
|
});
|
|
env.forEach(function (n) {
|
|
if (self.optionFromEnv[n]) {
|
|
throw new Error(format(
|
|
'option env collision: "%s" used in %j and %j',
|
|
n, self.optionFromEnv[n], o));
|
|
}
|
|
self.optionFromEnv[n] = o;
|
|
});
|
|
}
|
|
}
|
|
|
|
Parser.prototype.optionTakesArg = function optionTakesArg(option) {
|
|
return optionTypes[option.type].takesArg;
|
|
};
|
|
|
|
/**
|
|
* Parse options from the given argv.
|
|
*
|
|
* @param inputs {Object} Optional.
|
|
* - argv {Array} Optional. The argv to parse. Defaults to
|
|
* `process.argv`.
|
|
* - slice {Number} The index into argv at which options/args begin.
|
|
* Default is 2, as appropriate for `process.argv`.
|
|
* - env {Object} Optional. The env to use for 'env' entries in the
|
|
* option specs. Defaults to `process.env`.
|
|
* @returns {Object} Parsed `opts`. It has special keys `_args` (the
|
|
* remaining args from `argv`) and `_order` (gives the order that
|
|
* options were specified).
|
|
*/
|
|
Parser.prototype.parse = function parse(inputs) {
|
|
var self = this;
|
|
|
|
// Old API was `parse([argv, [slice]])`
|
|
if (Array.isArray(arguments[0])) {
|
|
inputs = {argv: arguments[0], slice: arguments[1]};
|
|
}
|
|
|
|
assert.optionalObject(inputs, 'inputs');
|
|
if (!inputs) {
|
|
inputs = {};
|
|
}
|
|
assert.optionalArrayOfString(inputs.argv, 'inputs.argv');
|
|
//assert.optionalNumber(slice, 'slice');
|
|
var argv = inputs.argv || process.argv;
|
|
var slice = inputs.slice !== undefined ? inputs.slice : 2;
|
|
var args = argv.slice(slice);
|
|
var env = inputs.env || process.env;
|
|
var opts = {};
|
|
var _order = [];
|
|
|
|
function addOpt(option, optstr, key, val, from) {
|
|
var type = optionTypes[option.type];
|
|
var parsedVal = type.parseArg(option, optstr, val);
|
|
if (type.array) {
|
|
if (!opts[key]) {
|
|
opts[key] = [];
|
|
}
|
|
if (type.arrayFlatten && Array.isArray(parsedVal)) {
|
|
for (var i = 0; i < parsedVal.length; i++) {
|
|
opts[key].push(parsedVal[i]);
|
|
}
|
|
} else {
|
|
opts[key].push(parsedVal);
|
|
}
|
|
} else {
|
|
opts[key] = parsedVal;
|
|
}
|
|
var item = { key: key, value: parsedVal, from: from };
|
|
_order.push(item);
|
|
}
|
|
|
|
// Parse args.
|
|
var _args = [];
|
|
var i = 0;
|
|
outer: while (i < args.length) {
|
|
var arg = args[i];
|
|
|
|
// End of options marker.
|
|
if (arg === '--') {
|
|
i++;
|
|
break;
|
|
|
|
// Long option
|
|
} else if (arg.slice(0, 2) === '--') {
|
|
var name = arg.slice(2);
|
|
var val = null;
|
|
var idx = name.indexOf('=');
|
|
if (idx !== -1) {
|
|
val = name.slice(idx + 1);
|
|
name = name.slice(0, idx);
|
|
}
|
|
var option = this.optionFromName[name];
|
|
if (!option) {
|
|
if (!this.allowUnknown)
|
|
throw new Error(format('unknown option: "--%s"', name));
|
|
else if (this.interspersed)
|
|
_args.push(arg);
|
|
else
|
|
break outer;
|
|
} else {
|
|
var takesArg = this.optionTakesArg(option);
|
|
if (val !== null && !takesArg) {
|
|
throw new Error(format('argument given to "--%s" option '
|
|
+ 'that does not take one: "%s"', name, arg));
|
|
}
|
|
if (!takesArg) {
|
|
addOpt(option, '--'+name, option.key, true, 'argv');
|
|
} else if (val !== null) {
|
|
addOpt(option, '--'+name, option.key, val, 'argv');
|
|
} else if (i + 1 >= args.length) {
|
|
throw new Error(format('do not have enough args for "--%s" '
|
|
+ 'option', name));
|
|
} else {
|
|
addOpt(option, '--'+name, option.key, args[i + 1], 'argv');
|
|
i++;
|
|
}
|
|
}
|
|
|
|
// Short option
|
|
} else if (arg[0] === '-' && arg.length > 1) {
|
|
var j = 1;
|
|
var allFound = true;
|
|
while (j < arg.length) {
|
|
var name = arg[j];
|
|
var option = this.optionFromName[name];
|
|
if (!option) {
|
|
allFound = false;
|
|
if (this.allowUnknown) {
|
|
if (this.interspersed) {
|
|
_args.push(arg);
|
|
break;
|
|
} else
|
|
break outer;
|
|
} else if (arg.length > 2) {
|
|
throw new Error(format(
|
|
'unknown option: "-%s" in "%s" group',
|
|
name, arg));
|
|
} else {
|
|
throw new Error(format('unknown option: "-%s"', name));
|
|
}
|
|
} else if (this.optionTakesArg(option)) {
|
|
break;
|
|
}
|
|
j++;
|
|
}
|
|
|
|
j = 1;
|
|
while (allFound && j < arg.length) {
|
|
var name = arg[j];
|
|
var val = arg.slice(j + 1); // option val if it takes an arg
|
|
var option = this.optionFromName[name];
|
|
var takesArg = this.optionTakesArg(option);
|
|
if (!takesArg) {
|
|
addOpt(option, '-'+name, option.key, true, 'argv');
|
|
} else if (val) {
|
|
addOpt(option, '-'+name, option.key, val, 'argv');
|
|
break;
|
|
} else {
|
|
if (i + 1 >= args.length) {
|
|
throw new Error(format('do not have enough args '
|
|
+ 'for "-%s" option', name));
|
|
}
|
|
addOpt(option, '-'+name, option.key, args[i + 1], 'argv');
|
|
i++;
|
|
break;
|
|
}
|
|
j++;
|
|
}
|
|
|
|
// An interspersed arg
|
|
} else if (this.interspersed) {
|
|
_args.push(arg);
|
|
|
|
// An arg and interspersed args are not allowed, so done options.
|
|
} else {
|
|
break outer;
|
|
}
|
|
i++;
|
|
}
|
|
_args = _args.concat(args.slice(i));
|
|
|
|
// Parse environment.
|
|
Object.keys(this.optionFromEnv).forEach(function (envname) {
|
|
var val = env[envname];
|
|
if (val === undefined)
|
|
return;
|
|
var option = self.optionFromEnv[envname];
|
|
if (opts[option.key] !== undefined)
|
|
return;
|
|
var takesArg = self.optionTakesArg(option);
|
|
if (takesArg) {
|
|
addOpt(option, envname, option.key, val, 'env');
|
|
} else if (val !== '') {
|
|
// Boolean envvar handling:
|
|
// - VAR=<empty-string> not set (as if the VAR was not set)
|
|
// - VAR=0 false
|
|
// - anything else true
|
|
addOpt(option, envname, option.key, (val !== '0'), 'env');
|
|
}
|
|
});
|
|
|
|
// Apply default values.
|
|
this.options.forEach(function (o) {
|
|
if (opts[o.key] === undefined) {
|
|
if (o.default !== undefined) {
|
|
opts[o.key] = o.default;
|
|
} else if (o.type && optionTypes[o.type].default !== undefined) {
|
|
opts[o.key] = optionTypes[o.type].default;
|
|
}
|
|
}
|
|
});
|
|
|
|
opts._order = _order;
|
|
opts._args = _args;
|
|
return opts;
|
|
};
|
|
|
|
|
|
/**
|
|
* Return help output for the current options.
|
|
*
|
|
* E.g.: if the current options are:
|
|
* [{names: ['help', 'h'], type: 'bool', help: 'Show help and exit.'}]
|
|
* then this would return:
|
|
* ' -h, --help Show help and exit.\n'
|
|
*
|
|
* @param config {Object} Config for controlling the option help output.
|
|
* - indent {Number|String} Default 4. An indent/prefix to use for
|
|
* each option line.
|
|
* - nameSort {String} Default is 'length'. By default the names are
|
|
* sorted to put the short opts first (i.e. '-h, --help' preferred
|
|
* to '--help, -h'). Set to 'none' to not do this sorting.
|
|
* - maxCol {Number} Default 80. Note that long tokens in a help string
|
|
* can go past this.
|
|
* - helpCol {Number} Set to specify a specific column at which
|
|
* option help will be aligned. By default this is determined
|
|
* automatically.
|
|
* - minHelpCol {Number} Default 20.
|
|
* - maxHelpCol {Number} Default 40.
|
|
* - includeEnv {Boolean} Default false. If true, a note stating the `env`
|
|
* envvar (if specified for this option) will be appended to the help
|
|
* output.
|
|
* - includeDefault {Boolean} Default false. If true, a note stating
|
|
* the `default` for this option, if any, will be appended to the help
|
|
* output.
|
|
* - helpWrap {Boolean} Default true. Wrap help text in helpCol..maxCol
|
|
* bounds.
|
|
* @returns {String}
|
|
*/
|
|
Parser.prototype.help = function help(config) {
|
|
config = config || {};
|
|
assert.object(config, 'config');
|
|
|
|
var indent = makeIndent(config.indent, 4, 'config.indent');
|
|
var headingIndent = makeIndent(config.headingIndent,
|
|
Math.round(indent.length / 2), 'config.headingIndent');
|
|
|
|
assert.optionalString(config.nameSort, 'config.nameSort');
|
|
var nameSort = config.nameSort || 'length';
|
|
assert.ok(~['length', 'none'].indexOf(nameSort),
|
|
'invalid "config.nameSort"');
|
|
assert.optionalNumber(config.maxCol, 'config.maxCol');
|
|
assert.optionalNumber(config.maxHelpCol, 'config.maxHelpCol');
|
|
assert.optionalNumber(config.minHelpCol, 'config.minHelpCol');
|
|
assert.optionalNumber(config.helpCol, 'config.helpCol');
|
|
assert.optionalBool(config.includeEnv, 'config.includeEnv');
|
|
assert.optionalBool(config.includeDefault, 'config.includeDefault');
|
|
assert.optionalBool(config.helpWrap, 'config.helpWrap');
|
|
var maxCol = config.maxCol || 80;
|
|
var minHelpCol = config.minHelpCol || 20;
|
|
var maxHelpCol = config.maxHelpCol || 40;
|
|
|
|
var lines = [];
|
|
var maxWidth = 0;
|
|
this.options.forEach(function (o) {
|
|
if (o.hidden) {
|
|
return;
|
|
}
|
|
if (o.group !== undefined && o.group !== null) {
|
|
// We deal with groups in the next pass
|
|
lines.push(null);
|
|
return;
|
|
}
|
|
var type = optionTypes[o.type];
|
|
var arg = o.helpArg || type.helpArg || 'ARG';
|
|
var line = '';
|
|
var names = o.names.slice();
|
|
if (nameSort === 'length') {
|
|
names.sort(function (a, b) {
|
|
if (a.length < b.length)
|
|
return -1;
|
|
else if (b.length < a.length)
|
|
return 1;
|
|
else
|
|
return 0;
|
|
})
|
|
}
|
|
names.forEach(function (name, i) {
|
|
if (i > 0)
|
|
line += ', ';
|
|
if (name.length === 1) {
|
|
line += '-' + name
|
|
if (type.takesArg)
|
|
line += ' ' + arg;
|
|
} else {
|
|
line += '--' + name
|
|
if (type.takesArg)
|
|
line += '=' + arg;
|
|
}
|
|
});
|
|
maxWidth = Math.max(maxWidth, line.length);
|
|
lines.push(line);
|
|
});
|
|
|
|
// Add help strings.
|
|
var helpCol = config.helpCol;
|
|
if (!helpCol) {
|
|
helpCol = maxWidth + indent.length + 2;
|
|
helpCol = Math.min(Math.max(helpCol, minHelpCol), maxHelpCol);
|
|
}
|
|
var i = -1;
|
|
this.options.forEach(function (o) {
|
|
if (o.hidden) {
|
|
return;
|
|
}
|
|
i++;
|
|
|
|
if (o.group !== undefined && o.group !== null) {
|
|
if (o.group === '') {
|
|
// Support a empty string "group" to have a blank line between
|
|
// sets of options.
|
|
lines[i] = '';
|
|
} else {
|
|
// Render the group heading with the heading-specific indent.
|
|
lines[i] = (i === 0 ? '' : '\n') + headingIndent +
|
|
o.group + ':';
|
|
}
|
|
return;
|
|
}
|
|
|
|
var helpDefault;
|
|
if (config.includeDefault) {
|
|
if (o.default !== undefined) {
|
|
helpDefault = format('Default: %j', o.default);
|
|
} else if (o.type && optionTypes[o.type].default !== undefined) {
|
|
helpDefault = format('Default: %j',
|
|
optionTypes[o.type].default);
|
|
}
|
|
}
|
|
|
|
var line = lines[i] = indent + lines[i];
|
|
if (!o.help && !(config.includeEnv && o.env) && !helpDefault) {
|
|
return;
|
|
}
|
|
var n = helpCol - line.length;
|
|
if (n >= 0) {
|
|
line += space(n);
|
|
} else {
|
|
line += '\n' + space(helpCol);
|
|
}
|
|
|
|
var helpEnv = '';
|
|
if (o.env && o.env.length && config.includeEnv) {
|
|
helpEnv += 'Environment: ';
|
|
var type = optionTypes[o.type];
|
|
var arg = o.helpArg || type.helpArg || 'ARG';
|
|
var envs = (Array.isArray(o.env) ? o.env : [o.env]).map(
|
|
function (e) {
|
|
if (type.takesArg) {
|
|
return e + '=' + arg;
|
|
} else {
|
|
return e + '=1';
|
|
}
|
|
}
|
|
);
|
|
helpEnv += envs.join(', ');
|
|
}
|
|
var help = (o.help || '').trim();
|
|
if (o.helpWrap !== false && config.helpWrap !== false) {
|
|
// Wrap help description normally.
|
|
if (help.length && !~'.!?"\''.indexOf(help.slice(-1))) {
|
|
help += '.';
|
|
}
|
|
if (help.length) {
|
|
help += ' ';
|
|
}
|
|
help += helpEnv;
|
|
if (helpDefault) {
|
|
if (helpEnv) {
|
|
help += '. ';
|
|
}
|
|
help += helpDefault;
|
|
}
|
|
line += textwrap(help, maxCol - helpCol).join(
|
|
'\n' + space(helpCol));
|
|
} else {
|
|
// Do not wrap help description, but indent newlines appropriately.
|
|
var helpLines = help.split('\n').filter(
|
|
function (ln) { return ln.length });
|
|
if (helpEnv !== '') {
|
|
helpLines.push(helpEnv);
|
|
}
|
|
if (helpDefault) {
|
|
helpLines.push(helpDefault);
|
|
}
|
|
line += helpLines.join('\n' + space(helpCol));
|
|
}
|
|
|
|
lines[i] = line;
|
|
});
|
|
|
|
var rv = '';
|
|
if (lines.length > 0) {
|
|
rv = lines.join('\n') + '\n';
|
|
}
|
|
return rv;
|
|
};
|
|
|
|
|
|
/**
|
|
* Return a string suitable for a Bash completion file for this tool.
|
|
*
|
|
* @param args.name {String} The tool name.
|
|
* @param args.specExtra {String} Optional. Extra Bash code content to add
|
|
* to the end of the "spec". Typically this is used to append Bash
|
|
* "complete_TYPE" functions for custom option types. See
|
|
* "examples/ddcompletion.js" for an example.
|
|
* @param args.argtypes {Array} Optional. Array of completion types for
|
|
* positional args (i.e. non-options). E.g.
|
|
* argtypes = ['fruit', 'veggie', 'file']
|
|
* will result in completion of fruits for the first arg, veggies for the
|
|
* second, and filenames for the third and subsequent positional args.
|
|
* If not given, positional args will use Bash's 'default' completion.
|
|
* See `specExtra` for providing Bash `complete_TYPE` functions, e.g.
|
|
* `complete_fruit` and `complete_veggie` in this example.
|
|
*/
|
|
Parser.prototype.bashCompletion = function bashCompletion(args) {
|
|
assert.object(args, 'args');
|
|
assert.string(args.name, 'args.name');
|
|
assert.optionalString(args.specExtra, 'args.specExtra');
|
|
assert.optionalArrayOfString(args.argtypes, 'args.argtypes');
|
|
|
|
return bashCompletionFromOptions({
|
|
name: args.name,
|
|
specExtra: args.specExtra,
|
|
argtypes: args.argtypes,
|
|
options: this.options
|
|
});
|
|
};
|
|
|
|
|
|
// ---- Bash completion
|
|
|
|
const BASH_COMPLETION_TEMPLATE_PATH = path.join(
|
|
__dirname, '../etc/dashdash.bash_completion.in');
|
|
|
|
/**
|
|
* Return the Bash completion "spec" (the string value for the "{{spec}}"
|
|
* var in the "dashdash.bash_completion.in" template) for this tool.
|
|
*
|
|
* The "spec" is Bash code that defines the CLI options and subcmds for
|
|
* the template's completion code. It looks something like this:
|
|
*
|
|
* local cmd_shortopts="-J ..."
|
|
* local cmd_longopts="--help ..."
|
|
* local cmd_optargs="-p=tritonprofile ..."
|
|
*
|
|
* @param args.options {Array} The array of dashdash option specs.
|
|
* @param args.context {String} Optional. A context string for the "local cmd*"
|
|
* vars in the spec. By default it is the empty string. When used to
|
|
* scope for completion on a *sub-command* (e.g. for "git log" on a "git"
|
|
* tool), then it would have a value (e.g. "__log"). See
|
|
* <http://github.com/trentm/node-cmdln> Bash completion for details.
|
|
* @param opts.includeHidden {Boolean} Optional. Default false. By default
|
|
* hidden options and subcmds are "excluded". Here excluded means they
|
|
* won't be offered as a completion, but if used, their argument type
|
|
* will be completed. "Hidden" options and subcmds are ones with the
|
|
* `hidden: true` attribute to exclude them from default help output.
|
|
* @param args.argtypes {Array} Optional. Array of completion types for
|
|
* positional args (i.e. non-options). E.g.
|
|
* argtypes = ['fruit', 'veggie', 'file']
|
|
* will result in completion of fruits for the first arg, veggies for the
|
|
* second, and filenames for the third and subsequent positional args.
|
|
* If not given, positional args will use Bash's 'default' completion.
|
|
* See `specExtra` for providing Bash `complete_TYPE` functions, e.g.
|
|
* `complete_fruit` and `complete_veggie` in this example.
|
|
*/
|
|
function bashCompletionSpecFromOptions(args) {
|
|
assert.object(args, 'args');
|
|
assert.object(args.options, 'args.options');
|
|
assert.optionalString(args.context, 'args.context');
|
|
assert.optionalBool(args.includeHidden, 'args.includeHidden');
|
|
assert.optionalArrayOfString(args.argtypes, 'args.argtypes');
|
|
|
|
var context = args.context || '';
|
|
var includeHidden = (args.includeHidden === undefined
|
|
? false : args.includeHidden);
|
|
|
|
var spec = [];
|
|
var shortopts = [];
|
|
var longopts = [];
|
|
var optargs = [];
|
|
(args.options || []).forEach(function (o) {
|
|
if (o.group !== undefined && o.group !== null) {
|
|
// Skip group headers.
|
|
return;
|
|
}
|
|
|
|
var optNames = o.names || [o.name];
|
|
var optType = getOptionType(o.type);
|
|
if (optType.takesArg) {
|
|
var completionType = o.completionType ||
|
|
optType.completionType || o.type;
|
|
optNames.forEach(function (optName) {
|
|
if (optName.length === 1) {
|
|
if (includeHidden || !o.hidden) {
|
|
shortopts.push('-' + optName);
|
|
}
|
|
// Include even hidden options in `optargs` so that bash
|
|
// completion of its arg still works.
|
|
optargs.push('-' + optName + '=' + completionType);
|
|
} else {
|
|
if (includeHidden || !o.hidden) {
|
|
longopts.push('--' + optName);
|
|
}
|
|
optargs.push('--' + optName + '=' + completionType);
|
|
}
|
|
});
|
|
} else {
|
|
optNames.forEach(function (optName) {
|
|
if (includeHidden || !o.hidden) {
|
|
if (optName.length === 1) {
|
|
shortopts.push('-' + optName);
|
|
} else {
|
|
longopts.push('--' + optName);
|
|
}
|
|
}
|
|
});
|
|
}
|
|
});
|
|
|
|
spec.push(format('local cmd%s_shortopts="%s"',
|
|
context, shortopts.sort().join(' ')));
|
|
spec.push(format('local cmd%s_longopts="%s"',
|
|
context, longopts.sort().join(' ')));
|
|
spec.push(format('local cmd%s_optargs="%s"',
|
|
context, optargs.sort().join(' ')));
|
|
if (args.argtypes) {
|
|
spec.push(format('local cmd%s_argtypes="%s"',
|
|
context, args.argtypes.join(' ')));
|
|
}
|
|
return spec.join('\n');
|
|
}
|
|
|
|
|
|
/**
|
|
* Return a string suitable for a Bash completion file for this tool.
|
|
*
|
|
* @param args.name {String} The tool name.
|
|
* @param args.options {Array} The array of dashdash option specs.
|
|
* @param args.specExtra {String} Optional. Extra Bash code content to add
|
|
* to the end of the "spec". Typically this is used to append Bash
|
|
* "complete_TYPE" functions for custom option types. See
|
|
* "examples/ddcompletion.js" for an example.
|
|
* @param args.argtypes {Array} Optional. Array of completion types for
|
|
* positional args (i.e. non-options). E.g.
|
|
* argtypes = ['fruit', 'veggie', 'file']
|
|
* will result in completion of fruits for the first arg, veggies for the
|
|
* second, and filenames for the third and subsequent positional args.
|
|
* If not given, positional args will use Bash's 'default' completion.
|
|
* See `specExtra` for providing Bash `complete_TYPE` functions, e.g.
|
|
* `complete_fruit` and `complete_veggie` in this example.
|
|
*/
|
|
function bashCompletionFromOptions(args) {
|
|
assert.object(args, 'args');
|
|
assert.object(args.options, 'args.options');
|
|
assert.string(args.name, 'args.name');
|
|
assert.optionalString(args.specExtra, 'args.specExtra');
|
|
assert.optionalArrayOfString(args.argtypes, 'args.argtypes');
|
|
|
|
// Gather template data.
|
|
var data = {
|
|
name: args.name,
|
|
date: new Date(),
|
|
spec: bashCompletionSpecFromOptions({
|
|
options: args.options,
|
|
argtypes: args.argtypes
|
|
}),
|
|
};
|
|
if (args.specExtra) {
|
|
data.spec += '\n\n' + args.specExtra;
|
|
}
|
|
|
|
// Render template.
|
|
var template = fs.readFileSync(BASH_COMPLETION_TEMPLATE_PATH, 'utf8');
|
|
return renderTemplate(template, data);
|
|
}
|
|
|
|
|
|
|
|
// ---- exports
|
|
|
|
function createParser(config) {
|
|
return new Parser(config);
|
|
}
|
|
|
|
/**
|
|
* Parse argv with the given options.
|
|
*
|
|
* @param config {Object} A merge of all the available fields from
|
|
* `dashdash.Parser` and `dashdash.Parser.parse`: options, interspersed,
|
|
* argv, env, slice.
|
|
*/
|
|
function parse(config) {
|
|
assert.object(config, 'config');
|
|
assert.optionalArrayOfString(config.argv, 'config.argv');
|
|
assert.optionalObject(config.env, 'config.env');
|
|
var config = shallowCopy(config);
|
|
var argv = config.argv;
|
|
delete config.argv;
|
|
var env = config.env;
|
|
delete config.env;
|
|
|
|
var parser = new Parser(config);
|
|
return parser.parse({argv: argv, env: env});
|
|
}
|
|
|
|
|
|
/**
|
|
* Add a new option type.
|
|
*
|
|
* @params optionType {Object}:
|
|
* - name {String} Required.
|
|
* - takesArg {Boolean} Required. Whether this type of option takes an
|
|
* argument on process.argv. Typically this is true for all but the
|
|
* "bool" type.
|
|
* - helpArg {String} Required iff `takesArg === true`. The string to
|
|
* show in generated help for options of this type.
|
|
* - parseArg {Function} Require. `function (option, optstr, arg)` parser
|
|
* that takes a string argument and returns an instance of the
|
|
* appropriate type, or throws an error if the arg is invalid.
|
|
* - array {Boolean} Optional. Set to true if this is an 'arrayOf' type
|
|
* that collects multiple usages of the option in process.argv and
|
|
* puts results in an array.
|
|
* - arrayFlatten {Boolean} Optional. XXX
|
|
* - default Optional. Default value for options of this type, if no
|
|
* default is specified in the option type usage.
|
|
*/
|
|
function addOptionType(optionType) {
|
|
assert.object(optionType, 'optionType');
|
|
assert.string(optionType.name, 'optionType.name');
|
|
assert.bool(optionType.takesArg, 'optionType.takesArg');
|
|
if (optionType.takesArg) {
|
|
assert.string(optionType.helpArg, 'optionType.helpArg');
|
|
}
|
|
assert.func(optionType.parseArg, 'optionType.parseArg');
|
|
assert.optionalBool(optionType.array, 'optionType.array');
|
|
assert.optionalBool(optionType.arrayFlatten, 'optionType.arrayFlatten');
|
|
|
|
optionTypes[optionType.name] = {
|
|
takesArg: optionType.takesArg,
|
|
helpArg: optionType.helpArg,
|
|
parseArg: optionType.parseArg,
|
|
array: optionType.array,
|
|
arrayFlatten: optionType.arrayFlatten,
|
|
default: optionType.default
|
|
}
|
|
}
|
|
|
|
|
|
function getOptionType(name) {
|
|
assert.string(name, 'name');
|
|
return optionTypes[name];
|
|
}
|
|
|
|
|
|
/**
|
|
* Return a synopsis string for the given option spec.
|
|
*
|
|
* Examples:
|
|
* > synopsisFromOpt({names: ['help', 'h'], type: 'bool'});
|
|
* '[ --help | -h ]'
|
|
* > synopsisFromOpt({name: 'file', type: 'string', helpArg: 'FILE'});
|
|
* '[ --file=FILE ]'
|
|
*/
|
|
function synopsisFromOpt(o) {
|
|
assert.object(o, 'o');
|
|
|
|
if (o.hasOwnProperty('group')) {
|
|
return null;
|
|
}
|
|
var names = o.names || [o.name];
|
|
// `type` here could be undefined if, for example, the command has a
|
|
// dashdash option spec with a bogus 'type'.
|
|
var type = getOptionType(o.type);
|
|
var helpArg = o.helpArg || (type && type.helpArg) || 'ARG';
|
|
var parts = [];
|
|
names.forEach(function (name) {
|
|
var part = (name.length === 1 ? '-' : '--') + name;
|
|
if (type && type.takesArg) {
|
|
part += (name.length === 1 ? ' ' + helpArg : '=' + helpArg);
|
|
}
|
|
parts.push(part);
|
|
});
|
|
return ('[ ' + parts.join(' | ') + ' ]');
|
|
};
|
|
|
|
|
|
module.exports = {
|
|
createParser: createParser,
|
|
Parser: Parser,
|
|
parse: parse,
|
|
addOptionType: addOptionType,
|
|
getOptionType: getOptionType,
|
|
synopsisFromOpt: synopsisFromOpt,
|
|
|
|
// Bash completion-related exports
|
|
BASH_COMPLETION_TEMPLATE_PATH: BASH_COMPLETION_TEMPLATE_PATH,
|
|
bashCompletionFromOptions: bashCompletionFromOptions,
|
|
bashCompletionSpecFromOptions: bashCompletionSpecFromOptions,
|
|
|
|
// Export the parseFoo parsers because they might be useful as primitives
|
|
// for custom option types.
|
|
parseBool: parseBool,
|
|
parseString: parseString,
|
|
parseNumber: parseNumber,
|
|
parseInteger: parseInteger,
|
|
parsePositiveInteger: parsePositiveInteger,
|
|
parseDate: parseDate
|
|
};
|