1225 lines
		
	
	
		
			27 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
		
		
			
		
	
	
			1225 lines
		
	
	
		
			27 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
|  | /** | ||
|  |  * Module dependencies. | ||
|  |  */ | ||
|  | 
 | ||
|  | var EventEmitter = require('events').EventEmitter; | ||
|  | var spawn = require('child_process').spawn; | ||
|  | var path = require('path'); | ||
|  | var dirname = path.dirname; | ||
|  | var basename = path.basename; | ||
|  | var fs = require('fs'); | ||
|  | 
 | ||
|  | /** | ||
|  |  * Inherit `Command` from `EventEmitter.prototype`. | ||
|  |  */ | ||
|  | 
 | ||
|  | require('util').inherits(Command, EventEmitter); | ||
|  | 
 | ||
|  | /** | ||
|  |  * 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('<') >= 0; | ||
|  |   this.optional = flags.indexOf('[') >= 0; | ||
|  |   this.bool = flags.indexOf('-no-') === -1; | ||
|  |   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-', ''); | ||
|  | }; | ||
|  | 
 | ||
|  | /** | ||
|  |  * Return option name, in a camelcase format that can be used | ||
|  |  * as a object attribute key. | ||
|  |  * | ||
|  |  * @return {String} | ||
|  |  * @api private | ||
|  |  */ | ||
|  | 
 | ||
|  | Option.prototype.attributeName = function() { | ||
|  |   return camelcase(this.name()); | ||
|  | }; | ||
|  | 
 | ||
|  | /** | ||
|  |  * Check if `arg` matches the short or long flag. | ||
|  |  * | ||
|  |  * @param {String} arg | ||
|  |  * @return {Boolean} | ||
|  |  * @api private | ||
|  |  */ | ||
|  | 
 | ||
|  | Option.prototype.is = function(arg) { | ||
|  |   return this.short === arg || this.long === arg; | ||
|  | }; | ||
|  | 
 | ||
|  | /** | ||
|  |  * 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 || ''; | ||
|  | } | ||
|  | 
 | ||
|  | /** | ||
|  |  * 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) { | ||
|  |   if (typeof desc === 'object' && desc !== null) { | ||
|  |     opts = desc; | ||
|  |     desc = null; | ||
|  |   } | ||
|  |   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; | ||
|  |     if (opts.isDefault) this.defaultExecutable = cmd._name; | ||
|  |   } | ||
|  |   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 && args[i] == null) { | ||
|  |         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('command:' + name, listener); | ||
|  |   if (this._alias) parent.on('command:' + 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|*} [fn] or default | ||
|  |  * @param {*} [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 = option.attributeName(); | ||
|  | 
 | ||
|  |   // 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 (!option.bool || option.optional || option.required) { | ||
|  |     // when --no-* we make sure default is true
 | ||
|  |     if (!option.bool) defaultValue = true; | ||
|  |     // preassign only if we have a default
 | ||
|  |     if (defaultValue !== undefined) { | ||
|  |       self[name] = defaultValue; | ||
|  |       option.defaultValue = defaultValue; | ||
|  |     } | ||
|  |   } | ||
|  | 
 | ||
|  |   // register the option
 | ||
|  |   this.options.push(option); | ||
|  | 
 | ||
|  |   // when it's passed assign the value
 | ||
|  |   // and conditionally invoke the callback
 | ||
|  |   this.on('option:' + oname, function(val) { | ||
|  |     // coercion
 | ||
|  |     if (val !== null && fn) { | ||
|  |       val = fn(val, self[name] === undefined ? defaultValue : self[name]); | ||
|  |     } | ||
|  | 
 | ||
|  |     // unassigned or bool
 | ||
|  |     if (typeof self[name] === 'boolean' || typeof self[name] === 'undefined') { | ||
|  |       // if no value, bool true, and we have a default, then use it!
 | ||
|  |       if (val == null) { | ||
|  |         self[name] = option.bool | ||
|  |           ? defaultValue || true | ||
|  |           : false; | ||
|  |       } else { | ||
|  |         self[name] = val; | ||
|  |       } | ||
|  |     } else if (val !== null) { | ||
|  |       // 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.defaultExecutable) { | ||
|  |     // 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]; | ||
|  | 
 | ||
|  |   var aliasCommand = null; | ||
|  |   // check alias of sub commands
 | ||
|  |   if (name) { | ||
|  |     aliasCommand = this.commands.filter(function(command) { | ||
|  |       return command.alias() === name; | ||
|  |     })[0]; | ||
|  |   } | ||
|  | 
 | ||
|  |   if (this._execs[name] && typeof this._execs[name] !== 'function') { | ||
|  |     return this.executeSubCommand(argv, args, parsed.unknown); | ||
|  |   } else if (aliasCommand) { | ||
|  |     // is alias of a subCommand
 | ||
|  |     args[0] = aliasCommand._name; | ||
|  |     return this.executeSubCommand(argv, args, parsed.unknown); | ||
|  |   } else if (this.defaultExecutable) { | ||
|  |     // use the default subcommand
 | ||
|  |     args.unshift(this.defaultExecutable); | ||
|  |     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 (args[0] === 'help' && args.length === 1) this.help(); | ||
|  | 
 | ||
|  |   // <cmd> --help
 | ||
|  |   if (args[0] === 'help') { | ||
|  |     args[0] = args[1]; | ||
|  |     args[1] = '--help'; | ||
|  |   } | ||
|  | 
 | ||
|  |   // executable
 | ||
|  |   var f = argv[1]; | ||
|  |   // name of the subcommand, link `pm-install`
 | ||
|  |   var bin = basename(f, path.extname(f)) + '-' + args[0]; | ||
|  | 
 | ||
|  |   // In case of globally installed, get the base dir where executable
 | ||
|  |   //  subcommand file should be located at
 | ||
|  |   var baseDir; | ||
|  | 
 | ||
|  |   var resolvedLink = fs.realpathSync(f); | ||
|  | 
 | ||
|  |   baseDir = dirname(resolvedLink); | ||
|  | 
 | ||
|  |   // prefer local `./<bin>` to bin in the $PATH
 | ||
|  |   var localBin = path.join(baseDir, bin); | ||
|  | 
 | ||
|  |   // whether bin file is a js script with explicit `.js` or `.ts` extension
 | ||
|  |   var isExplicitJS = false; | ||
|  |   if (exists(localBin + '.js')) { | ||
|  |     bin = localBin + '.js'; | ||
|  |     isExplicitJS = true; | ||
|  |   } else if (exists(localBin + '.ts')) { | ||
|  |     bin = localBin + '.ts'; | ||
|  |     isExplicitJS = true; | ||
|  |   } else if (exists(localBin)) { | ||
|  |     bin = localBin; | ||
|  |   } | ||
|  | 
 | ||
|  |   args = args.slice(1); | ||
|  | 
 | ||
|  |   var proc; | ||
|  |   if (process.platform !== 'win32') { | ||
|  |     if (isExplicitJS) { | ||
|  |       args.unshift(bin); | ||
|  |       // add executable arguments to spawn
 | ||
|  |       args = (process.execArgv || []).concat(args); | ||
|  | 
 | ||
|  |       proc = spawn(process.argv[0], args, { stdio: 'inherit', customFds: [0, 1, 2] }); | ||
|  |     } else { | ||
|  |       proc = spawn(bin, args, { stdio: 'inherit', customFds: [0, 1, 2] }); | ||
|  |     } | ||
|  |   } else { | ||
|  |     args.unshift(bin); | ||
|  |     proc = spawn(process.execPath, args, { stdio: 'inherit' }); | ||
|  |   } | ||
|  | 
 | ||
|  |   var signals = ['SIGUSR1', 'SIGUSR2', 'SIGTERM', 'SIGINT', 'SIGHUP']; | ||
|  |   signals.forEach(function(signal) { | ||
|  |     process.on(signal, function() { | ||
|  |       if (proc.killed === false && proc.exitCode === null) { | ||
|  |         proc.kill(signal); | ||
|  |       } | ||
|  |     }); | ||
|  |   }); | ||
|  |   proc.on('close', process.exit.bind(process)); | ||
|  |   proc.on('error', function(err) { | ||
|  |     if (err.code === 'ENOENT') { | ||
|  |       console.error('error: %s(1) does not exist, try --help', bin); | ||
|  |     } else if (err.code === 'EACCES') { | ||
|  |       console.error('error: %s(1) not executable. try chmod or run with root', bin); | ||
|  |     } | ||
|  |     process.exit(1); | ||
|  |   }); | ||
|  | 
 | ||
|  |   // Store the reference to the child process
 | ||
|  |   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('command:' + name).length) { | ||
|  |       this.emit('command:' + args.shift(), args, unknown); | ||
|  |     } else { | ||
|  |       this.emit('command:*', 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]); | ||
|  |     } | ||
|  |     if (this.commands.length === 0 && | ||
|  |         this._args.filter(function(a) { return a.required; }).length === 0) { | ||
|  |       this.emit('command:*'); | ||
|  |     } | ||
|  |   } | ||
|  | 
 | ||
|  |   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 (literal) { | ||
|  |       args.push(arg); | ||
|  |       continue; | ||
|  |     } | ||
|  | 
 | ||
|  |     if (arg === '--') { | ||
|  |       literal = true; | ||
|  |       continue; | ||
|  |     } | ||
|  | 
 | ||
|  |     // find matching Option
 | ||
|  |     option = this.optionFor(arg); | ||
|  | 
 | ||
|  |     // option is defined
 | ||
|  |     if (option) { | ||
|  |       // requires arg
 | ||
|  |       if (option.required) { | ||
|  |         arg = argv[++i]; | ||
|  |         if (arg == null) return this.optionMissingArgument(option); | ||
|  |         this.emit('option:' + option.name(), arg); | ||
|  |       // optional arg
 | ||
|  |       } else if (option.optional) { | ||
|  |         arg = argv[i + 1]; | ||
|  |         if (arg == null || (arg[0] === '-' && arg !== '-')) { | ||
|  |           arg = null; | ||
|  |         } else { | ||
|  |           ++i; | ||
|  |         } | ||
|  |         this.emit('option:' + option.name(), arg); | ||
|  |       // bool
 | ||
|  |       } else { | ||
|  |         this.emit('option:' + 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 ((i + 1) < argv.length && 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 = this.options[i].attributeName(); | ||
|  |     result[key] = key === this._versionOptionName ? this._version : this[key]; | ||
|  |   } | ||
|  |   return result; | ||
|  | }; | ||
|  | 
 | ||
|  | /** | ||
|  |  * Argument `name` is missing. | ||
|  |  * | ||
|  |  * @param {String} name | ||
|  |  * @api private | ||
|  |  */ | ||
|  | 
 | ||
|  | Command.prototype.missingArgument = function(name) { | ||
|  |   console.error("error: missing required argument `%s'", name); | ||
|  |   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) { | ||
|  |   if (flag) { | ||
|  |     console.error("error: option `%s' argument missing, got `%s'", option.flags, flag); | ||
|  |   } else { | ||
|  |     console.error("error: option `%s' argument missing", option.flags); | ||
|  |   } | ||
|  |   process.exit(1); | ||
|  | }; | ||
|  | 
 | ||
|  | /** | ||
|  |  * Unknown option `flag`. | ||
|  |  * | ||
|  |  * @param {String} flag | ||
|  |  * @api private | ||
|  |  */ | ||
|  | 
 | ||
|  | Command.prototype.unknownOption = function(flag) { | ||
|  |   if (this._allowUnknownOption) return; | ||
|  |   console.error("error: unknown option `%s'", flag); | ||
|  |   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("error: variadic arguments must be last `%s'", name); | ||
|  |   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 (arguments.length === 0) return this._version; | ||
|  |   this._version = str; | ||
|  |   flags = flags || '-V, --version'; | ||
|  |   var versionOption = new Option(flags, 'output the version number'); | ||
|  |   this._versionOptionName = versionOption.long.substr(2) || 'version'; | ||
|  |   this.options.push(versionOption); | ||
|  |   this.on('option:' + this._versionOptionName, function() { | ||
|  |     process.stdout.write(str + '\n'); | ||
|  |     process.exit(0); | ||
|  |   }); | ||
|  |   return this; | ||
|  | }; | ||
|  | 
 | ||
|  | /** | ||
|  |  * Set the description to `str`. | ||
|  |  * | ||
|  |  * @param {String} str | ||
|  |  * @param {Object} argsDescription | ||
|  |  * @return {String|Command} | ||
|  |  * @api public | ||
|  |  */ | ||
|  | 
 | ||
|  | Command.prototype.description = function(str, argsDescription) { | ||
|  |   if (arguments.length === 0) return this._description; | ||
|  |   this._description = str; | ||
|  |   this._argsDescription = argsDescription; | ||
|  |   return this; | ||
|  | }; | ||
|  | 
 | ||
|  | /** | ||
|  |  * Set an alias for the command | ||
|  |  * | ||
|  |  * @param {String} alias | ||
|  |  * @return {String|Command} | ||
|  |  * @api public | ||
|  |  */ | ||
|  | 
 | ||
|  | Command.prototype.alias = function(alias) { | ||
|  |   var command = this; | ||
|  |   if (this.commands.length !== 0) { | ||
|  |     command = this.commands[this.commands.length - 1]; | ||
|  |   } | ||
|  | 
 | ||
|  |   if (arguments.length === 0) return command._alias; | ||
|  | 
 | ||
|  |   if (alias === command._name) throw new Error('Command alias can\'t be the same as its name'); | ||
|  | 
 | ||
|  |   command._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 (arguments.length === 0) return this._usage || usage; | ||
|  |   this._usage = str; | ||
|  | 
 | ||
|  |   return this; | ||
|  | }; | ||
|  | 
 | ||
|  | /** | ||
|  |  * Get or set the name of the command | ||
|  |  * | ||
|  |  * @param {String} str | ||
|  |  * @return {String|Command} | ||
|  |  * @api public | ||
|  |  */ | ||
|  | 
 | ||
|  | Command.prototype.name = function(str) { | ||
|  |   if (arguments.length === 0) return this._name; | ||
|  |   this._name = str; | ||
|  |   return this; | ||
|  | }; | ||
|  | 
 | ||
|  | /** | ||
|  |  * Return prepared commands. | ||
|  |  * | ||
|  |  * @return {Array} | ||
|  |  * @api private | ||
|  |  */ | ||
|  | 
 | ||
|  | Command.prototype.prepareCommands = function() { | ||
|  |   return 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 ? ' ' + args : ''), | ||
|  |       cmd._description | ||
|  |     ]; | ||
|  |   }); | ||
|  | }; | ||
|  | 
 | ||
|  | /** | ||
|  |  * Return the largest command length. | ||
|  |  * | ||
|  |  * @return {Number} | ||
|  |  * @api private | ||
|  |  */ | ||
|  | 
 | ||
|  | Command.prototype.largestCommandLength = function() { | ||
|  |   var commands = this.prepareCommands(); | ||
|  |   return commands.reduce(function(max, command) { | ||
|  |     return Math.max(max, command[0].length); | ||
|  |   }, 0); | ||
|  | }; | ||
|  | 
 | ||
|  | /** | ||
|  |  * Return the largest option length. | ||
|  |  * | ||
|  |  * @return {Number} | ||
|  |  * @api private | ||
|  |  */ | ||
|  | 
 | ||
|  | Command.prototype.largestOptionLength = function() { | ||
|  |   var options = [].slice.call(this.options); | ||
|  |   options.push({ | ||
|  |     flags: '-h, --help' | ||
|  |   }); | ||
|  |   return options.reduce(function(max, option) { | ||
|  |     return Math.max(max, option.flags.length); | ||
|  |   }, 0); | ||
|  | }; | ||
|  | 
 | ||
|  | /** | ||
|  |  * Return the largest arg length. | ||
|  |  * | ||
|  |  * @return {Number} | ||
|  |  * @api private | ||
|  |  */ | ||
|  | 
 | ||
|  | Command.prototype.largestArgLength = function() { | ||
|  |   return this._args.reduce(function(max, arg) { | ||
|  |     return Math.max(max, arg.name.length); | ||
|  |   }, 0); | ||
|  | }; | ||
|  | 
 | ||
|  | /** | ||
|  |  * Return the pad width. | ||
|  |  * | ||
|  |  * @return {Number} | ||
|  |  * @api private | ||
|  |  */ | ||
|  | 
 | ||
|  | Command.prototype.padWidth = function() { | ||
|  |   var width = this.largestOptionLength(); | ||
|  |   if (this._argsDescription && this._args.length) { | ||
|  |     if (this.largestArgLength() > width) { | ||
|  |       width = this.largestArgLength(); | ||
|  |     } | ||
|  |   } | ||
|  | 
 | ||
|  |   if (this.commands && this.commands.length) { | ||
|  |     if (this.largestCommandLength() > width) { | ||
|  |       width = this.largestCommandLength(); | ||
|  |     } | ||
|  |   } | ||
|  | 
 | ||
|  |   return width; | ||
|  | }; | ||
|  | 
 | ||
|  | /** | ||
|  |  * Return help for options. | ||
|  |  * | ||
|  |  * @return {String} | ||
|  |  * @api private | ||
|  |  */ | ||
|  | 
 | ||
|  | Command.prototype.optionHelp = function() { | ||
|  |   var width = this.padWidth(); | ||
|  | 
 | ||
|  |   // Append the help information
 | ||
|  |   return this.options.map(function(option) { | ||
|  |     return pad(option.flags, width) + '  ' + option.description + | ||
|  |       ((option.bool && option.defaultValue !== undefined) ? ' (default: ' + JSON.stringify(option.defaultValue) + ')' : ''); | ||
|  |   }).concat([pad('-h, --help', width) + '  ' + 'output usage information']) | ||
|  |     .join('\n'); | ||
|  | }; | ||
|  | 
 | ||
|  | /** | ||
|  |  * Return command help documentation. | ||
|  |  * | ||
|  |  * @return {String} | ||
|  |  * @api private | ||
|  |  */ | ||
|  | 
 | ||
|  | Command.prototype.commandHelp = function() { | ||
|  |   if (!this.commands.length) return ''; | ||
|  | 
 | ||
|  |   var commands = this.prepareCommands(); | ||
|  |   var width = this.padWidth(); | ||
|  | 
 | ||
|  |   return [ | ||
|  |     'Commands:', | ||
|  |     commands.map(function(cmd) { | ||
|  |       var desc = cmd[1] ? '  ' + cmd[1] : ''; | ||
|  |       return (desc ? pad(cmd[0], width) : cmd[0]) + desc; | ||
|  |     }).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 argsDescription = this._argsDescription; | ||
|  |     if (argsDescription && this._args.length) { | ||
|  |       var width = this.padWidth(); | ||
|  |       desc.push('Arguments:'); | ||
|  |       desc.push(''); | ||
|  |       this._args.forEach(function(arg) { | ||
|  |         desc.push('  ' + pad(arg.name, width) + '  ' + argsDescription[arg.name]); | ||
|  |       }); | ||
|  |       desc.push(''); | ||
|  |     } | ||
|  |   } | ||
|  | 
 | ||
|  |   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(desc) | ||
|  |     .concat(options) | ||
|  |     .concat(cmds) | ||
|  |     .join('\n'); | ||
|  | }; | ||
|  | 
 | ||
|  | /** | ||
|  |  * Output help information for this command | ||
|  |  * | ||
|  |  * @api public | ||
|  |  */ | ||
|  | 
 | ||
|  | Command.prototype.outputHelp = function(cb) { | ||
|  |   if (!cb) { | ||
|  |     cb = function(passthru) { | ||
|  |       return passthru; | ||
|  |     }; | ||
|  |   } | ||
|  |   process.stdout.write(cb(this.helpInformation())); | ||
|  |   this.emit('--help'); | ||
|  | }; | ||
|  | 
 | ||
|  | /** | ||
|  |  * Output help information and exit. | ||
|  |  * | ||
|  |  * @api public | ||
|  |  */ | ||
|  | 
 | ||
|  | Command.prototype.help = function(cb) { | ||
|  |   this.outputHelp(cb); | ||
|  |   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; | ||
|  |   } | ||
|  | } |