Update RSCClientRoot and related files to use RSCPayloadChunk type for improved type safety in RSC streaming and handling.

Author: AbanoubGhadban

node_package/src/RSCClientRoot.ts

@@ -7,7 +7,7 @@ import * as ReactDOMClient from 'react-dom/client';
 import { createFromReadableStream } from 'react-on-rails-rsc/client';
 import { fetch } from './utils';
 import transformRSCStreamAndReplayConsoleLogs from './transformRSCStreamAndReplayConsoleLogs';
-import { RailsContext, RenderFunction, RenderResult } from './types';
+import { RailsContext, RenderFunction, RSCPayloadChunk } from './types';
 
 const { use } = React;
 
@@ -44,14 +44,14 @@ const fetchRSC = ({ componentName, rscPayloadGenerationUrlPath, componentProps }
 };
 
 const createRSCStreamFromPage = () => {
-  let streamController: ReadableStreamController<RenderResult> | undefined;
-  const stream = new ReadableStream<RenderResult>({
+  let streamController: ReadableStreamController<RSCPayloadChunk> | undefined;
+  const stream = new ReadableStream<RSCPayloadChunk>({
     start(controller) {
       if (typeof window === 'undefined') {
         return;
       }
       const handleChunk = (chunk: unknown) => {
-        controller.enqueue(chunk as RenderResult);
+        controller.enqueue(chunk as RSCPayloadChunk);
       };
       if (!window.__FLIGHT_DATA) {
         window.__FLIGHT_DATA = [];

node_package/src/streamServerRenderedReactComponent.ts

@@ -4,7 +4,7 @@ import { PassThrough, Readable } from 'stream';
 
 import ComponentRegistry from './ComponentRegistry';
 import createReactOutput from './createReactOutput';
-import { isServerRenderHash } from './isServerRenderResult';
+import { isServerRenderHash, isPromise } from './isServerRenderResult';
 import buildConsoleReplay from './buildConsoleReplay';
 import handleError from './handleError';
 import { createResultObject, convertToError, validateComponent } from './serverRenderUtils';

node_package/src/transformRSCStreamAndReplayConsoleLogs.ts

@@ -1,7 +1,7 @@
 import { RSCPayloadChunk } from './types';
 
 export default function transformRSCStreamAndReplayConsoleLogs(
-  stream: ReadableStream<Uint8Array | RenderResult>,
+  stream: ReadableStream<Uint8Array | RSCPayloadChunk>,
 ) {
   return new ReadableStream({
     async start(controller) {
@@ -12,7 +12,7 @@ export default function transformRSCStreamAndReplayConsoleLogs(
       let lastIncompleteChunk = '';
       let { value, done } = await reader.read();
 
-      const handleJsonChunk = (chunk: RenderResult) => {
+      const handleJsonChunk = (chunk: RSCPayloadChunk) => {
         const { html, consoleReplayScript = '' } = chunk;
         controller.enqueue(encoder.encode(html ?? ''));