| 1 | 'use strict';
|
|---|
| 2 | const isStream = require('is-stream');
|
|---|
| 3 | const getStream = require('get-stream');
|
|---|
| 4 | const mergeStream = require('merge-stream');
|
|---|
| 5 |
|
|---|
| 6 | // `input` option
|
|---|
| 7 | const handleInput = (spawned, input) => {
|
|---|
| 8 | // Checking for stdin is workaround for https://github.com/nodejs/node/issues/26852
|
|---|
| 9 | // @todo remove `|| spawned.stdin === undefined` once we drop support for Node.js <=12.2.0
|
|---|
| 10 | if (input === undefined || spawned.stdin === undefined) {
|
|---|
| 11 | return;
|
|---|
| 12 | }
|
|---|
| 13 |
|
|---|
| 14 | if (isStream(input)) {
|
|---|
| 15 | input.pipe(spawned.stdin);
|
|---|
| 16 | } else {
|
|---|
| 17 | spawned.stdin.end(input);
|
|---|
| 18 | }
|
|---|
| 19 | };
|
|---|
| 20 |
|
|---|
| 21 | // `all` interleaves `stdout` and `stderr`
|
|---|
| 22 | const makeAllStream = (spawned, {all}) => {
|
|---|
| 23 | if (!all || (!spawned.stdout && !spawned.stderr)) {
|
|---|
| 24 | return;
|
|---|
| 25 | }
|
|---|
| 26 |
|
|---|
| 27 | const mixed = mergeStream();
|
|---|
| 28 |
|
|---|
| 29 | if (spawned.stdout) {
|
|---|
| 30 | mixed.add(spawned.stdout);
|
|---|
| 31 | }
|
|---|
| 32 |
|
|---|
| 33 | if (spawned.stderr) {
|
|---|
| 34 | mixed.add(spawned.stderr);
|
|---|
| 35 | }
|
|---|
| 36 |
|
|---|
| 37 | return mixed;
|
|---|
| 38 | };
|
|---|
| 39 |
|
|---|
| 40 | // On failure, `result.stdout|stderr|all` should contain the currently buffered stream
|
|---|
| 41 | const getBufferedData = async (stream, streamPromise) => {
|
|---|
| 42 | if (!stream) {
|
|---|
| 43 | return;
|
|---|
| 44 | }
|
|---|
| 45 |
|
|---|
| 46 | stream.destroy();
|
|---|
| 47 |
|
|---|
| 48 | try {
|
|---|
| 49 | return await streamPromise;
|
|---|
| 50 | } catch (error) {
|
|---|
| 51 | return error.bufferedData;
|
|---|
| 52 | }
|
|---|
| 53 | };
|
|---|
| 54 |
|
|---|
| 55 | const getStreamPromise = (stream, {encoding, buffer, maxBuffer}) => {
|
|---|
| 56 | if (!stream || !buffer) {
|
|---|
| 57 | return;
|
|---|
| 58 | }
|
|---|
| 59 |
|
|---|
| 60 | if (encoding) {
|
|---|
| 61 | return getStream(stream, {encoding, maxBuffer});
|
|---|
| 62 | }
|
|---|
| 63 |
|
|---|
| 64 | return getStream.buffer(stream, {maxBuffer});
|
|---|
| 65 | };
|
|---|
| 66 |
|
|---|
| 67 | // Retrieve result of child process: exit code, signal, error, streams (stdout/stderr/all)
|
|---|
| 68 | const getSpawnedResult = async ({stdout, stderr, all}, {encoding, buffer, maxBuffer}, processDone) => {
|
|---|
| 69 | const stdoutPromise = getStreamPromise(stdout, {encoding, buffer, maxBuffer});
|
|---|
| 70 | const stderrPromise = getStreamPromise(stderr, {encoding, buffer, maxBuffer});
|
|---|
| 71 | const allPromise = getStreamPromise(all, {encoding, buffer, maxBuffer: maxBuffer * 2});
|
|---|
| 72 |
|
|---|
| 73 | try {
|
|---|
| 74 | return await Promise.all([processDone, stdoutPromise, stderrPromise, allPromise]);
|
|---|
| 75 | } catch (error) {
|
|---|
| 76 | return Promise.all([
|
|---|
| 77 | {error, signal: error.signal, timedOut: error.timedOut},
|
|---|
| 78 | getBufferedData(stdout, stdoutPromise),
|
|---|
| 79 | getBufferedData(stderr, stderrPromise),
|
|---|
| 80 | getBufferedData(all, allPromise)
|
|---|
| 81 | ]);
|
|---|
| 82 | }
|
|---|
| 83 | };
|
|---|
| 84 |
|
|---|
| 85 | const validateInputSync = ({input}) => {
|
|---|
| 86 | if (isStream(input)) {
|
|---|
| 87 | throw new TypeError('The `input` option cannot be a stream in sync mode');
|
|---|
| 88 | }
|
|---|
| 89 | };
|
|---|
| 90 |
|
|---|
| 91 | module.exports = {
|
|---|
| 92 | handleInput,
|
|---|
| 93 | makeAllStream,
|
|---|
| 94 | getSpawnedResult,
|
|---|
| 95 | validateInputSync
|
|---|
| 96 | };
|
|---|
| 97 |
|
|---|