import {once} from 'node:events'; import {DiscardedError} from '../return/final-error.js'; // If `error` is emitted before `spawn`, `exit` will never be emitted. // However, `error` might be emitted after `spawn`. // In that case, `exit` will still be emitted. // Since the `exit` event contains the signal name, we want to make sure we are listening for it. // This function also takes into account the following unlikely cases: // - `exit` being emitted in the same microtask as `spawn` // - `error` being emitted multiple times export const waitForExit = async (subprocess, context) => { const [exitCode, signal] = await waitForExitOrError(subprocess); context.isForcefullyTerminated ??= false; return [exitCode, signal]; }; const waitForExitOrError = async subprocess => { const [spawnPayload, exitPayload] = await Promise.allSettled([ once(subprocess, 'spawn'), once(subprocess, 'exit'), ]); if (spawnPayload.status === 'rejected') { return []; } return exitPayload.status === 'rejected' ? waitForSubprocessExit(subprocess) : exitPayload.value; }; const waitForSubprocessExit = async subprocess => { try { return await once(subprocess, 'exit'); } catch { return waitForSubprocessExit(subprocess); } }; // Retrieve the final exit code and|or signal name export const waitForSuccessfulExit = async exitPromise => { const [exitCode, signal] = await exitPromise; if (!isSubprocessErrorExit(exitCode, signal) && isFailedExit(exitCode, signal)) { throw new DiscardedError(); } return [exitCode, signal]; }; // When the subprocess fails due to an `error` event const isSubprocessErrorExit = (exitCode, signal) => exitCode === undefined && signal === undefined; // When the subprocess fails due to a non-0 exit code or to a signal termination export const isFailedExit = (exitCode, signal) => exitCode !== 0 || signal !== null;