yargs.js 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898
  1. 'use strict';
  2. let dest;
  3. if (typeof WScript !== "undefined") {
  4. dest = this;
  5. } else if (typeof require !== "undefined") {
  6. dest = module.exports;
  7. }
  8. (function() {
  9. dest.yargsParse = parse;
  10. //The MIT License (MIT)
  11. //
  12. //Copyright (c) Sindre Sorhus <[email protected]> (sindresorhus.com)
  13. //
  14. //Permission is hereby granted, free of charge, to any person obtaining a copy
  15. //of this software and associated documentation files (the "Software"), to deal
  16. //in the Software without restriction, including without limitation the rights
  17. //to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  18. //copies of the Software, and to permit persons to whom the Software is
  19. //furnished to do so, subject to the following conditions:
  20. //
  21. //The above copyright notice and this permission notice shall be included in
  22. //all copies or substantial portions of the Software.
  23. //
  24. //THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  25. //IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  26. //FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  27. //AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  28. //LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  29. //OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  30. //THE SOFTWARE.
  31. function preserveCamelCase(str) {
  32. let isLastCharLower = false;
  33. let isLastCharUpper = false;
  34. let isLastLastCharUpper = false;
  35. for (let i = 0; i < str.length; i++) {
  36. const c = str[i];
  37. if (isLastCharLower && /[a-zA-Z]/.test(c) && c.toUpperCase() === c) {
  38. str = str.substr(0, i) + '-' + str.substr(i);
  39. isLastCharLower = false;
  40. isLastLastCharUpper = isLastCharUpper;
  41. isLastCharUpper = true;
  42. i++;
  43. } else if (isLastCharUpper && isLastLastCharUpper && /[a-zA-Z]/.test(c) && c.toLowerCase() === c) {
  44. str = str.substr(0, i - 1) + '-' + str.substr(i - 1);
  45. isLastLastCharUpper = isLastCharUpper;
  46. isLastCharUpper = false;
  47. isLastCharLower = true;
  48. } else {
  49. isLastCharLower = c.toLowerCase() === c;
  50. isLastLastCharUpper = isLastCharUpper;
  51. isLastCharUpper = c.toUpperCase() === c;
  52. }
  53. }
  54. return str;
  55. }
  56. function camelCase(str) {
  57. if (arguments.length > 1) {
  58. str = Array.from(arguments)
  59. .map(x => x.trim())
  60. .filter(x => x.length)
  61. .join('-');
  62. } else {
  63. str = str.trim();
  64. }
  65. if (str.length === 0) {
  66. return '';
  67. }
  68. if (str.length === 1) {
  69. return str.toLowerCase();
  70. }
  71. if (/^[a-z0-9]+$/.test(str)) {
  72. return str;
  73. }
  74. const hasUpperCase = str !== str.toLowerCase();
  75. if (hasUpperCase) {
  76. str = preserveCamelCase(str);
  77. }
  78. return str
  79. .replace(/^[_.\- ]+/, '')
  80. .toLowerCase()
  81. .replace(/[_.\- ]+(\w|$)/g, (m, p1) => p1.toUpperCase());
  82. };
  83. //Copyright (c) 2016, Contributors
  84. //
  85. //Permission to use, copy, modify, and/or distribute this software
  86. //for any purpose with or without fee is hereby granted, provided
  87. //that the above copyright notice and this permission notice
  88. //appear in all copies.
  89. //
  90. //THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  91. //WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES
  92. //OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE
  93. //LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES
  94. //OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
  95. //WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION,
  96. //ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  97. // take an un-split argv string and tokenize it.
  98. function tokenizeArgString(argString) {
  99. if (Array.isArray(argString)) return argString
  100. var i = 0
  101. var c = null
  102. var opening = null
  103. var args = []
  104. for (var ii = 0; ii < argString.length; ii++) {
  105. c = argString.charAt(ii)
  106. // split on spaces unless we're in quotes.
  107. if (c === ' ' && !opening) {
  108. i++
  109. continue
  110. }
  111. // don't split the string if we're in matching
  112. // opening or closing single and double quotes.
  113. if (c === opening) {
  114. opening = null
  115. continue
  116. } else if ((c === "'" || c === '"') && !opening) {
  117. opening = c
  118. continue
  119. }
  120. if (!args[i]) args[i] = ''
  121. args[i] += c
  122. }
  123. return args
  124. }
  125. function parse (args, opts) {
  126. if (!opts) opts = {}
  127. // allow a string argument to be passed in rather
  128. // than an argv array.
  129. args = tokenizeArgString(args)
  130. // aliases might have transitive relationships, normalize this.
  131. var aliases = combineAliases(opts.alias || {})
  132. var configuration = assign({
  133. 'short-option-groups': true,
  134. 'camel-case-expansion': true,
  135. 'dot-notation': true,
  136. 'parse-numbers': true,
  137. 'boolean-negation': true,
  138. 'duplicate-arguments-array': true,
  139. 'flatten-duplicate-arrays': true,
  140. 'populate--': false
  141. }, opts.configuration)
  142. var defaults = opts.default || {}
  143. var configObjects = opts.configObjects || []
  144. var envPrefix = opts.envPrefix
  145. var notFlagsOption = configuration['populate--']
  146. var notFlagsArgv = notFlagsOption ? '--' : '_'
  147. var newAliases = {}
  148. // allow a i18n handler to be passed in, default to a fake one (util.format).
  149. var __ = opts.__ || function (str) {
  150. return str;
  151. }
  152. var error = null
  153. var flags = {
  154. aliases: {},
  155. arrays: {},
  156. bools: {},
  157. strings: {},
  158. numbers: {},
  159. counts: {},
  160. normalize: {},
  161. configs: {},
  162. defaulted: {},
  163. nargs: {},
  164. coercions: {}
  165. }
  166. var negative = /^-[0-9]+(\.[0-9]+)?/
  167. ;[].concat(opts.array).filter(Boolean).forEach(function (key) {
  168. flags.arrays[key] = true
  169. })
  170. ;[].concat(opts.boolean).filter(Boolean).forEach(function (key) {
  171. flags.bools[key] = true
  172. })
  173. ;[].concat(opts.string).filter(Boolean).forEach(function (key) {
  174. flags.strings[key] = true
  175. })
  176. ;[].concat(opts.number).filter(Boolean).forEach(function (key) {
  177. flags.numbers[key] = true
  178. })
  179. ;[].concat(opts.count).filter(Boolean).forEach(function (key) {
  180. flags.counts[key] = true
  181. })
  182. ;[].concat(opts.normalize).filter(Boolean).forEach(function (key) {
  183. flags.normalize[key] = true
  184. })
  185. Object.keys(opts.narg || {}).forEach(function (k) {
  186. flags.nargs[k] = opts.narg[k]
  187. })
  188. Object.keys(opts.coerce || {}).forEach(function (k) {
  189. flags.coercions[k] = opts.coerce[k]
  190. })
  191. if (Array.isArray(opts.config) || typeof opts.config === 'string') {
  192. ;[].concat(opts.config).filter(Boolean).forEach(function (key) {
  193. flags.configs[key] = true
  194. })
  195. } else {
  196. Object.keys(opts.config || {}).forEach(function (k) {
  197. flags.configs[k] = opts.config[k]
  198. })
  199. }
  200. // create a lookup table that takes into account all
  201. // combinations of aliases: {f: ['foo'], foo: ['f']}
  202. extendAliases(opts.key, aliases, opts.default, flags.arrays)
  203. // apply default values to all aliases.
  204. Object.keys(defaults).forEach(function (key) {
  205. (flags.aliases[key] || []).forEach(function (alias) {
  206. defaults[alias] = defaults[key]
  207. })
  208. })
  209. var argv = { _: [] }
  210. Object.keys(flags.bools).forEach(function (key) {
  211. setArg(key, !(key in defaults) ? false : defaults[key])
  212. setDefaulted(key)
  213. })
  214. var notFlags = []
  215. if (args.indexOf('--') !== -1) {
  216. notFlags = args.slice(args.indexOf('--') + 1)
  217. args = args.slice(0, args.indexOf('--'))
  218. }
  219. for (var i = 0; i < args.length; i++) {
  220. var arg = args[i]
  221. var broken
  222. var key
  223. var letters
  224. var m
  225. var next
  226. var value
  227. // -- seperated by =
  228. if (arg.match(/^--.+=/) || (
  229. !configuration['short-option-groups'] && arg.match(/^-.+=/)
  230. )) {
  231. // Using [\s\S] instead of . because js doesn't support the
  232. // 'dotall' regex modifier. See:
  233. // http://stackoverflow.com/a/1068308/13216
  234. m = arg.match(/^--?([^=]+)=([\s\S]*)$/)
  235. // nargs format = '--f=monkey washing cat'
  236. if (checkAllAliases(m[1], flags.nargs)) {
  237. args.splice(i + 1, 0, m[2])
  238. i = eatNargs(i, m[1], args)
  239. // arrays format = '--f=a b c'
  240. } else if (checkAllAliases(m[1], flags.arrays) && args.length > i + 1) {
  241. args.splice(i + 1, 0, m[2])
  242. i = eatArray(i, m[1], args)
  243. } else {
  244. setArg(m[1], m[2])
  245. }
  246. } else if (arg.match(/^--no-.+/) && configuration['boolean-negation']) {
  247. key = arg.match(/^--no-(.+)/)[1]
  248. setArg(key, false)
  249. // -- seperated by space.
  250. } else if (arg.match(/^--.+/) || (
  251. !configuration['short-option-groups'] && arg.match(/^-.+/)
  252. )) {
  253. key = arg.match(/^--?(.+)/)[1]
  254. // nargs format = '--foo a b c'
  255. if (checkAllAliases(key, flags.nargs)) {
  256. i = eatNargs(i, key, args)
  257. // array format = '--foo a b c'
  258. } else if (checkAllAliases(key, flags.arrays) && args.length > i + 1) {
  259. i = eatArray(i, key, args)
  260. } else {
  261. next = args[i + 1]
  262. if (next !== undefined && (!next.match(/^-/) ||
  263. next.match(negative)) &&
  264. !checkAllAliases(key, flags.bools) &&
  265. !checkAllAliases(key, flags.counts)) {
  266. setArg(key, next)
  267. i++
  268. } else if (/^(true|false)$/.test(next)) {
  269. setArg(key, next)
  270. i++
  271. } else {
  272. setArg(key, defaultForType(guessType(key, flags)))
  273. }
  274. }
  275. // dot-notation flag seperated by '='.
  276. } else if (arg.match(/^-.\..+=/)) {
  277. m = arg.match(/^-([^=]+)=([\s\S]*)$/)
  278. setArg(m[1], m[2])
  279. // dot-notation flag seperated by space.
  280. } else if (arg.match(/^-.\..+/)) {
  281. next = args[i + 1]
  282. key = arg.match(/^-(.\..+)/)[1]
  283. if (next !== undefined && !next.match(/^-/) &&
  284. !checkAllAliases(key, flags.bools) &&
  285. !checkAllAliases(key, flags.counts)) {
  286. setArg(key, next)
  287. i++
  288. } else {
  289. setArg(key, defaultForType(guessType(key, flags)))
  290. }
  291. } else if (arg.match(/^-[^-]+/) && !arg.match(negative)) {
  292. letters = arg.slice(1, -1).split('')
  293. broken = false
  294. for (var j = 0; j < letters.length; j++) {
  295. next = arg.slice(j + 2)
  296. if (letters[j + 1] && letters[j + 1] === '=') {
  297. value = arg.slice(j + 3)
  298. key = letters[j]
  299. // nargs format = '-f=monkey washing cat'
  300. if (checkAllAliases(key, flags.nargs)) {
  301. args.splice(i + 1, 0, value)
  302. i = eatNargs(i, key, args)
  303. // array format = '-f=a b c'
  304. } else if (checkAllAliases(key, flags.arrays) && args.length > i + 1) {
  305. args.splice(i + 1, 0, value)
  306. i = eatArray(i, key, args)
  307. } else {
  308. setArg(key, value)
  309. }
  310. broken = true
  311. break
  312. }
  313. if (next === '-') {
  314. setArg(letters[j], next)
  315. continue
  316. }
  317. // current letter is an alphabetic character and next value is a number
  318. if (/[A-Za-z]/.test(letters[j]) &&
  319. /^-?\d+(\.\d*)?(e-?\d+)?$/.test(next)) {
  320. setArg(letters[j], next)
  321. broken = true
  322. break
  323. }
  324. if (letters[j + 1] && letters[j + 1].match(/\W/)) {
  325. setArg(letters[j], next)
  326. broken = true
  327. break
  328. } else {
  329. setArg(letters[j], defaultForType(guessType(letters[j], flags)))
  330. }
  331. }
  332. key = arg.slice(-1)[0]
  333. if (!broken && key !== '-') {
  334. // nargs format = '-f a b c'
  335. if (checkAllAliases(key, flags.nargs)) {
  336. i = eatNargs(i, key, args)
  337. // array format = '-f a b c'
  338. } else if (checkAllAliases(key, flags.arrays) && args.length > i + 1) {
  339. i = eatArray(i, key, args)
  340. } else {
  341. next = args[i + 1]
  342. if (next !== undefined && (!/^(-|--)[^-]/.test(next) ||
  343. next.match(negative)) &&
  344. !checkAllAliases(key, flags.bools) &&
  345. !checkAllAliases(key, flags.counts)) {
  346. setArg(key, next)
  347. i++
  348. } else if (/^(true|false)$/.test(next)) {
  349. setArg(key, next)
  350. i++
  351. } else {
  352. setArg(key, defaultForType(guessType(key, flags)))
  353. }
  354. }
  355. }
  356. } else {
  357. argv._.push(
  358. flags.strings['_'] || !isNumber(arg) ? arg : Number(arg)
  359. )
  360. }
  361. }
  362. // order of precedence:
  363. // 1. command line arg
  364. // 2. value from env var
  365. // 3. value from config file
  366. // 4. value from config objects
  367. // 5. configured default value
  368. applyEnvVars(argv, true) // special case: check env vars that point to config file
  369. applyEnvVars(argv, false)
  370. setConfig(argv)
  371. setConfigObjects()
  372. applyDefaultsAndAliases(argv, flags.aliases, defaults)
  373. applyCoercions(argv)
  374. // for any counts either not in args or without an explicit default, set to 0
  375. Object.keys(flags.counts).forEach(function (key) {
  376. if (!hasKey(argv, key.split('.'))) setArg(key, 0)
  377. })
  378. // '--' defaults to undefined.
  379. if (notFlagsOption && notFlags.length) argv[notFlagsArgv] = []
  380. notFlags.forEach(function (key) {
  381. argv[notFlagsArgv].push(key)
  382. })
  383. // how many arguments should we consume, based
  384. // on the nargs option?
  385. function eatNargs (i, key, args) {
  386. var toEat = checkAllAliases(key, flags.nargs)
  387. if (args.length - (i + 1) < toEat) error = Error(__(`Not enough arguments following: ${key}`))
  388. for (var ii = i + 1; ii < (toEat + i + 1); ii++) {
  389. setArg(key, args[ii])
  390. }
  391. return (i + toEat)
  392. }
  393. // if an option is an array, eat all non-hyphenated arguments
  394. // following it... YUM!
  395. // e.g., --foo apple banana cat becomes ["apple", "banana", "cat"]
  396. function eatArray (i, key, args) {
  397. var start = i + 1
  398. var argsToSet = []
  399. var multipleArrayFlag = i > 0
  400. for (var ii = i + 1; ii < args.length; ii++) {
  401. if (/^-/.test(args[ii]) && !negative.test(args[ii])) {
  402. if (ii === start) {
  403. setArg(key, defaultForType('array'))
  404. }
  405. multipleArrayFlag = true
  406. break
  407. }
  408. i = ii
  409. argsToSet.push(args[ii])
  410. }
  411. if (multipleArrayFlag) {
  412. setArg(key, argsToSet.map(function (arg) {
  413. return processValue(key, arg)
  414. }))
  415. } else {
  416. argsToSet.forEach(function (arg) {
  417. setArg(key, arg)
  418. })
  419. }
  420. return i
  421. }
  422. function setArg (key, val) {
  423. unsetDefaulted(key)
  424. if (/-/.test(key) && !(flags.aliases[key] && flags.aliases[key].length) && configuration['camel-case-expansion']) {
  425. var c = camelCase(key)
  426. flags.aliases[key] = [c]
  427. newAliases[c] = true
  428. }
  429. var value = processValue(key, val)
  430. var splitKey = key.split('.')
  431. setKey(argv, splitKey, value)
  432. // handle populating aliases of the full key
  433. if (flags.aliases[key]) {
  434. flags.aliases[key].forEach(function (x) {
  435. x = x.split('.')
  436. setKey(argv, x, value)
  437. })
  438. }
  439. // handle populating aliases of the first element of the dot-notation key
  440. if (splitKey.length > 1 && configuration['dot-notation']) {
  441. ;(flags.aliases[splitKey[0]] || []).forEach(function (x) {
  442. x = x.split('.')
  443. // expand alias with nested objects in key
  444. var a = [].concat(splitKey)
  445. a.shift() // nuke the old key.
  446. x = x.concat(a)
  447. setKey(argv, x, value)
  448. })
  449. }
  450. // Set normalize getter and setter when key is in 'normalize' but isn't an array
  451. if (checkAllAliases(key, flags.normalize) && !checkAllAliases(key, flags.arrays)) {
  452. var keys = [key].concat(flags.aliases[key] || [])
  453. /*
  454. keys.forEach(function (key) {
  455. argv.__defineSetter__(key, function (v) {
  456. val = v
  457. })
  458. argv.__defineGetter__(key, function () {
  459. return typeof val === 'string' ? path.normalize(val) : val
  460. })
  461. })
  462. */
  463. }
  464. }
  465. function processValue (key, val) {
  466. // handle parsing boolean arguments --foo=true --bar false.
  467. if (checkAllAliases(key, flags.bools) || checkAllAliases(key, flags.counts)) {
  468. if (typeof val === 'string') val = val === 'true'
  469. }
  470. var value = val
  471. if (!checkAllAliases(key, flags.strings) && !checkAllAliases(key, flags.coercions)) {
  472. if (isNumber(val)) value = Number(val)
  473. if (!isUndefined(val) && !isNumber(val) && checkAllAliases(key, flags.numbers)) value = NaN
  474. }
  475. // increment a count given as arg (either no value or value parsed as boolean)
  476. if (checkAllAliases(key, flags.counts) && (isUndefined(value) || typeof value === 'boolean')) {
  477. value = increment
  478. }
  479. /*
  480. // Set normalized value when key is in 'normalize' and in 'arrays'
  481. if (checkAllAliases(key, flags.normalize) && checkAllAliases(key, flags.arrays)) {
  482. if (Array.isArray(val)) value = val.map(path.normalize)
  483. else value = path.normalize(val)
  484. }
  485. */
  486. return value
  487. }
  488. // set args from config.json file, this should be
  489. // applied last so that defaults can be applied.
  490. function setConfig (argv) {
  491. var configLookup = {}
  492. // expand defaults/aliases, in-case any happen to reference
  493. // the config.json file.
  494. applyDefaultsAndAliases(configLookup, flags.aliases, defaults)
  495. Object.keys(flags.configs).forEach(function (configKey) {
  496. var configPath = argv[configKey] || configLookup[configKey]
  497. if (configPath) {
  498. throw new Error("Not Supported in Chakra Host");
  499. try {
  500. var config = null
  501. var resolvedConfigPath = path.resolve(process.cwd(), configPath)
  502. if (typeof flags.configs[configKey] === 'function') {
  503. try {
  504. config = flags.configs[configKey](resolvedConfigPath)
  505. } catch (e) {
  506. config = e
  507. }
  508. if (config instanceof Error) {
  509. error = config
  510. return
  511. }
  512. } else {
  513. config = require(resolvedConfigPath)
  514. }
  515. setConfigObject(config)
  516. } catch (ex) {
  517. if (argv[configKey]) error = Error(__(`Invalid JSON config file: ${configPath}`))
  518. }
  519. }
  520. })
  521. }
  522. // set args from config object.
  523. // it recursively checks nested objects.
  524. function setConfigObject (config, prev) {
  525. Object.keys(config).forEach(function (key) {
  526. var value = config[key]
  527. var fullKey = prev ? prev + '.' + key : key
  528. // if the value is an inner object and we have dot-notation
  529. // enabled, treat inner objects in config the same as
  530. // heavily nested dot notations (foo.bar.apple).
  531. if (typeof value === 'object' && !Array.isArray(value) && configuration['dot-notation']) {
  532. // if the value is an object but not an array, check nested object
  533. setConfigObject(value, fullKey)
  534. } else {
  535. // setting arguments via CLI takes precedence over
  536. // values within the config file.
  537. if (!hasKey(argv, fullKey.split('.')) || (flags.defaulted[fullKey])) {
  538. setArg(fullKey, value)
  539. }
  540. }
  541. })
  542. }
  543. // set all config objects passed in opts
  544. function setConfigObjects () {
  545. if (typeof configObjects === 'undefined') return
  546. configObjects.forEach(function (configObject) {
  547. setConfigObject(configObject)
  548. })
  549. }
  550. function applyEnvVars (argv, configOnly) {
  551. if (typeof envPrefix === 'undefined') return
  552. var prefix = typeof envPrefix === 'string' ? envPrefix : ''
  553. Object.keys(process.env).forEach(function (envVar) {
  554. if (prefix === '' || envVar.lastIndexOf(prefix, 0) === 0) {
  555. // get array of nested keys and convert them to camel case
  556. var keys = envVar.split('__').map(function (key, i) {
  557. if (i === 0) {
  558. key = key.substring(prefix.length)
  559. }
  560. return camelCase(key)
  561. })
  562. if (((configOnly && flags.configs[keys.join('.')]) || !configOnly) && (!hasKey(argv, keys) || flags.defaulted[keys.join('.')])) {
  563. setArg(keys.join('.'), process.env[envVar])
  564. }
  565. }
  566. })
  567. }
  568. function applyCoercions (argv) {
  569. var coerce
  570. Object.keys(argv).forEach(function (key) {
  571. coerce = checkAllAliases(key, flags.coercions)
  572. if (typeof coerce === 'function') {
  573. try {
  574. argv[key] = coerce(argv[key])
  575. } catch (err) {
  576. error = err
  577. }
  578. }
  579. })
  580. }
  581. function applyDefaultsAndAliases (obj, aliases, defaults) {
  582. Object.keys(defaults).forEach(function (key) {
  583. if (!hasKey(obj, key.split('.'))) {
  584. setKey(obj, key.split('.'), defaults[key])
  585. ;(aliases[key] || []).forEach(function (x) {
  586. if (hasKey(obj, x.split('.'))) return
  587. setKey(obj, x.split('.'), defaults[key])
  588. })
  589. }
  590. })
  591. }
  592. function hasKey (obj, keys) {
  593. var o = obj
  594. if (!configuration['dot-notation']) keys = [keys.join('.')]
  595. keys.slice(0, -1).forEach(function (key) {
  596. o = (o[key] || {})
  597. })
  598. var key = keys[keys.length - 1]
  599. if (typeof o !== 'object') return false
  600. else return key in o
  601. }
  602. function setKey (obj, keys, value) {
  603. var o = obj
  604. if (!configuration['dot-notation']) keys = [keys.join('.')]
  605. keys.slice(0, -1).forEach(function (key) {
  606. if (o[key] === undefined) o[key] = {}
  607. o = o[key]
  608. })
  609. var key = keys[keys.length - 1]
  610. var isTypeArray = checkAllAliases(keys.join('.'), flags.arrays)
  611. var isValueArray = Array.isArray(value)
  612. var duplicate = configuration['duplicate-arguments-array']
  613. if (value === increment) {
  614. o[key] = increment(o[key])
  615. } else if (Array.isArray(o[key])) {
  616. if (duplicate && isTypeArray && isValueArray) {
  617. o[key] = configuration['flatten-duplicate-arrays'] ? o[key].concat(value) : [o[key]].concat([value])
  618. } else if (!duplicate && Boolean(isTypeArray) === Boolean(isValueArray)) {
  619. o[key] = value
  620. } else {
  621. o[key] = o[key].concat([value])
  622. }
  623. } else if (o[key] === undefined && isTypeArray) {
  624. o[key] = isValueArray ? value : [value]
  625. } else if (duplicate && !(o[key] === undefined || checkAllAliases(key, flags.bools) || checkAllAliases(keys.join('.'), flags.bools) || checkAllAliases(key, flags.counts))) {
  626. o[key] = [ o[key], value ]
  627. } else {
  628. o[key] = value
  629. }
  630. }
  631. // extend the aliases list with inferred aliases.
  632. function extendAliases () {
  633. Array.prototype.slice.call(arguments).forEach(function (obj) {
  634. Object.keys(obj || {}).forEach(function (key) {
  635. // short-circuit if we've already added a key
  636. // to the aliases array, for example it might
  637. // exist in both 'opts.default' and 'opts.key'.
  638. if (flags.aliases[key]) return
  639. flags.aliases[key] = [].concat(aliases[key] || [])
  640. // For "--option-name", also set argv.optionName
  641. flags.aliases[key].concat(key).forEach(function (x) {
  642. if (/-/.test(x) && configuration['camel-case-expansion']) {
  643. var c = camelCase(x)
  644. flags.aliases[key].push(c)
  645. newAliases[c] = true
  646. }
  647. })
  648. flags.aliases[key].forEach(function (x) {
  649. flags.aliases[x] = [key].concat(flags.aliases[key].filter(function (y) {
  650. return x !== y
  651. }))
  652. })
  653. })
  654. })
  655. }
  656. // check if a flag is set for any of a key's aliases.
  657. function checkAllAliases (key, flag) {
  658. var isSet = false
  659. var toCheck = [].concat(flags.aliases[key] || [], key)
  660. toCheck.forEach(function (key) {
  661. if (flag[key]) isSet = flag[key]
  662. })
  663. return isSet
  664. }
  665. function setDefaulted (key) {
  666. [].concat(flags.aliases[key] || [], key).forEach(function (k) {
  667. flags.defaulted[k] = true
  668. })
  669. }
  670. function unsetDefaulted (key) {
  671. [].concat(flags.aliases[key] || [], key).forEach(function (k) {
  672. delete flags.defaulted[k]
  673. })
  674. }
  675. // return a default value, given the type of a flag.,
  676. // e.g., key of type 'string' will default to '', rather than 'true'.
  677. function defaultForType (type) {
  678. var def = {
  679. boolean: true,
  680. string: '',
  681. number: undefined,
  682. array: []
  683. }
  684. return def[type]
  685. }
  686. // given a flag, enforce a default type.
  687. function guessType (key, flags) {
  688. var type = 'boolean'
  689. if (checkAllAliases(key, flags.strings)) type = 'string'
  690. else if (checkAllAliases(key, flags.numbers)) type = 'number'
  691. else if (checkAllAliases(key, flags.arrays)) type = 'array'
  692. return type
  693. }
  694. function isNumber (x) {
  695. if (!configuration['parse-numbers']) return false
  696. if (typeof x === 'number') return true
  697. if (/^0x[0-9a-f]+$/i.test(x)) return true
  698. return /^[-+]?(?:\d+(?:\.\d*)?|\.\d+)(e[-+]?\d+)?$/.test(x)
  699. }
  700. function isUndefined (num) {
  701. return num === undefined
  702. }
  703. return {
  704. argv: argv,
  705. error: error,
  706. aliases: flags.aliases,
  707. newAliases: newAliases,
  708. configuration: configuration
  709. }
  710. }
  711. // if any aliases reference each other, we should
  712. // merge them together.
  713. function combineAliases (aliases) {
  714. var aliasArrays = []
  715. var change = true
  716. var combined = {}
  717. // turn alias lookup hash {key: ['alias1', 'alias2']} into
  718. // a simple array ['key', 'alias1', 'alias2']
  719. Object.keys(aliases).forEach(function (key) {
  720. aliasArrays.push(
  721. [].concat(aliases[key], key)
  722. )
  723. })
  724. // combine arrays until zero changes are
  725. // made in an iteration.
  726. while (change) {
  727. change = false
  728. for (var i = 0; i < aliasArrays.length; i++) {
  729. for (var ii = i + 1; ii < aliasArrays.length; ii++) {
  730. var intersect = aliasArrays[i].filter(function (v) {
  731. return aliasArrays[ii].indexOf(v) !== -1
  732. })
  733. if (intersect.length) {
  734. aliasArrays[i] = aliasArrays[i].concat(aliasArrays[ii])
  735. aliasArrays.splice(ii, 1)
  736. change = true
  737. break
  738. }
  739. }
  740. }
  741. }
  742. // map arrays back to the hash-lookup (de-dupe while
  743. // we're at it).
  744. aliasArrays.forEach(function (aliasArray) {
  745. aliasArray = aliasArray.filter(function (v, i, self) {
  746. return self.indexOf(v) === i
  747. })
  748. combined[aliasArray.pop()] = aliasArray
  749. })
  750. return combined
  751. }
  752. function assign (defaults, configuration) {
  753. var o = {}
  754. configuration = configuration || {}
  755. Object.keys(defaults).forEach(function (k) {
  756. o[k] = defaults[k]
  757. })
  758. Object.keys(configuration).forEach(function (k) {
  759. o[k] = configuration[k]
  760. })
  761. return o
  762. }
  763. // this function should only be called when a count is given as an arg
  764. // it is NOT called to set a default value
  765. // thus we can start the count at 1 instead of 0
  766. function increment (orig) {
  767. return orig !== undefined ? orig + 1 : 1
  768. }
  769. })();