@omneedia/socketcluster/bin/cli.js

SocketCluster - A Highly parallelized WebSocket server cluster to make the most of multi-core machines/instances.

#!/usr/bin/env node

var fs = require('fs-extra');
var path = require('path');
var argv = require('minimist')(process.argv.slice(2));
var childProcess = require('child_process');
var inquirer = require('inquirer');
var prompt = inquirer.createPromptModule();
var exec = childProcess.exec;
var spawn = childProcess.spawn;
var fork = childProcess.fork;

var command = argv._[0];
var commandRawArgs = process.argv.slice(3);
var arg1 = argv._[1];
var force = argv.force ? true : false;

var fileExistsSync = function (filePath) {
  try {
    fs.accessSync(filePath, fs.constants.F_OK);
  } catch (err) {
    return false;
  }
  return true;
};

var parsePackageFile = function (moduleDir) {
  var packageFile = moduleDir + '/package.json';
  try {
    if (fileExistsSync(packageFile)) {
      return JSON.parse(fs.readFileSync(packageFile, {encoding: 'utf8'}));
    }
  } catch (e) {}

  return {};
};

var errorMessage = function (message) {
  console.log('\x1b[31m[Error]\x1b[0m ' + message);
};

var successMessage = function (message) {
  console.log('\x1b[32m[Success]\x1b[0m ' + message);
};

var warningMessage = function (message) {
  console.log('\x1b[33m[Warning]\x1b[0m ' + message);
};

var showCorrectUsage = function () {
  console.log('Usage: socketcluster [options] [command]\n');
  console.log('Options:');
  console.log("  -v            Get the version of the current SocketCluster installation");
  console.log('  --help        Get info on how to use this command');
  console.log('  --force       Force all necessary directory modifications without prompts');
  console.log();
  console.log('Commands:');
  console.log('  create <appname>            Create a new boilerplate app in working directory');
};

var failedToRemoveDirMessage = function (dirPath) {
  errorMessage('Failed to remove existing directory at ' + dirPath + '. This directory may be used by another program or you may not have the permission to remove it.');
};

var failedToCreateMessage = function () {
  errorMessage('Failed to create necessary files. Please check your permissions and try again.');
};

var promptConfirm = function (message, callback) {
  prompt([
    {
      type: 'confirm',
      message: message,
      name: 'result'
    }
  ]).then((answers) => {
    callback(answers.result);
  }).catch((err) => {
    errorMessage(err.message);
    process.exit();
  });
};

var copyDirRecursive = function (src, dest) {
  try {
    fs.copySync(src, dest);
    return true;
  } catch (e) {
    failedToCreateMessage();
  }
  return false;
};

var rmdirRecursive = function (dirname) {
  try {
    fs.removeSync(dirname);
    return true;
  } catch (e) {
    failedToRemoveDirMessage(dirname);
  }
  return false;
};

if (argv.help) {
  showCorrectUsage();
  process.exit();
};

if (argv.v) {
  var scDir = __dirname + '/../';
  var scPkg = parsePackageFile(scDir);
  console.log('v' + scPkg.version);
  process.exit();
};

var wd = process.cwd();

var sampleDir = __dirname + '/../sample';
var destDir = path.normalize(wd + '/' + arg1);
var clientFileSourcePath = path.normalize(destDir + '/node_modules/socketcluster-client/socketcluster.js');
var clientFileDestPath = path.normalize(destDir + '/public/socketcluster.js');

var createFail = function () {
  errorMessage("Failed to create SocketCluster sample app.");
  process.exit();
};

var createSuccess = function () {
  console.log('Installing app dependencies using npm. This could take a while...');

  var npmCommand = (process.platform === "win32" ? "npm.cmd" : "npm");
  var options = {
    cwd: destDir,
    maxBuffer: Infinity
  };

  var npmProcess = spawn(npmCommand, ['install'], options);

  npmProcess.stdout.on('data', function (data) {
    process.stdout.write(data);
  });

  npmProcess.stderr.on('data', function (data) {
    process.stderr.write(data);
  });

  npmProcess.on('close', function (code) {
    if (code) {
      errorMessage('Failed to install npm dependencies. Exited with code ' + code + '.');
    } else {
      try {
        fs.writeFileSync(clientFileDestPath, fs.readFileSync(clientFileSourcePath));
        successMessage("SocketCluster sample '" + destDir + "' was setup successfully.");
      } catch (err) {
        warningMessage("Failed to copy file from '" + clientFileSourcePath + "' to '" +
          clientFileDestPath + "' - Try copying it manually.");
      }
    }
    process.exit(code);
  });

  npmProcess.stdin.end();
};

var setupMessage = function () {
  console.log('Creating app structure...');
};

var confirmReplaceSetup = function (confirm) {
  if (confirm) {
    setupMessage();
    if (rmdirRecursive(destDir) && copyDirRecursive(sampleDir, destDir)) {
      createSuccess();
    } else {
      createFail();
    }
  } else {
    errorMessage("SocketCluster 'create' action was aborted.");
    process.exit();
  }
};

if (command == 'create') {
  if (arg1) {
    if (fileExistsSync(destDir)) {
      if (force) {
        confirmReplaceSetup(true);
      } else {
        var message = "There is already a directory at " + destDir + '. Do you want to overwrite it?';
        promptConfirm(message, confirmReplaceSetup);
      }
    } else {
      setupMessage();
      if (copyDirRecursive(sampleDir, destDir)) {
        createSuccess();
      } else {
        createFail();
      }
    }
  } else {
    errorMessage("The 'create' command requires a valid <appname> as argument.");
    showCorrectUsage();
    process.exit();
  }
} else {
  errorMessage("'" + command + "' is not a valid SocketCluster command.");
  showCorrectUsage();
  process.exit();
}