73 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			73 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
const command = require('./command')()
 | 
						|
const YError = require('./yerror')
 | 
						|
 | 
						|
const positionName = ['first', 'second', 'third', 'fourth', 'fifth', 'sixth']
 | 
						|
 | 
						|
module.exports = function (expected, callerArguments, length) {
 | 
						|
  // TODO: should this eventually raise an exception.
 | 
						|
  try {
 | 
						|
    // preface the argument description with "cmd", so
 | 
						|
    // that we can run it through yargs' command parser.
 | 
						|
    var position = 0
 | 
						|
    var parsed = {demanded: [], optional: []}
 | 
						|
    if (typeof expected === 'object') {
 | 
						|
      length = callerArguments
 | 
						|
      callerArguments = expected
 | 
						|
    } else {
 | 
						|
      parsed = command.parseCommand('cmd ' + expected)
 | 
						|
    }
 | 
						|
    const args = [].slice.call(callerArguments)
 | 
						|
 | 
						|
    while (args.length && args[args.length - 1] === undefined) args.pop()
 | 
						|
    length = length || args.length
 | 
						|
 | 
						|
    if (length < parsed.demanded.length) {
 | 
						|
      throw new YError('Not enough arguments provided. Expected ' + parsed.demanded.length +
 | 
						|
        ' but received ' + args.length + '.')
 | 
						|
    }
 | 
						|
 | 
						|
    const totalCommands = parsed.demanded.length + parsed.optional.length
 | 
						|
    if (length > totalCommands) {
 | 
						|
      throw new YError('Too many arguments provided. Expected max ' + totalCommands +
 | 
						|
        ' but received ' + length + '.')
 | 
						|
    }
 | 
						|
 | 
						|
    parsed.demanded.forEach(function (demanded) {
 | 
						|
      const arg = args.shift()
 | 
						|
      const observedType = guessType(arg)
 | 
						|
      const matchingTypes = demanded.cmd.filter(function (type) {
 | 
						|
        return type === observedType || type === '*'
 | 
						|
      })
 | 
						|
      if (matchingTypes.length === 0) argumentTypeError(observedType, demanded.cmd, position, false)
 | 
						|
      position += 1
 | 
						|
    })
 | 
						|
 | 
						|
    parsed.optional.forEach(function (optional) {
 | 
						|
      if (args.length === 0) return
 | 
						|
      const arg = args.shift()
 | 
						|
      const observedType = guessType(arg)
 | 
						|
      const matchingTypes = optional.cmd.filter(function (type) {
 | 
						|
        return type === observedType || type === '*'
 | 
						|
      })
 | 
						|
      if (matchingTypes.length === 0) argumentTypeError(observedType, optional.cmd, position, true)
 | 
						|
      position += 1
 | 
						|
    })
 | 
						|
  } catch (err) {
 | 
						|
    console.warn(err.stack)
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
function guessType (arg) {
 | 
						|
  if (Array.isArray(arg)) {
 | 
						|
    return 'array'
 | 
						|
  } else if (arg === null) {
 | 
						|
    return 'null'
 | 
						|
  }
 | 
						|
  return typeof arg
 | 
						|
}
 | 
						|
 | 
						|
function argumentTypeError (observedType, allowedTypes, position, optional) {
 | 
						|
  throw new YError('Invalid ' + (positionName[position] || 'manyith') + ' argument.' +
 | 
						|
    ' Expected ' + allowedTypes.join(' or ') + ' but received ' + observedType + '.')
 | 
						|
}
 |