1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168 |
- /**
- * class ArgumentParser
- *
- * Object for parsing command line strings into js objects.
- *
- * Inherited from [[ActionContainer]]
- **/
- 'use strict';
-
- var util = require('util');
- var format = require('util').format;
- var Path = require('path');
-
- var _ = require('underscore');
- _.str = require('underscore.string');
-
- // Constants
- var $$ = require('./const');
-
- var ActionContainer = require('./action_container');
-
- // Errors
- var argumentErrorHelper = require('./argument/error');
-
- var HelpFormatter = require('./help/formatter');
-
- var Namespace = require('./namespace');
-
-
- /**
- * new ArgumentParser(options)
- *
- * Create a new ArgumentParser object.
- *
- * ##### Options:
- * - `prog` The name of the program (default: Path.basename(process.argv[1]))
- * - `usage` A usage message (default: auto-generated from arguments)
- * - `description` A description of what the program does
- * - `epilog` Text following the argument descriptions
- * - `parents` Parsers whose arguments should be copied into this one
- * - `formatterClass` HelpFormatter class for printing help messages
- * - `prefixChars` Characters that prefix optional arguments
- * - `fromfilePrefixChars` Characters that prefix files containing additional arguments
- * - `argumentDefault` The default value for all arguments
- * - `addHelp` Add a -h/-help option
- * - `conflictHandler` Specifies how to handle conflicting argument names
- * - `debug` Enable debug mode. Argument errors throw exception in
- * debug mode and process.exit in normal. Used for development and
- * testing (default: false)
- *
- * See also [original guide][1]
- *
- * [1]:http://docs.python.org/dev/library/argparse.html#argumentparser-objects
- **/
- var ArgumentParser = module.exports = function ArgumentParser(options) {
- var self = this;
- options = options || {};
-
- options.description = (options.description || null);
- options.argumentDefault = (options.argumentDefault || null);
- options.prefixChars = (options.prefixChars || '-');
- options.conflictHandler = (options.conflictHandler || 'error');
- ActionContainer.call(this, options);
-
- options.addHelp = (options.addHelp === undefined || !!options.addHelp);
- options.parents = (options.parents || []);
- // default program name
- options.prog = (options.prog || Path.basename(process.argv[1]));
- this.prog = options.prog;
- this.usage = options.usage;
- this.epilog = options.epilog;
- this.version = options.version;
-
- this.debug = (options.debug === true);
-
- this.formatterClass = (options.formatterClass || HelpFormatter);
- this.fromfilePrefixChars = options.fromfilePrefixChars || null;
- this._positionals = this.addArgumentGroup({title: 'Positional arguments'});
- this._optionals = this.addArgumentGroup({title: 'Optional arguments'});
- this._subparsers = null;
-
- // register types
- var FUNCTION_IDENTITY = function (o) {
- return o;
- };
- this.register('type', 'auto', FUNCTION_IDENTITY);
- this.register('type', null, FUNCTION_IDENTITY);
- this.register('type', 'int', function (x) {
- var result = parseInt(x, 10);
- if (isNaN(result)) {
- throw new Error(x + ' is not a valid integer.');
- }
- return result;
- });
- this.register('type', 'float', function (x) {
- var result = parseFloat(x);
- if (isNaN(result)) {
- throw new Error(x + ' is not a valid float.');
- }
- return result;
- });
- this.register('type', 'string', function (x) {
- return '' + x;
- });
-
- // add help and version arguments if necessary
- var defaultPrefix = (this.prefixChars.indexOf('-') > -1) ? '-' : this.prefixChars[0];
- if (options.addHelp) {
- this.addArgument(
- [defaultPrefix + 'h', defaultPrefix + defaultPrefix + 'help'],
- {
- action: 'help',
- defaultValue: $$.SUPPRESS,
- help: 'Show this help message and exit.'
- }
- );
- }
- if (this.version !== undefined) {
- this.addArgument(
- [defaultPrefix + 'v', defaultPrefix + defaultPrefix + 'version'],
- {
- action: 'version',
- version: this.version,
- defaultValue: $$.SUPPRESS,
- help: "Show program's version number and exit."
- }
- );
- }
-
- // add parent arguments and defaults
- options.parents.forEach(function (parent) {
- self._addContainerActions(parent);
- if (parent._defaults !== undefined) {
- for (var defaultKey in parent._defaults) {
- if (parent._defaults.hasOwnProperty(defaultKey)) {
- self._defaults[defaultKey] = parent._defaults[defaultKey];
- }
- }
- }
- });
-
- };
- util.inherits(ArgumentParser, ActionContainer);
-
- /**
- * ArgumentParser#addSubparsers(options) -> [[ActionSubparsers]]
- * - options (object): hash of options see [[ActionSubparsers.new]]
- *
- * See also [subcommands][1]
- *
- * [1]:http://docs.python.org/dev/library/argparse.html#sub-commands
- **/
- ArgumentParser.prototype.addSubparsers = function (options) {
- if (!!this._subparsers) {
- this.error('Cannot have multiple subparser arguments.');
- }
-
- options = options || {};
- options.debug = (this.debug === true);
- options.optionStrings = [];
- options.parserClass = (options.parserClass || ArgumentParser);
-
-
- if (!!options.title || !!options.description) {
-
- this._subparsers = this.addArgumentGroup({
- title: (options.title || 'subcommands'),
- description: options.description
- });
- delete options.title;
- delete options.description;
-
- } else {
- this._subparsers = this._positionals;
- }
-
- // prog defaults to the usage message of this parser, skipping
- // optional arguments and with no "usage:" prefix
- if (!options.prog) {
- var formatter = this._getFormatter();
- var positionals = this._getPositionalActions();
- var groups = this._mutuallyExclusiveGroups;
- formatter.addUsage(this.usage, positionals, groups, '');
- options.prog = _.str.strip(formatter.formatHelp());
- }
-
- // create the parsers action and add it to the positionals list
- var ParsersClass = this._popActionClass(options, 'parsers');
- var action = new ParsersClass(options);
- this._subparsers._addAction(action);
-
- // return the created parsers action
- return action;
- };
-
- ArgumentParser.prototype._addAction = function (action) {
- if (action.isOptional()) {
- this._optionals._addAction(action);
- } else {
- this._positionals._addAction(action);
- }
- return action;
- };
-
- ArgumentParser.prototype._getOptionalActions = function () {
- return this._actions.filter(function (action) {
- return action.isOptional();
- });
- };
-
- ArgumentParser.prototype._getPositionalActions = function () {
- return this._actions.filter(function (action) {
- return action.isPositional();
- });
- };
-
-
- /**
- * ArgumentParser#parseArgs(args, namespace) -> Namespace|Object
- * - args (array): input elements
- * - namespace (Namespace|Object): result object
- *
- * Parsed args and throws error if some arguments are not recognized
- *
- * See also [original guide][1]
- *
- * [1]:http://docs.python.org/dev/library/argparse.html#the-parse-args-method
- **/
- ArgumentParser.prototype.parseArgs = function (args, namespace) {
- var argv;
- var result = this.parseKnownArgs(args, namespace);
-
- args = result[0];
- argv = result[1];
- if (argv && argv.length > 0) {
- this.error(
- format('Unrecognized arguments: %s.', argv.join(' '))
- );
- }
- return args;
- };
-
- /**
- * ArgumentParser#parseKnownArgs(args, namespace) -> array
- * - args (array): input options
- * - namespace (Namespace|Object): result object
- *
- * Parse known arguments and return tuple of result object
- * and unknown args
- *
- * See also [original guide][1]
- *
- * [1]:http://docs.python.org/dev/library/argparse.html#partial-parsing
- **/
- ArgumentParser.prototype.parseKnownArgs = function (args, namespace) {
- var self = this;
-
- // args default to the system args
- args = args || process.argv.slice(2);
-
- // default Namespace built from parser defaults
- namespace = namespace || new Namespace();
-
- self._actions.forEach(function (action) {
- if (action.dest !== $$.SUPPRESS) {
- if (!_.has(namespace, action.dest)) {
- if (action.defaultValue !== $$.SUPPRESS) {
- var defaultValue = action.defaultValue;
- if (_.isString(action.defaultValue)) {
- defaultValue = self._getValue(action, defaultValue);
- }
- namespace[action.dest] = defaultValue;
- }
- }
- }
- });
-
- _.keys(self._defaults).forEach(function (dest) {
- namespace[dest] = self._defaults[dest];
- });
-
- // parse the arguments and exit if there are any errors
- try {
- var res = this._parseKnownArgs(args, namespace);
-
- namespace = res[0];
- args = res[1];
- if (_.has(namespace, $$._UNRECOGNIZED_ARGS_ATTR)) {
- args = _.union(args, namespace[$$._UNRECOGNIZED_ARGS_ATTR]);
- delete namespace[$$._UNRECOGNIZED_ARGS_ATTR];
- }
- return [namespace, args];
- } catch (e) {
- this.error(e);
- }
- };
-
- ArgumentParser.prototype._parseKnownArgs = function (argStrings, namespace) {
- var self = this;
-
- var extras = [];
-
- // replace arg strings that are file references
- if (this.fromfilePrefixChars !== null) {
- argStrings = this._readArgsFromFiles(argStrings);
- }
- // map all mutually exclusive arguments to the other arguments
- // they can't occur with
- // Python has 'conflicts = action_conflicts.setdefault(mutex_action, [])'
- // though I can't conceive of a way in which an action could be a member
- // of two different mutually exclusive groups.
-
- function actionHash(action) {
- // some sort of hashable key for this action
- // action itself cannot be a key in actionConflicts
- // I think getName() (join of optionStrings) is unique enough
- return action.getName();
- }
-
- var conflicts, key;
- var actionConflicts = {};
-
- this._mutuallyExclusiveGroups.forEach(function (mutexGroup) {
- mutexGroup._groupActions.forEach(function (mutexAction, i, groupActions) {
- key = actionHash(mutexAction);
- if (!_.has(actionConflicts, key)) {
- actionConflicts[key] = [];
- }
- conflicts = actionConflicts[key];
- conflicts.push.apply(conflicts, groupActions.slice(0, i));
- conflicts.push.apply(conflicts, groupActions.slice(i + 1));
- });
- });
-
- // find all option indices, and determine the arg_string_pattern
- // which has an 'O' if there is an option at an index,
- // an 'A' if there is an argument, or a '-' if there is a '--'
- var optionStringIndices = {};
-
- var argStringPatternParts = [];
-
- argStrings.forEach(function (argString, argStringIndex) {
- if (argString === '--') {
- argStringPatternParts.push('-');
- while (argStringIndex < argStrings.length) {
- argStringPatternParts.push('A');
- argStringIndex++;
- }
- }
- // otherwise, add the arg to the arg strings
- // and note the index if it was an option
- else {
- var pattern;
- var optionTuple = self._parseOptional(argString);
- if (!optionTuple) {
- pattern = 'A';
- }
- else {
- optionStringIndices[argStringIndex] = optionTuple;
- pattern = 'O';
- }
- argStringPatternParts.push(pattern);
- }
- });
- var argStringsPattern = argStringPatternParts.join('');
-
- var seenActions = [];
- var seenNonDefaultActions = [];
-
-
- function takeAction(action, argumentStrings, optionString) {
- seenActions.push(action);
- var argumentValues = self._getValues(action, argumentStrings);
-
- // error if this argument is not allowed with other previously
- // seen arguments, assuming that actions that use the default
- // value don't really count as "present"
- if (argumentValues !== action.defaultValue) {
- seenNonDefaultActions.push(action);
- if (!!actionConflicts[actionHash(action)]) {
- actionConflicts[actionHash(action)].forEach(function (actionConflict) {
- if (seenNonDefaultActions.indexOf(actionConflict) >= 0) {
- throw argumentErrorHelper(
- action,
- format('Not allowed with argument "%s".', actionConflict.getName())
- );
- }
- });
- }
- }
-
- if (argumentValues !== $$.SUPPRESS) {
- action.call(self, namespace, argumentValues, optionString);
- }
- }
-
- function consumeOptional(startIndex) {
- // get the optional identified at this index
- var optionTuple = optionStringIndices[startIndex];
- var action = optionTuple[0];
- var optionString = optionTuple[1];
- var explicitArg = optionTuple[2];
-
- // identify additional optionals in the same arg string
- // (e.g. -xyz is the same as -x -y -z if no args are required)
- var actionTuples = [];
-
- var args, argCount, start, stop;
-
- while (true) {
- if (!action) {
- extras.push(argStrings[startIndex]);
- return startIndex + 1;
- }
- if (!!explicitArg) {
- argCount = self._matchArgument(action, 'A');
-
- // if the action is a single-dash option and takes no
- // arguments, try to parse more single-dash options out
- // of the tail of the option string
- var chars = self.prefixChars;
- if (argCount === 0 && chars.indexOf(optionString[1]) < 0) {
- actionTuples.push([action, [], optionString]);
- optionString = optionString[0] + explicitArg[0];
- var newExplicitArg = explicitArg.slice(1) || null;
- var optionalsMap = self._optionStringActions;
-
- if (_.keys(optionalsMap).indexOf(optionString) >= 0) {
- action = optionalsMap[optionString];
- explicitArg = newExplicitArg;
- }
- else {
- var msg = 'ignored explicit argument %r';
- throw argumentErrorHelper(action, msg);
- }
- }
- // if the action expect exactly one argument, we've
- // successfully matched the option; exit the loop
- else if (argCount === 1) {
- stop = startIndex + 1;
- args = [explicitArg];
- actionTuples.push([action, args, optionString]);
- break;
- }
- // error if a double-dash option did not use the
- // explicit argument
- else {
- var message = 'ignored explicit argument %r';
- throw argumentErrorHelper(action, _.str.sprintf(message, explicitArg));
- }
- }
- // if there is no explicit argument, try to match the
- // optional's string arguments with the following strings
- // if successful, exit the loop
- else {
-
- start = startIndex + 1;
- var selectedPatterns = argStringsPattern.substr(start);
-
- argCount = self._matchArgument(action, selectedPatterns);
- stop = start + argCount;
-
-
- args = argStrings.slice(start, stop);
-
- actionTuples.push([action, args, optionString]);
- break;
- }
-
- }
-
- // add the Optional to the list and return the index at which
- // the Optional's string args stopped
- if (actionTuples.length < 1) {
- throw new Error('length should be > 0');
- }
- for (var i = 0; i < actionTuples.length; i++) {
- takeAction.apply(self, actionTuples[i]);
- }
- return stop;
- }
-
- // the list of Positionals left to be parsed; this is modified
- // by consume_positionals()
- var positionals = self._getPositionalActions();
-
- function consumePositionals(startIndex) {
- // match as many Positionals as possible
- var selectedPattern = argStringsPattern.substr(startIndex);
- var argCounts = self._matchArgumentsPartial(positionals, selectedPattern);
-
- // slice off the appropriate arg strings for each Positional
- // and add the Positional and its args to the list
- _.zip(positionals, argCounts).forEach(function (item) {
- var action = item[0];
- var argCount = item[1];
- if (argCount === undefined) {
- return;
- }
- var args = argStrings.slice(startIndex, startIndex + argCount);
-
- startIndex += argCount;
- takeAction(action, args);
- });
-
- // slice off the Positionals that we just parsed and return the
- // index at which the Positionals' string args stopped
- positionals = positionals.slice(argCounts.length);
- return startIndex;
- }
-
- // consume Positionals and Optionals alternately, until we have
- // passed the last option string
- var startIndex = 0;
- var position;
-
- var maxOptionStringIndex = -1;
-
- Object.keys(optionStringIndices).forEach(function (position) {
- maxOptionStringIndex = Math.max(maxOptionStringIndex, parseInt(position, 10));
- });
-
- var positionalsEndIndex, nextOptionStringIndex;
-
- while (startIndex <= maxOptionStringIndex) {
- // consume any Positionals preceding the next option
- nextOptionStringIndex = null;
- for (position in optionStringIndices) {
- if (!optionStringIndices.hasOwnProperty(position)) { continue; }
-
- position = parseInt(position, 10);
- if (position >= startIndex) {
- if (nextOptionStringIndex !== null) {
- nextOptionStringIndex = Math.min(nextOptionStringIndex, position);
- }
- else {
- nextOptionStringIndex = position;
- }
- }
- }
-
- if (startIndex !== nextOptionStringIndex) {
- positionalsEndIndex = consumePositionals(startIndex);
- // only try to parse the next optional if we didn't consume
- // the option string during the positionals parsing
- if (positionalsEndIndex > startIndex) {
- startIndex = positionalsEndIndex;
- continue;
- }
- else {
- startIndex = positionalsEndIndex;
- }
- }
-
- // if we consumed all the positionals we could and we're not
- // at the index of an option string, there were extra arguments
- if (!optionStringIndices[startIndex]) {
- var strings = argStrings.slice(startIndex, nextOptionStringIndex);
- extras = extras.concat(strings);
- startIndex = nextOptionStringIndex;
- }
- // consume the next optional and any arguments for it
- startIndex = consumeOptional(startIndex);
- }
-
- // consume any positionals following the last Optional
- var stopIndex = consumePositionals(startIndex);
-
- // if we didn't consume all the argument strings, there were extras
- extras = extras.concat(_.rest(argStrings, stopIndex));
-
- // if we didn't use all the Positional objects, there were too few
- // arg strings supplied.
- if (positionals.length > 0) {
- self.error('too few arguments');
- }
-
- // make sure all required actions were present
- self._actions.forEach(function (action) {
- if (action.required) {
- if (_.indexOf(seenActions, action) < 0) {
- self.error(format('Argument "%s" is required', action.getName()));
- }
- }
- });
-
- // make sure all required groups have one option present
- var actionUsed = false;
- self._mutuallyExclusiveGroups.forEach(function (group) {
- if (group.required) {
- actionUsed = _.any(group._groupActions, function (action) {
- return _.contains(seenNonDefaultActions, action);
- });
-
- // if no actions were used, report the error
- if (!actionUsed) {
- var names = [];
- group._groupActions.forEach(function (action) {
- if (action.help !== $$.SUPPRESS) {
- names.push(action.getName());
- }
- });
- names = names.join(' ');
- var msg = 'one of the arguments ' + names + ' is required';
- self.error(msg);
- }
- }
- });
-
- // return the updated namespace and the extra arguments
- return [namespace, extras];
- };
-
- ArgumentParser.prototype._readArgsFromFiles = function (argStrings) {
- // expand arguments referencing files
- var _this = this;
- var fs = require('fs');
- var newArgStrings = [];
- argStrings.forEach(function (argString) {
- if (_this.fromfilePrefixChars.indexOf(argString[0]) < 0) {
- // for regular arguments, just add them back into the list
- newArgStrings.push(argString);
- } else {
- // replace arguments referencing files with the file content
- try {
- var argstrs = [];
- var filename = argString.slice(1);
- var content = fs.readFileSync(filename, 'utf8');
- content = content.trim().split('\n');
- content.forEach(function (argLine) {
- _this.convertArgLineToArgs(argLine).forEach(function (arg) {
- argstrs.push(arg);
- });
- argstrs = _this._readArgsFromFiles(argstrs);
- });
- newArgStrings.push.apply(newArgStrings, argstrs);
- } catch (error) {
- return _this.error(error.message);
- }
- }
- });
- return newArgStrings;
- };
-
- ArgumentParser.prototype.convertArgLineToArgs = function (argLine) {
- return [argLine];
- };
-
- ArgumentParser.prototype._matchArgument = function (action, regexpArgStrings) {
-
- // match the pattern for this action to the arg strings
- var regexpNargs = new RegExp('^' + this._getNargsPattern(action));
- var matches = regexpArgStrings.match(regexpNargs);
- var message;
-
- // throw an exception if we weren't able to find a match
- if (!matches) {
- switch (action.nargs) {
- case undefined:
- case null:
- message = 'Expected one argument.';
- break;
- case $$.OPTIONAL:
- message = 'Expected at most one argument.';
- break;
- case $$.ONE_OR_MORE:
- message = 'Expected at least one argument.';
- break;
- default:
- message = 'Expected %s argument(s)';
- }
-
- throw argumentErrorHelper(
- action,
- format(message, action.nargs)
- );
- }
- // return the number of arguments matched
- return matches[1].length;
- };
-
- ArgumentParser.prototype._matchArgumentsPartial = function (actions, regexpArgStrings) {
- // progressively shorten the actions list by slicing off the
- // final actions until we find a match
- var self = this;
- var result = [];
- var actionSlice, pattern, matches;
- var i, j;
-
- var getLength = function (string) {
- return string.length;
- };
-
- for (i = actions.length; i > 0; i--) {
- pattern = '';
- actionSlice = actions.slice(0, i);
- for (j = 0; j < actionSlice.length; j++) {
- pattern += self._getNargsPattern(actionSlice[j]);
- }
-
- pattern = new RegExp('^' + pattern);
- matches = regexpArgStrings.match(pattern);
-
- if (matches && matches.length > 0) {
- // need only groups
- matches = matches.splice(1);
- result = result.concat(matches.map(getLength));
- break;
- }
- }
-
- // return the list of arg string counts
- return result;
- };
-
- ArgumentParser.prototype._parseOptional = function (argString) {
- var action, optionString, argExplicit, optionTuples;
-
- // if it's an empty string, it was meant to be a positional
- if (!argString) {
- return null;
- }
-
- // if it doesn't start with a prefix, it was meant to be positional
- if (this.prefixChars.indexOf(argString[0]) < 0) {
- return null;
- }
-
- // if the option string is present in the parser, return the action
- if (!!this._optionStringActions[argString]) {
- return [this._optionStringActions[argString], argString, null];
- }
-
- // if it's just a single character, it was meant to be positional
- if (argString.length === 1) {
- return null;
- }
-
- // if the option string before the "=" is present, return the action
- if (argString.indexOf('=') >= 0) {
- var argStringSplit = argString.split('=');
- optionString = argStringSplit[0];
- argExplicit = argStringSplit[1];
-
- if (!!this._optionStringActions[optionString]) {
- action = this._optionStringActions[optionString];
- return [action, optionString, argExplicit];
- }
- }
-
- // search through all possible prefixes of the option string
- // and all actions in the parser for possible interpretations
- optionTuples = this._getOptionTuples(argString);
-
- // if multiple actions match, the option string was ambiguous
- if (optionTuples.length > 1) {
- var optionStrings = optionTuples.map(function (optionTuple) {
- return optionTuple[1];
- });
- this.error(format(
- 'Ambiguous option: "%s" could match %s.',
- argString, optionStrings.join(', ')
- ));
- // if exactly one action matched, this segmentation is good,
- // so return the parsed action
- } else if (optionTuples.length === 1) {
- return optionTuples[0];
- }
-
- // if it was not found as an option, but it looks like a negative
- // number, it was meant to be positional
- // unless there are negative-number-like options
- if (argString.match(this._regexpNegativeNumber)) {
- if (!_.any(this._hasNegativeNumberOptionals)) {
- return null;
- }
- }
- // if it contains a space, it was meant to be a positional
- if (argString.search(' ') >= 0) {
- return null;
- }
-
- // it was meant to be an optional but there is no such option
- // in this parser (though it might be a valid option in a subparser)
- return [null, argString, null];
- };
-
- ArgumentParser.prototype._getOptionTuples = function (optionString) {
- var result = [];
- var chars = this.prefixChars;
- var optionPrefix;
- var argExplicit;
- var action;
- var actionOptionString;
-
- // option strings starting with two prefix characters are only split at
- // the '='
- if (chars.indexOf(optionString[0]) >= 0 && chars.indexOf(optionString[1]) >= 0) {
- if (optionString.indexOf('=') >= 0) {
- var optionStringSplit = optionString.split('=', 1);
-
- optionPrefix = optionStringSplit[0];
- argExplicit = optionStringSplit[1];
- } else {
- optionPrefix = optionString;
- argExplicit = null;
- }
-
- for (actionOptionString in this._optionStringActions) {
- if (actionOptionString.substr(0, optionPrefix.length) === optionPrefix) {
- action = this._optionStringActions[actionOptionString];
- result.push([action, actionOptionString, argExplicit]);
- }
- }
-
- // single character options can be concatenated with their arguments
- // but multiple character options always have to have their argument
- // separate
- } else if (chars.indexOf(optionString[0]) >= 0 && chars.indexOf(optionString[1]) < 0) {
- optionPrefix = optionString;
- argExplicit = null;
- var optionPrefixShort = optionString.substr(0, 2);
- var argExplicitShort = optionString.substr(2);
-
- for (actionOptionString in this._optionStringActions) {
- action = this._optionStringActions[actionOptionString];
- if (actionOptionString === optionPrefixShort) {
- result.push([action, actionOptionString, argExplicitShort]);
- } else if (actionOptionString.substr(0, optionPrefix.length) === optionPrefix) {
- result.push([action, actionOptionString, argExplicit]);
- }
- }
-
- // shouldn't ever get here
- } else {
- throw new Error(format('Unexpected option string: %s.', optionString));
- }
- // return the collected option tuples
- return result;
- };
-
- ArgumentParser.prototype._getNargsPattern = function (action) {
- // in all examples below, we have to allow for '--' args
- // which are represented as '-' in the pattern
- var regexpNargs;
-
- switch (action.nargs) {
- // the default (null) is assumed to be a single argument
- case undefined:
- case null:
- regexpNargs = '(-*A-*)';
- break;
- // allow zero or more arguments
- case $$.OPTIONAL:
- regexpNargs = '(-*A?-*)';
- break;
- // allow zero or more arguments
- case $$.ZERO_OR_MORE:
- regexpNargs = '(-*[A-]*)';
- break;
- // allow one or more arguments
- case $$.ONE_OR_MORE:
- regexpNargs = '(-*A[A-]*)';
- break;
- // allow any number of options or arguments
- case $$.REMAINDER:
- regexpNargs = '([-AO]*)';
- break;
- // allow one argument followed by any number of options or arguments
- case $$.PARSER:
- regexpNargs = '(-*A[-AO]*)';
- break;
- // all others should be integers
- default:
- regexpNargs = '(-*' + _.str.repeat('-*A', action.nargs) + '-*)';
- }
-
- // if this is an optional action, -- is not allowed
- if (action.isOptional()) {
- regexpNargs = regexpNargs.replace(/-\*/g, '');
- regexpNargs = regexpNargs.replace(/-/g, '');
- }
-
- // return the pattern
- return regexpNargs;
- };
-
- //
- // Value conversion methods
- //
-
- ArgumentParser.prototype._getValues = function (action, argStrings) {
- var self = this;
-
- // for everything but PARSER args, strip out '--'
- if (action.nargs !== $$.PARSER && action.nargs !== $$.REMAINDER) {
- argStrings = argStrings.filter(function (arrayElement) {
- return arrayElement !== '--';
- });
- }
-
- var value, argString;
-
- // optional argument produces a default when not present
- if (argStrings.length === 0 && action.nargs === $$.OPTIONAL) {
-
- value = (action.isOptional()) ? action.constant: action.defaultValue;
-
- if (typeof(value) === 'string') {
- value = this._getValue(action, value);
- this._checkValue(action, value);
- }
-
- // when nargs='*' on a positional, if there were no command-line
- // args, use the default if it is anything other than None
- } else if (argStrings.length === 0 && action.nargs === $$.ZERO_OR_MORE &&
- action.optionStrings.length === 0) {
-
- value = (action.defaultValue || argStrings);
- this._checkValue(action, value);
-
- // single argument or optional argument produces a single value
- } else if (argStrings.length === 1 &&
- (!action.nargs || action.nargs === $$.OPTIONAL)) {
-
- argString = argStrings[0];
- value = this._getValue(action, argString);
- this._checkValue(action, value);
-
- // REMAINDER arguments convert all values, checking none
- } else if (action.nargs === $$.REMAINDER) {
- value = argStrings.map(function (v) {
- return self._getValue(action, v);
- });
-
- // PARSER arguments convert all values, but check only the first
- } else if (action.nargs === $$.PARSER) {
- value = argStrings.map(function (v) {
- return self._getValue(action, v);
- });
- this._checkValue(action, value[0]);
-
- // all other types of nargs produce a list
- } else {
- value = argStrings.map(function (v) {
- return self._getValue(action, v);
- });
- value.forEach(function (v) {
- self._checkValue(action, v);
- });
- }
-
- // return the converted value
- return value;
- };
-
- ArgumentParser.prototype._getValue = function (action, argString) {
- var result;
-
- var typeFunction = this._registryGet('type', action.type, action.type);
- if (!_.isFunction(typeFunction)) {
- var message = format('%s is not callable', typeFunction);
- throw argumentErrorHelper(action, message);
- }
-
- // convert the value to the appropriate type
- try {
- result = typeFunction(argString);
-
- // ArgumentTypeErrors indicate errors
- // If action.type is not a registered string, it is a function
- // Try to deduce its name for inclusion in the error message
- // Failing that, include the error message it raised.
- } catch (e) {
- var name = null;
- if (_.isString(action.type)) {
- name = action.type;
- } else {
- name = action.type.name || action.type.displayName || '<function>';
- }
- var msg = format('Invalid %s value: %s', name, argString);
- if (name === '<function>') {msg += '\n' + e.message; }
- throw argumentErrorHelper(action, msg);
- }
- // return the converted value
- return result;
- };
-
- ArgumentParser.prototype._checkValue = function (action, value) {
- // converted value must be one of the choices (if specified)
- var choices = action.choices;
- if (!!choices) {
- // choise for argument can by array or string
- if ((_.isString(choices) || _.isArray(choices)) &&
- choices.indexOf(value) !== -1) {
- return;
- }
- // choise for subparsers can by only hash
- if (_.isObject(choices) && !_.isArray(choices) && choices[value]) {
- return;
- }
-
- if (_.isString(choices)) {
- choices = choices.split('').join(', ');
- }
- else if (_.isArray(choices)) {
- choices = choices.join(', ');
- }
- else {
- choices = _.keys(choices).join(', ');
- }
- var message = format('Invalid choice: %s (choose from [%s])', value, choices);
- throw argumentErrorHelper(action, message);
- }
- };
-
- //
- // Help formatting methods
- //
-
- /**
- * ArgumentParser#formatUsage -> string
- *
- * Return usage string
- *
- * See also [original guide][1]
- *
- * [1]:http://docs.python.org/dev/library/argparse.html#printing-help
- **/
- ArgumentParser.prototype.formatUsage = function () {
- var formatter = this._getFormatter();
- formatter.addUsage(this.usage, this._actions, this._mutuallyExclusiveGroups);
- return formatter.formatHelp();
- };
-
- /**
- * ArgumentParser#formatHelp -> string
- *
- * Return help
- *
- * See also [original guide][1]
- *
- * [1]:http://docs.python.org/dev/library/argparse.html#printing-help
- **/
- ArgumentParser.prototype.formatHelp = function () {
- var formatter = this._getFormatter();
-
- // usage
- formatter.addUsage(this.usage, this._actions, this._mutuallyExclusiveGroups);
-
- // description
- formatter.addText(this.description);
-
- // positionals, optionals and user-defined groups
- this._actionGroups.forEach(function (actionGroup) {
- formatter.startSection(actionGroup.title);
- formatter.addText(actionGroup.description);
- formatter.addArguments(actionGroup._groupActions);
- formatter.endSection();
- });
-
- // epilog
- formatter.addText(this.epilog);
-
- // determine help from format above
- return formatter.formatHelp();
- };
-
- ArgumentParser.prototype._getFormatter = function () {
- var FormatterClass = this.formatterClass;
- var formatter = new FormatterClass({prog: this.prog});
- return formatter;
- };
-
- //
- // Print functions
- //
-
- /**
- * ArgumentParser#printUsage() -> Void
- *
- * Print usage
- *
- * See also [original guide][1]
- *
- * [1]:http://docs.python.org/dev/library/argparse.html#printing-help
- **/
- ArgumentParser.prototype.printUsage = function () {
- this._printMessage(this.formatUsage());
- };
-
- /**
- * ArgumentParser#printHelp() -> Void
- *
- * Print help
- *
- * See also [original guide][1]
- *
- * [1]:http://docs.python.org/dev/library/argparse.html#printing-help
- **/
- ArgumentParser.prototype.printHelp = function () {
- this._printMessage(this.formatHelp());
- };
-
- ArgumentParser.prototype._printMessage = function (message, stream) {
- if (!stream) {
- stream = process.stdout;
- }
- if (message) {
- stream.write('' + message);
- }
- };
-
- //
- // Exit functions
- //
-
- /**
- * ArgumentParser#exit(status=0, message) -> Void
- * - status (int): exit status
- * - message (string): message
- *
- * Print message in stderr/stdout and exit program
- **/
- ArgumentParser.prototype.exit = function (status, message) {
- if (!!message) {
- if (status === 0) {
- this._printMessage(message);
- }
- else {
- this._printMessage(message, process.stderr);
- }
- }
-
- process.exit(status);
- };
-
- /**
- * ArgumentParser#error(message) -> Void
- * - err (Error|string): message
- *
- * Error method Prints a usage message incorporating the message to stderr and
- * exits. If you override this in a subclass,
- * it should not return -- it should
- * either exit or throw an exception.
- *
- **/
- ArgumentParser.prototype.error = function (err) {
- var message;
- if (err instanceof Error) {
- if (this.debug === true) {
- throw err;
- }
- message = err.message;
- }
- else {
- message = err;
- }
- var msg = format('%s: error: %s', this.prog, message) + $$.EOL;
-
- if (this.debug === true) {
- throw new Error(msg);
- }
-
- this.printUsage(process.stderr);
-
- return this.exit(2, msg);
- };
|