Attempting to work around Node.js colossal "fuck you" re. https://github.com/joyent/node/issues/3584

exit.js

// The idea. This doesn't work, but basically comes
// from https://github.com/gruntjs/grunt/pull/708
var exitCode;
function exit(code) {
  if (exitCode === undefined) {
    exitcode = code || 0;
    tryToExit();
  }
}

function isDrained(stream) {
  var drained = stream.write('');
  if (!drained) { stream.on('drain', tryToExit); }
  return drained;
}

function tryToExit() {
  if (isDrained(process.stdout) && isDrained(process.stderr)) {
    process.exit(exitCode);
  }
}

// Test code
for (var i = 0; i < 10; i++) {
  console.log("testing %d", i);
}

exit(0);

problem.md

Both of these need to work in Windows, using Node.js 0.8.x and 0.10.x

C:\exit>node exit.js
testing 0
testing 1
testing 2
testing 3
testing 4
testing 5
testing 6
testing 7
testing 8
testing 9

C:\exit>node exit.js | find "t"

Ok, I've rewritten what you posted to be (I think) a little more robust. I've tested it in 0.8.25 and 0.10.18 and it appears to work fine with:

• node exit.js 10
• node exit.js 10 2>&1
• node exit.js 10 2>&1 | find "["
• node exit.js 10000
• node exit.js 10000 2>&1
• node exit.js 10000 2>&1 | find "["
• node exit.js 10000 2>&1 | find "log"
• node exit.js 10000 2>&1 | find "err"

What do you think? Can you test it out? If this works, I'll publish it as a standalone lib to npm.

function exit(exitCode) {
  var streams = [process.stdout, process.stderr];
  var drainCount = 0;
  // Actually exit if all streams are drained.
  function tryToExit() {
    if (drainCount === streams.length) {
      process.exit(exitCode);
    }
  }
  streams.forEach(function(stream) {
    // Prevent further writing.
    stream.write = function() {};
    // Count drained streams now, but monitor non-drained streams.
    if (stream.bufferSize === 0) {
      drainCount++;
    } else {
      stream.once('drain', function() {
        drainCount++;
        tryToExit();
      });
    }
  });
  // If all streams were already drained, exit now.
  tryToExit();
}

// Test code
var max = process.argv[2] || 10;
for (var i = 0; i < max; i++) {
  console.log("[log] testing %d", i);
  console.error("[err] testing %d", i);
}

exit(0);

console.log("[log] this shouldn't display");
console.error("[err] this shouldn't display");