File size: 3,079 Bytes
f500658 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 | interface StdoutTakeoverState {
rawStdoutWrite: (chunk: string, callback?: (error?: Error | null) => void) => boolean;
rawStderrWrite: (chunk: string, callback?: (error?: Error | null) => void) => boolean;
originalStdoutWrite: typeof process.stdout.write;
}
let stdoutTakeoverState: StdoutTakeoverState | undefined;
const RAW_STDOUT_RETRY_DELAY_MS = 10;
let rawStdoutWriteTail: Promise<void> = Promise.resolve();
function getRawStdoutWrite(): StdoutTakeoverState["rawStdoutWrite"] {
if (stdoutTakeoverState) {
return stdoutTakeoverState.rawStdoutWrite;
}
return process.stdout.write.bind(process.stdout) as StdoutTakeoverState["rawStdoutWrite"];
}
async function writeRawStdoutChunk(text: string): Promise<void> {
while (true) {
try {
await new Promise<void>((resolve, reject) => {
try {
getRawStdoutWrite()(text, (error) => {
if (error) reject(error);
else resolve();
});
} catch (error) {
reject(error instanceof Error ? error : new Error(String(error)));
}
});
return;
} catch (error) {
const writeError = error instanceof Error ? error : new Error(String(error));
const code = (writeError as Error & { code?: unknown }).code;
if (code !== "ENOBUFS" && code !== "EAGAIN" && code !== "EWOULDBLOCK") {
throw writeError;
}
await new Promise<void>((resolve) => setTimeout(resolve, RAW_STDOUT_RETRY_DELAY_MS));
}
}
}
export function takeOverStdout(): void {
if (stdoutTakeoverState) {
return;
}
const rawStdoutWrite = process.stdout.write.bind(process.stdout) as StdoutTakeoverState["rawStdoutWrite"];
const rawStderrWrite = process.stderr.write.bind(process.stderr) as StdoutTakeoverState["rawStderrWrite"];
const originalStdoutWrite = process.stdout.write;
process.stdout.write = ((
chunk: string | Uint8Array,
encodingOrCallback?: BufferEncoding | ((error?: Error | null) => void),
callback?: (error?: Error | null) => void,
): boolean => {
if (typeof encodingOrCallback === "function") {
return rawStderrWrite(String(chunk), encodingOrCallback);
}
return rawStderrWrite(String(chunk), callback);
}) as typeof process.stdout.write;
stdoutTakeoverState = {
rawStdoutWrite,
rawStderrWrite,
originalStdoutWrite,
};
}
export function restoreStdout(): void {
if (!stdoutTakeoverState) {
return;
}
process.stdout.write = stdoutTakeoverState.originalStdoutWrite;
stdoutTakeoverState = undefined;
}
export function isStdoutTakenOver(): boolean {
return stdoutTakeoverState !== undefined;
}
export function writeRawStdout(text: string): void {
if (text.length === 0) {
return;
}
rawStdoutWriteTail = rawStdoutWriteTail.then(() => writeRawStdoutChunk(text));
void rawStdoutWriteTail.catch(() => {
process.exit(1);
});
}
export async function waitForRawStdoutBackpressure(): Promise<void> {
while (true) {
const tail = rawStdoutWriteTail;
await tail;
if (tail === rawStdoutWriteTail) {
return;
}
}
}
export async function flushRawStdout(): Promise<void> {
await waitForRawStdoutBackpressure();
await writeRawStdoutChunk("");
}
|