| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482 | /** internal * class ActionContainer * * Action container. Parent for [[ArgumentParser]] and [[ArgumentGroup]] **/'use strict';var format = require('util').format;// Constantsvar c = require('./const');var $$ = require('./utils');//Actionsvar ActionHelp = require('./action/help');var ActionAppend = require('./action/append');var ActionAppendConstant = require('./action/append/constant');var ActionCount = require('./action/count');var ActionStore = require('./action/store');var ActionStoreConstant = require('./action/store/constant');var ActionStoreTrue = require('./action/store/true');var ActionStoreFalse = require('./action/store/false');var ActionVersion = require('./action/version');var ActionSubparsers = require('./action/subparsers');// Errorsvar argumentErrorHelper = require('./argument/error');/** * new ActionContainer(options) * * Action container. Parent for [[ArgumentParser]] and [[ArgumentGroup]] * * ##### Options: * * - `description` -- A description of what the program does * - `prefixChars`  -- Characters that prefix optional arguments * - `argumentDefault`  -- The default value for all arguments * - `conflictHandler` -- The conflict handler to use for duplicate arguments **/var ActionContainer = module.exports = function ActionContainer(options) {  options = options || {};  this.description = options.description;  this.argumentDefault = options.argumentDefault;  this.prefixChars = options.prefixChars || '';  this.conflictHandler = options.conflictHandler;  // set up registries  this._registries = {};  // register actions  this.register('action', null, ActionStore);  this.register('action', 'store', ActionStore);  this.register('action', 'storeConst', ActionStoreConstant);  this.register('action', 'storeTrue', ActionStoreTrue);  this.register('action', 'storeFalse', ActionStoreFalse);  this.register('action', 'append', ActionAppend);  this.register('action', 'appendConst', ActionAppendConstant);  this.register('action', 'count', ActionCount);  this.register('action', 'help', ActionHelp);  this.register('action', 'version', ActionVersion);  this.register('action', 'parsers', ActionSubparsers);  // raise an exception if the conflict handler is invalid  this._getHandler();  // action storage  this._actions = [];  this._optionStringActions = {};  // groups  this._actionGroups = [];  this._mutuallyExclusiveGroups = [];  // defaults storage  this._defaults = {};  // determines whether an "option" looks like a negative number  // -1, -1.5 -5e+4  this._regexpNegativeNumber = new RegExp('^[-]?[0-9]*\\.?[0-9]+([eE][-+]?[0-9]+)?$');  // whether or not there are any optionals that look like negative  // numbers -- uses a list so it can be shared and edited  this._hasNegativeNumberOptionals = [];};// Groups must be required, then ActionContainer already definedvar ArgumentGroup = require('./argument/group');var MutuallyExclusiveGroup = require('./argument/exclusive');//// Registration methods///** * ActionContainer#register(registryName, value, object) -> Void * - registryName (String) : object type action|type * - value (string) : keyword * - object (Object|Function) : handler * *  Register handlers **/ActionContainer.prototype.register = function (registryName, value, object) {  this._registries[registryName] = this._registries[registryName] || {};  this._registries[registryName][value] = object;};ActionContainer.prototype._registryGet = function (registryName, value, defaultValue) {  if (arguments.length < 3) {    defaultValue = null;  }  return this._registries[registryName][value] || defaultValue;};//// Namespace default accessor methods///** * ActionContainer#setDefaults(options) -> Void * - options (object):hash of options see [[Action.new]] * * Set defaults **/ActionContainer.prototype.setDefaults = function (options) {  options = options || {};  for (var property in options) {    if ($$.has(options, property)) {      this._defaults[property] = options[property];    }  }  // if these defaults match any existing arguments, replace the previous  // default on the object with the new one  this._actions.forEach(function (action) {    if ($$.has(options, action.dest)) {      action.defaultValue = options[action.dest];    }  });};/** * ActionContainer#getDefault(dest) -> Mixed * - dest (string): action destination * * Return action default value **/ActionContainer.prototype.getDefault = function (dest) {  var result = $$.has(this._defaults, dest) ? this._defaults[dest] : null;  this._actions.forEach(function (action) {    if (action.dest === dest && $$.has(action, 'defaultValue')) {      result = action.defaultValue;    }  });  return result;};//// Adding argument actions///** * ActionContainer#addArgument(args, options) -> Object * - args (String|Array): argument key, or array of argument keys * - options (Object): action objects see [[Action.new]] * * #### Examples * - addArgument([ '-f', '--foo' ], { action: 'store', defaultValue: 1, ... }) * - addArgument([ 'bar' ], { action: 'store', nargs: 1, ... }) * - addArgument('--baz', { action: 'store', nargs: 1, ... }) **/ActionContainer.prototype.addArgument = function (args, options) {  args = args;  options = options || {};  if (typeof args === 'string') {    args = [ args ];  }  if (!Array.isArray(args)) {    throw new TypeError('addArgument first argument should be a string or an array');  }  if (typeof options !== 'object' || Array.isArray(options)) {    throw new TypeError('addArgument second argument should be a hash');  }  // if no positional args are supplied or only one is supplied and  // it doesn't look like an option string, parse a positional argument  if (!args || args.length === 1 && this.prefixChars.indexOf(args[0][0]) < 0) {    if (args && !!options.dest) {      throw new Error('dest supplied twice for positional argument');    }    options = this._getPositional(args, options);    // otherwise, we're adding an optional argument  } else {    options = this._getOptional(args, options);  }  // if no default was supplied, use the parser-level default  if (typeof options.defaultValue === 'undefined') {    var dest = options.dest;    if ($$.has(this._defaults, dest)) {      options.defaultValue = this._defaults[dest];    } else if (typeof this.argumentDefault !== 'undefined') {      options.defaultValue = this.argumentDefault;    }  }  // create the action object, and add it to the parser  var ActionClass = this._popActionClass(options);  if (typeof ActionClass !== 'function') {    throw new Error(format('Unknown action "%s".', ActionClass));  }  var action = new ActionClass(options);  // throw an error if the action type is not callable  var typeFunction = this._registryGet('type', action.type, action.type);  if (typeof typeFunction !== 'function') {    throw new Error(format('"%s" is not callable', typeFunction));  }  return this._addAction(action);};/** * ActionContainer#addArgumentGroup(options) -> ArgumentGroup * - options (Object): hash of options see [[ArgumentGroup.new]] * * Create new arguments groups **/ActionContainer.prototype.addArgumentGroup = function (options) {  var group = new ArgumentGroup(this, options);  this._actionGroups.push(group);  return group;};/** * ActionContainer#addMutuallyExclusiveGroup(options) -> ArgumentGroup * - options (Object): {required: false} * * Create new mutual exclusive groups **/ActionContainer.prototype.addMutuallyExclusiveGroup = function (options) {  var group = new MutuallyExclusiveGroup(this, options);  this._mutuallyExclusiveGroups.push(group);  return group;};ActionContainer.prototype._addAction = function (action) {  var self = this;  // resolve any conflicts  this._checkConflict(action);  // add to actions list  this._actions.push(action);  action.container = this;  // index the action by any option strings it has  action.optionStrings.forEach(function (optionString) {    self._optionStringActions[optionString] = action;  });  // set the flag if any option strings look like negative numbers  action.optionStrings.forEach(function (optionString) {    if (optionString.match(self._regexpNegativeNumber)) {      if (!self._hasNegativeNumberOptionals.some(Boolean)) {        self._hasNegativeNumberOptionals.push(true);      }    }  });  // return the created action  return action;};ActionContainer.prototype._removeAction = function (action) {  var actionIndex = this._actions.indexOf(action);  if (actionIndex >= 0) {    this._actions.splice(actionIndex, 1);  }};ActionContainer.prototype._addContainerActions = function (container) {  // collect groups by titles  var titleGroupMap = {};  this._actionGroups.forEach(function (group) {    if (titleGroupMap[group.title]) {      throw new Error(format('Cannot merge actions - two groups are named "%s".', group.title));    }    titleGroupMap[group.title] = group;  });  // map each action to its group  var groupMap = {};  function actionHash(action) {    // unique (hopefully?) string suitable as dictionary key    return action.getName();  }  container._actionGroups.forEach(function (group) {    // if a group with the title exists, use that, otherwise    // create a new group matching the container's group    if (!titleGroupMap[group.title]) {      titleGroupMap[group.title] = this.addArgumentGroup({        title: group.title,        description: group.description      });    }    // map the actions to their new group    group._groupActions.forEach(function (action) {      groupMap[actionHash(action)] = titleGroupMap[group.title];    });  }, this);  // add container's mutually exclusive groups  // NOTE: if add_mutually_exclusive_group ever gains title= and  // description= then this code will need to be expanded as above  var mutexGroup;  container._mutuallyExclusiveGroups.forEach(function (group) {    mutexGroup = this.addMutuallyExclusiveGroup({      required: group.required    });    // map the actions to their new mutex group    group._groupActions.forEach(function (action) {      groupMap[actionHash(action)] = mutexGroup;    });  }, this);  // forEach takes a 'this' argument  // add all actions to this container or their group  container._actions.forEach(function (action) {    var key = actionHash(action);    if (groupMap[key]) {      groupMap[key]._addAction(action);    } else {      this._addAction(action);    }  });};ActionContainer.prototype._getPositional = function (dest, options) {  if (Array.isArray(dest)) {    dest = dest[0];  }  // make sure required is not specified  if (options.required) {    throw new Error('"required" is an invalid argument for positionals.');  }  // mark positional arguments as required if at least one is  // always required  if (options.nargs !== c.OPTIONAL && options.nargs !== c.ZERO_OR_MORE) {    options.required = true;  }  if (options.nargs === c.ZERO_OR_MORE && typeof options.defaultValue === 'undefined') {    options.required = true;  }  // return the keyword arguments with no option strings  options.dest = dest;  options.optionStrings = [];  return options;};ActionContainer.prototype._getOptional = function (args, options) {  var prefixChars = this.prefixChars;  var optionStrings = [];  var optionStringsLong = [];  // determine short and long option strings  args.forEach(function (optionString) {    // error on strings that don't start with an appropriate prefix    if (prefixChars.indexOf(optionString[0]) < 0) {      throw new Error(format('Invalid option string "%s": must start with a "%s".',        optionString,        prefixChars      ));    }    // strings starting with two prefix characters are long options    optionStrings.push(optionString);    if (optionString.length > 1 && prefixChars.indexOf(optionString[1]) >= 0) {      optionStringsLong.push(optionString);    }  });  // infer dest, '--foo-bar' -> 'foo_bar' and '-x' -> 'x'  var dest = options.dest || null;  delete options.dest;  if (!dest) {    var optionStringDest = optionStringsLong.length ? optionStringsLong[0] : optionStrings[0];    dest = $$.trimChars(optionStringDest, this.prefixChars);    if (dest.length === 0) {      throw new Error(        format('dest= is required for options like "%s"', optionStrings.join(', '))      );    }    dest = dest.replace(/-/g, '_');  }  // return the updated keyword arguments  options.dest = dest;  options.optionStrings = optionStrings;  return options;};ActionContainer.prototype._popActionClass = function (options, defaultValue) {  defaultValue = defaultValue || null;  var action = (options.action || defaultValue);  delete options.action;  var actionClass = this._registryGet('action', action, action);  return actionClass;};ActionContainer.prototype._getHandler = function () {  var handlerString = this.conflictHandler;  var handlerFuncName = '_handleConflict' + $$.capitalize(handlerString);  var func = this[handlerFuncName];  if (typeof func === 'undefined') {    var msg = 'invalid conflict resolution value: ' + handlerString;    throw new Error(msg);  } else {    return func;  }};ActionContainer.prototype._checkConflict = function (action) {  var optionStringActions = this._optionStringActions;  var conflictOptionals = [];  // find all options that conflict with this option  // collect pairs, the string, and an existing action that it conflicts with  action.optionStrings.forEach(function (optionString) {    var conflOptional = optionStringActions[optionString];    if (typeof conflOptional !== 'undefined') {      conflictOptionals.push([ optionString, conflOptional ]);    }  });  if (conflictOptionals.length > 0) {    var conflictHandler = this._getHandler();    conflictHandler.call(this, action, conflictOptionals);  }};ActionContainer.prototype._handleConflictError = function (action, conflOptionals) {  var conflicts = conflOptionals.map(function (pair) { return pair[0]; });  conflicts = conflicts.join(', ');  throw argumentErrorHelper(    action,    format('Conflicting option string(s): %s', conflicts)  );};ActionContainer.prototype._handleConflictResolve = function (action, conflOptionals) {  // remove all conflicting options  var self = this;  conflOptionals.forEach(function (pair) {    var optionString = pair[0];    var conflictingAction = pair[1];    // remove the conflicting option string    var i = conflictingAction.optionStrings.indexOf(optionString);    if (i >= 0) {      conflictingAction.optionStrings.splice(i, 1);    }    delete self._optionStringActions[optionString];    // if the option now has no option string, remove it from the    // container holding it    if (conflictingAction.optionStrings.length === 0) {      conflictingAction.container._removeAction(conflictingAction);    }  });};
 |