ryuu-beta/commands/login.js

var inquirer = require("inquirer");
var _ = require('lodash');
var chalk = require('chalk');
var shell = require('shelljs');
var multichoice = require('../util/multichoice');
var log = require('../util/log');
var Login = require('../util/login');
var evoke = require('../util/evoke');
var Domo = require('ryuu-client-beta');
var Promise = require('bluebird');
var path = require('path');
var request = Promise.promisify(require('request'));
var constants = require('../util/constants');

var logo = `
┏━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┓
┃                                                            ┃
┃                                                            ┃
┃                                                            ┃
┃                                                            ┃
┃                                                            ┃
┃                                                            ┃
┃                                                            ┃
┃                                                            ┃
┃                                                            ┃
┃                                                            ┃
┃██████████      ██████████  ███            ██   ██████████  ┃
┃████████████  ███████████████████        ███████████████████┃
┃███      ████████        ██████████    ██████████        ███┃
┃███       ██████          ██████████████████████          ██┃
┃███      ███████          ██████ ███████  ██████          ██┃
┃███     █████████       ████████   ███    ███████       ████┃
┃████████████  ██████████████████          ██████████████████┃
┃██████████      ██████████  ████          ███   ██████████  ┃
┃                                                            ┃
┃                                                            ┃
┃                                                            ┃
┃                                                            ┃
┃                                                            ┃
┃                                                            ┃
┃                                                            ┃
┃                                                            ┃
┃                                                            ┃
┃                                                            ┃
┗━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┛`;


module.exports = function(program){
  program
    .command('login')
    .description('login to Domo')
    .option('-i, --instance <value>', 'Domo instance')
    .option('-u, --user-email <value>', 'user email')
    .option('--no-upgrade-check', 'do not check for new versions')
    .action(function(args){

      var previousLogins = Login.getPreviousLogins();
      var mostRecent = Login.getMostRecentLogin();
      var prompts = [];

      var versionCheckFn = !args.upgradeCheck ?
        function (x) { return x; } :
        verifyVersion;

      if (!args.instance) {
        prompts.push(multichoice({
          type: "list",
          choices: function(){
            var options = _.pluck(previousLogins, 'instance');
            options.push('new instance');
            return options;
          },
          message: "Domo instance " + chalk.grey("e.g. company.domo.com "),
          name: 'instance',
          default: mostRecent.instance,
          // only show up if they have previous logins
          when: function(answers){
            return previousLogins.length > 0;
          }
        }),
        {
          type: "input",
          message: "Domo instance " + chalk.grey("e.g. company.domo.com "),
          name: "instance",
          when: function(answers){
            // only ask if the instance selector wasn't used or new one was selected
            return !answers.instance || answers.instance === 'new instance';
          }
        });
      }

      // prompt user
      inquirer.prompt(prompts, function(answers) {

        answers.instance = args.instance || answers.instance;

        if (!answers.instance.endsWith('domo.com') && !answers.instance.endsWith('domotech.io') && !answers.instance.endsWith('domorig.io')){
          log.fail('Please login using the full Domo instance.', 'Example: "company.domo.com"');
        }

        var instanceLogin = Login.getLogin(answers.instance);
        var login;
        var domo = new Domo(answers.instance, instanceLogin.refreshToken, constants.CLIENT_ID);
        domo
          .login()
          .then(function(l){
            return login = l;
          })
          .then(checkVersion.bind(null, answers.instance))
          .then(versionCheckFn)
          .then(function(){
            console.log(chalk.grey(logo));
            log.ok('Login to ' + answers.instance + ' successful.', 'Welcome, ' + login.displayName + '.');
            return Login.persistLogin(answers.instance)(login);
          })
          .catch(evoke(log.fail, 'Login unsuccessful'))
          .finally(process.exit)
      });
    });
}

/**
 * E.T. phone home
 */
function checkVersion(instance){
  var cliVersion = require(path.resolve(__dirname + '/../package.json')).version;
  var options = {
    url: 'https://' + instance + '/domoapps/admin/checkversion?version=' + cliVersion,
    method: 'GET',
    json: true
  };

  return request(options);
}


/**
 * log message based on the action
 * if action is 'block' then 'fail' will get called and end the process early
 */
function verifyVersion(res){

  var body = res[0].body;
  if (body.action === 'block'){
    log.fail(body.message);
  }

  if (body.action === 'warn'){
    log.warn(body.message);
    return promptUpgrade(body.latest);
  }

  if (body.action === 'info'){
    log.info(body.message);
    return promptUpgrade(body.latest);
  }
}

function promptUpgrade(latest){
  return new Promise(function(resolve, reject){
    inquirer.prompt([
      {
        type: "confirm",
        message: "Would you like to upgrade now?",
        name: "upgrade",
        default: true
      }

    ], function(answers) {
      if (answers.upgrade){
        log.info('Upgrading to latest version', 'Hang tight, this may take a moment...');
        var output = shell.exec('npm install ryuu -g', {async: false, silent: true});
        if (output.code === 0){
          log.ok('Upgrade to version '+latest+' successful');
          resolve();
        }
        else {
          log.warn('Upgrade unsuccessful', 'Please try upgrading manually with npm install -g ryuu');
        }
      }
      else{
        resolve();
      }
    });
  });
}