shx/lib/cli.js

#!/usr/bin/env node
'use strict';

var _shx = require('./shx');

// `input` is null if we're running from a TTY, or a string of all stdin if
// running from the right-hand side of a pipe
var run = function run(input) {
  // Pass stdin to shx as the 'this' parameter
  var code = _shx.shx.call(input, process.argv);

  // Make sure output is flushed before exiting the process. Please see:
  //  - https://github.com/shelljs/shx/issues/85
  //  - https://github.com/mochajs/mocha/issues/333
  var streamCount = 0;
  var streams = [process.stdout, process.stderr];
  streams.forEach(function (stream) {
    streamCount++; // count each stream
    stream.write('', function () {
      streamCount--; // mark each stream as finished
      if (streamCount === 0) process.exit(code);
    });
  });
};

// ShellJS doesn't support input streams, so we have to collect all input first
if (process.stdin.isTTY) {
  // There's no stdin, so we can immediately invoke the ShellJS function
  run(null);
} else {
  (function () {
    // Read all stdin first, and then pass that onto ShellJS
    var chunks = [];
    process.stdin.on('data', function (data) {
      return chunks.push(data);
    });
    process.stdin.on('end', function () {
      return run(chunks.join(''));
    });
  })();
}