chore: setFindSourceMapURLCallback

Author: SyMind

packages/react-server-dom-rspack/src/client/ReactFlightDOMClientBrowser.js

@@ -44,16 +44,26 @@ export {createTemporaryReferenceSet} from 'react-client/src/ReactFlightTemporary
 
 export type {TemporaryReferenceSet};
 
-function findSourceMapURL(filename: string, environmentName: string) {
-  const url = new URL('/__rspack_source_map', window.location.origin);
-  url.searchParams.set('filename', filename);
-  url.searchParams.set('environmentName', environmentName);
-  return url.toString();
+let findSourceMapURLCallback: FindSourceMapURLCallback | null = null;
+
+function setFindSourceMapURLCallback(fn: FindSourceMapURLCallback) {
+  findSourceMapURLCallback = fn;
+}
+
+function callCurrentFindSourceMapURLCallback(
+  fileName: string,
+  environmentName: string,
+): null | string {
+  if (findSourceMapURLCallback) {
+    return findSourceMapURLCallback(fileName, environmentName);
+  }
+  return null;
 }
 
 type CallServerCallback = <A, T>(string, args: A) => Promise<T>;
 
 let callServer: CallServerCallback | null = null;
+
 export function setServerCallback(fn: CallServerCallback) {
   callServer = fn;
 }
@@ -78,15 +88,14 @@ export function createServerReference<A: Iterable<any>, T>(
     id + '#' + exportName,
     callCurrentServerCallback,
     undefined,
-    findSourceMapURL,
+    callCurrentFindSourceMapURLCallback,
     exportName,
   );
 }
 
 export type Options = {
   debugChannel?: {writable?: WritableStream, readable?: ReadableStream, ...},
   temporaryReferences?: TemporaryReferenceSet,
-  findSourceMapURL?: FindSourceMapURLCallback,
   replayConsoleLogs?: boolean,
   environmentName?: string,
   startTime?: number,
@@ -135,9 +144,7 @@ function createResponseFromOptions(options: void | Options) {
     options && options.temporaryReferences
       ? options.temporaryReferences
       : undefined,
-    __DEV__ && options && options.findSourceMapURL
-      ? options.findSourceMapURL
-      : undefined,
+    __DEV__ && findSourceMapURLCallback ? findSourceMapURLCallback : undefined,
     __DEV__ ? (options ? options.replayConsoleLogs !== false : true) : false, // defaults to true
     __DEV__ && options && options.environmentName
       ? options.environmentName
@@ -323,7 +330,12 @@ function encodeReply(
   });
 }
 
-export {createFromFetch, createFromReadableStream, encodeReply};
+export {
+  createFromFetch,
+  createFromReadableStream,
+  encodeReply,
+  setFindSourceMapURLCallback,
+};
 
 if (__DEV__) {
   injectIntoDevTools();

packages/react-server-dom-rspack/src/client/ReactFlightDOMClientNode.js

@@ -21,12 +21,6 @@ import type {
   ServerManifest,
 } from 'react-client/src/ReactFlightClientConfig';
 
-type ServerConsumerManifest = {
-  moduleMap: ServerConsumerModuleMap,
-  moduleLoading: ModuleLoading,
-  serverModuleMap: null | ServerManifest,
-};
-
 import type {Readable} from 'stream';
 
 import {
@@ -96,7 +90,6 @@ declare const __rspack_rsc_manifest__: {
 
 function createFromNodeStream<T>(
   stream: Readable,
-  serverConsumerManifest: ServerConsumerManifest,
   options?: Options,
 ): Thenable<T> {
   const debugChannel: void | DebugChannel =