1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
import { createElement as h } from "react";import { RenderToReadableStreamOptions } from "react-dom/server";import { flushEffectHandler, UltraProvider } from "./provider.ts";import { fromFileUrl } from "./deps.ts";import type { Context } from "./types.ts";import { log } from "./logger.ts";import { continueFromInitialStream, renderToInitialStream } from "./stream.ts";import { ImportMap } from "./types.ts";
type RenderToStreamOptions = RenderToReadableStreamOptions & {  importMap: ImportMap | undefined;  assetManifest: Map<string, string> | undefined;  generateStaticHTML?: boolean;  flushEffectsToHead?: boolean;  disableHydration?: boolean;};
export async function renderToStream(  App: JSX.Element,  context: Context | undefined,  options: RenderToStreamOptions,) {  const {    generateStaticHTML = false,    disableHydration = false,    flushEffectsToHead = true,    importMap,    assetManifest,  } = options;
  /**   * For each bootstrapModule we convert from a file url (file:///project/client.tsx) to   * a path string (/project/client.tsx).   */  options.bootstrapModules = disableHydration    ? []    : options?.bootstrapModules?.map(      (url) => url.startsWith("file://") ? fromFileUrl(url) : url,    );
  options.onError = (error) => {    log.error(error);  };
  const renderStream = await renderToInitialStream({    element: h(      UltraProvider,      {        context,        assetManifest,        children: App,      },    ),    options,  });
  return await continueFromInitialStream(renderStream, {    generateStaticHTML,    disableHydration,    flushEffectsToHead,    flushEffectHandler,    importMap,  });}