| Leo Repp | 58b9f11 | 2021-11-22 11:57:47 +0100 | [diff] [blame^] | 1 | /* |
| 2 | * exit |
| 3 | * https://github.com/cowboy/node-exit |
| 4 | * |
| 5 | * Copyright (c) 2013 "Cowboy" Ben Alman |
| 6 | * Licensed under the MIT license. |
| 7 | */ |
| 8 | |
| 9 | 'use strict'; |
| 10 | |
| 11 | module.exports = function exit(exitCode, streams) { |
| 12 | if (!streams) { streams = [process.stdout, process.stderr]; } |
| 13 | var drainCount = 0; |
| 14 | // Actually exit if all streams are drained. |
| 15 | function tryToExit() { |
| 16 | if (drainCount === streams.length) { |
| 17 | process.exit(exitCode); |
| 18 | } |
| 19 | } |
| 20 | streams.forEach(function(stream) { |
| 21 | // Count drained streams now, but monitor non-drained streams. |
| 22 | if (stream.bufferSize === 0) { |
| 23 | drainCount++; |
| 24 | } else { |
| 25 | stream.write('', 'utf-8', function() { |
| 26 | drainCount++; |
| 27 | tryToExit(); |
| 28 | }); |
| 29 | } |
| 30 | // Prevent further writing. |
| 31 | stream.write = function() {}; |
| 32 | }); |
| 33 | // If all streams were already drained, exit now. |
| 34 | tryToExit(); |
| 35 | // In Windows, when run as a Node.js child process, a script utilizing |
| 36 | // this library might just exit with a 0 exit code, regardless. This code, |
| 37 | // despite the fact that it looks a bit crazy, appears to fix that. |
| 38 | process.on('exit', function() { |
| 39 | process.exit(exitCode); |
| 40 | }); |
| 41 | }; |