12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112 |
- /**
- * Copyright (c) 2010 Chris O'Hara <cohara87@gmail.com>
- *
- * Permission is hereby granted, free of charge, to any person obtaining
- * a copy of this software and associated documentation files (the
- * "Software"), to deal in the Software without restriction, including
- * without limitation the rights to use, copy, modify, merge, publish,
- * distribute, sublicense, and/or sell copies of the Software, and to
- * permit persons to whom the Software is furnished to do so, subject to
- * the following conditions:
- *
- * The above copyright notice and this permission notice shall be
- * included in all copies or substantial portions of the Software.
- *
- * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
- * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
- * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
- * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
- * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
- * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
- * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
- */
- //Note: cli includes kof/node-natives and creationix/stack. I couldn't find
- //license information for either - contact me if you want your license added
- var cli = exports,
- argv, curr_opt, curr_val, full_opt, is_long,
- short_tags = [], opt_list, parsed = {},
- usage, argv_parsed, command_list, commands,
- show_debug;
- cli.app = null;
- cli.version = null;
- cli.argv = [];
- cli.argc = 0;
- cli.options = {};
- cli.args = [];
- cli.command = null;
- cli.width = 70;
- cli.option_width = 25;
- /**
- * Bind kof's node-natives (https://github.com/kof/node-natives) to `cli.native`
- *
- * Rather than requiring node natives (e.g. var fs = require('fs')), all
- * native modules can be accessed like `cli.native.fs`
- */
- cli.native = {};
- var define_native = function (module) {
- Object.defineProperty(cli.native, module, {
- enumerable: true,
- configurable: true,
- get: function() {
- delete cli.native[module];
- return (cli.native[module] = require(module));
- }
- });
- };
- var natives = process.binding('natives');
- for (var module in natives) {
- define_native(module);
- }
- cli.output = console.log;
- cli.exit = require('exit');
- cli.no_color = false;
- if (process.env.NODE_DISABLE_COLORS || process.env.TERM === 'dumb') {
- cli.no_color = true;
- }
- /**
- * Define plugins. Plugins can be enabled and disabled by calling:
- *
- * `cli.enable(plugin1, [plugin2, ...])`
- * `cli.disable(plugin1, [plugin2, ...])`
- *
- * Methods are chainable - `cli.enable(plugin).disable(plugin2)`.
- *
- * The 'help' plugin is enabled by default.
- */
- var enable = {
- help: true, //Adds -h, --help
- version: false, //Adds -v,--version => gets version by parsing a nearby package.json
- status: false, //Adds -k,--no-color & --debug => display plain status messages /display debug messages
- timeout: false, //Adds -t,--timeout N => timeout the process after N seconds
- catchall: false, //Adds -c,--catch => catch and output uncaughtExceptions
- glob: false //Adds glob matching => use cli.glob(arg)
- }
- cli.enable = function (/*plugins*/) {
- Array.prototype.slice.call(arguments).forEach(function (plugin) {
- switch (plugin) {
- case 'catchall':
- process.on('uncaughtException', function (err) {
- cli.error('Uncaught exception: ' + (err.msg || err));
- });
- break;
- case 'help': case 'version': case 'status':
- case 'autocomplete': case 'timeout':
- //Just add switches.
- break;
- case 'glob':
- cli.glob = require('glob');
- break;
- default:
- cli.fatal('Unknown plugin "' + plugin + '"');
- break;
- }
- enable[plugin] = true;
- });
- return cli;
- }
- cli.disable = function (/*plugins*/) {
- Array.prototype.slice.call(arguments).forEach(function (plugin) {
- if (enable[plugin]) {
- enable[plugin] = false;
- }
- });
- return cli;
- }
- /**
- * Sets argv (default is process.argv).
- *
- * @param {Array|String} argv
- * @param {Boolean} keep_arg0 (optional - default is false)
- * @api public
- */
- cli.setArgv = function (arr, keep_arg0) {
- if (typeof arr == 'string') {
- arr = arr.split(' ');
- } else {
- arr = arr.slice();
- }
- cli.app = arr.shift();
- // Strip off argv[0] if it's a node binary
- // So this is still broken and will break if you are calling node through a
- // symlink, unless you are lucky enough to have it as 'node' literal. Latter
- // is a hack, but resolving abspaths/symlinks is an unportable can of worms.
- if (!keep_arg0 && (['node', 'node.exe'].indexOf(cli.native.path.basename(cli.app)) !== -1
- || cli.native.path.basename(process.execPath) === cli.app
- || process.execPath === cli.app)) {
- cli.app = arr.shift();
- }
- cli.app = cli.native.path.basename(cli.app);
- argv_parsed = false;
- cli.args = cli.argv = argv = arr;
- cli.argc = argv.length;
- cli.options = {};
- cli.command = null;
- };
- cli.setArgv(process.argv);
- /**
- * Returns the next opt, or false if no opts are found.
- *
- * @return {String} opt
- * @api public
- */
- cli.next = function () {
- if (!argv_parsed) {
- cli.args = [];
- argv_parsed = true;
- }
- curr_val = null;
- //If we're currently in a group of short opts (e.g. -abc), return the next opt
- if (short_tags.length) {
- curr_opt = short_tags.shift();
- full_opt = '-' + curr_opt;
- return curr_opt;
- }
- if (!argv.length) {
- return false;
- }
- curr_opt = argv.shift();
- //If an escape sequence is found (- or --), subsequent opts are ignored
- if (curr_opt === '-' || curr_opt === '--') {
- while (argv.length) {
- cli.args.push(argv.shift());
- }
- return false;
- }
- //If the next element in argv isn't an opt, add it to the list of args
- if (curr_opt[0] !== '-') {
- cli.args.push(curr_opt);
- return cli.next();
- } else {
- //Check if the opt is short/long
- is_long = curr_opt[1] === '-';
- curr_opt = curr_opt.substr(is_long ? 2 : 1);
- }
- //Accept grouped short opts, e.g. -abc => -a -b -c
- if (!is_long && curr_opt.length > 1) {
- short_tags = curr_opt.split('');
- return cli.next();
- }
- var eq, len;
- //Check if the long opt is in the form --option=VALUE
- if (is_long && (eq = curr_opt.indexOf('=')) >= 0) {
- curr_val = curr_opt.substr(eq + 1);
- curr_opt = curr_opt.substr(0, eq);
- len = curr_val.length;
- //Allow values to be quoted
- if ((curr_val[0] === '"' && curr_val[len - 1] === '"') ||
- (curr_val[0] === "'" && curr_val[len - 1] === "'"))
- {
- curr_val = curr_val.substr(1, len-2);
- }
- if (curr_val.match(/^[0-9]+$/)) {
- curr_val = parseInt(curr_val, 10);
- }
- }
- //Save the opt representation for later
- full_opt = (is_long ? '--' : '-') + curr_opt;
- return curr_opt;
- };
- /**
- * Parses command line opts.
- *
- * `opts` must be an object with opts defined like:
- * long_tag: [short_tag, description, value_type, default_value];
- *
- * `commands` is an optional array or object for apps that are of the form
- * my_app [OPTIONS] <command> [ARGS]
- * The command list is output with usage information + there is bundled
- * support for auto-completion, etc.
- *
- * See README.md for more information.
- *
- * @param {Object} opts
- * @param {Object} commands (optional)
- * @return {Object} opts (parsed)
- * @api public
- */
- cli.parse = function (opts, command_def) {
- var default_val, i, o, parsed = cli.options, seen,
- catch_all = !opts;
- opt_list = opts || {};
- commands = command_def;
- command_list = commands || [];
- if (commands && !Array.isArray(commands)) {
- command_list = Object.keys(commands);
- }
- while ((o = cli.next())) {
- seen = false;
- for (var opt in opt_list) {
- if (!(opt_list[opt] instanceof Array)) {
- continue;
- }
- if (!opt_list[opt][0]) {
- opt_list[opt][0] = opt;
- }
- if (o === opt || o === opt_list[opt][0]) {
- seen = true;
- if (opt_list[opt].length === 2) {
- parsed[opt] = true;
- break;
- }
- default_val = null;
- if (opt_list[opt].length === 4) {
- default_val = opt_list[opt][3];
- }
- if (opt_list[opt][2] instanceof Array) {
- for (i = 0, l = opt_list[opt][2].length; i < l; i++) {
- if (typeof opt_list[opt][2][i] === 'number') {
- opt_list[opt][2][i] += '';
- }
- }
- parsed[opt] = cli.getArrayValue(opt_list[opt][2], is_long ? null : default_val);
- break;
- }
- if (opt_list[opt][2].toLowerCase) {
- opt_list[opt][2] = opt_list[opt][2].toLowerCase();
- }
- switch (opt_list[opt][2]) {
- case 'string': case 1: case true:
- parsed[opt] = cli.getValue(default_val);
- break;
- case 'int': case 'number': case 'num':
- case 'time': case 'seconds': case 'secs': case 'minutes': case 'mins':
- case 'x': case 'n':
- parsed[opt] = cli.getInt(default_val);
- break;
- case 'date': case 'datetime': case 'date_time':
- parsed[opt] = cli.getDate(default_val);
- break;
- case 'float': case 'decimal':
- parsed[opt] = cli.getFloat(default_val);
- break;
- case 'path': case 'file': case 'directory': case 'dir':
- parsed[opt] = cli.getPath(default_val, opt_list[opt][2]);
- break;
- case 'email':
- parsed[opt] = cli.getEmail(default_val);
- break;
- case 'url': case 'uri': case 'domain': case 'host':
- parsed[opt] = cli.getUrl(default_val, opt_list[opt][2]);
- break;
- case 'ip':
- parsed[opt] = cli.getIp(default_val);
- break;
- case 'bool': case 'boolean': case 'on':
- parsed[opt] = true;
- break;
- case 'false': case 'off': case false: case 0:
- parsed[opt] = false;
- break;
- default:
- cli.fatal('Unknown opt type "' + opt_list[opt][2] + '"');
- }
- break;
- }
- }
- if (!seen) {
- if (enable.help && (o === 'h' || o === 'help')) {
- cli.getUsage();
- } else if (enable.version && (o === 'v' || o === 'version')) {
- if (cli.version == null) {
- cli.parsePackageJson();
- }
- console.error(cli.app + ' v' + cli.version);
- cli.exit();
- break;
- } else if (enable.catchall && (o === 'c' || o === 'catch')) {
- continue;
- } else if (enable.status && (o === 'k' || o === 'no-color')) {
- cli.no_color = (o === 'k' || o === 'no-color');
- continue;
- } else if (enable.status && (o === 'debug')) {
- show_debug = o === 'debug';
- continue;
- } else if (enable.timeout && (o === 't' || o === 'timeout')) {
- var secs = cli.getInt();
- setTimeout(function () {
- cli.fatal('Process timed out after ' + secs + 's');
- }, secs * 1000);
- continue;
- } else if (catch_all) {
- parsed[o] = curr_val || true;
- continue;
- }
- cli.fatal('Unknown option ' + full_opt);
- }
- }
- //Fill the remaining options with their default value or null
- for (var opt in opt_list) {
- default_val = opt_list[opt].length === 4 ? opt_list[opt][3] : null;
- if (!(opt_list[opt] instanceof Array)) {
- parsed[opt] = opt_list[opt];
- continue;
- } else if (typeof parsed[opt] === 'undefined') {
- parsed[opt] = default_val;
- }
- }
- if (command_list.length) {
- if (cli.args.length === 0) {
- if (enable.help) {
- cli.getUsage();
- } else {
- cli.fatal('A command is required (' + command_list.join(', ') + ').');
- }
- return cli.exit(1);
- } else {
- cli.command = cli.autocompleteCommand(cli.args.shift());
- }
- }
- cli.argc = cli.args.length;
- return parsed;
- };
- /**
- * Helper method for matching a command from the command list.
- *
- * @param {String} command
- * @return {String} full_command
- * @api public
- */
- cli.autocompleteCommand = function (command) {
- var list;
- if (!(command_list instanceof Array)) {
- list = Object.keys(command_list);
- } else {
- list = command_list;
- }
- var i, j = 0, c = command.length, tmp_list;
- if (list.length === 0 || list.indexOf(command) !== -1) {
- return command;
- }
- for (i = 0; i < c; i++) {
- tmp_list = [];
- l = list.length;
- if (l <= 1) break;
- for (j = 0; j < l; j++)
- if (list[j].length >= i && list[j][i] === command[i])
- tmp_list.push(list[j]);
- list = tmp_list;
- }
- l = list.length;
- if (l === 1) {
- return list[0];
- } else if (l === 0) {
- cli.fatal('Unknown command "' + command + '"' + (enable.help ? '. Please see --help for more information' : ''));
- } else {
- list.sort();
- cli.fatal('The command "' + command + '" is ambiguous and could mean "' + list.join('", "') + '"');
- }
- };
- /**
- * Adds methods to output styled status messages to stderr.
- *
- * Added methods are cli.info(msg), cli.error(msg), cli.ok(msg), and
- * cli.debug(msg).
- *
- * To control status messages, use the 'status' plugin
- * 1) debug() messages are hidden by default. Display them with
- * the --debug opt.
- * 2) to hide all status messages, use the -s or --silent opt.
- *
- * @api private
- */
- cli.status = function (msg, type) {
- var pre;
- switch (type) {
- case 'info':
- pre = cli.no_color ? 'INFO:' : '\x1B[33mINFO\x1B[0m:';
- break;
- case 'debug':
- pre = cli.no_color ? 'DEBUG:' : '\x1B[36mDEBUG\x1B[0m:';
- break;
- case 'error':
- case 'fatal':
- pre = cli.no_color ? 'ERROR:' : '\x1B[31mERROR\x1B[0m:';
- break;
- case 'ok':
- pre = cli.no_color ? 'OK:' : '\x1B[32mOK\x1B[0m:';
- break;
- }
- msg = pre + ' ' + msg;
- if (type === 'fatal') {
- console.error(msg);
- return cli.exit(1);
- }
- if (enable.status && !show_debug && type === 'debug') {
- return;
- }
- console.error(msg);
- };
- ['info','error','ok','debug','fatal'].forEach(function (type) {
- cli[type] = function (msg) {
- cli.status(msg, type);
- };
- });
- /**
- * Sets the app name and version.
- *
- * Usage:
- * setApp('myapp', '0.1.0');
- * setApp('./package.json'); //Pull name/version from package.json
- *
- * @param {String} name
- * @return cli (for chaining)
- * @api public
- */
- cli.setApp = function (name, version) {
- if (name.indexOf('package.json') !== -1) {
- cli.parsePackageJson(name);
- } else {
- cli.app = name;
- cli.version = version;
- }
- return cli;
- };
- /**
- * Parses the version number from package.json. If no path is specified, cli
- * will attempt to locate a package.json in ./, ../ or ../../
- *
- * @param {String} path (optional)
- * @api public
- */
- cli.parsePackageJson = function (path) {
- var parse_packagejson = function (path) {
- var packagejson = JSON.parse(cli.native.fs.readFileSync(path, 'utf8'));
- cli.version = packagejson.version;
- cli.app = packagejson.name;
- };
- var try_all = function (arr, func, err) {
- for (var i = 0, l = arr.length; i < l; i++) {
- try {
- func(arr[i]);
- return;
- } catch (e) {
- if (i === l-1) {
- cli.fatal(err);
- }
- }
- }
- };
- try {
- if (path) {
- return parse_packagejson(path);
- }
- try_all([
- __dirname + '/package.json',
- __dirname + '/../package.json',
- __dirname + '/../../package.json'
- ], parse_packagejson);
- } catch (e) {
- cli.fatal('Could not detect ' + cli.app + ' version');
- }
- };
- /**
- * Sets the usage string - default is `app [OPTIONS] [ARGS]`.
- *
- * @param {String} u
- * @return cli (for chaining)
- * @api public
- */
- cli.setUsage = function (u) {
- usage = u;
- return cli;
- };
- var pad = function (str, len) {
- if (typeof len === 'undefined') {
- len = str;
- str = '';
- }
- if (str.length < len) {
- len -= str.length;
- while (len--) str += ' ';
- }
- return str;
- };
- /**
- * Automatically build usage information from the opts list. If the help
- * plugin is enabled (default), this info is displayed with -h, --help.
- *
- * @api public
- */
- cli.getUsage = function (code) {
- var short, desc, optional, line, seen_opts = [],
- switch_pad = cli.option_width;
- var trunc_desc = function (pref, desc, len) {
- var pref_len = pref.length,
- desc_len = cli.width - pref_len,
- truncated = '';
- if (desc.length <= desc_len) {
- return desc;
- }
- var desc_words = (desc+'').split(' '), chars = 0, word;
- while (desc_words.length) {
- truncated += (word = desc_words.shift()) + ' ';
- chars += word.length;
- if (desc_words.length && chars + desc_words[0].length > desc_len) {
- truncated += '\n' + pad(pref_len);
- chars = 0;
- }
- }
- return truncated;
- };
- usage = usage || cli.app + ' [OPTIONS]' + (command_list.length ? ' <command>' : '') + ' [ARGS]';
- if (cli.no_color) {
- console.error('Usage:\n ' + usage);
- console.error('Options: ');
- } else {
- console.error('\x1b[1mUsage\x1b[0m:\n ' + usage);
- console.error('\n\x1b[1mOptions\x1b[0m: ');
- }
- for (var opt in opt_list) {
- if (opt.length === 1) {
- long = opt_list[opt][0];
- short = opt;
- } else {
- long = opt;
- short = opt_list[opt][0];
- }
- //Parse opt_list
- desc = opt_list[opt][1].trim();
- type = opt_list[opt].length >= 3 ? opt_list[opt][2] : null;
- optional = opt_list[opt].length === 4 ? opt_list[opt][3] : null;
- //Build usage line
- if (short === long) {
- if (short.length === 1) {
- line = ' -' + short;
- } else {
- line = ' --' + long;
- }
- } else if (short) {
- line = ' -' + short + ', --' + long;
- } else {
- line = ' --' + long;
- }
- line += ' ';
- if (type) {
- if (type instanceof Array) {
- desc += '. VALUE must be either [' + type.join('|') + ']';
- type = 'VALUE';
- }
- if (type === true || type === 1) {
- type = long.toUpperCase();
- }
- type = type.toUpperCase();
- if (type === 'FLOAT' || type === 'INT') {
- type = 'NUMBER';
- }
- line += optional ? '[' + type + ']' : type;
- }
- line = pad(line, switch_pad);
- line += trunc_desc(line, desc);
- line += optional ? ' (Default is ' + optional + ')' : '';
- console.error(line.replace('%s', '%\0s'));
- seen_opts.push(short);
- seen_opts.push(long);
- }
- if (enable.timeout && seen_opts.indexOf('t') === -1 && seen_opts.indexOf('timeout') === -1) {
- console.error(pad(' -t, --timeout N', switch_pad) + 'Exit if the process takes longer than N seconds');
- }
- if (enable.status) {
- if (seen_opts.indexOf('k') === -1 && seen_opts.indexOf('no-color') === -1) {
- console.error(pad(' -k, --no-color', switch_pad) + 'Omit color from output');
- }
- if (seen_opts.indexOf('debug') === -1) {
- console.error(pad(' --debug', switch_pad) + 'Show debug information');
- }
- }
- if (enable.catchall && seen_opts.indexOf('c') === -1 && seen_opts.indexOf('catch') === -1) {
- console.error(pad(' -c, --catch', switch_pad) + 'Catch unanticipated errors');
- }
- if (enable.version && seen_opts.indexOf('v') === -1 && seen_opts.indexOf('version') === -1) {
- console.error(pad(' -v, --version', switch_pad) + 'Display the current version');
- }
- if (enable.help && seen_opts.indexOf('h') === -1 && seen_opts.indexOf('help') === -1) {
- console.error(pad(' -h, --help', switch_pad) + 'Display help and usage details');
- }
- if (command_list.length) {
- console.error('\n\x1b[1mCommands\x1b[0m: ');
- if (!Array.isArray(commands)) {
- for (var c in commands) {
- line = ' ' + pad(c, switch_pad - 2);
- line += trunc_desc(line, commands[c]);
- console.error(line);
- }
- } else {
- command_list.sort();
- console.error(' ' + trunc_desc(' ', command_list.join(', ')));
- }
- }
- return cli.exit(code);
- };
- /**
- * Generates an error message when an opt is incorrectly used.
- *
- * @param {String} expects (e.g. 'a value')
- * @param {String} type (e.g. 'VALUE')
- * @api public
- */
- cli.getOptError = function (expects, type) {
- var err = full_opt + ' expects ' + expects
- + '. Use `' + cli.app + ' ' + full_opt + (is_long ? '=' : ' ') + type + '`';
- return err;
- };
- /**
- * Gets the next opt value and validates it with an optional validation
- * function. If validation fails or no value can be obtained, this method
- * will return the default value (if specified) or exit with err_msg.
- *
- * @param {String} default_val
- * @param {Function} validate_func
- * @param {String} err_msg
- * @api public
- */
- cli.getValue = function (default_val, validate_func, err_msg) {
- err_msg = err_msg || cli.getOptError('a value', 'VALUE');
- var value;
- try {
- if (curr_val) {
- if (validate_func) {
- curr_val = validate_func(curr_val);
- }
- return curr_val;
- }
- //Grouped short opts aren't allowed to have values
- if (short_tags.length) {
- throw 'Short tags';
- }
- //If there's no args left or the next arg is an opt, return the
- //default value (if specified) - otherwise fail
- if (!argv.length || (argv[0].length === 1 && argv[0][0] === '-')) {
- throw 'No value';
- }
- value = argv.shift();
- if (value.match(/^[0-9]+$/)) {
- value = parseInt(value, 10);
- }
- //Run the value through a validation/transformation function if specified
- if (validate_func) {
- value = validate_func(value);
- }
- } catch (e) {
- //The value didn't pass the validation/transformation. Unshift the value and
- //return the default value (if specified)
- if (value) {
- argv.unshift(value);
- }
- return default_val != null ? default_val : cli.fatal(err_msg);
- }
- return value;
- };
- cli.getInt = function (default_val) {
- return cli.getValue(default_val, function (value) {
- if (typeof value === 'number') return value;
- if (!value.match(/^(?:-?(?:0|[1-9][0-9]*))$/)) {
- throw 'Invalid int';
- }
- return parseInt(value);
- }, cli.getOptError('a number', 'NUMBER'));
- }
- cli.getDate = function (default_val) {
- return cli.getValue(default_val, function (value) {
- if (cli.toType(value) === 'date') return value;
- value = new Date(value);
- if ( ! value.getTime() ) {
- throw value.toString();
- }
- return value;
- }, cli.getOptError('a date', 'DATE'));
- }
- cli.getFloat = function (default_val) {
- return cli.getValue(default_val, function (value) {
- if (!value.match(/^(?:-?(?:0|[1-9][0-9]*))?(?:\.[0-9]*)?$/)) {
- throw 'Invalid float';
- }
- return parseFloat(value, 10);
- }, cli.getOptError('a number', 'NUMBER'));
- }
- cli.getUrl = function (default_val, identifier) {
- identifier = identifier || 'url';
- return cli.getValue(default_val, function (value) {
- if (!value.match(/^(?:(?:ht|f)tp(?:s?)\:\/\/|~\/|\/)?(?:\w+:\w+@)?((?:(?:[-\w\d{1-3}]+\.)+(?:com|org|net|gov|mil|biz|info|mobi|name|aero|jobs|edu|co\.uk|ac\.uk|it|fr|tv|museum|asia|local|travel|[a-z]{2})?)|((\b25[0-5]\b|\b[2][0-4][0-9]\b|\b[0-1]?[0-9]?[0-9]\b)(\.(\b25[0-5]\b|\b[2][0-4][0-9]\b|\b[0-1]?[0-9]?[0-9]\b)){3}))(?::[\d]{1,5})?(?:(?:(?:\/(?:[-\w~!$+|.,=]|%[a-f\d]{2})+)+|\/)+|\?|#)?(?:(?:\?(?:[-\w~!$+|.,*:]|%[a-f\d{2}])+=?(?:[-\w~!$+|.,*:=]|%[a-f\d]{2})*)(?:&(?:[-\w~!$+|.,*:]|%[a-f\d{2}])+=?(?:[-\w~!$+|.,*:=]|%[a-f\d]{2})*)*)*(?:#(?:[-\w~!$ |\/.,*:;=]|%[a-f\d]{2})*)?$/i)) {
- throw 'Invalid URL';
- }
- return value;
- }, cli.getOptError('a ' + identifier, identifier.toUpperCase()));
- }
- cli.getEmail = function (default_val) {
- return cli.getValue(default_val, function (value) {
- if (!value.match(/^(?:[\w\!\#\$\%\&\'\*\+\-\/\=\?\^\`\{\|\}\~]+\.)*[\w\!\#\$\%\&\'\*\+\-\/\=\?\^\`\{\|\}\~]+@(?:(?:(?:[a-zA-Z0-9](?:[a-zA-Z0-9\-](?!\.)){0,61}[a-zA-Z0-9]?\.)+[a-zA-Z0-9](?:[a-zA-Z0-9\-](?!$)){0,61}[a-zA-Z0-9]?)|(?:\[(?:(?:[01]?\d{1,2}|2[0-4]\d|25[0-5])\.){3}(?:[01]?\d{1,2}|2[0-4]\d|25[0-5])\]))$/)) {
- throw 'Invalid email';
- }
- return value;
- }, cli.getOptError('an email', 'EMAIL'));
- }
- cli.getIp = function (default_val) {
- return cli.getValue(default_val, function (value) {
- if (!value.match(/^(?:(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.){3}(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)$/)) {
- throw 'Invalid IP';
- }
- return value;
- }, cli.getOptError('an IP', 'IP'));
- }
- cli.getPath = function (default_val, identifier) {
- identifier = identifier || 'path';
- return cli.getValue(default_val, function (value) {
- if (value.match(/[?*;{}]/)) {
- throw 'Invalid path';
- }
- return value;
- }, cli.getOptError('a ' + identifier, identifier.toUpperCase()));
- }
- cli.getArrayValue = function (arr, default_val) {
- return cli.getValue(default_val, function (value) {
- if (arr.indexOf(value) === -1) {
- throw 'Unexpected value';
- }
- return value;
- }, cli.getOptError('either [' + arr.join('|') + ']', 'VALUE'));
- }
- /**
- * Gets all data from STDIN (with optional encoding) and sends it to callback.
- *
- * @param {String} encoding (optional - default is 'utf8')
- * @param {Function} callback
- * @api public
- */
- cli.withStdin = function (encoding, callback) {
- if (typeof encoding === 'function') {
- callback = encoding;
- encoding = 'utf8';
- }
- var stream = process.openStdin(), data = '';
- stream.setEncoding(encoding);
- stream.on('data', function (chunk) {
- data += chunk;
- });
- stream.on('end', function () {
- callback.apply(cli, [data]);
- });
- };
- /**
- * Gets all data from STDIN, splits the data into lines and sends it
- * to callback (callback isn't called until all of STDIN is read. To
- * process each line as it's received, see the method below
- *
- * @param {Function} callback
- * @api public
- */
- cli.withStdinLines = function (callback) {
- cli.withStdin(function (data) {
- var sep = data.indexOf('\r\n') !== -1 ? '\r\n' : '\n';
- callback.apply(cli, [data.split(sep), sep]);
- });
- };
- /**
- * Asynchronously reads a file line by line. When a line is received,
- * callback is called with (line, sep) - when EOF is reached, callback
- * receives (null, null, true)
- *
- * @param {String} file (optional - default is 'stdin')
- * @param {String} encoding (optional - default is 'utf8')
- * @param {Function} callback (line, sep, eof)
- * @api public
- */
- cli.withInput = function (file, encoding, callback) {
- if (typeof encoding === 'function') {
- callback = encoding;
- encoding = 'utf8';
- } else if (typeof file === 'function') {
- callback = file;
- encoding = 'utf8';
- file = 'stdin';
- }
- if (file === 'stdin') {
- file = process.openStdin();
- } else {
- try {
- file = cli.native.fs.createReadStream(file);
- file.on('error', cli.fatal);
- } catch (e) {
- return cli.fatal(e);
- }
- }
- file.setEncoding(encoding);
- var lines = [], data = '', eof, sep;
- file.on('data', function (chunk) {
- if (eof) return;
- data += chunk;
- if (!sep) {
- if (data.indexOf('\r\n') !== -1) {
- sep = '\r\n';
- } else if (data.indexOf('\n') !== -1) {
- sep = '\n';
- } else {
- last_line = data;
- return;
- }
- }
- lines = data.split(sep);
- data = eof ? null : lines.pop();
- while (lines.length) {
- callback.apply(cli, [lines.shift(), sep, false]);
- }
- });
- file.on('end', function () {
- eof = true;
- if (data.length) {
- callback.apply(cli, [data, sep || '', false]);
- }
- callback.apply(cli, [null, null, true]);
- });
- };
- /**
- * This function does a much better job at determining the object type than the typeof operator
- * @author Angus Croll - https://javascriptweblog.wordpress.com/2011/08/08/fixing-the-javascript-typeof-operator/
- * @param {Object} obj A Javascript object you wish to know the type of.
- * @return {string} A string describing the Object's type if it is indeed a built in JavaScript type.
- */
- cli.toType = function(obj) {
- var type = ({}).toString.call(obj).match(/\s([a-zA-Z]+)/)[1].toLowerCase();
- function isInt(n) {
- return Number(n) === n && n % 1 === 0;
- }
- function isFloat(n){
- return n === Number(n) && n % 1 !== 0;
- }
- if ( type === 'number' ) {
- if ( isInt(obj) ) {
- return 'integer';
- } else if ( isFloat(obj) ) {
- return 'float';
- }
- }
- return type;
- }
- /**
- * The main entry method. `callback` receives (args, options)
- *
- * @param {Function} callback
- * @api public
- */
- cli.main = function (callback) {
- callback.call(cli, cli.args, cli.options);
- }
- /**
- * Bind creationix's stack (https://github.com/creationix/stack).
- *
- * Create a simple middleware stack by calling:
- *
- * cli.createServer(middleware).listen(port);
- *
- * @return {Server} server
- * @api public
- */
- cli.createServer = function(/*layers*/) {
- var defaultStackErrorHandler = function (req, res, err) {
- if (err) {
- console.error(err.stack);
- res.writeHead(500, {"Content-Type": "text/plain"});
- return res.end(err.stack + "\n");
- }
- res.writeHead(404, {"Content-Type": "text/plain"});
- res.end("Not Found\n");
- };
- var handle, error;
- handle = error = defaultStackErrorHandler;
- var layers = Array.prototype.slice.call(arguments);
- //Allow createServer(a,b,c) and createServer([a,b,c])
- if (layers.length && layers[0] instanceof Array) {
- layers = layers[0];
- }
- layers.reverse().forEach(function (layer) {
- var child = handle;
- handle = function (req, res) {
- try {
- layer(req, res, function (err) {
- if (err) return error(req, res, err);
- child(req, res);
- });
- } catch (err) {
- error(req, res, err);
- }
- };
- });
- return cli.native.http.createServer(handle);
- };
- /**
- * A wrapper for child_process.exec().
- *
- * If the child_process exits successfully, `callback` receives an array of
- * stdout lines. The current process exits if the child process has an error
- * and `errback` isn't defined.
- *
- * @param {String} cmd
- * @param {Function} callback (optional)
- * @param {Function} errback (optional)
- * @api public
- */
- cli.exec = function (cmd, callback, errback) {
- cli.native.child_process.exec(cmd, function (err, stdout, stderr) {
- err = err || stderr;
- if (err) {
- if (errback) {
- return errback(err, stdout);
- }
- return cli.fatal('exec() failed\n' + err);
- }
- if (callback) {
- callback(stdout.split('\n'));
- }
- });
- };
- /**
- * Helper method for outputting a progress bar to the console.
- *
- * @param {Number} progress (0 <= progress <= 1)
- * @api public
- */
- var last_progress_call, progress_len = 74, min_progress_increase = 5, last_progress_percentage = 0;
- cli.progress = function (progress, decimals, stream) {
- stream = stream || process.stdout;
- if (progress < 0 || progress > 1 || isNaN(progress)) return;
- if (!decimals) decimals = 0;
- var now = (new Date()).getTime();
- if (last_progress_call && (now - last_progress_call) < 100 && progress !== 1) {
- return; //Throttle progress calls
- }
- last_progress_call = now;
- var pwr = Math.pow(10, decimals);
- var percentage_as_num = Math.floor(progress * 100 * pwr) / pwr;
- if (!stream.isTTY && percentage_as_num < 100 && percentage_as_num - last_progress_percentage < min_progress_increase) {
- return; //don't over-print if not TTY
- }
- last_progress_percentage = percentage_as_num;
- var percentage = percentage_as_num + '%';
- for (var i = 0; i < decimals; i++) {
- percentage += ' ';
- }
- if (!stream.isTTY) {
- if (percentage_as_num < 100) {
- stream.write(percentage + '...');
- }
- else {
- stream.write(percentage + '\n');
- last_progress_percentage = 0;
- }
- return;
- }
- var bar_length = Math.floor(progress_len * progress),
- str = '';
- if (bar_length == 0 && progress > 0) {
- bar_length = 1;
- }
- for (i = 1; i <= progress_len; i++) {
- str += i <= bar_length ? '#' : ' ';
- }
- stream.clearLine();
- stream.write('[' + str + '] ' + percentage);
- if (progress === 1) {
- stream.write('\n');
- } else {
- stream.cursorTo(0);
- }
- };
- /**
- * Helper method for outputting a spinner to the console.
- *
- * @param {String|Boolean} prefix (optional)
- * @api public
- */
- var spinner_interval;
- cli.spinner = function (prefix, end, stream) {
- stream = stream || process.stdout;
- if(!stream.isTTY) {
- stream.write(prefix + '\n');
- return;
- }
- if (end) {
- stream.clearLine();
- stream.cursorTo(0);
- stream.write(prefix + '\n');
- return clearInterval(spinner_interval);
- }
- prefix = prefix + ' ' || '';
- var spinner = ['-','\\','|','/'], i = 0, l = spinner.length;
- spinner_interval = setInterval(function () {
- stream.clearLine();
- stream.cursorTo(0);
- stream.write(prefix + spinner[i++]);
- if (i == l) i = 0;
- }, 200);
- };
|