ReactDOMFizzServerNode.js

/**
 * Copyright (c) Facebook, Inc. and its affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 *
 * @flow
 */

import type {ReactNodeList} from 'shared/ReactTypes';
import type {Writable} from 'stream';

import {
  createRequest,
  startWork,
  startFlowing,
  abort,
} from 'react-server/src/ReactFizzServer';

import {createResponseState} from './ReactDOMServerFormatConfig';

function createDrainHandler(destination, request) {
  return () => startFlowing(request);
}

type Options = {
  identifierPrefix?: string,
  progressiveChunkSize?: number,
};

type Controls = {
  // Cancel any pending I/O and put anything remaining into
  // client rendered mode.
  abort(): void,
};

function pipeToNodeWritable(
  children: ReactNodeList,
  destination: Writable,
  options?: Options,
): Controls {
  const request = createRequest(
    children,
    destination,
    createResponseState(options ? options.identifierPrefix : undefined),
    options ? options.progressiveChunkSize : undefined,
  );
  let hasStartedFlowing = false;
  startWork(request);
  return {
    startWriting() {
      if (hasStartedFlowing) {
        return;
      }
      hasStartedFlowing = true;
      startFlowing(request);
      destination.on('drain', createDrainHandler(destination, request));
    },
    abort() {
      abort(request);
    },
  };
}

export {pipeToNodeWritable};