12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103 |
- 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');
- exports = module.exports = new Command();
- exports.Command = Command;
- exports.Option = Option;
- 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 || '';
- }
- Option.prototype.name = function() {
- return this.long
- .replace('--', '')
- .replace('no-', '');
- };
- Option.prototype.is = function(arg) {
- return arg == this.short || arg == this.long;
- };
- function Command(name) {
- this.commands = [];
- this.options = [];
- this._execs = [];
- this._allowUnknownOption = false;
- this._args = [];
- this._name = name;
- }
- Command.prototype.__proto__ = EventEmitter.prototype;
- 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;
- };
- Command.prototype.arguments = function (desc) {
- return this.parseExpectedArgs(desc.split(/ +/));
- }
- Command.prototype.addImplicitHelpCommand = function() {
- this.command('help [cmd]', 'display help for [cmd]');
- };
- 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;
- };
- Command.prototype.action = function(fn) {
- var self = this;
- var listener = function(args, unknown) {
-
- args = args || [];
- unknown = unknown || [];
- var parsed = self.parseOptions(unknown);
-
- outputHelpIfNecessary(self, parsed.unknown);
-
-
-
- if (parsed.unknown.length > 0) {
- self.unknownOption(parsed.unknown[0]);
- }
-
- 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);
- }
- });
-
-
-
- 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;
- };
- Command.prototype.option = function(flags, description, fn, defaultValue) {
- var self = this
- , option = new Option(flags, description)
- , oname = option.name()
- , name = camelcase(oname);
-
- 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;
- }
- }
-
- if (false == option.bool || option.optional || option.required) {
-
- if (false == option.bool) defaultValue = true;
-
- if (undefined !== defaultValue) self[name] = defaultValue;
- }
-
- this.options.push(option);
-
-
- this.on(oname, function(val) {
-
- if (null !== val && fn) val = fn(val, undefined === self[name]
- ? defaultValue
- : self[name]);
-
- if ('boolean' == typeof self[name] || 'undefined' == typeof self[name]) {
-
- if (null == val) {
- self[name] = option.bool
- ? defaultValue || true
- : false;
- } else {
- self[name] = val;
- }
- } else if (null !== val) {
-
- self[name] = val;
- }
- });
- return this;
- };
- Command.prototype.allowUnknownOption = function(arg) {
- this._allowUnknownOption = arguments.length === 0 || arg;
- return this;
- };
- Command.prototype.parse = function(argv) {
-
- if (this.executables) this.addImplicitHelpCommand();
-
- this.rawArgs = argv;
-
- this._name = this._name || basename(argv[1], '.js');
-
- if (this.executables && argv.length < 3) {
-
- argv.push('--help');
- }
-
- var parsed = this.parseOptions(this.normalize(argv.slice(2)));
- var args = this.args = parsed.args;
- var result = this.parseArgs(this.args, parsed.unknown);
-
- var name = result.args[0];
- if (this._execs[name] && typeof this._execs[name] != "function") {
- return this.executeSubCommand(argv, args, parsed.unknown);
- }
- return result;
- };
- 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();
-
- if ('help' == args[0]) {
- args[0] = args[1];
- args[1] = '--help';
- }
-
- var f = argv[1];
-
- var bin = basename(f, '.js') + '-' + args[0];
-
-
- var baseDir
- , link = readlink(f);
-
- if (link !== f && link.charAt(0) !== '/') {
- link = path.join(dirname(f), link)
- }
- baseDir = dirname(link);
-
- var localBin = path.join(baseDir, bin);
-
- 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);
-
- 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;
- };
- 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 === '--') {
-
- 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;
- };
- 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 (unknown.length > 0) {
- this.unknownOption(unknown[0]);
- }
- }
- return this;
- };
- 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];
- }
- }
- };
- Command.prototype.parseOptions = function(argv) {
- var args = []
- , len = argv.length
- , literal
- , option
- , arg;
- var unknownOptions = [];
-
- for (var i = 0; i < len; ++i) {
- arg = argv[i];
-
- if ('--' == arg) {
- literal = true;
- continue;
- }
- if (literal) {
- args.push(arg);
- continue;
- }
-
- option = this.optionFor(arg);
-
- if (option) {
-
- if (option.required) {
- arg = argv[++i];
- if (null == arg) return this.optionMissingArgument(option);
- this.emit(option.name(), arg);
-
- } else if (option.optional) {
- arg = argv[i+1];
- if (null == arg || ('-' == arg[0] && '-' != arg)) {
- arg = null;
- } else {
- ++i;
- }
- this.emit(option.name(), arg);
-
- } else {
- this.emit(option.name());
- }
- continue;
- }
-
- if (arg.length > 1 && '-' == arg[0]) {
- unknownOptions.push(arg);
-
-
-
- if (argv[i+1] && '-' != argv[i+1][0]) {
- unknownOptions.push(argv[++i]);
- }
- continue;
- }
-
- args.push(arg);
- }
- return { args: args, unknown: unknownOptions };
- };
- 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;
- };
- Command.prototype.missingArgument = function(name) {
- console.error();
- console.error(" error: missing required argument `%s'", name);
- console.error();
- process.exit(1);
- };
- 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);
- };
- Command.prototype.unknownOption = function(flag) {
- if (this._allowUnknownOption) return;
- console.error();
- console.error(" error: unknown option `%s'", flag);
- console.error();
- process.exit(1);
- };
- Command.prototype.variadicArgNotLast = function(name) {
- console.error();
- console.error(" error: variadic arguments must be last `%s'", name);
- console.error();
- process.exit(1);
- };
- 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;
- };
- Command.prototype.description = function(str) {
- if (0 == arguments.length) return this._description;
- this._description = str;
- return this;
- };
- Command.prototype.alias = function(alias) {
- if (0 == arguments.length) return this._alias;
- this._alias = alias;
- return this;
- };
- 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;
- };
- Command.prototype.name = function() {
- return this._name;
- };
- Command.prototype.largestOptionLength = function() {
- return this.options.reduce(function(max, option) {
- return Math.max(max, option.flags.length);
- }, 0);
- };
- Command.prototype.optionHelp = function() {
- var width = this.largestOptionLength();
-
- return [pad('-h, --help', width) + ' ' + 'output usage information']
- .concat(this.options.map(function(option) {
- return pad(option.flags, width) + ' ' + option.description;
- }))
- .join('\n');
- };
- 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');
- };
- 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');
- };
- Command.prototype.outputHelp = function() {
- process.stdout.write(this.helpInformation());
- this.emit('--help');
- };
- Command.prototype.help = function() {
- this.outputHelp();
- process.exit();
- };
- function camelcase(flag) {
- return flag.split('-').reduce(function(str, word) {
- return str + word[0].toUpperCase() + word.slice(1);
- });
- }
- function pad(str, width) {
- var len = Math.max(0, width - str.length);
- return str + Array(len + 1).join(' ');
- }
- 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);
- }
- }
- }
- function humanReadableArgName(arg) {
- var nameOutput = arg.name + (arg.variadic === true ? '...' : '');
- return arg.required
- ? '<' + nameOutput + '>'
- : '[' + nameOutput + ']'
- }
- function exists(file) {
- try {
- if (fs.statSync(file).isFile()) {
- return true;
- }
- } catch (e) {
- return false;
- }
- }
|