123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275 |
- // 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.
- // grab a reference to node's real process object right away
- import { signals } from './signals.js';
- export { signals };
- const processOk = (process) => !!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';
- const kExitEmitter = Symbol.for('signal-exit emitter');
- const global = globalThis;
- const ObjectDefineProperty = Object.defineProperty.bind(Object);
- // teeny special purpose ee
- class Emitter {
- emitted = {
- afterExit: false,
- exit: false,
- };
- listeners = {
- afterExit: [],
- exit: [],
- };
- count = 0;
- id = Math.random();
- constructor() {
- if (global[kExitEmitter]) {
- return global[kExitEmitter];
- }
- ObjectDefineProperty(global, kExitEmitter, {
- value: this,
- writable: false,
- enumerable: false,
- configurable: false,
- });
- }
- on(ev, fn) {
- this.listeners[ev].push(fn);
- }
- removeListener(ev, fn) {
- const list = this.listeners[ev];
- const i = list.indexOf(fn);
- /* c8 ignore start */
- if (i === -1) {
- return;
- }
- /* c8 ignore stop */
- if (i === 0 && list.length === 1) {
- list.length = 0;
- }
- else {
- list.splice(i, 1);
- }
- }
- emit(ev, code, signal) {
- if (this.emitted[ev]) {
- return false;
- }
- this.emitted[ev] = true;
- let ret = false;
- for (const fn of this.listeners[ev]) {
- ret = fn(code, signal) === true || ret;
- }
- if (ev === 'exit') {
- ret = this.emit('afterExit', code, signal) || ret;
- }
- return ret;
- }
- }
- class SignalExitBase {
- }
- const signalExitWrap = (handler) => {
- return {
- onExit(cb, opts) {
- return handler.onExit(cb, opts);
- },
- load() {
- return handler.load();
- },
- unload() {
- return handler.unload();
- },
- };
- };
- class SignalExitFallback extends SignalExitBase {
- onExit() {
- return () => { };
- }
- load() { }
- unload() { }
- }
- class SignalExit extends SignalExitBase {
- // "SIGHUP" throws an `ENOSYS` error on Windows,
- // so use a supported signal instead
- /* c8 ignore start */
- #hupSig = process.platform === 'win32' ? 'SIGINT' : 'SIGHUP';
- /* c8 ignore stop */
- #emitter = new Emitter();
- #process;
- #originalProcessEmit;
- #originalProcessReallyExit;
- #sigListeners = {};
- #loaded = false;
- constructor(process) {
- super();
- this.#process = process;
- // { <signal>: <listener fn>, ... }
- this.#sigListeners = {};
- for (const sig of signals) {
- this.#sigListeners[sig] = () => {
- // 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.
- const listeners = this.#process.listeners(sig);
- let { count } = this.#emitter;
- // This is a workaround for the fact that signal-exit v3 and signal
- // exit v4 are not aware of each other, and each will attempt to let
- // the other handle it, so neither of them do. To correct this, we
- // detect if we're the only handler *except* for previous versions
- // of signal-exit, and increment by the count of listeners it has
- // created.
- /* c8 ignore start */
- const p = process;
- if (typeof p.__signal_exit_emitter__ === 'object' &&
- typeof p.__signal_exit_emitter__.count === 'number') {
- count += p.__signal_exit_emitter__.count;
- }
- /* c8 ignore stop */
- if (listeners.length === count) {
- this.unload();
- const ret = this.#emitter.emit('exit', null, sig);
- /* c8 ignore start */
- const s = sig === 'SIGHUP' ? this.#hupSig : sig;
- if (!ret)
- process.kill(process.pid, s);
- /* c8 ignore stop */
- }
- };
- }
- this.#originalProcessReallyExit = process.reallyExit;
- this.#originalProcessEmit = process.emit;
- }
- onExit(cb, opts) {
- /* c8 ignore start */
- if (!processOk(this.#process)) {
- return () => { };
- }
- /* c8 ignore stop */
- if (this.#loaded === false) {
- this.load();
- }
- const ev = opts?.alwaysLast ? 'afterExit' : 'exit';
- this.#emitter.on(ev, cb);
- return () => {
- this.#emitter.removeListener(ev, cb);
- if (this.#emitter.listeners['exit'].length === 0 &&
- this.#emitter.listeners['afterExit'].length === 0) {
- this.unload();
- }
- };
- }
- load() {
- if (this.#loaded) {
- return;
- }
- this.#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.
- this.#emitter.count += 1;
- for (const sig of signals) {
- try {
- const fn = this.#sigListeners[sig];
- if (fn)
- this.#process.on(sig, fn);
- }
- catch (_) { }
- }
- this.#process.emit = (ev, ...a) => {
- return this.#processEmit(ev, ...a);
- };
- this.#process.reallyExit = (code) => {
- return this.#processReallyExit(code);
- };
- }
- unload() {
- if (!this.#loaded) {
- return;
- }
- this.#loaded = false;
- signals.forEach(sig => {
- const listener = this.#sigListeners[sig];
- /* c8 ignore start */
- if (!listener) {
- throw new Error('Listener not defined for signal: ' + sig);
- }
- /* c8 ignore stop */
- try {
- this.#process.removeListener(sig, listener);
- /* c8 ignore start */
- }
- catch (_) { }
- /* c8 ignore stop */
- });
- this.#process.emit = this.#originalProcessEmit;
- this.#process.reallyExit = this.#originalProcessReallyExit;
- this.#emitter.count -= 1;
- }
- #processReallyExit(code) {
- /* c8 ignore start */
- if (!processOk(this.#process)) {
- return 0;
- }
- this.#process.exitCode = code || 0;
- /* c8 ignore stop */
- this.#emitter.emit('exit', this.#process.exitCode, null);
- return this.#originalProcessReallyExit.call(this.#process, this.#process.exitCode);
- }
- #processEmit(ev, ...args) {
- const og = this.#originalProcessEmit;
- if (ev === 'exit' && processOk(this.#process)) {
- if (typeof args[0] === 'number') {
- this.#process.exitCode = args[0];
- /* c8 ignore start */
- }
- /* c8 ignore start */
- const ret = og.call(this.#process, ev, ...args);
- /* c8 ignore start */
- this.#emitter.emit('exit', this.#process.exitCode, null);
- /* c8 ignore stop */
- return ret;
- }
- else {
- return og.call(this.#process, ev, ...args);
- }
- }
- }
- const process = globalThis.process;
- // wrap so that we call the method on the actual handler, without
- // exporting it directly.
- export const {
- /**
- * Called when the process is exiting, whether via signal, explicit
- * exit, or running out of stuff to do.
- *
- * If the global process object is not suitable for instrumentation,
- * then this will be a no-op.
- *
- * Returns a function that may be used to unload signal-exit.
- */
- onExit,
- /**
- * Load the listeners. Likely you never need to call this, unless
- * doing a rather deep integration with signal-exit functionality.
- * Mostly exposed for the benefit of testing.
- *
- * @internal
- */
- load,
- /**
- * Unload the listeners. Likely you never need to call this, unless
- * doing a rather deep integration with signal-exit functionality.
- * Mostly exposed for the benefit of testing.
- *
- * @internal
- */
- unload, } = signalExitWrap(processOk(process) ? new SignalExit(process) : new SignalExitFallback());
- //# sourceMappingURL=index.js.map
|