| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103 | 
/** * Module dependencies. */var EventEmitter = require('events').EventEmitter;var spawn = require('child_process').spawn;var readlink = require('graceful-readlink').readlinkSync;var path = require('path');var dirname = path.dirname;var basename = path.basename;var fs = require('fs');/** * Expose the root command. */exports = module.exports = new Command();/** * Expose `Command`. */exports.Command = Command;/** * Expose `Option`. */exports.Option = Option;/** * Initialize a new `Option` with the given `flags` and `description`. * * @param {String} flags * @param {String} description * @api public */function Option(flags, description) {  this.flags = flags;  this.required = ~flags.indexOf('<');  this.optional = ~flags.indexOf('[');  this.bool = !~flags.indexOf('-no-');  flags = flags.split(/[ ,|]+/);  if (flags.length > 1 && !/^[[<]/.test(flags[1])) this.short = flags.shift();  this.long = flags.shift();  this.description = description || '';}/** * Return option name. * * @return {String} * @api private */Option.prototype.name = function() {  return this.long    .replace('--', '')    .replace('no-', '');};/** * Check if `arg` matches the short or long flag. * * @param {String} arg * @return {Boolean} * @api private */Option.prototype.is = function(arg) {  return arg == this.short || arg == this.long;};/** * Initialize a new `Command`. * * @param {String} name * @api public */function Command(name) {  this.commands = [];  this.options = [];  this._execs = [];  this._allowUnknownOption = false;  this._args = [];  this._name = name;}/** * Inherit from `EventEmitter.prototype`. */Command.prototype.__proto__ = EventEmitter.prototype;/** * Add command `name`. * * The `.action()` callback is invoked when the * command `name` is specified via __ARGV__, * and the remaining arguments are applied to the * function for access. * * When the `name` is "*" an un-matched command * will be passed as the first arg, followed by * the rest of __ARGV__ remaining. * * Examples: * *      program *        .version('0.0.1') *        .option('-C, --chdir <path>', 'change the working directory') *        .option('-c, --config <path>', 'set config path. defaults to ./deploy.conf') *        .option('-T, --no-tests', 'ignore test hook') * *      program *        .command('setup') *        .description('run remote setup commands') *        .action(function() { *          console.log('setup'); *        }); * *      program *        .command('exec <cmd>') *        .description('run the given remote command') *        .action(function(cmd) { *          console.log('exec "%s"', cmd); *        }); * *      program *        .command('teardown <dir> [otherDirs...]') *        .description('run teardown commands') *        .action(function(dir, otherDirs) { *          console.log('dir "%s"', dir); *          if (otherDirs) { *            otherDirs.forEach(function (oDir) { *              console.log('dir "%s"', oDir); *            }); *          } *        }); * *      program *        .command('*') *        .description('deploy the given env') *        .action(function(env) { *          console.log('deploying "%s"', env); *        }); * *      program.parse(process.argv);  * * @param {String} name * @param {String} [desc] for git-style sub-commands * @return {Command} the new command * @api public */Command.prototype.command = function(name, desc, opts) {  opts = opts || {};  var args = name.split(/ +/);  var cmd = new Command(args.shift());  if (desc) {    cmd.description(desc);    this.executables = true;    this._execs[cmd._name] = true;  }  cmd._noHelp = !!opts.noHelp;  this.commands.push(cmd);  cmd.parseExpectedArgs(args);  cmd.parent = this;  if (desc) return this;  return cmd;};/** * Define argument syntax for the top-level command. * * @api public */Command.prototype.arguments = function (desc) {  return this.parseExpectedArgs(desc.split(/ +/));}/** * Add an implicit `help [cmd]` subcommand * which invokes `--help` for the given command. * * @api private */Command.prototype.addImplicitHelpCommand = function() {  this.command('help [cmd]', 'display help for [cmd]');};/** * Parse expected `args`. * * For example `["[type]"]` becomes `[{ required: false, name: 'type' }]`. * * @param {Array} args * @return {Command} for chaining * @api public */Command.prototype.parseExpectedArgs = function(args) {  if (!args.length) return;  var self = this;  args.forEach(function(arg) {    var argDetails = {      required: false,      name: '',      variadic: false    };    switch (arg[0]) {      case '<':        argDetails.required = true;        argDetails.name = arg.slice(1, -1);        break;      case '[':        argDetails.name = arg.slice(1, -1);        break;    }    if (argDetails.name.length > 3 && argDetails.name.slice(-3) === '...') {      argDetails.variadic = true;      argDetails.name = argDetails.name.slice(0, -3);    }    if (argDetails.name) {      self._args.push(argDetails);    }  });  return this;};/** * Register callback `fn` for the command. * * Examples: * *      program *        .command('help') *        .description('display verbose help') *        .action(function() { *           // output help here *        }); * * @param {Function} fn * @return {Command} for chaining * @api public */Command.prototype.action = function(fn) {  var self = this;  var listener = function(args, unknown) {    // Parse any so-far unknown options    args = args || [];    unknown = unknown || [];    var parsed = self.parseOptions(unknown);    // Output help if necessary    outputHelpIfNecessary(self, parsed.unknown);    // If there are still any unknown options, then we simply    // die, unless someone asked for help, in which case we give it    // to them, and then we die.    if (parsed.unknown.length > 0) {      self.unknownOption(parsed.unknown[0]);    }    // Leftover arguments need to be pushed back. Fixes issue #56    if (parsed.args.length) args = parsed.args.concat(args);    self._args.forEach(function(arg, i) {      if (arg.required && null == args[i]) {        self.missingArgument(arg.name);      } else if (arg.variadic) {        if (i !== self._args.length - 1) {          self.variadicArgNotLast(arg.name);        }        args[i] = args.splice(i);      }    });    // Always append ourselves to the end of the arguments,    // to make sure we match the number of arguments the user    // expects    if (self._args.length) {      args[self._args.length] = self;    } else {      args.push(self);    }    fn.apply(self, args);  };  var parent = this.parent || this;  var name = parent === this ? '*' : this._name;  parent.on(name, listener);  if (this._alias) parent.on(this._alias, listener);  return this;};/** * Define option with `flags`, `description` and optional * coercion `fn`. * * The `flags` string should contain both the short and long flags, * separated by comma, a pipe or space. The following are all valid * all will output this way when `--help` is used. * *    "-p, --pepper" *    "-p|--pepper" *    "-p --pepper" * * Examples: * *     // simple boolean defaulting to false *     program.option('-p, --pepper', 'add pepper'); * *     --pepper *     program.pepper *     // => Boolean * *     // simple boolean defaulting to true *     program.option('-C, --no-cheese', 'remove cheese'); * *     program.cheese *     // => true * *     --no-cheese *     program.cheese *     // => false * *     // required argument *     program.option('-C, --chdir <path>', 'change the working directory'); * *     --chdir /tmp *     program.chdir *     // => "/tmp" * *     // optional argument *     program.option('-c, --cheese [type]', 'add cheese [marble]'); * * @param {String} flags * @param {String} description * @param {Function|Mixed} fn or default * @param {Mixed} defaultValue * @return {Command} for chaining * @api public */Command.prototype.option = function(flags, description, fn, defaultValue) {  var self = this    , option = new Option(flags, description)    , oname = option.name()    , name = camelcase(oname);  // default as 3rd arg  if (typeof fn != 'function') {    if (fn instanceof RegExp) {      var regex = fn;      fn = function(val, def) {        var m = regex.exec(val);        return m ? m[0] : def;      }    }    else {      defaultValue = fn;      fn = null;    }  }  // preassign default value only for --no-*, [optional], or <required>  if (false == option.bool || option.optional || option.required) {    // when --no-* we make sure default is true    if (false == option.bool) defaultValue = true;    // preassign only if we have a default    if (undefined !== defaultValue) self[name] = defaultValue;  }  // register the option  this.options.push(option);  // when it's passed assign the value  // and conditionally invoke the callback  this.on(oname, function(val) {    // coercion    if (null !== val && fn) val = fn(val, undefined === self[name]      ? defaultValue      : self[name]);    // unassigned or bool    if ('boolean' == typeof self[name] || 'undefined' == typeof self[name]) {      // if no value, bool true, and we have a default, then use it!      if (null == val) {        self[name] = option.bool          ? defaultValue || true          : false;      } else {        self[name] = val;      }    } else if (null !== val) {      // reassign      self[name] = val;    }  });  return this;};/** * Allow unknown options on the command line. * * @param {Boolean} arg if `true` or omitted, no error will be thrown * for unknown options. * @api public */Command.prototype.allowUnknownOption = function(arg) {    this._allowUnknownOption = arguments.length === 0 || arg;    return this;};/** * Parse `argv`, settings options and invoking commands when defined. * * @param {Array} argv * @return {Command} for chaining * @api public */Command.prototype.parse = function(argv) {  // implicit help  if (this.executables) this.addImplicitHelpCommand();  // store raw args  this.rawArgs = argv;  // guess name  this._name = this._name || basename(argv[1], '.js');  // github-style sub-commands with no sub-command  if (this.executables && argv.length < 3) {    // this user needs help    argv.push('--help');  }  // process argv  var parsed = this.parseOptions(this.normalize(argv.slice(2)));  var args = this.args = parsed.args;  var result = this.parseArgs(this.args, parsed.unknown);  // executable sub-commands  var name = result.args[0];  if (this._execs[name] && typeof this._execs[name] != "function") {    return this.executeSubCommand(argv, args, parsed.unknown);  }  return result;};/** * Execute a sub-command executable. * * @param {Array} argv * @param {Array} args * @param {Array} unknown * @api private */Command.prototype.executeSubCommand = function(argv, args, unknown) {  args = args.concat(unknown);  if (!args.length) this.help();  if ('help' == args[0] && 1 == args.length) this.help();  // <cmd> --help  if ('help' == args[0]) {    args[0] = args[1];    args[1] = '--help';  }  // executable  var f = argv[1];  // name of the subcommand, link `pm-install`  var bin = basename(f, '.js') + '-' + args[0];  // In case of globally installed, get the base dir where executable  //  subcommand file should be located at  var baseDir    , link = readlink(f);  // when symbolink is relative path  if (link !== f && link.charAt(0) !== '/') {    link = path.join(dirname(f), link)  }  baseDir = dirname(link);  // prefer local `./<bin>` to bin in the $PATH  var localBin = path.join(baseDir, bin);  // whether bin file is a js script with explicit `.js` extension  var isExplicitJS = false;  if (exists(localBin + '.js')) {    bin = localBin + '.js';    isExplicitJS = true;  } else if (exists(localBin)) {    bin = localBin;  }  args = args.slice(1);  var proc;  if (process.platform !== 'win32') {    if (isExplicitJS) {      args.unshift(localBin);      // add executable arguments to spawn      args = (process.execArgv || []).concat(args);      proc = spawn('node', args, { stdio: 'inherit', customFds: [0, 1, 2] });    } else {      proc = spawn(bin, args, { stdio: 'inherit', customFds: [0, 1, 2] });    }  } else {    args.unshift(localBin);    proc = spawn(process.execPath, args, { stdio: 'inherit'});  }  proc.on('close', process.exit.bind(process));  proc.on('error', function(err) {    if (err.code == "ENOENT") {      console.error('\n  %s(1) does not exist, try --help\n', bin);    } else if (err.code == "EACCES") {      console.error('\n  %s(1) not executable. try chmod or run with root\n', bin);    }    process.exit(1);  });  this.runningCommand = proc;};/** * Normalize `args`, splitting joined short flags. For example * the arg "-abc" is equivalent to "-a -b -c". * This also normalizes equal sign and splits "--abc=def" into "--abc def". * * @param {Array} args * @return {Array} * @api private */Command.prototype.normalize = function(args) {  var ret = []    , arg    , lastOpt    , index;  for (var i = 0, len = args.length; i < len; ++i) {    arg = args[i];    if (i > 0) {      lastOpt = this.optionFor(args[i-1]);    }    if (arg === '--') {      // Honor option terminator      ret = ret.concat(args.slice(i));      break;    } else if (lastOpt && lastOpt.required) {      ret.push(arg);    } else if (arg.length > 1 && '-' == arg[0] && '-' != arg[1]) {      arg.slice(1).split('').forEach(function(c) {        ret.push('-' + c);      });    } else if (/^--/.test(arg) && ~(index = arg.indexOf('='))) {      ret.push(arg.slice(0, index), arg.slice(index + 1));    } else {      ret.push(arg);    }  }  return ret;};/** * Parse command `args`. * * When listener(s) are available those * callbacks are invoked, otherwise the "*" * event is emitted and those actions are invoked. * * @param {Array} args * @return {Command} for chaining * @api private */Command.prototype.parseArgs = function(args, unknown) {  var name;  if (args.length) {    name = args[0];    if (this.listeners(name).length) {      this.emit(args.shift(), args, unknown);    } else {      this.emit('*', args);    }  } else {    outputHelpIfNecessary(this, unknown);    // If there were no args and we have unknown options,    // then they are extraneous and we need to error.    if (unknown.length > 0) {      this.unknownOption(unknown[0]);    }  }  return this;};/** * Return an option matching `arg` if any. * * @param {String} arg * @return {Option} * @api private */Command.prototype.optionFor = function(arg) {  for (var i = 0, len = this.options.length; i < len; ++i) {    if (this.options[i].is(arg)) {      return this.options[i];    }  }};/** * Parse options from `argv` returning `argv` * void of these options. * * @param {Array} argv * @return {Array} * @api public */Command.prototype.parseOptions = function(argv) {  var args = []    , len = argv.length    , literal    , option    , arg;  var unknownOptions = [];  // parse options  for (var i = 0; i < len; ++i) {    arg = argv[i];    // literal args after --    if ('--' == arg) {      literal = true;      continue;    }    if (literal) {      args.push(arg);      continue;    }    // find matching Option    option = this.optionFor(arg);    // option is defined    if (option) {      // requires arg      if (option.required) {        arg = argv[++i];        if (null == arg) return this.optionMissingArgument(option);        this.emit(option.name(), arg);      // optional arg      } else if (option.optional) {        arg = argv[i+1];        if (null == arg || ('-' == arg[0] && '-' != arg)) {          arg = null;        } else {          ++i;        }        this.emit(option.name(), arg);      // bool      } else {        this.emit(option.name());      }      continue;    }    // looks like an option    if (arg.length > 1 && '-' == arg[0]) {      unknownOptions.push(arg);      // If the next argument looks like it might be      // an argument for this option, we pass it on.      // If it isn't, then it'll simply be ignored      if (argv[i+1] && '-' != argv[i+1][0]) {        unknownOptions.push(argv[++i]);      }      continue;    }    // arg    args.push(arg);  }  return { args: args, unknown: unknownOptions };};/** * Return an object containing options as key-value pairs * * @return {Object} * @api public */Command.prototype.opts = function() {  var result = {}    , len = this.options.length;  for (var i = 0 ; i < len; i++) {    var key = camelcase(this.options[i].name());    result[key] = key === 'version' ? this._version : this[key];  }  return result;};/** * Argument `name` is missing. * * @param {String} name * @api private */Command.prototype.missingArgument = function(name) {  console.error();  console.error("  error: missing required argument `%s'", name);  console.error();  process.exit(1);};/** * `Option` is missing an argument, but received `flag` or nothing. * * @param {String} option * @param {String} flag * @api private */Command.prototype.optionMissingArgument = function(option, flag) {  console.error();  if (flag) {    console.error("  error: option `%s' argument missing, got `%s'", option.flags, flag);  } else {    console.error("  error: option `%s' argument missing", option.flags);  }  console.error();  process.exit(1);};/** * Unknown option `flag`. * * @param {String} flag * @api private */Command.prototype.unknownOption = function(flag) {  if (this._allowUnknownOption) return;  console.error();  console.error("  error: unknown option `%s'", flag);  console.error();  process.exit(1);};/** * Variadic argument with `name` is not the last argument as required. * * @param {String} name * @api private */Command.prototype.variadicArgNotLast = function(name) {  console.error();  console.error("  error: variadic arguments must be last `%s'", name);  console.error();  process.exit(1);};/** * Set the program version to `str`. * * This method auto-registers the "-V, --version" flag * which will print the version number when passed. * * @param {String} str * @param {String} flags * @return {Command} for chaining * @api public */Command.prototype.version = function(str, flags) {  if (0 == arguments.length) return this._version;  this._version = str;  flags = flags || '-V, --version';  this.option(flags, 'output the version number');  this.on('version', function() {    process.stdout.write(str + '\n');    process.exit(0);  });  return this;};/** * Set the description to `str`. * * @param {String} str * @return {String|Command} * @api public */Command.prototype.description = function(str) {  if (0 == arguments.length) return this._description;  this._description = str;  return this;};/** * Set an alias for the command * * @param {String} alias * @return {String|Command} * @api public */Command.prototype.alias = function(alias) {  if (0 == arguments.length) return this._alias;  this._alias = alias;  return this;};/** * Set / get the command usage `str`. * * @param {String} str * @return {String|Command} * @api public */Command.prototype.usage = function(str) {  var args = this._args.map(function(arg) {    return humanReadableArgName(arg);  });  var usage = '[options]'    + (this.commands.length ? ' [command]' : '')    + (this._args.length ? ' ' + args.join(' ') : '');  if (0 == arguments.length) return this._usage || usage;  this._usage = str;  return this;};/** * Get the name of the command * * @param {String} name * @return {String|Command} * @api public */Command.prototype.name = function() {  return this._name;};/** * Return the largest option length. * * @return {Number} * @api private */Command.prototype.largestOptionLength = function() {  return this.options.reduce(function(max, option) {    return Math.max(max, option.flags.length);  }, 0);};/** * Return help for options. * * @return {String} * @api private */Command.prototype.optionHelp = function() {  var width = this.largestOptionLength();  // Prepend the help information  return [pad('-h, --help', width) + '  ' + 'output usage information']    .concat(this.options.map(function(option) {      return pad(option.flags, width) + '  ' + option.description;      }))    .join('\n');};/** * Return command help documentation. * * @return {String} * @api private */Command.prototype.commandHelp = function() {  if (!this.commands.length) return '';  var commands = this.commands.filter(function(cmd) {    return !cmd._noHelp;  }).map(function(cmd) {    var args = cmd._args.map(function(arg) {      return humanReadableArgName(arg);    }).join(' ');    return [      cmd._name        + (cmd._alias          ? '|' + cmd._alias          : '')        + (cmd.options.length          ? ' [options]'          : '')        + ' ' + args    , cmd.description()    ];  });  var width = commands.reduce(function(max, command) {    return Math.max(max, command[0].length);  }, 0);  return [      ''    , '  Commands:'    , ''    , commands.map(function(cmd) {      return pad(cmd[0], width) + '  ' + cmd[1];    }).join('\n').replace(/^/gm, '    ')    , ''  ].join('\n');};/** * Return program help documentation. * * @return {String} * @api private */Command.prototype.helpInformation = function() {  var desc = [];  if (this._description) {    desc = [      '  ' + this._description      , ''    ];  }  var cmdName = this._name;  if (this._alias) {    cmdName = cmdName + '|' + this._alias;  }  var usage = [    ''    ,'  Usage: ' + cmdName + ' ' + this.usage()    , ''  ];  var cmds = [];  var commandHelp = this.commandHelp();  if (commandHelp) cmds = [commandHelp];  var options = [    '  Options:'    , ''    , '' + this.optionHelp().replace(/^/gm, '    ')    , ''    , ''  ];  return usage    .concat(cmds)    .concat(desc)    .concat(options)    .join('\n');};/** * Output help information for this command * * @api public */Command.prototype.outputHelp = function() {  process.stdout.write(this.helpInformation());  this.emit('--help');};/** * Output help information and exit. * * @api public */Command.prototype.help = function() {  this.outputHelp();  process.exit();};/** * Camel-case the given `flag` * * @param {String} flag * @return {String} * @api private */function camelcase(flag) {  return flag.split('-').reduce(function(str, word) {    return str + word[0].toUpperCase() + word.slice(1);  });}/** * Pad `str` to `width`. * * @param {String} str * @param {Number} width * @return {String} * @api private */function pad(str, width) {  var len = Math.max(0, width - str.length);  return str + Array(len + 1).join(' ');}/** * Output help information if necessary * * @param {Command} command to output help for * @param {Array} array of options to search for -h or --help * @api private */function outputHelpIfNecessary(cmd, options) {  options = options || [];  for (var i = 0; i < options.length; i++) {    if (options[i] == '--help' || options[i] == '-h') {      cmd.outputHelp();      process.exit(0);    }  }}/** * Takes an argument an returns its human readable equivalent for help usage. * * @param {Object} arg * @return {String} * @api private */function humanReadableArgName(arg) {  var nameOutput = arg.name + (arg.variadic === true ? '...' : '');  return arg.required    ? '<' + nameOutput + '>'    : '[' + nameOutput + ']'}// for versions before node v0.8 when there weren't `fs.existsSync`function exists(file) {  try {    if (fs.statSync(file).isFile()) {      return true;    }  } catch (e) {    return false;  }}
 |