Spaces:
Runtime error
Runtime error
| /** | |
| * 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] === true) { | |
| 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; | |
| } | |
| } | |