serverless/lib/classes/CLI.js

'use strict';

const version = require('../../package.json').version;
const Spinner = require('cli-spinner').Spinner;
const minimist = require('minimist');
const _ = require('lodash');
const os = require('os');
const chalk = require('chalk');

class CLI {
  constructor(serverless, interactive, inputArray) {
    this.serverless = serverless;
    this.interactive = interactive;
    this.inputArray = (typeof inputArray !== 'undefined' ? inputArray : []);
    this.loadedPlugins = [];
  }

  setLoadedPlugins(plugins) {
    this.loadedPlugins = plugins;
  }

  processInput() {
    let inputArray;

    // check if commands are passed externally (e.g. used by tests)
    // otherwise use process.argv to receive the commands
    if (this.inputArray.length) {
      inputArray = this.inputArray;
    } else {
      inputArray = process.argv.slice(2);
    }

    const argv = minimist(inputArray);

    const commandsAndOptions = {};
    commandsAndOptions.commands = [];
    commandsAndOptions.options = {};

    // if only "help" or "h" was entered
    if ((argv._.length === 0 && (argv.help || argv.h)) ||
      (argv._.length === 1 && (argv._.indexOf('help') > -1))) {
      this.generateMainHelp();
      return commandsAndOptions;
    }
    // if only "version" or "v" was entered
    if ((argv._.length === 0 && (argv.version || argv.v)) ||
      (argv._.length === 1 && (argv._.indexOf('version') > -1))) {
      this.getVersionNumber();
      return commandsAndOptions;
    }
    // if "help" was entered in combination with commands (or one command)
    if (argv._.length >= 1 && (argv.help || argv.h)) {
      const commands = argv._;
      this.generateCommandsHelp(commands);
      return commandsAndOptions;
    }

    const commands = [];
    const options = {};

    // get all the commands
    argv._.forEach((command) => {
      commands.push(command);
    });

    // remove the array which holds the commands from the argv object
    const optionsObject = _.omit(argv, ['_']);

    // get all the options
    _.forEach(optionsObject, (value, key) => {
      options[key] = value;
    });

    commandsAndOptions.commands = commands;
    commandsAndOptions.options = options;

    return commandsAndOptions;
  }

  generateMainHelp() {
    this.asciiGreeting();

    console.log('');

    console.log(chalk.yellow.underline('Commands'));
    console.log(chalk.dim('* Serverless documentation: http://docs.serverless.com'));
    console.log(chalk.dim('* You can run commands with "serverless" or the shortcut "sls"'));
    console.log(chalk.dim('* Pass "--help" after any <command> for contextual help'));

    console.log('');

    const sortedPlugins = this.loadedPlugins.sort();

    // TODO: implement recursive command exploration (now only 2 steps are possible)
    const dotsLength = 25;
    sortedPlugins.forEach((plugin) => {
      _.forEach(plugin.commands,
        (firstLevelCommandObject, firstLevelCommand) => {
          // check if command has lifecycleEvents (can be execute)
          if (firstLevelCommandObject.lifecycleEvents) {
            const command = firstLevelCommand;
            const usage = firstLevelCommandObject.usage;
            const dots = _.repeat('.', dotsLength - command.length);
            console.log('%s %s %s', chalk.yellow(command), chalk.dim(dots), usage);
          }
          _.forEach(firstLevelCommandObject.commands,
            (secondLevelCommandObject, secondLevelCommand) => {
              // check if command has lifecycleEvents (can be executed)
              if (secondLevelCommandObject.lifecycleEvents) {
                const command = `${firstLevelCommand} ${secondLevelCommand}`;
                const usage = secondLevelCommandObject.usage;
                const dots = _.repeat('.', dotsLength - command.length);
                console.log('%s %s %s', chalk.yellow(command), chalk.dim(dots), usage);
              }
            });
        });
    });

    console.log('');

    // print all the installed plugins
    console.log(chalk.yellow.underline('Plugins'));
    if (sortedPlugins.length) {
      console.log(sortedPlugins.map((plugin) => plugin.constructor.name).join(', '));
    } else {
      console.log('No plugins added yet');
    }
  }

  generateCommandsHelp(commands) {
    const dotsLength = 25;

    // TODO: use lodash utility functions to reduce loop usage
    // TODO: support more than 2 levels of nested commands
    if (commands.length === 1) {
      this.loadedPlugins.forEach((plugin) => {
        _.forEach(plugin.commands, (commandObject, command) => {
          if (command === commands[0]) {
            if (commandObject.lifecycleEvents) {
              // print the name of the plugin
              console.log(chalk.yellow.underline(`Plugin: ${plugin.constructor.name}`));
              // print the command with the corresponding usage
              const commandsDots = _.repeat('.', dotsLength - command.length);
              const commandsUsage = commandObject.usage;
              console.log('%s %s %s',
                chalk.yellow(command),
                chalk.dim(commandsDots),
                commandsUsage);
              // print all options
              _.forEach(commandObject.options, (optionsObject, option) => {
                const optionsDots = _.repeat('.', dotsLength - option.length);
                const optionsUsage = optionsObject.usage;
                console.log('    %s %s %s',
                  chalk.yellow(`--${option}`),
                  chalk.dim(optionsDots.slice(0, optionsDots.length - 6)),
                  optionsUsage);
              });
            }
          }
        });
      });
    } else {
      this.loadedPlugins.forEach((plugin) => {
        _.forEach(plugin.commands,
          (firstLevelCommandObject, firstLevelCommand) => {
            if (firstLevelCommand === commands[0]) {
              _.forEach(firstLevelCommandObject.commands,
                (secondLevelCommandObject, secondLevelCommand) => {
                  if (secondLevelCommand === commands[1]) {
                    if (secondLevelCommandObject.lifecycleEvents) {
                      // print the name of the plugin
                      console.log(chalk.yellow.underline(`Plugin: ${plugin.constructor.name}`));
                      // print the command with the corresponding usage
                      const commandsDots = _.repeat('.', dotsLength - secondLevelCommand.length);
                      const commandsUsage = secondLevelCommandObject.usage;
                      console.log('%s %s %s',
                        chalk.yellow(secondLevelCommand),
                        chalk.dim(commandsDots),
                        commandsUsage);
                      // print all options
                      _.forEach(secondLevelCommandObject.options, (optionsObject, option) => {
                        const optionsDots = _.repeat('.', dotsLength - option.length);
                        const optionsUsage = optionsObject.usage;
                        console.log('    %s %s %s',
                          chalk.yellow(`--${option}`),
                          chalk.dim(optionsDots.slice(0, optionsDots.length - 6)),
                          optionsUsage);
                      });
                    }
                  }
                });
            }
          });
      });
    }

    console.log('');
  }

  getVersionNumber() {
    console.log(version);
  }

  asciiGreeting() {
    let art = '';
    art = art + ' _______                             __' + os.EOL;
    art = art + '|   _   .-----.----.--.--.-----.----|  .-----.-----.-----.' + os.EOL;
    art = art + '|   |___|  -__|   _|  |  |  -__|   _|  |  -__|__ --|__ --|' + os.EOL;
    art = art + '|____   |_____|__|  \\___/|_____|__| |__|_____|_____|_____|' + os.EOL;
    art = art + '|   |   |             The Serverless Application Framework' + os.EOL;
    art = art + '|       |                           serverless.com, v' + version + os.EOL;
    art = art + '`-------\'';

    console.log(chalk.yellow(art));
    console.log('');
  }

  log(message) {
    console.log(`Serverless: ${chalk.yellow(`${message}`)}`);
  }

  isInteractive() {
    return (process.stdout.isTTY && !process.env.CI) || this.interactive;
  }

  spinner(message) {
    let spinner;

    if (this.isInteractive()) {
      message = message ? message : '';
      spinner = new Spinner('Serverless: ' + chalk.yellow('%s ' + message));
      spinner.setSpinnerString('|/-\\');
    } else {
      // Non-interactive spinner object
      spinner = {
        start: (message) => {
          message = message || 'Loading... ';
          process.stdout.write(`Serverless: ${message}`);
        },
        stop: (message) => {
          // Because of how spinner is used with normal library
          // we do a small hack and still allow for setting message
          if (message === true || message === false) {
            message = 'Done!\n';
          }
          message = message || 'Done!\n';
          process.stdout.write(message);
        },
      };
    }
    return spinner;
  }
}

module.exports = CLI;