You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					
					
						
							1055 lines
						
					
					
						
							34 KiB
						
					
					
				
			
		
		
		
			
			
			
		
		
	
	
							1055 lines
						
					
					
						
							34 KiB
						
					
					
				| /** | |
|  * dashdash - A light, featureful and explicit option parsing library for | |
|  * node.js. | |
|  */ | |
| // vim: set ts=4 sts=4 sw=4 et: | |
|  | |
| var assert = require('assert-plus'); | |
| var format = require('util').format; | |
| var fs = require('fs'); | |
| var path = require('path'); | |
| 
 | |
| 
 | |
| var DEBUG = true; | |
| if (DEBUG) { | |
|     var debug = console.warn; | |
| } else { | |
|     var debug = function () {}; | |
| } | |
| 
 | |
| 
 | |
| 
 | |
| // ---- internal support stuff | |
|  | |
| // Replace {{variable}} in `s` with the template data in `d`. | |
| function renderTemplate(s, d) { | |
|     return s.replace(/{{([a-zA-Z]+)}}/g, function (match, key) { | |
|         return d.hasOwnProperty(key) ? d[key] : match; | |
|     }); | |
| } | |
| 
 | |
| /** | |
|  * Return a shallow copy of the given object; | |
|  */ | |
| function shallowCopy(obj) { | |
|     if (!obj) { | |
|         return (obj); | |
|     } | |
|     var copy = {}; | |
|     Object.keys(obj).forEach(function (k) { | |
|         copy[k] = obj[k]; | |
|     }); | |
|     return (copy); | |
| } | |
| 
 | |
| 
 | |
| function space(n) { | |
|     var s = ''; | |
|     for (var i = 0; i < n; i++) { | |
|         s += ' '; | |
|     } | |
|     return s; | |
| } | |
| 
 | |
| 
 | |
| function makeIndent(arg, deflen, name) { | |
|     if (arg === null || arg === undefined) | |
|         return space(deflen); | |
|     else if (typeof (arg) === 'number') | |
|         return space(arg); | |
|     else if (typeof (arg) === 'string') | |
|         return arg; | |
|     else | |
|         assert.fail('invalid "' + name + '": not a string or number: ' + arg); | |
| } | |
| 
 | |
| 
 | |
| /** | |
|  * Return an array of lines wrapping the given text to the given width. | |
|  * This splits on whitespace. Single tokens longer than `width` are not | |
|  * broken up. | |
|  */ | |
| function textwrap(s, width) { | |
|     var words = s.trim().split(/\s+/); | |
|     var lines = []; | |
|     var line = ''; | |
|     words.forEach(function (w) { | |
|         var newLength = line.length + w.length; | |
|         if (line.length > 0) | |
|             newLength += 1; | |
|         if (newLength > width) { | |
|             lines.push(line); | |
|             line = ''; | |
|         } | |
|         if (line.length > 0) | |
|             line += ' '; | |
|         line += w; | |
|     }); | |
|     lines.push(line); | |
|     return lines; | |
| } | |
| 
 | |
| 
 | |
| /** | |
|  * Transform an option name to a "key" that is used as the field | |
|  * on the `opts` object returned from `<parser>.parse()`. | |
|  * | |
|  * Transformations: | |
|  * - '-' -> '_': This allow one to use hyphen in option names (common) | |
|  *   but not have to do silly things like `opt["dry-run"]` to access the | |
|  *   parsed results. | |
|  */ | |
| function optionKeyFromName(name) { | |
|     return name.replace(/-/g, '_'); | |
| } | |
| 
 | |
| 
 | |
| 
 | |
| // ---- Option types | |
|  | |
| function parseBool(option, optstr, arg) { | |
|     return Boolean(arg); | |
| } | |
| 
 | |
| function parseString(option, optstr, arg) { | |
|     assert.string(arg, 'arg'); | |
|     return arg; | |
| } | |
| 
 | |
| function parseNumber(option, optstr, arg) { | |
|     assert.string(arg, 'arg'); | |
|     var num = Number(arg); | |
|     if (isNaN(num)) { | |
|         throw new Error(format('arg for "%s" is not a number: "%s"', | |
|             optstr, arg)); | |
|     } | |
|     return num; | |
| } | |
| 
 | |
| function parseInteger(option, optstr, arg) { | |
|     assert.string(arg, 'arg'); | |
|     var num = Number(arg); | |
|     if (!/^[0-9-]+$/.test(arg) || isNaN(num)) { | |
|         throw new Error(format('arg for "%s" is not an integer: "%s"', | |
|             optstr, arg)); | |
|     } | |
|     return num; | |
| } | |
| 
 | |
| function parsePositiveInteger(option, optstr, arg) { | |
|     assert.string(arg, 'arg'); | |
|     var num = Number(arg); | |
|     if (!/^[0-9]+$/.test(arg) || isNaN(num) || num === 0) { | |
|         throw new Error(format('arg for "%s" is not a positive integer: "%s"', | |
|             optstr, arg)); | |
|     } | |
|     return num; | |
| } | |
| 
 | |
| /** | |
|  * Supported date args: | |
|  * - epoch second times (e.g. 1396031701) | |
|  * - ISO 8601 format: YYYY-MM-DD[THH:MM:SS[.sss][Z]] | |
|  *      2014-03-28T18:35:01.489Z | |
|  *      2014-03-28T18:35:01.489 | |
|  *      2014-03-28T18:35:01Z | |
|  *      2014-03-28T18:35:01 | |
|  *      2014-03-28 | |
|  */ | |
| function parseDate(option, optstr, arg) { | |
|     assert.string(arg, 'arg'); | |
|     var date; | |
|     if (/^\d+$/.test(arg)) { | |
|         // epoch seconds | |
|         date = new Date(Number(arg) * 1000); | |
|     /* JSSTYLED */ | |
|     } else if (/^\d{4}-\d{2}-\d{2}(T\d{2}:\d{2}:\d{2}(\.\d+)?Z?)?$/i.test(arg)) { | |
|         // ISO 8601 format | |
|         date = new Date(arg); | |
|     } else { | |
|         throw new Error(format('arg for "%s" is not a valid date format: "%s"', | |
|             optstr, arg)); | |
|     } | |
|     if (date.toString() === 'Invalid Date') { | |
|         throw new Error(format('arg for "%s" is an invalid date: "%s"', | |
|             optstr, arg)); | |
|     } | |
|     return date; | |
| } | |
| 
 | |
| var optionTypes = { | |
|     bool: { | |
|         takesArg: false, | |
|         parseArg: parseBool | |
|     }, | |
|     string: { | |
|         takesArg: true, | |
|         helpArg: 'ARG', | |
|         parseArg: parseString | |
|     }, | |
|     number: { | |
|         takesArg: true, | |
|         helpArg: 'NUM', | |
|         parseArg: parseNumber | |
|     }, | |
|     integer: { | |
|         takesArg: true, | |
|         helpArg: 'INT', | |
|         parseArg: parseInteger | |
|     }, | |
|     positiveInteger: { | |
|         takesArg: true, | |
|         helpArg: 'INT', | |
|         parseArg: parsePositiveInteger | |
|     }, | |
|     date: { | |
|         takesArg: true, | |
|         helpArg: 'DATE', | |
|         parseArg: parseDate | |
|     }, | |
|     arrayOfBool: { | |
|         takesArg: false, | |
|         array: true, | |
|         parseArg: parseBool | |
|     }, | |
|     arrayOfString: { | |
|         takesArg: true, | |
|         helpArg: 'ARG', | |
|         array: true, | |
|         parseArg: parseString | |
|     }, | |
|     arrayOfNumber: { | |
|         takesArg: true, | |
|         helpArg: 'NUM', | |
|         array: true, | |
|         parseArg: parseNumber | |
|     }, | |
|     arrayOfInteger: { | |
|         takesArg: true, | |
|         helpArg: 'INT', | |
|         array: true, | |
|         parseArg: parseInteger | |
|     }, | |
|     arrayOfPositiveInteger: { | |
|         takesArg: true, | |
|         helpArg: 'INT', | |
|         array: true, | |
|         parseArg: parsePositiveInteger | |
|     }, | |
|     arrayOfDate: { | |
|         takesArg: true, | |
|         helpArg: 'INT', | |
|         array: true, | |
|         parseArg: parseDate | |
|     }, | |
| }; | |
| 
 | |
| 
 | |
| 
 | |
| // ---- Parser | |
|  | |
| /** | |
|  * Parser constructor. | |
|  * | |
|  * @param config {Object} The parser configuration | |
|  *      - options {Array} Array of option specs. See the README for how to | |
|  *        specify each option spec. | |
|  *      - allowUnknown {Boolean} Default false. Whether to throw on unknown | |
|  *        options. If false, then unknown args are included in the _args array. | |
|  *      - interspersed {Boolean} Default true. Whether to allow interspersed | |
|  *        arguments (non-options) and options. E.g.: | |
|  *              node tool.js arg1 arg2 -v | |
|  *        '-v' is after some args here. If `interspersed: false` then '-v' | |
|  *        would not be parsed out. Note that regardless of `interspersed` | |
|  *        the presence of '--' will stop option parsing, as all good | |
|  *        option parsers should. | |
|  */ | |
| function Parser(config) { | |
|     assert.object(config, 'config'); | |
|     assert.arrayOfObject(config.options, 'config.options'); | |
|     assert.optionalBool(config.interspersed, 'config.interspersed'); | |
|     var self = this; | |
| 
 | |
|     // Allow interspersed arguments (true by default). | |
|     this.interspersed = (config.interspersed !== undefined | |
|         ? config.interspersed : true); | |
| 
 | |
|     // Don't allow unknown flags (true by default). | |
|     this.allowUnknown = (config.allowUnknown !== undefined | |
|         ? config.allowUnknown : false); | |
| 
 | |
|     this.options = config.options.map(function (o) { return shallowCopy(o); }); | |
|     this.optionFromName = {}; | |
|     this.optionFromEnv = {}; | |
|     for (var i = 0; i < this.options.length; i++) { | |
|         var o = this.options[i]; | |
|         if (o.group !== undefined && o.group !== null) { | |
|             assert.optionalString(o.group, | |
|                 format('config.options.%d.group', i)); | |
|             continue; | |
|         } | |
|         assert.ok(optionTypes[o.type], | |
|             format('invalid config.options.%d.type: "%s" in %j', | |
|                    i, o.type, o)); | |
|         assert.optionalString(o.name, format('config.options.%d.name', i)); | |
|         assert.optionalArrayOfString(o.names, | |
|             format('config.options.%d.names', i)); | |
|         assert.ok((o.name || o.names) && !(o.name && o.names), | |
|             format('exactly one of "name" or "names" required: %j', o)); | |
|         assert.optionalString(o.help, format('config.options.%d.help', i)); | |
|         var env = o.env || []; | |
|         if (typeof (env) === 'string') { | |
|             env = [env]; | |
|         } | |
|         assert.optionalArrayOfString(env, format('config.options.%d.env', i)); | |
|         assert.optionalString(o.helpGroup, | |
|             format('config.options.%d.helpGroup', i)); | |
|         assert.optionalBool(o.helpWrap, | |
|             format('config.options.%d.helpWrap', i)); | |
|         assert.optionalBool(o.hidden, format('config.options.%d.hidden', i)); | |
| 
 | |
|         if (o.name) { | |
|             o.names = [o.name]; | |
|         } else { | |
|             assert.string(o.names[0], | |
|                 format('config.options.%d.names is empty', i)); | |
|         } | |
|         o.key = optionKeyFromName(o.names[0]); | |
|         o.names.forEach(function (n) { | |
|             if (self.optionFromName[n]) { | |
|                 throw new Error(format( | |
|                     'option name collision: "%s" used in %j and %j', | |
|                     n, self.optionFromName[n], o)); | |
|             } | |
|             self.optionFromName[n] = o; | |
|         }); | |
|         env.forEach(function (n) { | |
|             if (self.optionFromEnv[n]) { | |
|                 throw new Error(format( | |
|                     'option env collision: "%s" used in %j and %j', | |
|                     n, self.optionFromEnv[n], o)); | |
|             } | |
|             self.optionFromEnv[n] = o; | |
|         }); | |
|     } | |
| } | |
| 
 | |
| Parser.prototype.optionTakesArg = function optionTakesArg(option) { | |
|     return optionTypes[option.type].takesArg; | |
| }; | |
| 
 | |
| /** | |
|  * Parse options from the given argv. | |
|  * | |
|  * @param inputs {Object} Optional. | |
|  *      - argv {Array} Optional. The argv to parse. Defaults to | |
|  *        `process.argv`. | |
|  *      - slice {Number} The index into argv at which options/args begin. | |
|  *        Default is 2, as appropriate for `process.argv`. | |
|  *      - env {Object} Optional. The env to use for 'env' entries in the | |
|  *        option specs. Defaults to `process.env`. | |
|  * @returns {Object} Parsed `opts`. It has special keys `_args` (the | |
|  *      remaining args from `argv`) and `_order` (gives the order that | |
|  *      options were specified). | |
|  */ | |
| Parser.prototype.parse = function parse(inputs) { | |
|     var self = this; | |
| 
 | |
|     // Old API was `parse([argv, [slice]])` | |
|     if (Array.isArray(arguments[0])) { | |
|         inputs = {argv: arguments[0], slice: arguments[1]}; | |
|     } | |
| 
 | |
|     assert.optionalObject(inputs, 'inputs'); | |
|     if (!inputs) { | |
|         inputs = {}; | |
|     } | |
|     assert.optionalArrayOfString(inputs.argv, 'inputs.argv'); | |
|     //assert.optionalNumber(slice, 'slice'); | |
|     var argv = inputs.argv || process.argv; | |
|     var slice = inputs.slice !== undefined ? inputs.slice : 2; | |
|     var args = argv.slice(slice); | |
|     var env = inputs.env || process.env; | |
|     var opts = {}; | |
|     var _order = []; | |
| 
 | |
|     function addOpt(option, optstr, key, val, from) { | |
|         var type = optionTypes[option.type]; | |
|         var parsedVal = type.parseArg(option, optstr, val); | |
|         if (type.array) { | |
|             if (!opts[key]) { | |
|                 opts[key] = []; | |
|             } | |
|             if (type.arrayFlatten && Array.isArray(parsedVal)) { | |
|                 for (var i = 0; i < parsedVal.length; i++) { | |
|                     opts[key].push(parsedVal[i]); | |
|                 } | |
|             } else { | |
|                 opts[key].push(parsedVal); | |
|             } | |
|         } else { | |
|             opts[key] = parsedVal; | |
|         } | |
|         var item = { key: key, value: parsedVal, from: from }; | |
|         _order.push(item); | |
|     } | |
| 
 | |
|     // Parse args. | |
|     var _args = []; | |
|     var i = 0; | |
|     outer: while (i < args.length) { | |
|         var arg = args[i]; | |
| 
 | |
|         // End of options marker. | |
|         if (arg === '--') { | |
|             i++; | |
|             break; | |
| 
 | |
|         // Long option | |
|         } else if (arg.slice(0, 2) === '--') { | |
|             var name = arg.slice(2); | |
|             var val = null; | |
|             var idx = name.indexOf('='); | |
|             if (idx !== -1) { | |
|                 val = name.slice(idx + 1); | |
|                 name = name.slice(0, idx); | |
|             } | |
|             var option = this.optionFromName[name]; | |
|             if (!option) { | |
|                 if (!this.allowUnknown) | |
|                     throw new Error(format('unknown option: "--%s"', name)); | |
|                 else if (this.interspersed) | |
|                     _args.push(arg); | |
|                 else | |
|                     break outer; | |
|             } else { | |
|                 var takesArg = this.optionTakesArg(option); | |
|                 if (val !== null && !takesArg) { | |
|                     throw new Error(format('argument given to "--%s" option ' | |
|                         + 'that does not take one: "%s"', name, arg)); | |
|                 } | |
|                 if (!takesArg) { | |
|                     addOpt(option, '--'+name, option.key, true, 'argv'); | |
|                 } else if (val !== null) { | |
|                     addOpt(option, '--'+name, option.key, val, 'argv'); | |
|                 } else if (i + 1 >= args.length) { | |
|                     throw new Error(format('do not have enough args for "--%s" ' | |
|                         + 'option', name)); | |
|                 } else { | |
|                     addOpt(option, '--'+name, option.key, args[i + 1], 'argv'); | |
|                     i++; | |
|                 } | |
|             } | |
| 
 | |
|         // Short option | |
|         } else if (arg[0] === '-' && arg.length > 1) { | |
|             var j = 1; | |
|             var allFound = true; | |
|             while (j < arg.length) { | |
|                 var name = arg[j]; | |
|                 var option = this.optionFromName[name]; | |
|                 if (!option) { | |
|                     allFound = false; | |
|                     if (this.allowUnknown) { | |
|                         if (this.interspersed) { | |
|                             _args.push(arg); | |
|                             break; | |
|                         } else | |
|                             break outer; | |
|                     } else if (arg.length > 2) { | |
|                         throw new Error(format( | |
|                             'unknown option: "-%s" in "%s" group', | |
|                             name, arg)); | |
|                     } else { | |
|                         throw new Error(format('unknown option: "-%s"', name)); | |
|                     } | |
|                 } else if (this.optionTakesArg(option)) { | |
|                     break; | |
|                 } | |
|                 j++; | |
|             } | |
| 
 | |
|             j = 1; | |
|             while (allFound && j < arg.length) { | |
|                 var name = arg[j]; | |
|                 var val = arg.slice(j + 1);  // option val if it takes an arg | |
|                 var option = this.optionFromName[name]; | |
|                 var takesArg = this.optionTakesArg(option); | |
|                 if (!takesArg) { | |
|                     addOpt(option, '-'+name, option.key, true, 'argv'); | |
|                 } else if (val) { | |
|                     addOpt(option, '-'+name, option.key, val, 'argv'); | |
|                     break; | |
|                 } else { | |
|                     if (i + 1 >= args.length) { | |
|                         throw new Error(format('do not have enough args ' | |
|                             + 'for "-%s" option', name)); | |
|                     } | |
|                     addOpt(option, '-'+name, option.key, args[i + 1], 'argv'); | |
|                     i++; | |
|                     break; | |
|                 } | |
|                 j++; | |
|             } | |
| 
 | |
|         // An interspersed arg | |
|         } else if (this.interspersed) { | |
|             _args.push(arg); | |
| 
 | |
|         // An arg and interspersed args are not allowed, so done options. | |
|         } else { | |
|             break outer; | |
|         } | |
|         i++; | |
|     } | |
|     _args = _args.concat(args.slice(i)); | |
| 
 | |
|     // Parse environment. | |
|     Object.keys(this.optionFromEnv).forEach(function (envname) { | |
|         var val = env[envname]; | |
|         if (val === undefined) | |
|             return; | |
|         var option = self.optionFromEnv[envname]; | |
|         if (opts[option.key] !== undefined) | |
|             return; | |
|         var takesArg = self.optionTakesArg(option); | |
|         if (takesArg) { | |
|             addOpt(option, envname, option.key, val, 'env'); | |
|         } else if (val !== '') { | |
|             // Boolean envvar handling: | |
|             // - VAR=<empty-string>     not set (as if the VAR was not set) | |
|             // - VAR=0                  false | |
|             // - anything else          true | |
|             addOpt(option, envname, option.key, (val !== '0'), 'env'); | |
|         } | |
|     }); | |
| 
 | |
|     // Apply default values. | |
|     this.options.forEach(function (o) { | |
|         if (opts[o.key] === undefined) { | |
|             if (o.default !== undefined) { | |
|                 opts[o.key] = o.default; | |
|             } else if (o.type && optionTypes[o.type].default !== undefined) { | |
|                 opts[o.key] = optionTypes[o.type].default; | |
|             } | |
|         } | |
|     }); | |
| 
 | |
|     opts._order = _order; | |
|     opts._args = _args; | |
|     return opts; | |
| }; | |
| 
 | |
| 
 | |
| /** | |
|  * Return help output for the current options. | |
|  * | |
|  * E.g.: if the current options are: | |
|  *      [{names: ['help', 'h'], type: 'bool', help: 'Show help and exit.'}] | |
|  * then this would return: | |
|  *      '  -h, --help     Show help and exit.\n' | |
|  * | |
|  * @param config {Object} Config for controlling the option help output. | |
|  *      - indent {Number|String} Default 4. An indent/prefix to use for | |
|  *        each option line. | |
|  *      - nameSort {String} Default is 'length'. By default the names are | |
|  *        sorted to put the short opts first (i.e. '-h, --help' preferred | |
|  *        to '--help, -h'). Set to 'none' to not do this sorting. | |
|  *      - maxCol {Number} Default 80. Note that long tokens in a help string | |
|  *        can go past this. | |
|  *      - helpCol {Number} Set to specify a specific column at which | |
|  *        option help will be aligned. By default this is determined | |
|  *        automatically. | |
|  *      - minHelpCol {Number} Default 20. | |
|  *      - maxHelpCol {Number} Default 40. | |
|  *      - includeEnv {Boolean} Default false. If true, a note stating the `env` | |
|  *        envvar (if specified for this option) will be appended to the help | |
|  *        output. | |
|  *      - includeDefault {Boolean} Default false. If true, a note stating | |
|  *        the `default` for this option, if any, will be appended to the help | |
|  *        output. | |
|  *      - helpWrap {Boolean} Default true. Wrap help text in helpCol..maxCol | |
|  *        bounds. | |
|  * @returns {String} | |
|  */ | |
| Parser.prototype.help = function help(config) { | |
|     config = config || {}; | |
|     assert.object(config, 'config'); | |
| 
 | |
|     var indent = makeIndent(config.indent, 4, 'config.indent'); | |
|     var headingIndent = makeIndent(config.headingIndent, | |
|         Math.round(indent.length / 2), 'config.headingIndent'); | |
| 
 | |
|     assert.optionalString(config.nameSort, 'config.nameSort'); | |
|     var nameSort = config.nameSort || 'length'; | |
|     assert.ok(~['length', 'none'].indexOf(nameSort), | |
|         'invalid "config.nameSort"'); | |
|     assert.optionalNumber(config.maxCol, 'config.maxCol'); | |
|     assert.optionalNumber(config.maxHelpCol, 'config.maxHelpCol'); | |
|     assert.optionalNumber(config.minHelpCol, 'config.minHelpCol'); | |
|     assert.optionalNumber(config.helpCol, 'config.helpCol'); | |
|     assert.optionalBool(config.includeEnv, 'config.includeEnv'); | |
|     assert.optionalBool(config.includeDefault, 'config.includeDefault'); | |
|     assert.optionalBool(config.helpWrap, 'config.helpWrap'); | |
|     var maxCol = config.maxCol || 80; | |
|     var minHelpCol = config.minHelpCol || 20; | |
|     var maxHelpCol = config.maxHelpCol || 40; | |
| 
 | |
|     var lines = []; | |
|     var maxWidth = 0; | |
|     this.options.forEach(function (o) { | |
|         if (o.hidden) { | |
|             return; | |
|         } | |
|         if (o.group !== undefined && o.group !== null) { | |
|             // We deal with groups in the next pass | |
|             lines.push(null); | |
|             return; | |
|         } | |
|         var type = optionTypes[o.type]; | |
|         var arg = o.helpArg || type.helpArg || 'ARG'; | |
|         var line = ''; | |
|         var names = o.names.slice(); | |
|         if (nameSort === 'length') { | |
|             names.sort(function (a, b) { | |
|                 if (a.length < b.length) | |
|                     return -1; | |
|                 else if (b.length < a.length) | |
|                     return 1; | |
|                 else | |
|                     return 0; | |
|             }) | |
|         } | |
|         names.forEach(function (name, i) { | |
|             if (i > 0) | |
|                 line += ', '; | |
|             if (name.length === 1) { | |
|                 line += '-' + name | |
|                 if (type.takesArg) | |
|                     line += ' ' + arg; | |
|             } else { | |
|                 line += '--' + name | |
|                 if (type.takesArg) | |
|                     line += '=' + arg; | |
|             } | |
|         }); | |
|         maxWidth = Math.max(maxWidth, line.length); | |
|         lines.push(line); | |
|     }); | |
| 
 | |
|     // Add help strings. | |
|     var helpCol = config.helpCol; | |
|     if (!helpCol) { | |
|         helpCol = maxWidth + indent.length + 2; | |
|         helpCol = Math.min(Math.max(helpCol, minHelpCol), maxHelpCol); | |
|     } | |
|     var i = -1; | |
|     this.options.forEach(function (o) { | |
|         if (o.hidden) { | |
|             return; | |
|         } | |
|         i++; | |
| 
 | |
|         if (o.group !== undefined && o.group !== null) { | |
|             if (o.group === '') { | |
|                 // Support a empty string "group" to have a blank line between | |
|                 // sets of options. | |
|                 lines[i] = ''; | |
|             } else { | |
|                 // Render the group heading with the heading-specific indent. | |
|                 lines[i] = (i === 0 ? '' : '\n') + headingIndent + | |
|                     o.group + ':'; | |
|             } | |
|             return; | |
|         } | |
| 
 | |
|         var helpDefault; | |
|         if (config.includeDefault) { | |
|             if (o.default !== undefined) { | |
|                 helpDefault = format('Default: %j', o.default); | |
|             } else if (o.type && optionTypes[o.type].default !== undefined) { | |
|                 helpDefault = format('Default: %j', | |
|                     optionTypes[o.type].default); | |
|             } | |
|         } | |
| 
 | |
|         var line = lines[i] = indent + lines[i]; | |
|         if (!o.help && !(config.includeEnv && o.env) && !helpDefault) { | |
|             return; | |
|         } | |
|         var n = helpCol - line.length; | |
|         if (n >= 0) { | |
|             line += space(n); | |
|         } else { | |
|             line += '\n' + space(helpCol); | |
|         } | |
| 
 | |
|         var helpEnv = ''; | |
|         if (o.env && o.env.length && config.includeEnv) { | |
|             helpEnv += 'Environment: '; | |
|             var type = optionTypes[o.type]; | |
|             var arg = o.helpArg || type.helpArg || 'ARG'; | |
|             var envs = (Array.isArray(o.env) ? o.env : [o.env]).map( | |
|                 function (e) { | |
|                     if (type.takesArg) { | |
|                         return e + '=' + arg; | |
|                     } else { | |
|                         return e + '=1'; | |
|                     } | |
|                 } | |
|             ); | |
|             helpEnv += envs.join(', '); | |
|         } | |
|         var help = (o.help || '').trim(); | |
|         if (o.helpWrap !== false && config.helpWrap !== false) { | |
|             // Wrap help description normally. | |
|             if (help.length && !~'.!?"\''.indexOf(help.slice(-1))) { | |
|                 help += '.'; | |
|             } | |
|             if (help.length) { | |
|                 help += ' '; | |
|             } | |
|             help += helpEnv; | |
|             if (helpDefault) { | |
|                 if (helpEnv) { | |
|                     help += '. '; | |
|                 } | |
|                 help += helpDefault; | |
|             } | |
|             line += textwrap(help, maxCol - helpCol).join( | |
|                 '\n' + space(helpCol)); | |
|         } else { | |
|             // Do not wrap help description, but indent newlines appropriately. | |
|             var helpLines = help.split('\n').filter( | |
|                     function (ln) { return ln.length }); | |
|             if (helpEnv !== '') { | |
|                 helpLines.push(helpEnv); | |
|             } | |
|             if (helpDefault) { | |
|                 helpLines.push(helpDefault); | |
|             } | |
|             line += helpLines.join('\n' + space(helpCol)); | |
|         } | |
| 
 | |
|         lines[i] = line; | |
|     }); | |
| 
 | |
|     var rv = ''; | |
|     if (lines.length > 0) { | |
|         rv = lines.join('\n') + '\n'; | |
|     } | |
|     return rv; | |
| }; | |
| 
 | |
| 
 | |
| /** | |
|  * Return a string suitable for a Bash completion file for this tool. | |
|  * | |
|  * @param args.name {String} The tool name. | |
|  * @param args.specExtra {String} Optional. Extra Bash code content to add | |
|  *      to the end of the "spec". Typically this is used to append Bash | |
|  *      "complete_TYPE" functions for custom option types. See | |
|  *      "examples/ddcompletion.js" for an example. | |
|  * @param args.argtypes {Array} Optional. Array of completion types for | |
|  *      positional args (i.e. non-options). E.g. | |
|  *          argtypes = ['fruit', 'veggie', 'file'] | |
|  *      will result in completion of fruits for the first arg, veggies for the | |
|  *      second, and filenames for the third and subsequent positional args. | |
|  *      If not given, positional args will use Bash's 'default' completion. | |
|  *      See `specExtra` for providing Bash `complete_TYPE` functions, e.g. | |
|  *      `complete_fruit` and `complete_veggie` in this example. | |
|  */ | |
| Parser.prototype.bashCompletion = function bashCompletion(args) { | |
|     assert.object(args, 'args'); | |
|     assert.string(args.name, 'args.name'); | |
|     assert.optionalString(args.specExtra, 'args.specExtra'); | |
|     assert.optionalArrayOfString(args.argtypes, 'args.argtypes'); | |
| 
 | |
|     return bashCompletionFromOptions({ | |
|         name: args.name, | |
|         specExtra: args.specExtra, | |
|         argtypes: args.argtypes, | |
|         options: this.options | |
|     }); | |
| }; | |
| 
 | |
| 
 | |
| // ---- Bash completion | |
|  | |
| const BASH_COMPLETION_TEMPLATE_PATH = path.join( | |
|     __dirname, '../etc/dashdash.bash_completion.in'); | |
| 
 | |
| /** | |
|  * Return the Bash completion "spec" (the string value for the "{{spec}}" | |
|  * var in the "dashdash.bash_completion.in" template) for this tool. | |
|  * | |
|  * The "spec" is Bash code that defines the CLI options and subcmds for | |
|  * the template's completion code. It looks something like this: | |
|  * | |
|  *      local cmd_shortopts="-J ..." | |
|  *      local cmd_longopts="--help ..." | |
|  *      local cmd_optargs="-p=tritonprofile ..." | |
|  * | |
|  * @param args.options {Array} The array of dashdash option specs. | |
|  * @param args.context {String} Optional. A context string for the "local cmd*" | |
|  *      vars in the spec. By default it is the empty string. When used to | |
|  *      scope for completion on a *sub-command* (e.g. for "git log" on a "git" | |
|  *      tool), then it would have a value (e.g. "__log"). See | |
|  *      <http://github.com/trentm/node-cmdln> Bash completion for details. | |
|  * @param opts.includeHidden {Boolean} Optional. Default false. By default | |
|  *      hidden options and subcmds are "excluded". Here excluded means they | |
|  *      won't be offered as a completion, but if used, their argument type | |
|  *      will be completed. "Hidden" options and subcmds are ones with the | |
|  *      `hidden: true` attribute to exclude them from default help output. | |
|  * @param args.argtypes {Array} Optional. Array of completion types for | |
|  *      positional args (i.e. non-options). E.g. | |
|  *          argtypes = ['fruit', 'veggie', 'file'] | |
|  *      will result in completion of fruits for the first arg, veggies for the | |
|  *      second, and filenames for the third and subsequent positional args. | |
|  *      If not given, positional args will use Bash's 'default' completion. | |
|  *      See `specExtra` for providing Bash `complete_TYPE` functions, e.g. | |
|  *      `complete_fruit` and `complete_veggie` in this example. | |
|  */ | |
| function bashCompletionSpecFromOptions(args) { | |
|     assert.object(args, 'args'); | |
|     assert.object(args.options, 'args.options'); | |
|     assert.optionalString(args.context, 'args.context'); | |
|     assert.optionalBool(args.includeHidden, 'args.includeHidden'); | |
|     assert.optionalArrayOfString(args.argtypes, 'args.argtypes'); | |
| 
 | |
|     var context = args.context || ''; | |
|     var includeHidden = (args.includeHidden === undefined | |
|         ? false : args.includeHidden); | |
| 
 | |
|     var spec = []; | |
|     var shortopts = []; | |
|     var longopts = []; | |
|     var optargs = []; | |
|     (args.options || []).forEach(function (o) { | |
|         if (o.group !== undefined && o.group !== null) { | |
|             // Skip group headers. | |
|             return; | |
|         } | |
| 
 | |
|         var optNames = o.names || [o.name]; | |
|         var optType = getOptionType(o.type); | |
|         if (optType.takesArg) { | |
|             var completionType = o.completionType || | |
|                 optType.completionType || o.type; | |
|             optNames.forEach(function (optName) { | |
|                 if (optName.length === 1) { | |
|                     if (includeHidden || !o.hidden) { | |
|                         shortopts.push('-' + optName); | |
|                     } | |
|                     // Include even hidden options in `optargs` so that bash | |
|                     // completion of its arg still works. | |
|                     optargs.push('-' + optName + '=' + completionType); | |
|                 } else { | |
|                     if (includeHidden || !o.hidden) { | |
|                         longopts.push('--' + optName); | |
|                     } | |
|                     optargs.push('--' + optName + '=' + completionType); | |
|                 } | |
|             }); | |
|         } else { | |
|             optNames.forEach(function (optName) { | |
|                 if (includeHidden || !o.hidden) { | |
|                     if (optName.length === 1) { | |
|                         shortopts.push('-' + optName); | |
|                     } else { | |
|                         longopts.push('--' + optName); | |
|                     } | |
|                 } | |
|             }); | |
|         } | |
|     }); | |
| 
 | |
|     spec.push(format('local cmd%s_shortopts="%s"', | |
|         context, shortopts.sort().join(' '))); | |
|     spec.push(format('local cmd%s_longopts="%s"', | |
|         context, longopts.sort().join(' '))); | |
|     spec.push(format('local cmd%s_optargs="%s"', | |
|         context, optargs.sort().join(' '))); | |
|     if (args.argtypes) { | |
|         spec.push(format('local cmd%s_argtypes="%s"', | |
|             context, args.argtypes.join(' '))); | |
|     } | |
|     return spec.join('\n'); | |
| } | |
| 
 | |
| 
 | |
| /** | |
|  * Return a string suitable for a Bash completion file for this tool. | |
|  * | |
|  * @param args.name {String} The tool name. | |
|  * @param args.options {Array} The array of dashdash option specs. | |
|  * @param args.specExtra {String} Optional. Extra Bash code content to add | |
|  *      to the end of the "spec". Typically this is used to append Bash | |
|  *      "complete_TYPE" functions for custom option types. See | |
|  *      "examples/ddcompletion.js" for an example. | |
|  * @param args.argtypes {Array} Optional. Array of completion types for | |
|  *      positional args (i.e. non-options). E.g. | |
|  *          argtypes = ['fruit', 'veggie', 'file'] | |
|  *      will result in completion of fruits for the first arg, veggies for the | |
|  *      second, and filenames for the third and subsequent positional args. | |
|  *      If not given, positional args will use Bash's 'default' completion. | |
|  *      See `specExtra` for providing Bash `complete_TYPE` functions, e.g. | |
|  *      `complete_fruit` and `complete_veggie` in this example. | |
|  */ | |
| function bashCompletionFromOptions(args) { | |
|     assert.object(args, 'args'); | |
|     assert.object(args.options, 'args.options'); | |
|     assert.string(args.name, 'args.name'); | |
|     assert.optionalString(args.specExtra, 'args.specExtra'); | |
|     assert.optionalArrayOfString(args.argtypes, 'args.argtypes'); | |
| 
 | |
|     // Gather template data. | |
|     var data = { | |
|         name: args.name, | |
|         date: new Date(), | |
|         spec: bashCompletionSpecFromOptions({ | |
|             options: args.options, | |
|             argtypes: args.argtypes | |
|         }), | |
|     }; | |
|     if (args.specExtra) { | |
|         data.spec += '\n\n' + args.specExtra; | |
|     } | |
| 
 | |
|     // Render template. | |
|     var template = fs.readFileSync(BASH_COMPLETION_TEMPLATE_PATH, 'utf8'); | |
|     return renderTemplate(template, data); | |
| } | |
| 
 | |
| 
 | |
| 
 | |
| // ---- exports | |
|  | |
| function createParser(config) { | |
|     return new Parser(config); | |
| } | |
| 
 | |
| /** | |
|  * Parse argv with the given options. | |
|  * | |
|  * @param config {Object} A merge of all the available fields from | |
|  *      `dashdash.Parser` and `dashdash.Parser.parse`: options, interspersed, | |
|  *      argv, env, slice. | |
|  */ | |
| function parse(config) { | |
|     assert.object(config, 'config'); | |
|     assert.optionalArrayOfString(config.argv, 'config.argv'); | |
|     assert.optionalObject(config.env, 'config.env'); | |
|     var config = shallowCopy(config); | |
|     var argv = config.argv; | |
|     delete config.argv; | |
|     var env = config.env; | |
|     delete config.env; | |
| 
 | |
|     var parser = new Parser(config); | |
|     return parser.parse({argv: argv, env: env}); | |
| } | |
| 
 | |
| 
 | |
| /** | |
|  * Add a new option type. | |
|  * | |
|  * @params optionType {Object}: | |
|  *      - name {String} Required. | |
|  *      - takesArg {Boolean} Required. Whether this type of option takes an | |
|  *        argument on process.argv. Typically this is true for all but the | |
|  *        "bool" type. | |
|  *      - helpArg {String} Required iff `takesArg === true`. The string to | |
|  *        show in generated help for options of this type. | |
|  *      - parseArg {Function} Require. `function (option, optstr, arg)` parser | |
|  *        that takes a string argument and returns an instance of the | |
|  *        appropriate type, or throws an error if the arg is invalid. | |
|  *      - array {Boolean} Optional. Set to true if this is an 'arrayOf' type | |
|  *        that collects multiple usages of the option in process.argv and | |
|  *        puts results in an array. | |
|  *      - arrayFlatten {Boolean} Optional. XXX | |
|  *      - default Optional. Default value for options of this type, if no | |
|  *        default is specified in the option type usage. | |
|  */ | |
| function addOptionType(optionType) { | |
|     assert.object(optionType, 'optionType'); | |
|     assert.string(optionType.name, 'optionType.name'); | |
|     assert.bool(optionType.takesArg, 'optionType.takesArg'); | |
|     if (optionType.takesArg) { | |
|         assert.string(optionType.helpArg, 'optionType.helpArg'); | |
|     } | |
|     assert.func(optionType.parseArg, 'optionType.parseArg'); | |
|     assert.optionalBool(optionType.array, 'optionType.array'); | |
|     assert.optionalBool(optionType.arrayFlatten, 'optionType.arrayFlatten'); | |
| 
 | |
|     optionTypes[optionType.name] = { | |
|         takesArg: optionType.takesArg, | |
|         helpArg: optionType.helpArg, | |
|         parseArg: optionType.parseArg, | |
|         array: optionType.array, | |
|         arrayFlatten: optionType.arrayFlatten, | |
|         default: optionType.default | |
|     } | |
| } | |
| 
 | |
| 
 | |
| function getOptionType(name) { | |
|     assert.string(name, 'name'); | |
|     return optionTypes[name]; | |
| } | |
| 
 | |
| 
 | |
| /** | |
|  * Return a synopsis string for the given option spec. | |
|  * | |
|  * Examples: | |
|  *      > synopsisFromOpt({names: ['help', 'h'], type: 'bool'}); | |
|  *      '[ --help | -h ]' | |
|  *      > synopsisFromOpt({name: 'file', type: 'string', helpArg: 'FILE'}); | |
|  *      '[ --file=FILE ]' | |
|  */ | |
| function synopsisFromOpt(o) { | |
|     assert.object(o, 'o'); | |
| 
 | |
|     if (o.hasOwnProperty('group')) { | |
|         return null; | |
|     } | |
|     var names = o.names || [o.name]; | |
|     // `type` here could be undefined if, for example, the command has a | |
|     // dashdash option spec with a bogus 'type'. | |
|     var type = getOptionType(o.type); | |
|     var helpArg = o.helpArg || (type && type.helpArg) || 'ARG'; | |
|     var parts = []; | |
|     names.forEach(function (name) { | |
|         var part = (name.length === 1 ? '-' : '--') + name; | |
|         if (type && type.takesArg) { | |
|             part += (name.length === 1 ? ' ' + helpArg : '=' + helpArg); | |
|         } | |
|         parts.push(part); | |
|     }); | |
|     return ('[ ' + parts.join(' | ') + ' ]'); | |
| }; | |
| 
 | |
| 
 | |
| module.exports = { | |
|     createParser: createParser, | |
|     Parser: Parser, | |
|     parse: parse, | |
|     addOptionType: addOptionType, | |
|     getOptionType: getOptionType, | |
|     synopsisFromOpt: synopsisFromOpt, | |
| 
 | |
|     // Bash completion-related exports | |
|     BASH_COMPLETION_TEMPLATE_PATH: BASH_COMPLETION_TEMPLATE_PATH, | |
|     bashCompletionFromOptions: bashCompletionFromOptions, | |
|     bashCompletionSpecFromOptions: bashCompletionSpecFromOptions, | |
| 
 | |
|     // Export the parseFoo parsers because they might be useful as primitives | |
|     // for custom option types. | |
|     parseBool: parseBool, | |
|     parseString: parseString, | |
|     parseNumber: parseNumber, | |
|     parseInteger: parseInteger, | |
|     parsePositiveInteger: parsePositiveInteger, | |
|     parseDate: parseDate | |
| };
 | |
| 
 |