cli.js

#!/usr/bin/env node
var path = require('path'),
  fs = require('fs'),
  program = require('commander'),
  ini = require('ini'),
  extend = require('extend'),
  async = require('async'),
  request = require('request'),
  exec = require('child_process').exec,
  registries = require('./registries.json'),
  PKG = require('./package.json'),
  CGRRC = path.join(process.env.HOME, '.cgrrc'),
  npmRe = {
    get: 'npm config get registry',
    set: 'npm config set registry'
  },
  yarnRe = {
    get: 'yarn config get registry',
    set: 'yarn config set registry'
  };

program.version(PKG.version);

program
  .command('ls')
  .description('List all the registries')
  .action(onList);

program
  .command('current')
  .description('Show current registry name')
  .action(showCurrent);

program
  .command('use <registry> [type]')
  .description('Change registry to registry')
  .action(onUse);

program
  .command('add <registry> <url> [home]')
  .description('Add one custom registry')
  .action(onAdd);

program
  .command('del <registry>')
  .description('Delete one custom registry')
  .action(onDel);

program
  .command('test [registry]')
  .description('Show response time for specific or all registries')
  .action(onTest);

program
  .command('help')
  .description('Print this help')
  .action(function() {
    program.outputHelp();
  });

program.parse(process.argv);

if (process.argv.length === 2) {
  program.outputHelp();
}

/*//////////////// cmd methods /////////////////*/

function onList() {
  getCurrentRegistry(function(curArr) {
    var info = [''],
      allRegistries = getAllRegistry();

    Object.keys(allRegistries).forEach(function(key) {
      var item = allRegistries[key],
        registry = String(item.registry),
        prefixIndex = curArr.indexOf(registry),
        prefix =
          prefixIndex === -1
            ? '  '
            : `${curArr.length === 1 ? '*' : prefixIndex === 0 ? 'N' : 'Y'} `;
      info.push(prefix + key + line(key, 8) + registry);
    });

    info.push('');
    printMsg(info);
  });
}

function showCurrent() {
  getCurrentRegistry(function(curArr) {
    var info = [''],
      allRegistries = getAllRegistry();

    Object.keys(allRegistries).forEach(function(key) {
      var item = allRegistries[key],
        registry = String(item.registry),
        prefixIndex = curArr.indexOf(registry);
      if (prefixIndex !== -1) {
        info.push(
          `${curArr.length === 1 ? '*' : prefixIndex === 0 ? 'N' : 'Y'} ${key}${line(
            key,
            8
          )}${registry}`
        );
      }
    });

    info.push('');
    printMsg(info);
  });
}

function onUse(name, type) {
  var allRegistries = getAllRegistry();
  if (allRegistries.hasOwnProperty(name)) {
    var registry = allRegistries[name],
      info = [''];
    if (!type) {
      exec(`${npmRe.set} ${registry.registry}`, function(errN, stdoutN, stderrN) {
        exec(`${yarnRe.set} ${registry.registry}`, function(errY, stdoutY, stderrY) {
          if (errN && errY) return exit([stderrN, stderrY]);
          if (errN) info.push(stderrN);
          if (errY) info.push(stderrY);
          if (!errN) info.push(`   npm registry has been set to: ${registry.registry}`);
          if (!errY) info.push(`   yarn registry has been set to: ${registry.registry}`);
          info.push('');
          printMsg(info);
        });
      });
    } else {
      var smType = type.toLowerCase();
      if (smType === 'npm' || smType === 'n') {
        exec(`${npmRe.set} ${registry.registry}`, function(err, stdout, stderr) {
          if (err) return exit([stderr]);
          info.push(`   npm registry has been set to: ${registry.registry}`);
          info.push('');
          printMsg(info);
        });
      } else if (smType === 'yarn' || smType === 'y') {
        exec(`${yarnRe.set} ${registry.registry}`, function(err, stdout, stderr) {
          if (err) return exit([stderr]);
          info.push(`   yarn registry has been set to: ${registry.registry}`);
          info.push('');
          printMsg(info);
        });
      } else {
        info.push('   cgr use <registry> [type]');
        info.push('   type must be oneOf yarn | y | npm | n');
        info.push('');
        printMsg(info);
      }
    }
  } else {
    printMsg(['', '   Not find registry: ' + name, '']);
  }
}

function onDel(name) {
  var customRegistries = getCustomRegistry();
  if (!customRegistries.hasOwnProperty(name)) return;
  getCurrentRegistry(function(curArr) {
    if (curArr.indexOf(customRegistries[name].registry) !== -1) {
      onUse('npm');
    }
    delete customRegistries[name];
    setCustomRegistry(customRegistries, function(err) {
      if (err) return exit([err]);
      printMsg(['', '    delete registry ' + name + ' success', '']);
    });
  });
}

function onAdd(name, url, home) {
  var customRegistries = getCustomRegistry();
  if (customRegistries.hasOwnProperty(name)) return;
  var config = (customRegistries[name] = {});
  if (url[url.length - 1] !== '/') url += '/'; // ensure url end with /
  config.registry = url;
  if (home) {
    config.home = home;
  }
  setCustomRegistry(customRegistries, function(err) {
    if (err) return exit([err]);
    printMsg(['', '    add registry ' + name + ' success', '']);
  });
}

function onTest(registry) {
  var allRegistries = getAllRegistry(),
    toTest;

  if (registry) {
    if (!allRegistries.hasOwnProperty(registry)) {
      return;
    }
    toTest = { [registry]: allRegistries[registry] };
  } else {
    toTest = allRegistries;
  }

  async.map(
    Object.keys(toTest),
    function(name, cbk) {
      var registry = toTest[name],
        start = +new Date();
      request(registry.registry + 'pedding', function(error) {
        cbk(null, {
          name: name,
          registry: registry.registry,
          time: +new Date() - start,
          error: error ? true : false
        });
      });
    },
    function(err, results) {
      getCurrentRegistry(function(curArr) {
        var msg = [''];
        results.forEach(function(result) {
          var registry = String(result.registry),
            prefixIndex = curArr.indexOf(registry),
            prefix =
              prefixIndex === -1
                ? '  '
                : `${curArr.length === 1 ? '*' : prefixIndex === 0 ? 'N' : 'Y'} `,
            suffix = result.error ? 'Fetch Error' : result.time + 'ms';
          msg.push(prefix + result.name + line(result.name, 8) + suffix);
        });
        msg.push('');
        printMsg(msg);
      });
    }
  );
}

/*//////////////// helper methods /////////////////*/

/*
 * get current registry
 */
function getCurrentRegistry(cbk) {
  exec(npmRe.get, function(errN, stdoutN, stderrN) {
    exec(yarnRe.get, function(errY, stdoutY, stderrY) {
      if (errN && errY) return exit([stderrN, stderrY]);
      if (errN) console.log(stderrN);
      if (errY) console.log(stderrY);
      var npmRegistry = stdoutN.trim(),
        yarnRegistry = stdoutY.trim();
      if (npmRegistry[npmRegistry.length - 1] !== '/') {
        npmRegistry = npmRegistry + '/';
      }
      if (yarnRegistry[yarnRegistry.length - 1] !== '/') {
        yarnRegistry = yarnRegistry + '/';
      }
      if (npmRegistry === yarnRegistry) {
        cbk([npmRegistry]);
      } else {
        cbk([npmRegistry, yarnRegistry]);
      }
    });
  });
}

function getCustomRegistry() {
  return fs.existsSync(CGRRC) ? ini.parse(fs.readFileSync(CGRRC, 'utf-8')) : {};
}

function setCustomRegistry(config, cbk) {
  fs.writeFile(CGRRC, ini.stringify(config), 'utf-8', cbk);
}

function getAllRegistry() {
  return extend({}, registries, getCustomRegistry());
}

function printMsg(infos) {
  infos.forEach(function(info) {
    console.log(info);
  });
}

/*
 * print message & exit
 */
function exit(errs) {
  printMsg(errs);
  process.exit(1);
}

function line(str, len) {
  var line = new Array(Math.max(1, len - str.length)).join('-');
  return ' ' + line + ' ';
}