feat: expose request header x-react-server-render-mode: prerender

hi-ogawa · Jul 1, 2024 · d9e2f13 · d9e2f13
1 parent cd33c0a
commit d9e2f13
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 6 deletions.
diff --git a/packages/react-server/examples/prerender/src/routes/posts/[id]/page.tsx b/packages/react-server/examples/prerender/src/routes/posts/[id]/page.tsx
@@ -20,7 +20,7 @@ export default async function Page(props: PageProps) {
       <div>{post.body}</div>
       <pre>
         [
-        {globalThis?.process?.env?.["REACT_SERVER_PRERENDER"]
+        {props.request.headers["x-react-server-render-mode"] === "prerender"
           ? "prerendered at"
           : "dynamically rendered at"}{" "}
         {new Date().toISOString()}]

diff --git a/packages/react-server/examples/prerender/vite.config.ts b/packages/react-server/examples/prerender/vite.config.ts
@@ -12,10 +12,7 @@ export default defineConfig({
   plugins: [
     react(),
     vitePluginReactServer({
-      prerender: async (_manifest, presets) => {
-        process.env["REACT_SERVER_PRERENDER"] = "1";
-        return presets.default();
-      },
+      prerender: (_manifest, presets) => presets.default(),
     }),
     vitePluginLogger(),
     vitePluginSsrMiddleware({

diff --git a/packages/react-server/src/features/prerender/plugin.ts b/packages/react-server/src/features/prerender/plugin.ts
@@ -48,7 +48,11 @@ export function prerenderPlugin({
         for (const route of routes) {
           console.log(`  • ${route}`);
           const url = new URL(route, "https://prerender.local");
-          const request = new Request(url);
+          const request = new Request(url, {
+            headers: {
+              "x-react-server-render-mode": "prerender",
+            },
+          });
           const { stream, html } = await entry.prerender(request);
           const data = Readable.from(stream as any);
           const htmlFile =