feat(plugin-rsc): expose Node.js stream APIs

packages/plugin-rsc/src/react/rsc.node.ts

@@ -0,0 +1,25 @@
+// @ts-ignore
+import * as ReactServerNode from '@vitejs/plugin-rsc/vendor/react-server-dom/server.node'
+import { createClientManifest } from '../core/rsc'
+import type { PipeableStream, RenderToPipeableStreamOptions } from '../types'
+
+export * from './rsc'
+
+export function renderToPipeableStream<T>(
+  data: T,
+  options?: RenderToPipeableStreamOptions,
+  extraOptions?: {
+    /**
+     * @internal
+     */
+    onClientReference?: (metadata: { id: string; name: string }) => void
+  },
+): PipeableStream {
+  return ReactServerNode.renderToPipeableStream(
+    data,
+    createClientManifest({
+      onClientReference: extraOptions?.onClientReference,
+    }),
+    options,
+  )
+}

packages/plugin-rsc/src/react/ssr.node.ts

@@ -0,0 +1,16 @@
+// @ts-ignore
+import * as ReactClientNode from '@vitejs/plugin-rsc/vendor/react-server-dom/client.node'
+import { createServerConsumerManifest } from '../core/ssr'
+import type { CreateFromNodeStreamOptions } from '../types'
+
+export * from './ssr'
+
+export function createFromNodeStream<T>(
+  stream: import('node:stream').Readable,
+  options: CreateFromNodeStreamOptions = {},
+): Promise<T> {
+  return ReactClientNode.createFromNodeStream(stream, {
+    serverConsumerManifest: createServerConsumerManifest(),
+    ...options,
+  })
+}

packages/plugin-rsc/src/rsc.node.tsx

@@ -0,0 +1,35 @@
+import assetsManifest from 'virtual:vite-rsc/assets-manifest'
+import type { ResolvedAssetDeps } from './plugin'
+import { renderToPipeableStream as originalRenderToPipeableStream } from './react/rsc.node'
+import type { PipeableStream } from './types'
+
+export * from './rsc'
+
+export function renderToPipeableStream<T>(
+  data: T,
+  options?: object,
+  extraOptions?: {
+    /**
+     * @experimental
+     */
+    onClientReference?: (metadata: {
+      id: string
+      name: string
+      deps: ResolvedAssetDeps
+    }) => void
+  },
+): PipeableStream {
+  return originalRenderToPipeableStream(data, options, {
+    onClientReference(metadata) {
+      const deps = assetsManifest.clientReferenceDeps[metadata.id] ?? {
+        js: [],
+        css: [],
+      }
+      extraOptions?.onClientReference?.({
+        id: metadata.id,
+        name: metadata.name,
+        deps,
+      })
+    },
+  })
+}

packages/plugin-rsc/src/ssr.node.tsx

@@ -0,0 +1,3 @@
+export * from './ssr'
+
+export { createFromNodeStream } from './react/ssr.node'

packages/plugin-rsc/src/types/index.ts

@@ -93,6 +93,15 @@ type DebugChannel = {
   writable?: WritableStream<Uint8Array>
 }
 
+export interface PipeableStream {
+  pipe<T extends NodeJS.WritableStream>(destination: T): T
+  abort(reason?: unknown): void
+}
+
+export type RenderToPipeableStreamOptions = RenderToReadableStreamOptions
+
+export type CreateFromNodeStreamOptions = CreateFromReadableStreamEdgeOptions
+
 // TODO: for now keep them unknown
 // export type ServerTemporaryReferenceSet = WeakMap<object, string>
 // export type ClientTemporaryReferenceSet = Map<unknown, unknown>

packages/plugin-rsc/tsdown.config.ts

@@ -7,14 +7,18 @@ export default defineConfig({
     'src/plugin.ts',
     'src/browser.ts',
     'src/ssr.tsx',
+    'src/ssr.node.tsx',
     'src/rsc.tsx',
+    'src/rsc.node.tsx',
     'src/core/browser.ts',
     'src/core/ssr.ts',
     'src/core/rsc.ts',
     'src/core/plugin.ts',
     'src/react/browser.ts',
     'src/react/ssr.ts',
+    'src/react/ssr.node.ts',
     'src/react/rsc.ts',
+    'src/react/rsc.node.ts',
     'src/transforms/index.ts',
     'src/plugins/cjs.ts',
     'src/utils/rpc.ts',