sundry/cli/index.js

A dynamically configurable, load balancing, reverse proxy.

#! /usr/bin/env node
process.stdout.write('\033c');

//var env = require('../lib/config');

var app = require('commander');
var _ = require('lodash');
var chalk = require('chalk');
var path = require('path');
var url = require('url');
var logger = require(path.join(__dirname, '../','lib', 'logger'))
var redis = require('redis');
var packageVersion = require(path.join(__dirname, '../', 'package.json')).version;

app
  .version(packageVersion)
  .option('-i --interactive', 'Start interactive management mode.');

app
  .command('start')
  .description('Start the Sundry proxy server.')
  .action(function(env, options){
    require(path.join(__dirname, '../index'))
  })
  .on('--help', function(){
    console.log('  Example:');
    console.log();
    console.log('    sundry start');
    console.log();
  });

app
  .command('list [host]')
  .description('Quickly view the routes registered with the Sundry server. ' +
  'Optionally list backend routes for [host].')
  .action(function(host){
    connectRedis(function(red){
      if(host){
        var fhost = 'route:' + host;
        return red.listBackends(fhost).then(function(keys){
          console.log(chalk.green('Registered backends for ' + host));
          _.each(keys, function(k){
            console.log(chalk.green('  '+ k));
          });
          red.quit()
        })
      }
      return red.listHosts().then(function(keys){
        console.log(chalk.green('Registered hosts'));
        _.each(keys, function(k){
          console.log(chalk.green('  '+ k.split(':')[1]));
        });
        red.quit()
      })
    });
  })
  .on('--help', function(){
    console.log('  Example:');
    console.log();
    console.log('    list');
    console.log('    list some.host.com')
  });

app
  .command('add <host> <ip>')
  .description('Quickly add a host and backend.')
  .action(function(host, ip){
    connectRedis(function(red){
      red.addHost(host, ip).then(function(status){
        if(status){
          console.log(chalk.green('Added host '+ host));
          return red.quit()
        }
        console.log( chalk.red(
          'Failed to add host ' + host +
          '. Does it already exist?'
        ));
        return red.quit()
      })
    })
  })
  .on('--help', function(){
    console.log('  Example:');
    console.log();
    console.log('    add some.host.com 127.0.0.1:10256')
  });

app
  .command('config [action] [key] [val]')
  .description('Allows editing of the config file located in ~/.sundry')
  .action(function(action, k, v){
    switch(action) {
      case 'build':
        logger.enableLogging(true)
        require('./../lib/Configuration/configManager').buildConfig(function(err, created){
          if(err){
            console.log('Error: ' + err.message)
          } else {
            if(created){
              console.log('Success: Config file created in ' + process.env.HOME + './sundry')
            }
          }
        })
        break;
      case 'edit':
        require('./../lib/Configuration/configManager').editConfig(k,v, function(err, success){
          if(err){
            console.log('Error: ' + err.message)
          } else {
            if(success){
              console.log(success)
            }
          }
        })
        break
      case 'list':
        require('./../lib/Configuration/configManager').list()
        break
      default:
        console.log('No options given to config command');
        console.log('sundry config --help for more info.')

    }
  })
  .on('--help', function(){
    console.log('  Example:');
    console.log();
    console.log('    sundry config build');
    console.log('      Builds skeleton config in ~/.sundry');
    console.log();
    console.log('    sundry config edit [key] [value]');
    console.log('      Sets config [key] to [value].');
    console.log();
    console.log('    sundry config list');
    console.log('      Lists config keys, values.');
    console.log();
  });

app.parse(process.argv);

if(!process.argv.slice(2).length){
  app.outputHelp()
}

if(app.interactive){
  require('./interactive')
}

function connectRedis(cb){
  var config = require('../lib/Configuration/configLoader');
  config.on('ready', function(){
    var env = config.loadedConfig
    var redisHost = url.parse(env.sundry_redis_url);
    var client = redis.createClient(redisHost.port, redisHost.hostname, {max_attempts: 1});
    client.on('error', redisError)
    var redisModel = require('./redis')(client)
    client.on('ready', function() {
      cb(redisModel)
    })
  })
}

function redisError(e){
  console.log(chalk.red("Please make sure the value of 'sundry_redis_url' is correct in ~/.sundry/config.json \n" +
  "Or the sundry_redis_url environment variable is set to a valid redis url."))
}