@tanstack/router-core 1.121.33 → 1.121.39

package/dist/esm/ssr/ssr-client.js

@@ -0,0 +1,131 @@
+import invariant from "tiny-invariant";
+import { isPlainObject } from "../utils.js";
+import { tsrSerializer } from "../serializer.js";
+async function hydrate(router) {
+  var _a, _b, _c;
+  invariant(
+    (_a = window.__TSR_SSR__) == null ? void 0 : _a.dehydrated,
+    "Expected to find a dehydrated data on window.__TSR_SSR__.dehydrated... but we did not. Please file an issue!"
+  );
+  const { manifest, dehydratedData, lastMatchId } = tsrSerializer.parse(
+    window.__TSR_SSR__.dehydrated
+  );
+  router.ssr = {
+    manifest,
+    serializer: tsrSerializer
+  };
+  router.clientSsr = {
+    getStreamedValue: (key) => {
+      var _a2;
+      if (router.isServer) {
+        return void 0;
+      }
+      const streamedValue = (_a2 = window.__TSR_SSR__) == null ? void 0 : _a2.streamedValues[key];
+      if (!streamedValue) {
+        return;
+      }
+      if (!streamedValue.parsed) {
+        streamedValue.parsed = router.ssr.serializer.parse(streamedValue.value);
+      }
+      return streamedValue.parsed;
+    }
+  };
+  const matches = router.matchRoutes(router.state.location);
+  const routeChunkPromise = Promise.all(
+    matches.map((match) => {
+      const route = router.looseRoutesById[match.routeId];
+      return router.loadRouteChunk(route);
+    })
+  );
+  matches.forEach((match) => {
+    var _a2;
+    const dehydratedMatch = window.__TSR_SSR__.matches.find(
+      (d) => d.id === match.id
+    );
+    if (!dehydratedMatch) {
+      return;
+    }
+    Object.assign(match, dehydratedMatch);
+    if (dehydratedMatch.__beforeLoadContext) {
+      match.__beforeLoadContext = router.ssr.serializer.parse(
+        dehydratedMatch.__beforeLoadContext
+      );
+    }
+    if (dehydratedMatch.loaderData) {
+      match.loaderData = router.ssr.serializer.parse(
+        dehydratedMatch.loaderData
+      );
+    }
+    if (dehydratedMatch.error) {
+      match.error = router.ssr.serializer.parse(dehydratedMatch.error);
+    }
+    (_a2 = match.extracted) == null ? void 0 : _a2.forEach((ex) => {
+      deepMutableSetByPath(match, ["loaderData", ...ex.path], ex.value);
+    });
+    return match;
+  });
+  router.__store.setState((s) => {
+    return {
+      ...s,
+      matches
+    };
+  });
+  await ((_c = (_b = router.options).hydrate) == null ? void 0 : _c.call(_b, dehydratedData));
+  await Promise.all(
+    router.state.matches.map(async (match) => {
+      var _a2, _b2, _c2, _d, _e, _f;
+      const route = router.looseRoutesById[match.routeId];
+      const parentMatch = router.state.matches[match.index - 1];
+      const parentContext = (parentMatch == null ? void 0 : parentMatch.context) ?? router.options.context ?? {};
+      const contextFnContext = {
+        deps: match.loaderDeps,
+        params: match.params,
+        context: parentContext,
+        location: router.state.location,
+        navigate: (opts) => router.navigate({ ...opts, _fromLocation: router.state.location }),
+        buildLocation: router.buildLocation,
+        cause: match.cause,
+        abortController: match.abortController,
+        preload: false,
+        matches
+      };
+      match.__routeContext = ((_b2 = (_a2 = route.options).context) == null ? void 0 : _b2.call(_a2, contextFnContext)) ?? {};
+      match.context = {
+        ...parentContext,
+        ...match.__routeContext,
+        ...match.__beforeLoadContext
+      };
+      const assetContext = {
+        matches: router.state.matches,
+        match,
+        params: match.params,
+        loaderData: match.loaderData
+      };
+      const headFnContent = await ((_d = (_c2 = route.options).head) == null ? void 0 : _d.call(_c2, assetContext));
+      const scripts = await ((_f = (_e = route.options).scripts) == null ? void 0 : _f.call(_e, assetContext));
+      match.meta = headFnContent == null ? void 0 : headFnContent.meta;
+      match.links = headFnContent == null ? void 0 : headFnContent.links;
+      match.headScripts = headFnContent == null ? void 0 : headFnContent.scripts;
+      match.scripts = scripts;
+    })
+  );
+  if (matches[matches.length - 1].id !== lastMatchId) {
+    return await Promise.all([routeChunkPromise, router.load()]);
+  }
+  return routeChunkPromise;
+}
+function deepMutableSetByPath(obj, path, value) {
+  if (path.length === 1) {
+    obj[path[0]] = value;
+  }
+  const [key, ...rest] = path;
+  if (Array.isArray(obj)) {
+    deepMutableSetByPath(obj[Number(key)], rest, value);
+  } else if (isPlainObject(obj)) {
+    deepMutableSetByPath(obj[key], rest, value);
+  }
+}
+export {
+  hydrate
+};
+//# sourceMappingURL=ssr-client.js.map

package/dist/esm/ssr/ssr-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ssr-client.js","sources":["../../../src/ssr/ssr-client.ts"],"sourcesContent":["import invariant from 'tiny-invariant'\nimport { isPlainObject } from '../utils'\nimport { tsrSerializer } from '../serializer'\nimport type { DeferredPromiseState } from '../defer'\nimport type { MakeRouteMatch } from '../Matches'\nimport type { AnyRouter, ControllablePromise } from '../router'\nimport type { Manifest } from '../manifest'\nimport type { RouteContextOptions } from '../route'\n\ndeclare global {\n  interface Window {\n    __TSR_SSR__?: TsrSsrGlobal\n  }\n}\n\nexport interface TsrSsrGlobal {\n  matches: Array<SsrMatch>\n  streamedValues: Record<\n    string,\n    {\n      value: any\n      parsed: any\n    }\n  >\n  cleanScripts: () => void\n  dehydrated?: any\n  initMatch: (match: SsrMatch) => void\n  resolvePromise: (opts: {\n    matchId: string\n    id: number\n    promiseState: DeferredPromiseState<any>\n  }) => void\n  injectChunk: (opts: { matchId: string; id: number; chunk: string }) => void\n  closeStream: (opts: { matchId: string; id: number }) => void\n}\n\nexport interface SsrMatch {\n  id: string\n  __beforeLoadContext: string\n  loaderData?: string\n  error?: string\n  extracted?: Array<ClientExtractedEntry>\n  updatedAt: MakeRouteMatch['updatedAt']\n  status: MakeRouteMatch['status']\n}\n\nexport type ClientExtractedEntry =\n  | ClientExtractedStream\n  | ClientExtractedPromise\n\nexport interface ClientExtractedPromise extends ClientExtractedBaseEntry {\n  type: 'promise'\n  value?: ControllablePromise<any>\n}\n\nexport interface ClientExtractedStream extends ClientExtractedBaseEntry {\n  type: 'stream'\n  value?: ReadableStream & { controller?: ReadableStreamDefaultController }\n}\n\nexport interface ClientExtractedBaseEntry {\n  type: string\n  path: Array<string>\n}\n\nexport interface ResolvePromiseState {\n  matchId: string\n  id: number\n  promiseState: DeferredPromiseState<any>\n}\n\nexport interface DehydratedRouter {\n  manifest: Manifest | undefined\n  dehydratedData: any\n  lastMatchId: string\n}\n\nexport async function hydrate(router: AnyRouter): Promise<any> {\n  invariant(\n    window.__TSR_SSR__?.dehydrated,\n    'Expected to find a dehydrated data on window.__TSR_SSR__.dehydrated... but we did not. Please file an issue!',\n  )\n\n  const { manifest, dehydratedData, lastMatchId } = tsrSerializer.parse(\n    window.__TSR_SSR__.dehydrated,\n  ) as DehydratedRouter\n\n  router.ssr = {\n    manifest,\n    serializer: tsrSerializer,\n  }\n\n  router.clientSsr = {\n    getStreamedValue: <T>(key: string): T | undefined => {\n      if (router.isServer) {\n        return undefined\n      }\n\n      const streamedValue = window.__TSR_SSR__?.streamedValues[key]\n\n      if (!streamedValue) {\n        return\n      }\n\n      if (!streamedValue.parsed) {\n        streamedValue.parsed = router.ssr!.serializer.parse(streamedValue.value)\n      }\n\n      return streamedValue.parsed\n    },\n  }\n\n  // Hydrate the router state\n  const matches = router.matchRoutes(router.state.location)\n\n  // kick off loading the route chunks\n  const routeChunkPromise = Promise.all(\n    matches.map((match) => {\n      const route = router.looseRoutesById[match.routeId]!\n      return router.loadRouteChunk(route)\n    }),\n  )\n\n  // Right after hydration and before the first render, we need to rehydrate each match\n  // First step is to reyhdrate loaderData and __beforeLoadContext\n  matches.forEach((match) => {\n    const dehydratedMatch = window.__TSR_SSR__!.matches.find(\n      (d) => d.id === match.id,\n    )\n\n    if (!dehydratedMatch) {\n      return\n    }\n\n    Object.assign(match, dehydratedMatch)\n\n    // Handle beforeLoadContext\n    if (dehydratedMatch.__beforeLoadContext) {\n      match.__beforeLoadContext = router.ssr!.serializer.parse(\n        dehydratedMatch.__beforeLoadContext,\n      ) as any\n    }\n\n    // Handle loaderData\n    if (dehydratedMatch.loaderData) {\n      match.loaderData = router.ssr!.serializer.parse(\n        dehydratedMatch.loaderData,\n      )\n    }\n\n    // Handle error\n    if (dehydratedMatch.error) {\n      match.error = router.ssr!.serializer.parse(dehydratedMatch.error)\n    }\n\n    // Handle extracted\n    ;(match as unknown as SsrMatch).extracted?.forEach((ex) => {\n      deepMutableSetByPath(match, ['loaderData', ...ex.path], ex.value)\n    })\n\n    return match\n  })\n\n  router.__store.setState((s) => {\n    return {\n      ...s,\n      matches,\n    }\n  })\n\n  // Allow the user to handle custom hydration data\n  await router.options.hydrate?.(dehydratedData)\n\n  // now that all necessary data is hydrated:\n  // 1) fully reconstruct the route context\n  // 2) execute `head()` and `scripts()` for each match\n  await Promise.all(\n    router.state.matches.map(async (match) => {\n      const route = router.looseRoutesById[match.routeId]!\n\n      const parentMatch = router.state.matches[match.index - 1]\n      const parentContext = parentMatch?.context ?? router.options.context ?? {}\n\n      // `context()` was already executed by `matchRoutes`, however route context was not yet fully reconstructed\n      // so run it again and merge route context\n      const contextFnContext: RouteContextOptions<any, any, any, any> = {\n        deps: match.loaderDeps,\n        params: match.params,\n        context: parentContext,\n        location: router.state.location,\n        navigate: (opts: any) =>\n          router.navigate({ ...opts, _fromLocation: router.state.location }),\n        buildLocation: router.buildLocation,\n        cause: match.cause,\n        abortController: match.abortController,\n        preload: false,\n        matches,\n      }\n      match.__routeContext = route.options.context?.(contextFnContext) ?? {}\n\n      match.context = {\n        ...parentContext,\n        ...match.__routeContext,\n        ...match.__beforeLoadContext,\n      }\n\n      const assetContext = {\n        matches: router.state.matches,\n        match,\n        params: match.params,\n        loaderData: match.loaderData,\n      }\n      const headFnContent = await route.options.head?.(assetContext)\n\n      const scripts = await route.options.scripts?.(assetContext)\n\n      match.meta = headFnContent?.meta\n      match.links = headFnContent?.links\n      match.headScripts = headFnContent?.scripts\n      match.scripts = scripts\n    }),\n  )\n\n  if (matches[matches.length - 1]!.id !== lastMatchId) {\n    return await Promise.all([routeChunkPromise, router.load()])\n  }\n\n  return routeChunkPromise\n}\n\nfunction deepMutableSetByPath<T>(obj: T, path: Array<string>, value: any) {\n  // mutable set by path retaining array and object references\n  if (path.length === 1) {\n    ;(obj as any)[path[0]!] = value\n  }\n\n  const [key, ...rest] = path\n\n  if (Array.isArray(obj)) {\n    deepMutableSetByPath(obj[Number(key)], rest, value)\n  } else if (isPlainObject(obj)) {\n    deepMutableSetByPath((obj as any)[key!], rest, value)\n  }\n}\n"],"names":["_a","_b","_c"],"mappings":";;;AA6EA,eAAsB,QAAQ,QAAiC;;AAC7D;AAAA,KACE,YAAO,gBAAP,mBAAoB;AAAA,IACpB;AAAA,EACF;AAEA,QAAM,EAAE,UAAU,gBAAgB,gBAAgB,cAAc;AAAA,IAC9D,OAAO,YAAY;AAAA,EACrB;AAEA,SAAO,MAAM;AAAA,IACX;AAAA,IACA,YAAY;AAAA,EACd;AAEA,SAAO,YAAY;AAAA,IACjB,kBAAkB,CAAI,QAA+B;;AACnD,UAAI,OAAO,UAAU;AACZ,eAAA;AAAA,MAAA;AAGT,YAAM,iBAAgBA,MAAA,OAAO,gBAAP,gBAAAA,IAAoB,eAAe;AAEzD,UAAI,CAAC,eAAe;AAClB;AAAA,MAAA;AAGE,UAAA,CAAC,cAAc,QAAQ;AACzB,sBAAc,SAAS,OAAO,IAAK,WAAW,MAAM,cAAc,KAAK;AAAA,MAAA;AAGzE,aAAO,cAAc;AAAA,IAAA;AAAA,EAEzB;AAGA,QAAM,UAAU,OAAO,YAAY,OAAO,MAAM,QAAQ;AAGxD,QAAM,oBAAoB,QAAQ;AAAA,IAChC,QAAQ,IAAI,CAAC,UAAU;AACrB,YAAM,QAAQ,OAAO,gBAAgB,MAAM,OAAO;AAC3C,aAAA,OAAO,eAAe,KAAK;AAAA,IACnC,CAAA;AAAA,EACH;AAIQ,UAAA,QAAQ,CAAC,UAAU;;AACnB,UAAA,kBAAkB,OAAO,YAAa,QAAQ;AAAA,MAClD,CAAC,MAAM,EAAE,OAAO,MAAM;AAAA,IACxB;AAEA,QAAI,CAAC,iBAAiB;AACpB;AAAA,IAAA;AAGK,WAAA,OAAO,OAAO,eAAe;AAGpC,QAAI,gBAAgB,qBAAqB;AACjC,YAAA,sBAAsB,OAAO,IAAK,WAAW;AAAA,QACjD,gBAAgB;AAAA,MAClB;AAAA,IAAA;AAIF,QAAI,gBAAgB,YAAY;AACxB,YAAA,aAAa,OAAO,IAAK,WAAW;AAAA,QACxC,gBAAgB;AAAA,MAClB;AAAA,IAAA;AAIF,QAAI,gBAAgB,OAAO;AACzB,YAAM,QAAQ,OAAO,IAAK,WAAW,MAAM,gBAAgB,KAAK;AAAA,IAAA;AAIhE,KAAAA,MAAA,MAA8B,cAA9B,gBAAAA,IAAyC,QAAQ,CAAC,OAAO;AACpC,2BAAA,OAAO,CAAC,cAAc,GAAG,GAAG,IAAI,GAAG,GAAG,KAAK;AAAA,IAAA;AAG3D,WAAA;AAAA,EAAA,CACR;AAEM,SAAA,QAAQ,SAAS,CAAC,MAAM;AACtB,WAAA;AAAA,MACL,GAAG;AAAA,MACH;AAAA,IACF;AAAA,EAAA,CACD;AAGK,UAAA,kBAAO,SAAQ,YAAf,4BAAyB;AAK/B,QAAM,QAAQ;AAAA,IACZ,OAAO,MAAM,QAAQ,IAAI,OAAO,UAAU;;AACxC,YAAM,QAAQ,OAAO,gBAAgB,MAAM,OAAO;AAElD,YAAM,cAAc,OAAO,MAAM,QAAQ,MAAM,QAAQ,CAAC;AACxD,YAAM,iBAAgB,2CAAa,YAAW,OAAO,QAAQ,WAAW,CAAC;AAIzE,YAAM,mBAA4D;AAAA,QAChE,MAAM,MAAM;AAAA,QACZ,QAAQ,MAAM;AAAA,QACd,SAAS;AAAA,QACT,UAAU,OAAO,MAAM;AAAA,QACvB,UAAU,CAAC,SACT,OAAO,SAAS,EAAE,GAAG,MAAM,eAAe,OAAO,MAAM,SAAA,CAAU;AAAA,QACnE,eAAe,OAAO;AAAA,QACtB,OAAO,MAAM;AAAA,QACb,iBAAiB,MAAM;AAAA,QACvB,SAAS;AAAA,QACT;AAAA,MACF;AACA,YAAM,mBAAiBC,OAAAD,MAAA,MAAM,SAAQ,YAAd,gBAAAC,IAAA,KAAAD,KAAwB,sBAAqB,CAAC;AAErE,YAAM,UAAU;AAAA,QACd,GAAG;AAAA,QACH,GAAG,MAAM;AAAA,QACT,GAAG,MAAM;AAAA,MACX;AAEA,YAAM,eAAe;AAAA,QACnB,SAAS,OAAO,MAAM;AAAA,QACtB;AAAA,QACA,QAAQ,MAAM;AAAA,QACd,YAAY,MAAM;AAAA,MACpB;AACA,YAAM,gBAAgB,QAAM,MAAAE,MAAA,MAAM,SAAQ,SAAd,wBAAAA,KAAqB;AAEjD,YAAM,UAAU,QAAM,iBAAM,SAAQ,YAAd,4BAAwB;AAE9C,YAAM,OAAO,+CAAe;AAC5B,YAAM,QAAQ,+CAAe;AAC7B,YAAM,cAAc,+CAAe;AACnC,YAAM,UAAU;AAAA,IACjB,CAAA;AAAA,EACH;AAEA,MAAI,QAAQ,QAAQ,SAAS,CAAC,EAAG,OAAO,aAAa;AAC5C,WAAA,MAAM,QAAQ,IAAI,CAAC,mBAAmB,OAAO,KAAA,CAAM,CAAC;AAAA,EAAA;AAGtD,SAAA;AACT;AAEA,SAAS,qBAAwB,KAAQ,MAAqB,OAAY;AAEpE,MAAA,KAAK,WAAW,GAAG;AACnB,QAAY,KAAK,CAAC,CAAE,IAAI;AAAA,EAAA;AAG5B,QAAM,CAAC,KAAK,GAAG,IAAI,IAAI;AAEnB,MAAA,MAAM,QAAQ,GAAG,GAAG;AACtB,yBAAqB,IAAI,OAAO,GAAG,CAAC,GAAG,MAAM,KAAK;AAAA,EAAA,WACzC,cAAc,GAAG,GAAG;AAC7B,yBAAsB,IAAY,GAAI,GAAG,MAAM,KAAK;AAAA,EAAA;AAExD;"}

package/dist/esm/ssr/ssr-server.d.ts

@@ -0,0 +1,32 @@
+import { DeferredPromise } from '../defer.js';
+import { ClientExtractedBaseEntry } from './ssr-client.js';
+import { AnyRouter } from '../router.js';
+import { Manifest } from '../manifest.js';
+import { AnyRouteMatch } from '../Matches.js';
+export type ServerExtractedEntry = ServerExtractedStream | ServerExtractedPromise;
+export interface ServerExtractedBaseEntry extends ClientExtractedBaseEntry {
+    id: number;
+    matchIndex: number;
+}
+export interface ServerExtractedStream extends ServerExtractedBaseEntry {
+    type: 'stream';
+    stream: ReadableStream;
+}
+export interface ServerExtractedPromise extends ServerExtractedBaseEntry {
+    type: 'promise';
+    promise: DeferredPromise<any>;
+}
+export declare function attachRouterServerSsrUtils(router: AnyRouter, manifest: Manifest | undefined): void;
+export declare function dehydrateRouter(router: AnyRouter): void;
+export declare function extractAsyncLoaderData(loaderData: any, ctx: {
+    match: AnyRouteMatch;
+    router: AnyRouter;
+}): {
+    replaced: any;
+    extracted: ServerExtractedEntry[];
+};
+export declare function onMatchSettled(opts: {
+    router: AnyRouter;
+    match: AnyRouteMatch;
+}): void;
+export declare function replaceBy<T>(obj: T, cb: (value: any, path: Array<string>) => any, path?: Array<string>): T;

package/dist/esm/ssr/ssr-server.js

@@ -0,0 +1,248 @@
+import warning from "tiny-warning";
+import jsesc from "jsesc";
+import { TSR_DEFERRED_PROMISE, defer } from "../defer.js";
+import { pick, isPlainArray, isPlainObject } from "../utils.js";
+import { tsrSerializer } from "../serializer.js";
+import minifiedTsrBootStrapScript from "./tsrScript.js";
+function attachRouterServerSsrUtils(router, manifest) {
+  router.ssr = {
+    manifest,
+    serializer: tsrSerializer
+  };
+  router.serverSsr = {
+    injectedHtml: [],
+    streamedKeys: /* @__PURE__ */ new Set(),
+    injectHtml: (getHtml) => {
+      const promise = Promise.resolve().then(getHtml);
+      router.serverSsr.injectedHtml.push(promise);
+      router.emit({
+        type: "onInjectedHtml",
+        promise
+      });
+      return promise.then(() => {
+      });
+    },
+    injectScript: (getScript, opts) => {
+      return router.serverSsr.injectHtml(async () => {
+        const script = await getScript();
+        return `<script class='tsr-once'>${script}${process.env.NODE_ENV === "development" && ((opts == null ? void 0 : opts.logScript) ?? true) ? `; console.info(\`Injected From Server:
+${jsesc(script, { quotes: "backtick" })}\`)` : ""}; if (typeof __TSR_SSR__ !== 'undefined') __TSR_SSR__.cleanScripts()<\/script>`;
+      });
+    },
+    streamValue: (key, value) => {
+      warning(
+        !router.serverSsr.streamedKeys.has(key),
+        "Key has already been streamed: " + key
+      );
+      router.serverSsr.streamedKeys.add(key);
+      router.serverSsr.injectScript(
+        () => `__TSR_SSR__.streamedValues['${key}'] = { value: ${jsesc(
+          router.ssr.serializer.stringify(value),
+          {
+            isScriptContext: true,
+            wrap: true,
+            json: true
+          }
+        )}}`
+      );
+    },
+    onMatchSettled
+  };
+  router.serverSsr.injectScript(() => minifiedTsrBootStrapScript, {
+    logScript: false
+  });
+}
+function dehydrateRouter(router) {
+  var _a, _b, _c;
+  const dehydratedRouter = {
+    manifest: router.ssr.manifest,
+    dehydratedData: (_b = (_a = router.options).dehydrate) == null ? void 0 : _b.call(_a),
+    lastMatchId: ((_c = router.state.matches[router.state.matches.length - 1]) == null ? void 0 : _c.id) || ""
+  };
+  router.serverSsr.injectScript(
+    () => `__TSR_SSR__.dehydrated = ${jsesc(
+      router.ssr.serializer.stringify(dehydratedRouter),
+      {
+        isScriptContext: true,
+        wrap: true,
+        json: true
+      }
+    )}`
+  );
+}
+function extractAsyncLoaderData(loaderData, ctx) {
+  const extracted = [];
+  const replaced = replaceBy(loaderData, (value, path) => {
+    if (value instanceof ReadableStream) {
+      const [copy1, copy2] = value.tee();
+      const entry = {
+        type: "stream",
+        path,
+        id: extracted.length,
+        matchIndex: ctx.match.index,
+        stream: copy2
+      };
+      extracted.push(entry);
+      return copy1;
+    } else if (value instanceof Promise) {
+      const deferredPromise = defer(value);
+      const entry = {
+        type: "promise",
+        path,
+        id: extracted.length,
+        matchIndex: ctx.match.index,
+        promise: deferredPromise
+      };
+      extracted.push(entry);
+    }
+    return value;
+  });
+  return { replaced, extracted };
+}
+function onMatchSettled(opts) {
+  const { router, match } = opts;
+  let extracted = void 0;
+  let serializedLoaderData = void 0;
+  if (match.loaderData !== void 0) {
+    const result = extractAsyncLoaderData(match.loaderData, {
+      match
+    });
+    match.loaderData = result.replaced;
+    extracted = result.extracted;
+    serializedLoaderData = extracted.reduce(
+      (acc, entry) => {
+        return deepImmutableSetByPath(acc, ["temp", ...entry.path], void 0);
+      },
+      { temp: result.replaced }
+    ).temp;
+  }
+  const initCode = `__TSR_SSR__.initMatch(${jsesc(
+    {
+      id: match.id,
+      __beforeLoadContext: router.ssr.serializer.stringify(
+        match.__beforeLoadContext
+      ),
+      loaderData: router.ssr.serializer.stringify(serializedLoaderData),
+      error: router.ssr.serializer.stringify(match.error),
+      extracted: extracted == null ? void 0 : extracted.map((entry) => pick(entry, ["type", "path"])),
+      updatedAt: match.updatedAt,
+      status: match.status
+    },
+    {
+      isScriptContext: true,
+      wrap: true,
+      json: true
+    }
+  )})`;
+  router.serverSsr.injectScript(() => initCode);
+  if (extracted) {
+    extracted.forEach((entry) => {
+      if (entry.type === "promise") return injectPromise(entry);
+      return injectStream(entry);
+    });
+  }
+  function injectPromise(entry) {
+    router.serverSsr.injectScript(async () => {
+      await entry.promise;
+      return `__TSR_SSR__.resolvePromise(${jsesc(
+        {
+          matchId: match.id,
+          id: entry.id,
+          promiseState: entry.promise[TSR_DEFERRED_PROMISE]
+        },
+        {
+          isScriptContext: true,
+          wrap: true,
+          json: true
+        }
+      )})`;
+    });
+  }
+  function injectStream(entry) {
+    router.serverSsr.injectHtml(async () => {
+      try {
+        const reader = entry.stream.getReader();
+        let chunk = null;
+        while (!(chunk = await reader.read()).done) {
+          if (chunk.value) {
+            const code = `__TSR_SSR__.injectChunk(${jsesc(
+              {
+                matchId: match.id,
+                id: entry.id,
+                chunk: chunk.value
+              },
+              {
+                isScriptContext: true,
+                wrap: true,
+                json: true
+              }
+            )})`;
+            router.serverSsr.injectScript(() => code);
+          }
+        }
+        router.serverSsr.injectScript(
+          () => `__TSR_SSR__.closeStream(${jsesc(
+            {
+              matchId: match.id,
+              id: entry.id
+            },
+            {
+              isScriptContext: true,
+              wrap: true,
+              json: true
+            }
+          )})`
+        );
+      } catch (err) {
+        console.error("stream read error", err);
+      }
+      return "";
+    });
+  }
+}
+function deepImmutableSetByPath(obj, path, value) {
+  if (path.length === 0) {
+    return value;
+  }
+  const [key, ...rest] = path;
+  if (Array.isArray(obj)) {
+    return obj.map((item, i) => {
+      if (i === Number(key)) {
+        return deepImmutableSetByPath(item, rest, value);
+      }
+      return item;
+    });
+  }
+  if (isPlainObject(obj)) {
+    return {
+      ...obj,
+      [key]: deepImmutableSetByPath(obj[key], rest, value)
+    };
+  }
+  return obj;
+}
+function replaceBy(obj, cb, path = []) {
+  if (isPlainArray(obj)) {
+    return obj.map((value, i) => replaceBy(value, cb, [...path, `${i}`]));
+  }
+  if (isPlainObject(obj)) {
+    const newObj2 = {};
+    for (const key in obj) {
+      newObj2[key] = replaceBy(obj[key], cb, [...path, key]);
+    }
+    return newObj2;
+  }
+  const newObj = cb(obj, path);
+  if (newObj !== obj) {
+    return newObj;
+  }
+  return obj;
+}
+export {
+  attachRouterServerSsrUtils,
+  dehydrateRouter,
+  extractAsyncLoaderData,
+  onMatchSettled,
+  replaceBy
+};
+//# sourceMappingURL=ssr-server.js.map

package/dist/esm/ssr/ssr-server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ssr-server.js","sources":["../../../src/ssr/ssr-server.ts"],"sourcesContent":["import { default as warning } from 'tiny-warning'\nimport jsesc from 'jsesc'\nimport { TSR_DEFERRED_PROMISE, defer } from '../defer'\nimport { isPlainArray, isPlainObject, pick } from '../utils'\nimport { tsrSerializer } from '../serializer'\nimport minifiedTsrBootStrapScript from './tsrScript?script-string'\nimport type { DeferredPromise } from '../defer'\nimport type {\n  ClientExtractedBaseEntry,\n  DehydratedRouter,\n  ResolvePromiseState,\n  SsrMatch,\n} from './ssr-client'\nimport type { AnyRouter } from '../router'\nimport type { Manifest } from '../manifest'\nimport type { AnyRouteMatch } from '../Matches'\n\nexport type ServerExtractedEntry =\n  | ServerExtractedStream\n  | ServerExtractedPromise\n\nexport interface ServerExtractedBaseEntry extends ClientExtractedBaseEntry {\n  id: number\n  matchIndex: number\n}\n\nexport interface ServerExtractedStream extends ServerExtractedBaseEntry {\n  type: 'stream'\n  stream: ReadableStream\n}\n\nexport interface ServerExtractedPromise extends ServerExtractedBaseEntry {\n  type: 'promise'\n  promise: DeferredPromise<any>\n}\n\nexport function attachRouterServerSsrUtils(\n  router: AnyRouter,\n  manifest: Manifest | undefined,\n) {\n  router.ssr = {\n    manifest,\n    serializer: tsrSerializer,\n  }\n\n  router.serverSsr = {\n    injectedHtml: [],\n    streamedKeys: new Set(),\n    injectHtml: (getHtml) => {\n      const promise = Promise.resolve().then(getHtml)\n      router.serverSsr!.injectedHtml.push(promise)\n      router.emit({\n        type: 'onInjectedHtml',\n        promise,\n      })\n\n      return promise.then(() => {})\n    },\n    injectScript: (getScript, opts) => {\n      return router.serverSsr!.injectHtml(async () => {\n        const script = await getScript()\n        return `<script class='tsr-once'>${script}${\n          process.env.NODE_ENV === 'development' && (opts?.logScript ?? true)\n            ? `; console.info(\\`Injected From Server:\n${jsesc(script, { quotes: 'backtick' })}\\`)`\n            : ''\n        }; if (typeof __TSR_SSR__ !== 'undefined') __TSR_SSR__.cleanScripts()</script>`\n      })\n    },\n    streamValue: (key, value) => {\n      warning(\n        !router.serverSsr!.streamedKeys.has(key),\n        'Key has already been streamed: ' + key,\n      )\n\n      router.serverSsr!.streamedKeys.add(key)\n      router.serverSsr!.injectScript(\n        () =>\n          `__TSR_SSR__.streamedValues['${key}'] = { value: ${jsesc(\n            router.ssr!.serializer.stringify(value),\n            {\n              isScriptContext: true,\n              wrap: true,\n              json: true,\n            },\n          )}}`,\n      )\n    },\n    onMatchSettled,\n  }\n\n  router.serverSsr.injectScript(() => minifiedTsrBootStrapScript, {\n    logScript: false,\n  })\n}\n\nexport function dehydrateRouter(router: AnyRouter) {\n  const dehydratedRouter: DehydratedRouter = {\n    manifest: router.ssr!.manifest,\n    dehydratedData: router.options.dehydrate?.(),\n    lastMatchId:\n      router.state.matches[router.state.matches.length - 1]?.id || '',\n  }\n\n  router.serverSsr!.injectScript(\n    () =>\n      `__TSR_SSR__.dehydrated = ${jsesc(\n        router.ssr!.serializer.stringify(dehydratedRouter),\n        {\n          isScriptContext: true,\n          wrap: true,\n          json: true,\n        },\n      )}`,\n  )\n}\n\nexport function extractAsyncLoaderData(\n  loaderData: any,\n  ctx: {\n    match: AnyRouteMatch\n    router: AnyRouter\n  },\n) {\n  const extracted: Array<ServerExtractedEntry> = []\n\n  const replaced = replaceBy(loaderData, (value, path) => {\n    // If it's a stream, we need to tee it so we can read it multiple times\n    if (value instanceof ReadableStream) {\n      const [copy1, copy2] = value.tee()\n      const entry: ServerExtractedStream = {\n        type: 'stream',\n        path,\n        id: extracted.length,\n        matchIndex: ctx.match.index,\n        stream: copy2,\n      }\n\n      extracted.push(entry)\n      return copy1\n    } else if (value instanceof Promise) {\n      const deferredPromise = defer(value)\n      const entry: ServerExtractedPromise = {\n        type: 'promise',\n        path,\n        id: extracted.length,\n        matchIndex: ctx.match.index,\n        promise: deferredPromise,\n      }\n      extracted.push(entry)\n    }\n\n    return value\n  })\n\n  return { replaced, extracted }\n}\n\nexport function onMatchSettled(opts: {\n  router: AnyRouter\n  match: AnyRouteMatch\n}) {\n  const { router, match } = opts\n\n  let extracted: Array<ServerExtractedEntry> | undefined = undefined\n  let serializedLoaderData: any = undefined\n  if (match.loaderData !== undefined) {\n    const result = extractAsyncLoaderData(match.loaderData, {\n      router,\n      match,\n    })\n    match.loaderData = result.replaced\n    extracted = result.extracted\n    serializedLoaderData = extracted.reduce(\n      (acc: any, entry: ServerExtractedEntry) => {\n        return deepImmutableSetByPath(acc, ['temp', ...entry.path], undefined)\n      },\n      { temp: result.replaced },\n    ).temp\n  }\n\n  const initCode = `__TSR_SSR__.initMatch(${jsesc(\n    {\n      id: match.id,\n      __beforeLoadContext: router.ssr!.serializer.stringify(\n        match.__beforeLoadContext,\n      ),\n      loaderData: router.ssr!.serializer.stringify(serializedLoaderData),\n      error: router.ssr!.serializer.stringify(match.error),\n      extracted: extracted?.map((entry) => pick(entry, ['type', 'path'])),\n      updatedAt: match.updatedAt,\n      status: match.status,\n    } satisfies SsrMatch,\n    {\n      isScriptContext: true,\n      wrap: true,\n      json: true,\n    },\n  )})`\n\n  router.serverSsr!.injectScript(() => initCode)\n\n  if (extracted) {\n    extracted.forEach((entry) => {\n      if (entry.type === 'promise') return injectPromise(entry)\n      return injectStream(entry)\n    })\n  }\n\n  function injectPromise(entry: ServerExtractedPromise) {\n    router.serverSsr!.injectScript(async () => {\n      await entry.promise\n\n      return `__TSR_SSR__.resolvePromise(${jsesc(\n        {\n          matchId: match.id,\n          id: entry.id,\n          promiseState: entry.promise[TSR_DEFERRED_PROMISE],\n        } satisfies ResolvePromiseState,\n        {\n          isScriptContext: true,\n          wrap: true,\n          json: true,\n        },\n      )})`\n    })\n  }\n\n  function injectStream(entry: ServerExtractedStream) {\n    // Inject a promise that resolves when the stream is done\n    // We do this to keep the stream open until we're done\n    router.serverSsr!.injectHtml(async () => {\n      //\n      try {\n        const reader = entry.stream.getReader()\n        let chunk: ReadableStreamReadResult<any> | null = null\n        while (!(chunk = await reader.read()).done) {\n          if (chunk.value) {\n            const code = `__TSR_SSR__.injectChunk(${jsesc(\n              {\n                matchId: match.id,\n                id: entry.id,\n                chunk: chunk.value,\n              },\n              {\n                isScriptContext: true,\n                wrap: true,\n                json: true,\n              },\n            )})`\n\n            router.serverSsr!.injectScript(() => code)\n          }\n        }\n\n        router.serverSsr!.injectScript(\n          () =>\n            `__TSR_SSR__.closeStream(${jsesc(\n              {\n                matchId: match.id,\n                id: entry.id,\n              },\n              {\n                isScriptContext: true,\n                wrap: true,\n                json: true,\n              },\n            )})`,\n        )\n      } catch (err) {\n        console.error('stream read error', err)\n      }\n\n      return ''\n    })\n  }\n}\n\nfunction deepImmutableSetByPath<T>(obj: T, path: Array<string>, value: any): T {\n  // immutable set by path retaining array and object references\n  if (path.length === 0) {\n    return value\n  }\n\n  const [key, ...rest] = path\n\n  if (Array.isArray(obj)) {\n    return obj.map((item, i) => {\n      if (i === Number(key)) {\n        return deepImmutableSetByPath(item, rest, value)\n      }\n      return item\n    }) as T\n  }\n\n  if (isPlainObject(obj)) {\n    return {\n      ...obj,\n      [key!]: deepImmutableSetByPath((obj as any)[key!], rest, value),\n    }\n  }\n\n  return obj\n}\n\nexport function replaceBy<T>(\n  obj: T,\n  cb: (value: any, path: Array<string>) => any,\n  path: Array<string> = [],\n): T {\n  if (isPlainArray(obj)) {\n    return obj.map((value, i) => replaceBy(value, cb, [...path, `${i}`])) as any\n  }\n\n  if (isPlainObject(obj)) {\n    // Do not allow objects with illegal\n    const newObj: any = {}\n\n    for (const key in obj) {\n      newObj[key] = replaceBy(obj[key], cb, [...path, key])\n    }\n\n    return newObj\n  }\n\n  // // Detect classes, functions, and other non-serializable objects\n  // // and return undefined. Exclude some known types that are serializable\n  // if (\n  //   typeof obj === 'function' ||\n  //   (typeof obj === 'object' &&\n  //     ![Object, Promise, ReadableStream].includes((obj as any)?.constructor))\n  // ) {\n  //   console.info(obj)\n  //   warning(false, `Non-serializable value ☝️ found at ${path.join('.')}`)\n  //   return undefined as any\n  // }\n\n  const newObj = cb(obj, path)\n\n  if (newObj !== obj) {\n    return newObj\n  }\n\n  return obj\n}\n"],"names":["newObj"],"mappings":";;;;;;AAoCgB,SAAA,2BACd,QACA,UACA;AACA,SAAO,MAAM;AAAA,IACX;AAAA,IACA,YAAY;AAAA,EACd;AAEA,SAAO,YAAY;AAAA,IACjB,cAAc,CAAC;AAAA,IACf,kCAAkB,IAAI;AAAA,IACtB,YAAY,CAAC,YAAY;AACvB,YAAM,UAAU,QAAQ,QAAQ,EAAE,KAAK,OAAO;AACvC,aAAA,UAAW,aAAa,KAAK,OAAO;AAC3C,aAAO,KAAK;AAAA,QACV,MAAM;AAAA,QACN;AAAA,MAAA,CACD;AAEM,aAAA,QAAQ,KAAK,MAAM;AAAA,MAAA,CAAE;AAAA,IAC9B;AAAA,IACA,cAAc,CAAC,WAAW,SAAS;AAC1B,aAAA,OAAO,UAAW,WAAW,YAAY;AACxC,cAAA,SAAS,MAAM,UAAU;AACxB,eAAA,4BAA4B,MAAM,GACvC,QAAQ,IAAI,aAAa,mBAAkB,6BAAM,cAAa,QAC1D;AAAA,EACZ,MAAM,QAAQ,EAAE,QAAQ,WAAY,CAAA,CAAC,QACzB,EACN;AAAA,MAAA,CACD;AAAA,IACH;AAAA,IACA,aAAa,CAAC,KAAK,UAAU;AAC3B;AAAA,QACE,CAAC,OAAO,UAAW,aAAa,IAAI,GAAG;AAAA,QACvC,oCAAoC;AAAA,MACtC;AAEO,aAAA,UAAW,aAAa,IAAI,GAAG;AACtC,aAAO,UAAW;AAAA,QAChB,MACE,+BAA+B,GAAG,iBAAiB;AAAA,UACjD,OAAO,IAAK,WAAW,UAAU,KAAK;AAAA,UACtC;AAAA,YACE,iBAAiB;AAAA,YACjB,MAAM;AAAA,YACN,MAAM;AAAA,UAAA;AAAA,QACR,CACD;AAAA,MACL;AAAA,IACF;AAAA,IACA;AAAA,EACF;AAEO,SAAA,UAAU,aAAa,MAAM,4BAA4B;AAAA,IAC9D,WAAW;AAAA,EAAA,CACZ;AACH;AAEO,SAAS,gBAAgB,QAAmB;;AACjD,QAAM,mBAAqC;AAAA,IACzC,UAAU,OAAO,IAAK;AAAA,IACtB,iBAAgB,kBAAO,SAAQ,cAAf;AAAA,IAChB,eACE,YAAO,MAAM,QAAQ,OAAO,MAAM,QAAQ,SAAS,CAAC,MAApD,mBAAuD,OAAM;AAAA,EACjE;AAEA,SAAO,UAAW;AAAA,IAChB,MACE,4BAA4B;AAAA,MAC1B,OAAO,IAAK,WAAW,UAAU,gBAAgB;AAAA,MACjD;AAAA,QACE,iBAAiB;AAAA,QACjB,MAAM;AAAA,QACN,MAAM;AAAA,MAAA;AAAA,IACR,CACD;AAAA,EACL;AACF;AAEgB,SAAA,uBACd,YACA,KAIA;AACA,QAAM,YAAyC,CAAC;AAEhD,QAAM,WAAW,UAAU,YAAY,CAAC,OAAO,SAAS;AAEtD,QAAI,iBAAiB,gBAAgB;AACnC,YAAM,CAAC,OAAO,KAAK,IAAI,MAAM,IAAI;AACjC,YAAM,QAA+B;AAAA,QACnC,MAAM;AAAA,QACN;AAAA,QACA,IAAI,UAAU;AAAA,QACd,YAAY,IAAI,MAAM;AAAA,QACtB,QAAQ;AAAA,MACV;AAEA,gBAAU,KAAK,KAAK;AACb,aAAA;AAAA,IAAA,WACE,iBAAiB,SAAS;AAC7B,YAAA,kBAAkB,MAAM,KAAK;AACnC,YAAM,QAAgC;AAAA,QACpC,MAAM;AAAA,QACN;AAAA,QACA,IAAI,UAAU;AAAA,QACd,YAAY,IAAI,MAAM;AAAA,QACtB,SAAS;AAAA,MACX;AACA,gBAAU,KAAK,KAAK;AAAA,IAAA;AAGf,WAAA;AAAA,EAAA,CACR;AAEM,SAAA,EAAE,UAAU,UAAU;AAC/B;AAEO,SAAS,eAAe,MAG5B;AACK,QAAA,EAAE,QAAQ,MAAA,IAAU;AAE1B,MAAI,YAAqD;AACzD,MAAI,uBAA4B;AAC5B,MAAA,MAAM,eAAe,QAAW;AAC5B,UAAA,SAAS,uBAAuB,MAAM,YAAY;AAAA,MAEtD;AAAA,IAAA,CACD;AACD,UAAM,aAAa,OAAO;AAC1B,gBAAY,OAAO;AACnB,2BAAuB,UAAU;AAAA,MAC/B,CAAC,KAAU,UAAgC;AAClC,eAAA,uBAAuB,KAAK,CAAC,QAAQ,GAAG,MAAM,IAAI,GAAG,MAAS;AAAA,MACvE;AAAA,MACA,EAAE,MAAM,OAAO,SAAS;AAAA,IAAA,EACxB;AAAA,EAAA;AAGJ,QAAM,WAAW,yBAAyB;AAAA,IACxC;AAAA,MACE,IAAI,MAAM;AAAA,MACV,qBAAqB,OAAO,IAAK,WAAW;AAAA,QAC1C,MAAM;AAAA,MACR;AAAA,MACA,YAAY,OAAO,IAAK,WAAW,UAAU,oBAAoB;AAAA,MACjE,OAAO,OAAO,IAAK,WAAW,UAAU,MAAM,KAAK;AAAA,MACnD,WAAW,uCAAW,IAAI,CAAC,UAAU,KAAK,OAAO,CAAC,QAAQ,MAAM,CAAC;AAAA,MACjE,WAAW,MAAM;AAAA,MACjB,QAAQ,MAAM;AAAA,IAChB;AAAA,IACA;AAAA,MACE,iBAAiB;AAAA,MACjB,MAAM;AAAA,MACN,MAAM;AAAA,IAAA;AAAA,EAET,CAAA;AAEM,SAAA,UAAW,aAAa,MAAM,QAAQ;AAE7C,MAAI,WAAW;AACH,cAAA,QAAQ,CAAC,UAAU;AAC3B,UAAI,MAAM,SAAS,UAAW,QAAO,cAAc,KAAK;AACxD,aAAO,aAAa,KAAK;AAAA,IAAA,CAC1B;AAAA,EAAA;AAGH,WAAS,cAAc,OAA+B;AAC7C,WAAA,UAAW,aAAa,YAAY;AACzC,YAAM,MAAM;AAEZ,aAAO,8BAA8B;AAAA,QACnC;AAAA,UACE,SAAS,MAAM;AAAA,UACf,IAAI,MAAM;AAAA,UACV,cAAc,MAAM,QAAQ,oBAAoB;AAAA,QAClD;AAAA,QACA;AAAA,UACE,iBAAiB;AAAA,UACjB,MAAM;AAAA,UACN,MAAM;AAAA,QAAA;AAAA,MAET,CAAA;AAAA,IAAA,CACF;AAAA,EAAA;AAGH,WAAS,aAAa,OAA8B;AAG3C,WAAA,UAAW,WAAW,YAAY;AAEnC,UAAA;AACI,cAAA,SAAS,MAAM,OAAO,UAAU;AACtC,YAAI,QAA8C;AAClD,eAAO,EAAE,QAAQ,MAAM,OAAO,KAAA,GAAQ,MAAM;AAC1C,cAAI,MAAM,OAAO;AACf,kBAAM,OAAO,2BAA2B;AAAA,cACtC;AAAA,gBACE,SAAS,MAAM;AAAA,gBACf,IAAI,MAAM;AAAA,gBACV,OAAO,MAAM;AAAA,cACf;AAAA,cACA;AAAA,gBACE,iBAAiB;AAAA,gBACjB,MAAM;AAAA,gBACN,MAAM;AAAA,cAAA;AAAA,YAET,CAAA;AAEM,mBAAA,UAAW,aAAa,MAAM,IAAI;AAAA,UAAA;AAAA,QAC3C;AAGF,eAAO,UAAW;AAAA,UAChB,MACE,2BAA2B;AAAA,YACzB;AAAA,cACE,SAAS,MAAM;AAAA,cACf,IAAI,MAAM;AAAA,YACZ;AAAA,YACA;AAAA,cACE,iBAAiB;AAAA,cACjB,MAAM;AAAA,cACN,MAAM;AAAA,YAAA;AAAA,UACR,CACD;AAAA,QACL;AAAA,eACO,KAAK;AACJ,gBAAA,MAAM,qBAAqB,GAAG;AAAA,MAAA;AAGjC,aAAA;AAAA,IAAA,CACR;AAAA,EAAA;AAEL;AAEA,SAAS,uBAA0B,KAAQ,MAAqB,OAAe;AAEzE,MAAA,KAAK,WAAW,GAAG;AACd,WAAA;AAAA,EAAA;AAGT,QAAM,CAAC,KAAK,GAAG,IAAI,IAAI;AAEnB,MAAA,MAAM,QAAQ,GAAG,GAAG;AACtB,WAAO,IAAI,IAAI,CAAC,MAAM,MAAM;AACtB,UAAA,MAAM,OAAO,GAAG,GAAG;AACd,eAAA,uBAAuB,MAAM,MAAM,KAAK;AAAA,MAAA;AAE1C,aAAA;AAAA,IAAA,CACR;AAAA,EAAA;AAGC,MAAA,cAAc,GAAG,GAAG;AACf,WAAA;AAAA,MACL,GAAG;AAAA,MACH,CAAC,GAAI,GAAG,uBAAwB,IAAY,GAAI,GAAG,MAAM,KAAK;AAAA,IAChE;AAAA,EAAA;AAGK,SAAA;AACT;AAEO,SAAS,UACd,KACA,IACA,OAAsB,CAAA,GACnB;AACC,MAAA,aAAa,GAAG,GAAG;AACrB,WAAO,IAAI,IAAI,CAAC,OAAO,MAAM,UAAU,OAAO,IAAI,CAAC,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,CAAC;AAAA,EAAA;AAGlE,MAAA,cAAc,GAAG,GAAG;AAEtB,UAAMA,UAAc,CAAC;AAErB,eAAW,OAAO,KAAK;AACrBA,cAAO,GAAG,IAAI,UAAU,IAAI,GAAG,GAAG,IAAI,CAAC,GAAG,MAAM,GAAG,CAAC;AAAA,IAAA;AAG/CA,WAAAA;AAAAA,EAAA;AAeH,QAAA,SAAS,GAAG,KAAK,IAAI;AAE3B,MAAI,WAAW,KAAK;AACX,WAAA;AAAA,EAAA;AAGF,SAAA;AACT;"}

package/dist/esm/ssr/transformStreamWithRouter.d.ts

@@ -0,0 +1,6 @@
+import { ReadableStream } from 'node:stream/web';
+import { Readable } from 'node:stream';
+import { AnyRouter } from '../router.js';
+export declare function transformReadableStreamWithRouter(router: AnyRouter, routerStream: ReadableStream): ReadableStream<any>;
+export declare function transformPipeableStreamWithRouter(router: AnyRouter, routerStream: Readable): Readable;
+export declare function transformStreamWithRouter(router: AnyRouter, appStream: ReadableStream): ReadableStream<any>;

package/dist/esm/ssr/transformStreamWithRouter.js

@@ -0,0 +1,183 @@
+import { ReadableStream } from "node:stream/web";
+import { Readable } from "node:stream";
+import { createControlledPromise } from "../utils.js";
+function transformReadableStreamWithRouter(router, routerStream) {
+  return transformStreamWithRouter(router, routerStream);
+}
+function transformPipeableStreamWithRouter(router, routerStream) {
+  return Readable.fromWeb(
+    transformStreamWithRouter(router, Readable.toWeb(routerStream))
+  );
+}
+const patternBodyStart = /(<body)/;
+const patternBodyEnd = /(<\/body>)/;
+const patternHtmlEnd = /(<\/html>)/;
+const patternHeadStart = /(<head.*?>)/;
+const patternClosingTag = /(<\/[a-zA-Z][\w:.-]*?>)/g;
+const textDecoder = new TextDecoder();
+function createPassthrough() {
+  let controller;
+  const encoder = new TextEncoder();
+  const stream = new ReadableStream({
+    start(c) {
+      controller = c;
+    }
+  });
+  const res = {
+    stream,
+    write: (chunk) => {
+      controller.enqueue(encoder.encode(chunk));
+    },
+    end: (chunk) => {
+      if (chunk) {
+        controller.enqueue(encoder.encode(chunk));
+      }
+      controller.close();
+      res.destroyed = true;
+    },
+    destroy: (error) => {
+      controller.error(error);
+    },
+    destroyed: false
+  };
+  return res;
+}
+async function readStream(stream, opts) {
+  var _a, _b, _c;
+  try {
+    const reader = stream.getReader();
+    let chunk;
+    while (!(chunk = await reader.read()).done) {
+      (_a = opts.onData) == null ? void 0 : _a.call(opts, chunk);
+    }
+    (_b = opts.onEnd) == null ? void 0 : _b.call(opts);
+  } catch (error) {
+    (_c = opts.onError) == null ? void 0 : _c.call(opts, error);
+  }
+}
+function transformStreamWithRouter(router, appStream) {
+  const finalPassThrough = createPassthrough();
+  let isAppRendering = true;
+  let routerStreamBuffer = "";
+  let pendingClosingTags = "";
+  let bodyStarted = false;
+  let headStarted = false;
+  let leftover = "";
+  let leftoverHtml = "";
+  function getBufferedRouterStream() {
+    const html = routerStreamBuffer;
+    routerStreamBuffer = "";
+    return html;
+  }
+  function decodeChunk(chunk) {
+    if (chunk instanceof Uint8Array) {
+      return textDecoder.decode(chunk);
+    }
+    return String(chunk);
+  }
+  const injectedHtmlDonePromise = createControlledPromise();
+  let processingCount = 0;
+  router.serverSsr.injectedHtml.forEach((promise) => {
+    handleInjectedHtml(promise);
+  });
+  const stopListeningToInjectedHtml = router.subscribe(
+    "onInjectedHtml",
+    (e) => {
+      handleInjectedHtml(e.promise);
+    }
+  );
+  function handleInjectedHtml(promise) {
+    processingCount++;
+    promise.then((html) => {
+      if (!bodyStarted) {
+        routerStreamBuffer += html;
+      } else {
+        finalPassThrough.write(html);
+      }
+    }).catch(injectedHtmlDonePromise.reject).finally(() => {
+      processingCount--;
+      if (!isAppRendering && processingCount === 0) {
+        stopListeningToInjectedHtml();
+        injectedHtmlDonePromise.resolve();
+      }
+    });
+  }
+  injectedHtmlDonePromise.then(() => {
+    const finalHtml = leftoverHtml + getBufferedRouterStream() + pendingClosingTags;
+    finalPassThrough.end(finalHtml);
+  }).catch((err) => {
+    console.error("Error reading routerStream:", err);
+    finalPassThrough.destroy(err);
+  });
+  readStream(appStream, {
+    onData: (chunk) => {
+      const text = decodeChunk(chunk.value);
+      let chunkString = leftover + text;
+      const bodyEndMatch = chunkString.match(patternBodyEnd);
+      const htmlEndMatch = chunkString.match(patternHtmlEnd);
+      if (!bodyStarted) {
+        const bodyStartMatch = chunkString.match(patternBodyStart);
+        if (bodyStartMatch) {
+          bodyStarted = true;
+        }
+      }
+      if (!headStarted) {
+        const headStartMatch = chunkString.match(patternHeadStart);
+        if (headStartMatch) {
+          headStarted = true;
+          const index = headStartMatch.index;
+          const headTag = headStartMatch[0];
+          const remaining = chunkString.slice(index + headTag.length);
+          finalPassThrough.write(
+            chunkString.slice(0, index) + headTag + getBufferedRouterStream()
+          );
+          chunkString = remaining;
+        }
+      }
+      if (!bodyStarted) {
+        finalPassThrough.write(chunkString);
+        leftover = "";
+        return;
+      }
+      if (bodyEndMatch && htmlEndMatch && bodyEndMatch.index < htmlEndMatch.index) {
+        const bodyEndIndex = bodyEndMatch.index;
+        pendingClosingTags = chunkString.slice(bodyEndIndex);
+        finalPassThrough.write(
+          chunkString.slice(0, bodyEndIndex) + getBufferedRouterStream()
+        );
+        leftover = "";
+        return;
+      }
+      let result;
+      let lastIndex = 0;
+      while ((result = patternClosingTag.exec(chunkString)) !== null) {
+        lastIndex = result.index + result[0].length;
+      }
+      if (lastIndex > 0) {
+        const processed = chunkString.slice(0, lastIndex) + getBufferedRouterStream() + leftoverHtml;
+        finalPassThrough.write(processed);
+        leftover = chunkString.slice(lastIndex);
+      } else {
+        leftover = chunkString;
+        leftoverHtml += getBufferedRouterStream();
+      }
+    },
+    onEnd: () => {
+      isAppRendering = false;
+      if (processingCount === 0) {
+        injectedHtmlDonePromise.resolve();
+      }
+    },
+    onError: (error) => {
+      console.error("Error reading appStream:", error);
+      finalPassThrough.destroy(error);
+    }
+  });
+  return finalPassThrough.stream;
+}
+export {
+  transformPipeableStreamWithRouter,
+  transformReadableStreamWithRouter,
+  transformStreamWithRouter
+};
+//# sourceMappingURL=transformStreamWithRouter.js.map