@rangojs/router 0.0.0-experimental.9 → 0.0.0-experimental.a5f27bd5

package/src/router/segment-resolution/static-store.ts

@@ -0,0 +1,67 @@
+/**
+ * Static Store
+ *
+ * Lazy-initialized static store for production Static handler interception.
+ * Manages the build-time static manifest lookup for pre-rendered components.
+ */
+
+import type { ReactNode } from "react";
+import { _getRequestContext } from "../../server/request-context.js";
+import type { StaticStore } from "../../prerender/store.js";
+
+// Lazy-initialized static store for production Static handler interception.
+// Remains undefined until first check; null means checked but no manifest.
+// When no __STATIC_MANIFEST exists (dev mode), set to null eagerly to avoid
+// the dynamic import() in ensureStaticDeps() which disrupts AsyncLocalStorage
+// in workerd/Cloudflare runtime.
+let _staticStore: StaticStore | null | undefined =
+  typeof globalThis !== "undefined" && globalThis.__STATIC_MANIFEST
+    ? undefined
+    : null;
+let _deserializeComponent: ((encoded: string) => Promise<unknown>) | undefined;
+
+async function ensureStaticDeps(): Promise<void> {
+  if (_staticStore === undefined) {
+    const { createStaticStore } = await import("../../prerender/store.js");
+    _staticStore = createStaticStore();
+  }
+  if (!_deserializeComponent && _staticStore) {
+    const { deserializeComponent } =
+      await import("../../cache/segment-codec.js");
+    _deserializeComponent = deserializeComponent;
+  }
+}
+
+/**
+ * Try to load a pre-rendered Static component from the build-time store.
+ * Returns the deserialized React element, or undefined if not available.
+ * Also replays any handle data captured at build time into the request's HandleStore.
+ *
+ * @param handlerId - The handler's $$id used to look up the static store entry.
+ * @param segmentId - The runtime segment shortCode. Handle data is replayed under
+ *   this key so that useHandle's segmentOrder matching works correctly.
+ */
+export async function tryStaticLookup(
+  handlerId: string,
+  segmentId: string,
+): Promise<ReactNode | undefined> {
+  // Fast path: already checked and no manifest available (dev mode or no Static handlers).
+  // Avoids await which can disrupt AsyncLocalStorage in workerd/Cloudflare runtime.
+  if (_staticStore === null) return undefined;
+  await ensureStaticDeps();
+  if (!_staticStore || !_deserializeComponent) return undefined;
+  const entry = await _staticStore.get(handlerId);
+  if (!entry) return undefined;
+
+  // Replay handle data captured during build-time rendering.
+  // The data was keyed by handlerId at build time; replay under segmentId
+  // so it matches the segment order used by useHandle on the client.
+  if (entry.handles && Object.keys(entry.handles).length > 0) {
+    const handleStore = _getRequestContext()?._handleStore;
+    if (handleStore) {
+      handleStore.replaySegmentData(segmentId, entry.handles);
+    }
+  }
+
+  return _deserializeComponent(entry.encoded) as Promise<ReactNode>;
+}