123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416 |
- "use strict";
- Object.defineProperty(exports, "__esModule", { value: true });
- exports.isCommandBuilderCallback = exports.isCommandBuilderDefinition = exports.isCommandHandlerDefinition = exports.command = void 0;
- const common_types_1 = require("./common-types");
- const is_promise_1 = require("./is-promise");
- const middleware_1 = require("./middleware");
- const parse_command_1 = require("./parse-command");
- const path = require("path");
- const util_1 = require("util");
- const yargs_1 = require("./yargs");
- const requireDirectory = require("require-directory");
- const whichModule = require("which-module");
- const Parser = require("yargs-parser");
- const DEFAULT_MARKER = /(^\*)|(^\$0)/;
- // handles parsing positional arguments,
- // and populating argv with said positional
- // arguments.
- function command(yargs, usage, validation, globalMiddleware = []) {
- const self = {};
- let handlers = {};
- let aliasMap = {};
- let defaultCommand;
- self.addHandler = function addHandler(cmd, description, builder, handler, commandMiddleware, deprecated) {
- let aliases = [];
- const middlewares = middleware_1.commandMiddlewareFactory(commandMiddleware);
- handler = handler || (() => { });
- if (Array.isArray(cmd)) {
- aliases = cmd.slice(1);
- cmd = cmd[0];
- }
- else if (isCommandHandlerDefinition(cmd)) {
- let command = (Array.isArray(cmd.command) || typeof cmd.command === 'string') ? cmd.command : moduleName(cmd);
- if (cmd.aliases)
- command = [].concat(command).concat(cmd.aliases);
- self.addHandler(command, extractDesc(cmd), cmd.builder, cmd.handler, cmd.middlewares, cmd.deprecated);
- return;
- }
- // allow a module to be provided instead of separate builder and handler
- if (isCommandBuilderDefinition(builder)) {
- self.addHandler([cmd].concat(aliases), description, builder.builder, builder.handler, builder.middlewares, builder.deprecated);
- return;
- }
- // parse positionals out of cmd string
- const parsedCommand = parse_command_1.parseCommand(cmd);
- // remove positional args from aliases only
- aliases = aliases.map(alias => parse_command_1.parseCommand(alias).cmd);
- // check for default and filter out '*''
- let isDefault = false;
- const parsedAliases = [parsedCommand.cmd].concat(aliases).filter((c) => {
- if (DEFAULT_MARKER.test(c)) {
- isDefault = true;
- return false;
- }
- return true;
- });
- // standardize on $0 for default command.
- if (parsedAliases.length === 0 && isDefault)
- parsedAliases.push('$0');
- // shift cmd and aliases after filtering out '*'
- if (isDefault) {
- parsedCommand.cmd = parsedAliases[0];
- aliases = parsedAliases.slice(1);
- cmd = cmd.replace(DEFAULT_MARKER, parsedCommand.cmd);
- }
- // populate aliasMap
- aliases.forEach((alias) => {
- aliasMap[alias] = parsedCommand.cmd;
- });
- if (description !== false) {
- usage.command(cmd, description, isDefault, aliases, deprecated);
- }
- handlers[parsedCommand.cmd] = {
- original: cmd,
- description,
- handler,
- builder: builder || {},
- middlewares,
- deprecated,
- demanded: parsedCommand.demanded,
- optional: parsedCommand.optional
- };
- if (isDefault)
- defaultCommand = handlers[parsedCommand.cmd];
- };
- self.addDirectory = function addDirectory(dir, context, req, callerFile, opts) {
- opts = opts || {};
- // disable recursion to support nested directories of subcommands
- if (typeof opts.recurse !== 'boolean')
- opts.recurse = false;
- // exclude 'json', 'coffee' from require-directory defaults
- if (!Array.isArray(opts.extensions))
- opts.extensions = ['js'];
- // allow consumer to define their own visitor function
- const parentVisit = typeof opts.visit === 'function' ? opts.visit : (o) => o;
- // call addHandler via visitor function
- opts.visit = function visit(obj, joined, filename) {
- const visited = parentVisit(obj, joined, filename);
- // allow consumer to skip modules with their own visitor
- if (visited) {
- // check for cyclic reference
- // each command file path should only be seen once per execution
- if (~context.files.indexOf(joined))
- return visited;
- // keep track of visited files in context.files
- context.files.push(joined);
- self.addHandler(visited);
- }
- return visited;
- };
- requireDirectory({ require: req, filename: callerFile }, dir, opts);
- };
- // lookup module object from require()d command and derive name
- // if module was not require()d and no name given, throw error
- function moduleName(obj) {
- const mod = whichModule(obj);
- if (!mod)
- throw new Error(`No command name given for module: ${util_1.inspect(obj)}`);
- return commandFromFilename(mod.filename);
- }
- // derive command name from filename
- function commandFromFilename(filename) {
- return path.basename(filename, path.extname(filename));
- }
- function extractDesc({ describe, description, desc }) {
- for (const test of [describe, description, desc]) {
- if (typeof test === 'string' || test === false)
- return test;
- common_types_1.assertNotStrictEqual(test, true);
- }
- return false;
- }
- self.getCommands = () => Object.keys(handlers).concat(Object.keys(aliasMap));
- self.getCommandHandlers = () => handlers;
- self.hasDefaultCommand = () => !!defaultCommand;
- self.runCommand = function runCommand(command, yargs, parsed, commandIndex) {
- let aliases = parsed.aliases;
- const commandHandler = handlers[command] || handlers[aliasMap[command]] || defaultCommand;
- const currentContext = yargs.getContext();
- let numFiles = currentContext.files.length;
- const parentCommands = currentContext.commands.slice();
- // what does yargs look like after the builder is run?
- let innerArgv = parsed.argv;
- let positionalMap = {};
- if (command) {
- currentContext.commands.push(command);
- currentContext.fullCommands.push(commandHandler.original);
- }
- const builder = commandHandler.builder;
- if (isCommandBuilderCallback(builder)) {
- // a function can be provided, which builds
- // up a yargs chain and possibly returns it.
- const builderOutput = builder(yargs.reset(parsed.aliases));
- const innerYargs = yargs_1.isYargsInstance(builderOutput) ? builderOutput : yargs;
- if (shouldUpdateUsage(innerYargs)) {
- innerYargs.getUsageInstance().usage(usageFromParentCommandsCommandHandler(parentCommands, commandHandler), commandHandler.description);
- }
- innerArgv = innerYargs._parseArgs(null, null, true, commandIndex);
- aliases = innerYargs.parsed.aliases;
- }
- else if (isCommandBuilderOptionDefinitions(builder)) {
- // as a short hand, an object can instead be provided, specifying
- // the options that a command takes.
- const innerYargs = yargs.reset(parsed.aliases);
- if (shouldUpdateUsage(innerYargs)) {
- innerYargs.getUsageInstance().usage(usageFromParentCommandsCommandHandler(parentCommands, commandHandler), commandHandler.description);
- }
- Object.keys(commandHandler.builder).forEach((key) => {
- innerYargs.option(key, builder[key]);
- });
- innerArgv = innerYargs._parseArgs(null, null, true, commandIndex);
- aliases = innerYargs.parsed.aliases;
- }
- if (!yargs._hasOutput()) {
- positionalMap = populatePositionals(commandHandler, innerArgv, currentContext);
- }
- const middlewares = globalMiddleware.slice(0).concat(commandHandler.middlewares);
- middleware_1.applyMiddleware(innerArgv, yargs, middlewares, true);
- // we apply validation post-hoc, so that custom
- // checks get passed populated positional arguments.
- if (!yargs._hasOutput()) {
- yargs._runValidation(innerArgv, aliases, positionalMap, yargs.parsed.error, !command);
- }
- if (commandHandler.handler && !yargs._hasOutput()) {
- yargs._setHasOutput();
- // to simplify the parsing of positionals in commands,
- // we temporarily populate '--' rather than _, with arguments
- const populateDoubleDash = !!yargs.getOptions().configuration['populate--'];
- if (!populateDoubleDash)
- yargs._copyDoubleDash(innerArgv);
- innerArgv = middleware_1.applyMiddleware(innerArgv, yargs, middlewares, false);
- let handlerResult;
- if (is_promise_1.isPromise(innerArgv)) {
- handlerResult = innerArgv.then(argv => commandHandler.handler(argv));
- }
- else {
- handlerResult = commandHandler.handler(innerArgv);
- }
- const handlerFinishCommand = yargs.getHandlerFinishCommand();
- if (is_promise_1.isPromise(handlerResult)) {
- yargs.getUsageInstance().cacheHelpMessage();
- handlerResult
- .then(value => {
- if (handlerFinishCommand) {
- handlerFinishCommand(value);
- }
- })
- .catch(error => {
- try {
- yargs.getUsageInstance().fail(null, error);
- }
- catch (err) {
- // fail's throwing would cause an unhandled rejection.
- }
- })
- .then(() => {
- yargs.getUsageInstance().clearCachedHelpMessage();
- });
- }
- else {
- if (handlerFinishCommand) {
- handlerFinishCommand(handlerResult);
- }
- }
- }
- if (command) {
- currentContext.commands.pop();
- currentContext.fullCommands.pop();
- }
- numFiles = currentContext.files.length - numFiles;
- if (numFiles > 0)
- currentContext.files.splice(numFiles * -1, numFiles);
- return innerArgv;
- };
- function shouldUpdateUsage(yargs) {
- return !yargs.getUsageInstance().getUsageDisabled() &&
- yargs.getUsageInstance().getUsage().length === 0;
- }
- function usageFromParentCommandsCommandHandler(parentCommands, commandHandler) {
- const c = DEFAULT_MARKER.test(commandHandler.original) ? commandHandler.original.replace(DEFAULT_MARKER, '').trim() : commandHandler.original;
- const pc = parentCommands.filter((c) => { return !DEFAULT_MARKER.test(c); });
- pc.push(c);
- return `$0 ${pc.join(' ')}`;
- }
- self.runDefaultBuilderOn = function (yargs) {
- common_types_1.assertNotStrictEqual(defaultCommand, undefined);
- if (shouldUpdateUsage(yargs)) {
- // build the root-level command string from the default string.
- const commandString = DEFAULT_MARKER.test(defaultCommand.original)
- ? defaultCommand.original : defaultCommand.original.replace(/^[^[\]<>]*/, '$0 ');
- yargs.getUsageInstance().usage(commandString, defaultCommand.description);
- }
- const builder = defaultCommand.builder;
- if (isCommandBuilderCallback(builder)) {
- builder(yargs);
- }
- else {
- Object.keys(builder).forEach((key) => {
- yargs.option(key, builder[key]);
- });
- }
- };
- // transcribe all positional arguments "command <foo> <bar> [apple]"
- // onto argv.
- function populatePositionals(commandHandler, argv, context) {
- argv._ = argv._.slice(context.commands.length); // nuke the current commands
- const demanded = commandHandler.demanded.slice(0);
- const optional = commandHandler.optional.slice(0);
- const positionalMap = {};
- validation.positionalCount(demanded.length, argv._.length);
- while (demanded.length) {
- const demand = demanded.shift();
- populatePositional(demand, argv, positionalMap);
- }
- while (optional.length) {
- const maybe = optional.shift();
- populatePositional(maybe, argv, positionalMap);
- }
- argv._ = context.commands.concat(argv._);
- postProcessPositionals(argv, positionalMap, self.cmdToParseOptions(commandHandler.original));
- return positionalMap;
- }
- function populatePositional(positional, argv, positionalMap) {
- const cmd = positional.cmd[0];
- if (positional.variadic) {
- positionalMap[cmd] = argv._.splice(0).map(String);
- }
- else {
- if (argv._.length)
- positionalMap[cmd] = [String(argv._.shift())];
- }
- }
- // we run yargs-parser against the positional arguments
- // applying the same parsing logic used for flags.
- function postProcessPositionals(argv, positionalMap, parseOptions) {
- // combine the parsing hints we've inferred from the command
- // string with explicitly configured parsing hints.
- const options = Object.assign({}, yargs.getOptions());
- options.default = Object.assign(parseOptions.default, options.default);
- for (const key of Object.keys(parseOptions.alias)) {
- options.alias[key] = (options.alias[key] || []).concat(parseOptions.alias[key]);
- }
- options.array = options.array.concat(parseOptions.array);
- delete options.config; // don't load config when processing positionals.
- const unparsed = [];
- Object.keys(positionalMap).forEach((key) => {
- positionalMap[key].map((value) => {
- if (options.configuration['unknown-options-as-args'])
- options.key[key] = true;
- unparsed.push(`--${key}`);
- unparsed.push(value);
- });
- });
- // short-circuit parse.
- if (!unparsed.length)
- return;
- const config = Object.assign({}, options.configuration, {
- 'populate--': true
- });
- const parsed = Parser.detailed(unparsed, Object.assign({}, options, {
- configuration: config
- }));
- if (parsed.error) {
- yargs.getUsageInstance().fail(parsed.error.message, parsed.error);
- }
- else {
- // only copy over positional keys (don't overwrite
- // flag arguments that were already parsed).
- const positionalKeys = Object.keys(positionalMap);
- Object.keys(positionalMap).forEach((key) => {
- positionalKeys.push(...parsed.aliases[key]);
- });
- Object.keys(parsed.argv).forEach((key) => {
- if (positionalKeys.indexOf(key) !== -1) {
- // any new aliases need to be placed in positionalMap, which
- // is used for validation.
- if (!positionalMap[key])
- positionalMap[key] = parsed.argv[key];
- argv[key] = parsed.argv[key];
- }
- });
- }
- }
- self.cmdToParseOptions = function (cmdString) {
- const parseOptions = {
- array: [],
- default: {},
- alias: {},
- demand: {}
- };
- const parsed = parse_command_1.parseCommand(cmdString);
- parsed.demanded.forEach((d) => {
- const [cmd, ...aliases] = d.cmd;
- if (d.variadic) {
- parseOptions.array.push(cmd);
- parseOptions.default[cmd] = [];
- }
- parseOptions.alias[cmd] = aliases;
- parseOptions.demand[cmd] = true;
- });
- parsed.optional.forEach((o) => {
- const [cmd, ...aliases] = o.cmd;
- if (o.variadic) {
- parseOptions.array.push(cmd);
- parseOptions.default[cmd] = [];
- }
- parseOptions.alias[cmd] = aliases;
- });
- return parseOptions;
- };
- self.reset = () => {
- handlers = {};
- aliasMap = {};
- defaultCommand = undefined;
- return self;
- };
- // used by yargs.parse() to freeze
- // the state of commands such that
- // we can apply .parse() multiple times
- // with the same yargs instance.
- const frozens = [];
- self.freeze = () => {
- frozens.push({
- handlers,
- aliasMap,
- defaultCommand
- });
- };
- self.unfreeze = () => {
- const frozen = frozens.pop();
- common_types_1.assertNotStrictEqual(frozen, undefined);
- ({
- handlers,
- aliasMap,
- defaultCommand
- } = frozen);
- };
- return self;
- }
- exports.command = command;
- function isCommandHandlerDefinition(cmd) {
- return typeof cmd === 'object';
- }
- exports.isCommandHandlerDefinition = isCommandHandlerDefinition;
- function isCommandBuilderDefinition(builder) {
- return typeof builder === 'object' &&
- !!builder.builder &&
- typeof builder.handler === 'function';
- }
- exports.isCommandBuilderDefinition = isCommandBuilderDefinition;
- function isCommandBuilderCallback(builder) {
- return typeof builder === 'function';
- }
- exports.isCommandBuilderCallback = isCommandBuilderCallback;
- function isCommandBuilderOptionDefinitions(builder) {
- return typeof builder === 'object';
- }
|