You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
157 lines
3.8 KiB
157 lines
3.8 KiB
// 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 EE = require('events') |
|
/* istanbul ignore if */ |
|
if (typeof EE !== 'function') { |
|
EE = EE.EventEmitter |
|
} |
|
|
|
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 = {} |
|
} |
|
|
|
// 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') |
|
|
|
if (loaded === false) { |
|
load() |
|
} |
|
|
|
var ev = 'exit' |
|
if (opts && opts.alwaysLast) { |
|
ev = 'afterexit' |
|
} |
|
|
|
var remove = function () { |
|
emitter.removeListener(ev, cb) |
|
if (emitter.listeners('exit').length === 0 && |
|
emitter.listeners('afterexit').length === 0) { |
|
unload() |
|
} |
|
} |
|
emitter.on(ev, cb) |
|
|
|
return remove |
|
} |
|
|
|
module.exports.unload = unload |
|
function unload () { |
|
if (!loaded) { |
|
return |
|
} |
|
loaded = false |
|
|
|
signals.forEach(function (sig) { |
|
try { |
|
process.removeListener(sig, sigListeners[sig]) |
|
} catch (er) {} |
|
}) |
|
process.emit = originalProcessEmit |
|
process.reallyExit = originalProcessReallyExit |
|
emitter.count -= 1 |
|
} |
|
|
|
function emit (event, code, signal) { |
|
if (emitter.emitted[event]) { |
|
return |
|
} |
|
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 */ |
|
process.kill(process.pid, sig) |
|
} |
|
} |
|
}) |
|
|
|
module.exports.signals = function () { |
|
return signals |
|
} |
|
|
|
module.exports.load = load |
|
|
|
var loaded = false |
|
|
|
function load () { |
|
if (loaded) { |
|
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 |
|
} |
|
|
|
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 originalProcessEmit = process.emit |
|
function processEmit (ev, arg) { |
|
if (ev === 'exit') { |
|
if (arg !== undefined) { |
|
process.exitCode = arg |
|
} |
|
var ret = originalProcessEmit.apply(this, arguments) |
|
emit('exit', process.exitCode, null) |
|
/* istanbul ignore next */ |
|
emit('afterexit', process.exitCode, null) |
|
return ret |
|
} else { |
|
return originalProcessEmit.apply(this, arguments) |
|
} |
|
}
|
|
|