wip: prerender

Author: hi-ogawa

packages/plugin-rsc/examples/react-router/nitro.ts

@@ -3,10 +3,16 @@ import {
   copyPublicAssets,
   createNitro,
   prepare,
+  type $Fetch,
+  type Nitro,
   type NitroConfig,
 } from 'nitropack'
 import type { PresetName } from 'nitropack/presets'
+import path from 'node:path'
+import { pathToFileURL } from 'node:url'
 import type { Plugin, ViteBuilder } from 'vite'
+import { joinURL, withBase, withoutBase } from 'ufo'
+import fsp from 'node:fs/promises'
 
 // Using Nitro as post-build to target deployment platform. Inspired by Tanstack Start's approach.
 // https://github.com/TanStack/router/blob/5fd079e482b1252b8b11a936f1524a0dee368cae/packages/start-plugin-core/src/nitro-plugin/plugin.ts
@@ -18,6 +24,7 @@ type NitroPluginOptions = {
   preset: PresetName
   clientDir: string
   serverEntry: string
+  prerender?: string[]
 }
 
 export function nitroPlugin(nitroPluginOptions: NitroPluginOptions): Plugin[] {
@@ -121,10 +128,93 @@ export default defineEventHandler((event) => handler(toWebRequest(event)))
   const nitro = await createNitro(nitroConfig)
   await prepare(nitro)
   await copyPublicAssets(nitro)
-  await prerender()
+  if (nitroPluginOptions.prerender?.length) {
+    await prerender(nitro, nitroPluginOptions.prerender)
+  }
   await build(nitro)
   await nitro.close()
+}
+
+// TODO
+// https://github.com/TanStack/router/blob/5fd079e482b1252b8b11a936f1524a0dee368cae/packages/start-plugin-core/src/nitro-plugin/prerender.ts#L53
+// https://github.com/nitrojs/nitro/blob/c468de271cff8d56361c3b09ea1071ed545a550f/src/prerender/prerender.ts#L62-L74
+async function prerender(nitro: Nitro, pages: string[]) {
+  const nodeNitro = await createNitro({
+    ...nitro.options._config,
+    preset: 'nitro-prerender',
+    // logLevel: 0,
+    output: {
+      dir: 'dist/nitro/prerender',
+      // serverDir: path.resolve(prerenderOutputDir, 'server'),
+      // publicDir: path.resolve(prerenderOutputDir, 'public'),
+    },
+  })
+  await build(nodeNitro)
+
+  const serverEntrypoint = pathToFileURL(
+    path.resolve(nodeNitro.options.output.serverDir, 'index.mjs'),
+  ).toString()
+
+  const { closePrerenderer, localFetch } = (await import(serverEntrypoint)) as {
+    closePrerenderer: () => void
+    localFetch: $Fetch
+  }
+
+  for (const page of pages) {
+    await prerenderPage({ path: page })
+  }
+  closePrerenderer()
+  await nodeNitro.close()
+
+  async function prerenderPage(page: { path: string }) {
+    const encodedRoute = encodeURI(page.path)
+
+    const res = await localFetch<Response>(
+      withBase(encodedRoute, nodeNitro.options.baseURL),
+      {
+        headers: { 'x-nitro-prerender': encodedRoute },
+      },
+    )
+
+    if (!res.ok) {
+      throw new Error(`Failed to fetch ${page.path}: ${res.statusText}`, {
+        cause: res,
+      })
+    }
+
+    // const cleanPagePath = (prerenderOptions.outputPath || page.path).split(
+    //   /[?#]/,
+    // )[0]!
+    const cleanPagePath = page.path
+
+    // Guess route type and populate fileName
+    const contentType = res.headers.get('content-type') || ''
+    const isImplicitHTML =
+      !cleanPagePath.endsWith('.html') && contentType.includes('html')
+    // &&
+    // !JsonSigRx.test(dataBuff.subarray(0, 32).toString('utf8'))
+    const routeWithIndex = cleanPagePath.endsWith('/')
+      ? cleanPagePath + 'index'
+      : cleanPagePath
+
+    const htmlPath = cleanPagePath.endsWith('/')
+      ? // || prerenderOptions.autoSubfolderIndex
+        joinURL(cleanPagePath, 'index.html')
+      : cleanPagePath + '.html'
 
-  // TODO
-  async function prerender() {}
+    const filename = withoutBase(
+      isImplicitHTML ? htmlPath : routeWithIndex,
+      nitro.options.baseURL,
+    )
+
+    const html = await res.text()
+
+    const filepath = path.join(nitro.options.output.publicDir, filename)
+
+    await fsp.mkdir(path.dirname(filepath), {
+      recursive: true,
+    })
+
+    await fsp.writeFile(filepath, html)
+  }
 }

packages/plugin-rsc/examples/react-router/package.json

@@ -29,6 +29,7 @@
     "h3": "^1.15.3",
     "nitropack": "^2.11.13",
     "tailwindcss": "^4.1.11",
+    "ufo": "^1.6.1",
     "vite": "^7.0.2",
     "vite-plugin-inspect": "^11.3.0",
     "wrangler": "^4.23.0"

packages/plugin-rsc/examples/react-router/vite.config.ts

@@ -35,6 +35,7 @@ export default defineConfig({
         // - builder.environments.client.config.build.outDir
         clientDir: path.resolve('./dist/client'),
         serverEntry: path.resolve('./dist/ssr/index.js'),
+        // prerender: ['/'],
       }),
   ],
   optimizeDeps: {