divshot-cli/lib/commands/protect.js

CLI for Divshot

var regular = require('regular');
var format = require('chalk');
var _ = require('lodash');

module.exports = function (cli) {
  var command = cli.command('protect <environment> <username:password>');
  
  command.before('authenticate', 'isApp');
  command.description('add http basic auth to any environment');
  
  command.handler(function (env, credentials, done) {
    
    cli.log();
    
    if (!env) {
      return done(cli.errors.MISSING_ENVIRONMENT + protectUsage());
    }
    
    if (!credentials) {
      return done(cli.errors.MISSING_CREDENTIALS + protectUsage());
    }
    
    var username = credentials.split(':')[0];
    var password = credentials.split(':')[1];
    
    if (!regular.basicAuth.test(credentials) || !username || !password) {
      return done(cli.errors.INVALID_CREDENTIALS);
    }
    
    cli.commands.apps(function (err, apps) {
      
      // TODO: refactor this out for reuse
      var name = cli.cwd.getConfig().name;
      var appObj;
      
      _.each(apps, function (orgApps) {
        var app = _.find(orgApps, function (app) {
          return app.name === name;
        });
        
        if (app) appObj = app;
      });
      
      if (!appObj) return done('The app ' + format.bold(name) + ' does not exist');
      
      cli.api.apps.id(appObj.name).env(env).config({
        auth: credentials
      }, function (err, response, body) {
        
        if (body && body.error) {
          return done(body.error)
        }
        
        // Assuming that environment is invalid with a 500 error
        if (response.statusCode == 500) {
          return done(cli.errors.INVALID_ENVIRONMENT);
        }
        
        if (response.statusCode == 400) {
          return done(cli.errors.BLANK_BUILD);
        }
        
        if (err || response.statusCode !== +200) {
          return done(cli.errors.DEFAULT);
        } 
        
        cli.log(format.bold(env) + ' has been protected.', {success: true});
        done(err, response);
      });
    });
  });
};

function protectUsage () {
  
  return '\n\n Example Usage:\n\n ' + format.bold('divshot protect [environment] [username:password]');
}