初始化

康凯

2022-05-20 aa4c235a8ca67ea8b731f90c951a465e92c0a865

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
const path = require('path')
const microargs = require('microargs')
const { get, difference, isEmpty, padEnd, forEach, capitalize, omit, isString } = require('lodash')
 
class CLIError extends Error { }
 
function optionToString (optionName) {
  return optionName.length === 1 ? `-${optionName}` : `--${optionName}`
}
 
function optionsToString (optionsKeys) {
  return optionsKeys.map(optionToString).join(' ')
}
 
function printHelp (scriptName, annotations, logger) {
  if (isEmpty(annotations)) {
    logger.log('Documentation not found')
    return null
  }
 
  const { description, params, options } = annotations
  const extra = omit(annotations, ['description', 'params', 'options'])
  const usageOptions = isEmpty(options) ? '' : '[options]'
  const usageParams = isEmpty(params) ? '' : `[${params.join(' ')}]`
 
  logger.log(`Usage: ${path.basename(scriptName)} ${usageOptions} ${usageParams}\n`)
 
  if (description) {
    logger.log(`${description}\n`)
  }
 
  if (!isEmpty(options)) {
    logger.log('Options:\n')
    forEach(options, (value, key) => {
      logger.log(`  ${padEnd(optionToString(key), 12)}${value}`)
    })
  }
 
  forEach(extra, (value, key) => {
    logger.log(`\n${capitalize(key)}:\n`)
    logger.log(`${value}\n`)
  })
}
 
module.exports = (argv, annotations = {}, help, logger = console) => {
  help = help || printHelp
 
  return (callback) => {
    const { params, options } = microargs(argv.slice(2))
    const scriptName = path.basename(argv[1])
 
    if (isString(annotations)) {
      annotations = {
        description: annotations
      }
    }
 
    if (options.help) {
      return help(scriptName, annotations, logger)
    }
 
    const annotatedOptionsKeys = (get(annotations, 'options') && Object.keys(annotations.options)) || []
    const optionsKeys = Object.keys(options)
    const illegalOptionsKeys = difference(optionsKeys, annotatedOptionsKeys)
 
    if (annotatedOptionsKeys.length && illegalOptionsKeys.length) {
      const msg = `Illegal option: ${optionsToString(illegalOptionsKeys)}\n` +
        `Available options: ${optionsToString(annotatedOptionsKeys)}\n` +
        `Type "${scriptName} --help" for more information`
      throw new CLIError(msg)
    }
 
    return callback(options, ...params)
  }
}
 
module.exports.CLIError = CLIError