134 lines
3.4 KiB
JavaScript
134 lines
3.4 KiB
JavaScript
|
import {createReadStream, readFileSync} from 'node:fs';
|
||
|
import {setTimeout} from 'node:timers/promises';
|
||
|
import {isStream} from 'is-stream';
|
||
|
import getStream, {getStreamAsBuffer} from 'get-stream';
|
||
|
import mergeStream from 'merge-stream';
|
||
|
|
||
|
const validateInputOptions = input => {
|
||
|
if (input !== undefined) {
|
||
|
throw new TypeError('The `input` and `inputFile` options cannot be both set.');
|
||
|
}
|
||
|
};
|
||
|
|
||
|
const getInputSync = ({input, inputFile}) => {
|
||
|
if (typeof inputFile !== 'string') {
|
||
|
return input;
|
||
|
}
|
||
|
|
||
|
validateInputOptions(input);
|
||
|
return readFileSync(inputFile);
|
||
|
};
|
||
|
|
||
|
// `input` and `inputFile` option in sync mode
|
||
|
export const handleInputSync = options => {
|
||
|
const input = getInputSync(options);
|
||
|
|
||
|
if (isStream(input)) {
|
||
|
throw new TypeError('The `input` option cannot be a stream in sync mode');
|
||
|
}
|
||
|
|
||
|
return input;
|
||
|
};
|
||
|
|
||
|
const getInput = ({input, inputFile}) => {
|
||
|
if (typeof inputFile !== 'string') {
|
||
|
return input;
|
||
|
}
|
||
|
|
||
|
validateInputOptions(input);
|
||
|
return createReadStream(inputFile);
|
||
|
};
|
||
|
|
||
|
// `input` and `inputFile` option in async mode
|
||
|
export const handleInput = (spawned, options) => {
|
||
|
const input = getInput(options);
|
||
|
|
||
|
if (input === undefined) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
if (isStream(input)) {
|
||
|
input.pipe(spawned.stdin);
|
||
|
} else {
|
||
|
spawned.stdin.end(input);
|
||
|
}
|
||
|
};
|
||
|
|
||
|
// `all` interleaves `stdout` and `stderr`
|
||
|
export const makeAllStream = (spawned, {all}) => {
|
||
|
if (!all || (!spawned.stdout && !spawned.stderr)) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
const mixed = mergeStream();
|
||
|
|
||
|
if (spawned.stdout) {
|
||
|
mixed.add(spawned.stdout);
|
||
|
}
|
||
|
|
||
|
if (spawned.stderr) {
|
||
|
mixed.add(spawned.stderr);
|
||
|
}
|
||
|
|
||
|
return mixed;
|
||
|
};
|
||
|
|
||
|
// On failure, `result.stdout|stderr|all` should contain the currently buffered stream
|
||
|
const getBufferedData = async (stream, streamPromise) => {
|
||
|
// When `buffer` is `false`, `streamPromise` is `undefined` and there is no buffered data to retrieve
|
||
|
if (!stream || streamPromise === undefined) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
// Wait for the `all` stream to receive the last chunk before destroying the stream
|
||
|
await setTimeout(0);
|
||
|
|
||
|
stream.destroy();
|
||
|
|
||
|
try {
|
||
|
return await streamPromise;
|
||
|
} catch (error) {
|
||
|
return error.bufferedData;
|
||
|
}
|
||
|
};
|
||
|
|
||
|
const getStreamPromise = (stream, {encoding, buffer, maxBuffer}) => {
|
||
|
if (!stream || !buffer) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
// eslint-disable-next-line unicorn/text-encoding-identifier-case
|
||
|
if (encoding === 'utf8' || encoding === 'utf-8') {
|
||
|
return getStream(stream, {maxBuffer});
|
||
|
}
|
||
|
|
||
|
if (encoding === null || encoding === 'buffer') {
|
||
|
return getStreamAsBuffer(stream, {maxBuffer});
|
||
|
}
|
||
|
|
||
|
return applyEncoding(stream, maxBuffer, encoding);
|
||
|
};
|
||
|
|
||
|
const applyEncoding = async (stream, maxBuffer, encoding) => {
|
||
|
const buffer = await getStreamAsBuffer(stream, {maxBuffer});
|
||
|
return buffer.toString(encoding);
|
||
|
};
|
||
|
|
||
|
// Retrieve result of child process: exit code, signal, error, streams (stdout/stderr/all)
|
||
|
export const getSpawnedResult = async ({stdout, stderr, all}, {encoding, buffer, maxBuffer}, processDone) => {
|
||
|
const stdoutPromise = getStreamPromise(stdout, {encoding, buffer, maxBuffer});
|
||
|
const stderrPromise = getStreamPromise(stderr, {encoding, buffer, maxBuffer});
|
||
|
const allPromise = getStreamPromise(all, {encoding, buffer, maxBuffer: maxBuffer * 2});
|
||
|
|
||
|
try {
|
||
|
return await Promise.all([processDone, stdoutPromise, stderrPromise, allPromise]);
|
||
|
} catch (error) {
|
||
|
return Promise.all([
|
||
|
{error, signal: error.signal, timedOut: error.timedOut},
|
||
|
getBufferedData(stdout, stdoutPromise),
|
||
|
getBufferedData(stderr, stderrPromise),
|
||
|
getBufferedData(all, allPromise),
|
||
|
]);
|
||
|
}
|
||
|
};
|