-
Notifications
You must be signed in to change notification settings - Fork 3.2k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
5 changed files
with
1,043 additions
and
1,052 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,163 +1,200 @@ | ||
| // Note: since nyc uses this module to output coverage, any lines | ||
| // that are in the direct sync flow of nyc's outputCoverage are | ||
| // ignored, since we can never get coverage for them. | ||
| var assert = require('assert') | ||
| var signals = require('./signals.js') | ||
| var isWin = /^win/i.test(process.platform) | ||
|
|
||
| var EE = require('events') | ||
| /* istanbul ignore if */ | ||
| if (typeof EE !== 'function') { | ||
| EE = EE.EventEmitter | ||
| // grab a reference to node's real process object right away | ||
| var process = global.process | ||
|
|
||
| const processOk = function (process) { | ||
| return process && | ||
| typeof process === 'object' && | ||
| typeof process.removeListener === 'function' && | ||
| typeof process.emit === 'function' && | ||
| typeof process.reallyExit === 'function' && | ||
| typeof process.listeners === 'function' && | ||
| typeof process.kill === 'function' && | ||
| typeof process.pid === 'number' && | ||
| typeof process.on === 'function' | ||
| } | ||
|
|
||
| var emitter | ||
| if (process.__signal_exit_emitter__) { | ||
| emitter = process.__signal_exit_emitter__ | ||
| // some kind of non-node environment, just no-op | ||
| /* istanbul ignore if */ | ||
| if (!processOk(process)) { | ||
| module.exports = function () {} | ||
| } else { | ||
| emitter = process.__signal_exit_emitter__ = new EE() | ||
| emitter.count = 0 | ||
| emitter.emitted = {} | ||
| } | ||
|
|
||
| // Because this emitter is a global, we have to check to see if a | ||
| // previous version of this library failed to enable infinite listeners. | ||
| // I know what you're about to say. But literally everything about | ||
| // signal-exit is a compromise with evil. Get used to it. | ||
| if (!emitter.infinite) { | ||
| emitter.setMaxListeners(Infinity) | ||
| emitter.infinite = true | ||
| } | ||
|
|
||
| module.exports = function (cb, opts) { | ||
| assert.equal(typeof cb, 'function', 'a callback must be provided for exit handler') | ||
| var assert = require('assert') | ||
| var signals = require('./signals.js') | ||
| var isWin = /^win/i.test(process.platform) | ||
|
|
||
| var EE = require('events') | ||
| /* istanbul ignore if */ | ||
| if (typeof EE !== 'function') { | ||
| EE = EE.EventEmitter | ||
| } | ||
|
|
||
| if (loaded === false) { | ||
| load() | ||
| var emitter | ||
| if (process.__signal_exit_emitter__) { | ||
| emitter = process.__signal_exit_emitter__ | ||
| } else { | ||
| emitter = process.__signal_exit_emitter__ = new EE() | ||
| emitter.count = 0 | ||
| emitter.emitted = {} | ||
| } | ||
|
|
||
| var ev = 'exit' | ||
| if (opts && opts.alwaysLast) { | ||
| ev = 'afterexit' | ||
| // Because this emitter is a global, we have to check to see if a | ||
| // previous version of this library failed to enable infinite listeners. | ||
| // I know what you're about to say. But literally everything about | ||
| // signal-exit is a compromise with evil. Get used to it. | ||
| if (!emitter.infinite) { | ||
| emitter.setMaxListeners(Infinity) | ||
| emitter.infinite = true | ||
| } | ||
|
|
||
| var remove = function () { | ||
| emitter.removeListener(ev, cb) | ||
| if (emitter.listeners('exit').length === 0 && | ||
| emitter.listeners('afterexit').length === 0) { | ||
| unload() | ||
| module.exports = function (cb, opts) { | ||
| /* istanbul ignore if */ | ||
| if (!processOk(global.process)) { | ||
| return | ||
| } | ||
| } | ||
| emitter.on(ev, cb) | ||
| assert.equal(typeof cb, 'function', 'a callback must be provided for exit handler') | ||
|
|
||
| return remove | ||
| } | ||
| if (loaded === false) { | ||
| load() | ||
| } | ||
|
|
||
| module.exports.unload = unload | ||
| function unload () { | ||
| if (!loaded) { | ||
| return | ||
| } | ||
| loaded = false | ||
| var ev = 'exit' | ||
| if (opts && opts.alwaysLast) { | ||
| ev = 'afterexit' | ||
| } | ||
|
|
||
| signals.forEach(function (sig) { | ||
| try { | ||
| process.removeListener(sig, sigListeners[sig]) | ||
| } catch (er) {} | ||
| }) | ||
| process.emit = originalProcessEmit | ||
| process.reallyExit = originalProcessReallyExit | ||
| emitter.count -= 1 | ||
| } | ||
| var remove = function () { | ||
| emitter.removeListener(ev, cb) | ||
| if (emitter.listeners('exit').length === 0 && | ||
| emitter.listeners('afterexit').length === 0) { | ||
| unload() | ||
| } | ||
| } | ||
| emitter.on(ev, cb) | ||
|
|
||
| function emit (event, code, signal) { | ||
| if (emitter.emitted[event]) { | ||
| return | ||
| return remove | ||
| } | ||
| emitter.emitted[event] = true | ||
| emitter.emit(event, code, signal) | ||
| } | ||
|
|
||
| // { <signal>: <listener fn>, ... } | ||
| var sigListeners = {} | ||
| signals.forEach(function (sig) { | ||
| sigListeners[sig] = function listener () { | ||
| // If there are no other listeners, an exit is coming! | ||
| // Simplest way: remove us and then re-send the signal. | ||
| // We know that this will kill the process, so we can | ||
| // safely emit now. | ||
| var listeners = process.listeners(sig) | ||
| if (listeners.length === emitter.count) { | ||
| unload() | ||
| emit('exit', null, sig) | ||
| /* istanbul ignore next */ | ||
| emit('afterexit', null, sig) | ||
| /* istanbul ignore next */ | ||
| if (isWin && sig === 'SIGHUP') { | ||
| // "SIGHUP" throws an `ENOSYS` error on Windows, | ||
| // so use a supported signal instead | ||
| sig = 'SIGINT' | ||
| } | ||
| process.kill(process.pid, sig) | ||
| var unload = function unload () { | ||
| if (!loaded || !processOk(global.process)) { | ||
| return | ||
| } | ||
| loaded = false | ||
|
|
||
| signals.forEach(function (sig) { | ||
| try { | ||
| process.removeListener(sig, sigListeners[sig]) | ||
| } catch (er) {} | ||
| }) | ||
| process.emit = originalProcessEmit | ||
| process.reallyExit = originalProcessReallyExit | ||
| emitter.count -= 1 | ||
| } | ||
| }) | ||
|
|
||
| module.exports.signals = function () { | ||
| return signals | ||
| } | ||
| module.exports.unload = unload | ||
|
|
||
| module.exports.load = load | ||
|
|
||
| var loaded = false | ||
|
|
||
| function load () { | ||
| if (loaded) { | ||
| return | ||
| var emit = function emit (event, code, signal) { | ||
| /* istanbul ignore if */ | ||
| if (emitter.emitted[event]) { | ||
| return | ||
| } | ||
| emitter.emitted[event] = true | ||
| emitter.emit(event, code, signal) | ||
| } | ||
| loaded = true | ||
|
|
||
| // This is the number of onSignalExit's that are in play. | ||
| // It's important so that we can count the correct number of | ||
| // listeners on signals, and don't wait for the other one to | ||
| // handle it instead of us. | ||
| emitter.count += 1 | ||
|
|
||
| signals = signals.filter(function (sig) { | ||
| try { | ||
| process.on(sig, sigListeners[sig]) | ||
| return true | ||
| } catch (er) { | ||
| return false | ||
|
|
||
| // { <signal>: <listener fn>, ... } | ||
| var sigListeners = {} | ||
| signals.forEach(function (sig) { | ||
| sigListeners[sig] = function listener () { | ||
| /* istanbul ignore if */ | ||
| if (!processOk(global.process)) { | ||
| return | ||
| } | ||
| // If there are no other listeners, an exit is coming! | ||
| // Simplest way: remove us and then re-send the signal. | ||
| // We know that this will kill the process, so we can | ||
| // safely emit now. | ||
| var listeners = process.listeners(sig) | ||
| if (listeners.length === emitter.count) { | ||
| unload() | ||
| emit('exit', null, sig) | ||
| /* istanbul ignore next */ | ||
| emit('afterexit', null, sig) | ||
| /* istanbul ignore next */ | ||
| if (isWin && sig === 'SIGHUP') { | ||
| // "SIGHUP" throws an `ENOSYS` error on Windows, | ||
| // so use a supported signal instead | ||
| sig = 'SIGINT' | ||
| } | ||
| /* istanbul ignore next */ | ||
| process.kill(process.pid, sig) | ||
| } | ||
| } | ||
| }) | ||
|
|
||
| process.emit = processEmit | ||
| process.reallyExit = processReallyExit | ||
| } | ||
| module.exports.signals = function () { | ||
| return signals | ||
| } | ||
|
|
||
| var originalProcessReallyExit = process.reallyExit | ||
| function processReallyExit (code) { | ||
| process.exitCode = code || 0 | ||
| emit('exit', process.exitCode, null) | ||
| /* istanbul ignore next */ | ||
| emit('afterexit', process.exitCode, null) | ||
| /* istanbul ignore next */ | ||
| originalProcessReallyExit.call(process, process.exitCode) | ||
| } | ||
| var loaded = false | ||
|
|
||
| var load = function load () { | ||
| if (loaded || !processOk(global.process)) { | ||
| return | ||
| } | ||
| loaded = true | ||
|
|
||
| // This is the number of onSignalExit's that are in play. | ||
| // It's important so that we can count the correct number of | ||
| // listeners on signals, and don't wait for the other one to | ||
| // handle it instead of us. | ||
| emitter.count += 1 | ||
|
|
||
| signals = signals.filter(function (sig) { | ||
| try { | ||
| process.on(sig, sigListeners[sig]) | ||
| return true | ||
| } catch (er) { | ||
| return false | ||
| } | ||
| }) | ||
|
|
||
| process.emit = processEmit | ||
| process.reallyExit = processReallyExit | ||
| } | ||
| module.exports.load = load | ||
|
|
||
| var originalProcessEmit = process.emit | ||
| function processEmit (ev, arg) { | ||
| if (ev === 'exit') { | ||
| if (arg !== undefined) { | ||
| process.exitCode = arg | ||
| var originalProcessReallyExit = process.reallyExit | ||
| var processReallyExit = function processReallyExit (code) { | ||
| /* istanbul ignore if */ | ||
| if (!processOk(global.process)) { | ||
| return | ||
| } | ||
| var ret = originalProcessEmit.apply(this, arguments) | ||
| process.exitCode = code || /* istanbul ignore next */ 0 | ||
| emit('exit', process.exitCode, null) | ||
| /* istanbul ignore next */ | ||
| emit('afterexit', process.exitCode, null) | ||
| return ret | ||
| } else { | ||
| return originalProcessEmit.apply(this, arguments) | ||
| /* istanbul ignore next */ | ||
| originalProcessReallyExit.call(process, process.exitCode) | ||
| } | ||
|
|
||
| var originalProcessEmit = process.emit | ||
| var processEmit = function processEmit (ev, arg) { | ||
| if (ev === 'exit' && processOk(global.process)) { | ||
| /* istanbul ignore else */ | ||
| if (arg !== undefined) { | ||
| process.exitCode = arg | ||
| } | ||
| var ret = originalProcessEmit.apply(this, arguments) | ||
| /* istanbul ignore next */ | ||
| emit('exit', process.exitCode, null) | ||
| /* istanbul ignore next */ | ||
| emit('afterexit', process.exitCode, null) | ||
| /* istanbul ignore next */ | ||
| return ret | ||
| } else { | ||
| return originalProcessEmit.apply(this, arguments) | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.