@rangojs/router 0.0.0-experimental.60 → 0.0.0-experimental.60c5dbde

package/dist/vite/index.js

@@ -18,6 +18,9 @@ function hashId(filePath, exportName) {
   const hash = crypto.createHash("sha256").update(input).digest("hex");
   return `${hash.slice(0, 8)}#${exportName}`;
 }
+function makeStubId(filePath, exportName, isBuild) {
+  return isBuild ? hashId(filePath, exportName) : `${filePath}#${exportName}`;
+}
 function hashInlineId(filePath, lineNumber, index) {
   const input = index !== void 0 && index > 0 ? `${filePath}:${lineNumber}:${index}` : `${filePath}:${lineNumber}`;
   return crypto.createHash("sha256").update(input).digest("hex").slice(0, 8);
@@ -910,9 +913,7 @@ function generateWholeFileStubs(cfg, bindings, code, filePath, isBuild) {
   });
   return { code: stubs.join("\n") + "\n", map: null };
 }
-function generateExprStubs(cfg, bindings, code, filePath, sourceId, isBuild) {
-  if (bindings.length === 0) return null;
-  const s = new MagicString2(code);
+function stubHandlerExprs(cfg, bindings, s, filePath, isBuild) {
   let hasChanges = false;
   for (const binding of bindings) {
     const exportName = binding.exportNames[0];
@@ -924,15 +925,7 @@ function generateExprStubs(cfg, bindings, code, filePath, sourceId, isBuild) {
     );
     hasChanges = true;
   }
-  if (!hasChanges) return null;
-  return {
-    code: s.toString(),
-    map: s.generateMap({
-      source: sourceId,
-      includeContent: true,
-      hires: "boundary"
-    })
-  };
+  return hasChanges;
 }
 function transformHandlerIds(cfg, bindings, s, filePath, isBuild) {
   let hasChanges = false;
@@ -1269,15 +1262,6 @@ ${lazyImports.join(",\n")}
           isBuild
         );
         if (wholeFile) return wholeFile;
-        const exprStubs = generateExprStubs(
-          PRERENDER_CONFIG,
-          bindings,
-          code,
-          filePath,
-          id,
-          isBuild
-        );
-        if (exprStubs) return exprStubs;
       }
       if (hasPrerenderHandlerCode && isRscEnv && isBuild) {
         const fnNames = getFnNames(PRERENDER_CONFIG.fnName);
@@ -1329,15 +1313,134 @@ ${lazyImports.join(",\n")}
           isBuild
         );
         if (wholeFile) return wholeFile;
-        const exprStubs = generateExprStubs(
-          STATIC_CONFIG,
-          bindings,
-          code,
-          filePath,
-          id,
-          isBuild
-        );
-        if (exprStubs) return exprStubs;
+      }
+      if (!isRscEnv && (hasPrerenderHandlerCode || hasStaticHandlerCode)) {
+        const prerenderFnNames = hasPrerenderHandlerCode ? getFnNames(PRERENDER_CONFIG.fnName) : [];
+        const staticFnNames = hasStaticHandlerCode ? getFnNames(STATIC_CONFIG.fnName) : [];
+        const loaderFnNames = hasLoaderCode ? getFnNames("createLoader") : [];
+        const handleFnNames = hasHandleCode ? getFnNames("createHandle") : [];
+        const lsFnNames = hasLocationStateCode ? getFnNames("createLocationState") : [];
+        const allBindings = [];
+        for (const fnNames of [
+          prerenderFnNames,
+          staticFnNames,
+          loaderFnNames,
+          handleFnNames,
+          lsFnNames
+        ]) {
+          if (fnNames.length > 0) {
+            allBindings.push(...getBindings(code, fnNames));
+          }
+        }
+        let canStubWholeFile = allBindings.length > 0 && isExportOnlyFile(code, allBindings);
+        if (canStubWholeFile && (handleFnNames.length > 0 || lsFnNames.length > 0)) {
+          const exportedLocals = new Set(allBindings.map((b) => b.localName));
+          const strippedBindings = [];
+          const localDeclPattern = /(?:^|;|\n)\s*(?:const|let|var|function)\s+(\w+)/g;
+          let declMatch;
+          while ((declMatch = localDeclPattern.exec(code)) !== null) {
+            const name = declMatch[1];
+            if (!exportedLocals.has(name) && !/^_c\d*$/.test(name)) {
+              strippedBindings.push(name);
+            }
+          }
+          const importPattern = /import\s*\{([^}]*)\}\s*from\s*["'](?!@rangojs\/router)[^"']*["']/g;
+          let importMatch;
+          while ((importMatch = importPattern.exec(code)) !== null) {
+            for (const spec of importMatch[1].split(",")) {
+              const m = spec.trim().match(/^[A-Za-z_$][\w$]*(?:\s+as\s+([A-Za-z_$][\w$]*))?$/);
+              if (m) strippedBindings.push(m[1] || m[0].trim().split(/\s/)[0]);
+            }
+          }
+          const defaultImportPattern = /import\s+([A-Za-z_$][\w$]*)\s+from\s*["'](?!@rangojs\/router)[^"']*["']/g;
+          while ((importMatch = defaultImportPattern.exec(code)) !== null) {
+            strippedBindings.push(importMatch[1]);
+          }
+          const nsImportPattern = /import\s+\*\s+as\s+([A-Za-z_$][\w$]*)\s+from\s*["'](?!@rangojs\/router)[^"']*["']/g;
+          while ((importMatch = nsImportPattern.exec(code)) !== null) {
+            strippedBindings.push(importMatch[1]);
+          }
+          if (strippedBindings.length > 0) {
+            const preservedBindings = allBindings.filter((b) => {
+              const fc = code.slice(b.callExprStart, b.callOpenParenPos + 1);
+              return handleFnNames.some((n) => fc.includes(n)) || lsFnNames.some((n) => fc.includes(n));
+            });
+            const strippedRe = new RegExp(
+              `\\b(?:${strippedBindings.join("|")})\\b`
+            );
+            canStubWholeFile = !preservedBindings.some((b) => {
+              const expr = code.slice(b.callExprStart, b.callCloseParenPos + 1);
+              return strippedRe.test(expr);
+            });
+          }
+        }
+        if (canStubWholeFile) {
+          const lines = [];
+          const neededImports = [];
+          if (handleFnNames.length > 0) neededImports.push("createHandle");
+          if (lsFnNames.length > 0) neededImports.push("createLocationState");
+          if (neededImports.length > 0) {
+            lines.push(
+              `import { ${neededImports.join(", ")} } from "@rangojs/router";`
+            );
+          }
+          for (const binding of allBindings) {
+            const fnCall = code.slice(
+              binding.callExprStart,
+              binding.callOpenParenPos + 1
+            );
+            const isHandle = handleFnNames.some((n) => fnCall.includes(n));
+            const isLocationState = lsFnNames.some((n) => fnCall.includes(n));
+            const primaryName = binding.exportNames[0];
+            const stubId = makeStubId(filePath, primaryName, isBuild);
+            if (isHandle || isLocationState) {
+              const rawArgs = code.slice(binding.callOpenParenPos + 1, binding.callCloseParenPos).replace(/\b_c\d*\s*=\s*/g, "");
+              const canonicalName = isHandle ? "createHandle" : "createLocationState";
+              const activeFnNames = isHandle ? handleFnNames : lsFnNames;
+              let rawCallee = code.slice(
+                binding.callExprStart,
+                binding.callOpenParenPos
+              );
+              for (const alias of activeFnNames) {
+                if (alias !== canonicalName && rawCallee.startsWith(alias)) {
+                  rawCallee = canonicalName + rawCallee.slice(alias.length);
+                  break;
+                }
+              }
+              if (isHandle) {
+                const idParam = binding.argCount === 0 ? `undefined, "${stubId}"` : `, "${stubId}"`;
+                lines.push(
+                  `export const ${primaryName} = ${rawCallee}(${rawArgs}${idParam});`
+                );
+                lines.push(`${primaryName}.$$id = "${stubId}";`);
+              } else {
+                lines.push(
+                  `export const ${primaryName} = ${rawCallee}(${rawArgs});`
+                );
+                lines.push(
+                  `${primaryName}.__rsc_ls_key = "__rsc_ls_${stubId}";`
+                );
+              }
+              for (const name of binding.exportNames.slice(1)) {
+                lines.push(`export const ${name} = ${primaryName};`);
+              }
+            } else {
+              let brand = "loader";
+              if (prerenderFnNames.some((n) => fnCall.includes(n))) {
+                brand = PRERENDER_CONFIG.brand;
+              } else if (staticFnNames.some((n) => fnCall.includes(n))) {
+                brand = STATIC_CONFIG.brand;
+              }
+              lines.push(
+                `export const ${primaryName} = { __brand: "${brand}", $$id: "${stubId}" };`
+              );
+              for (const name of binding.exportNames.slice(1)) {
+                lines.push(`export const ${name} = ${primaryName};`);
+              }
+            }
+          }
+          return { code: lines.join("\n") + "\n", map: null };
+        }
       }
       if (hasStaticHandlerCode && isRscEnv && isBuild) {
         const fnNames = getFnNames(STATIC_CONFIG.fnName);
@@ -1372,25 +1475,41 @@ ${lazyImports.join(",\n")}
           isBuild
         ) || changed;
       }
-      if (hasPrerenderHandlerCode && isRscEnv) {
+      if (hasPrerenderHandlerCode) {
         const fnNames = getFnNames(PRERENDER_CONFIG.fnName);
-        changed = transformHandlerIds(
-          PRERENDER_CONFIG,
-          getBindings(code, fnNames),
-          s,
-          filePath,
-          isBuild
-        ) || changed;
+        const bindings = getBindings(code, fnNames);
+        if (isRscEnv) {
+          changed = transformHandlerIds(
+            PRERENDER_CONFIG,
+            bindings,
+            s,
+            filePath,
+            isBuild
+          ) || changed;
+        } else {
+          changed = stubHandlerExprs(
+            PRERENDER_CONFIG,
+            bindings,
+            s,
+            filePath,
+            isBuild
+          ) || changed;
+        }
       }
-      if (hasStaticHandlerCode && isRscEnv) {
+      if (hasStaticHandlerCode) {
         const fnNames = getFnNames(STATIC_CONFIG.fnName);
-        changed = transformHandlerIds(
-          STATIC_CONFIG,
-          getBindings(code, fnNames),
-          s,
-          filePath,
-          isBuild
-        ) || changed;
+        const bindings = getBindings(code, fnNames);
+        if (isRscEnv) {
+          changed = transformHandlerIds(
+            STATIC_CONFIG,
+            bindings,
+            s,
+            filePath,
+            isBuild
+          ) || changed;
+        } else {
+          changed = stubHandlerExprs(STATIC_CONFIG, bindings, s, filePath, isBuild) || changed;
+        }
       }
       if (!changed) return;
       return {
@@ -1745,7 +1864,7 @@ import { resolve } from "node:path";
 // package.json
 var package_default = {
   name: "@rangojs/router",
-  version: "0.0.0-experimental.60",
+  version: "0.0.0-experimental.60c5dbde",
   description: "Django-inspired RSC router with composable URL patterns",
   keywords: [
     "react",
@@ -1887,7 +2006,7 @@ var package_default = {
     "test:unit:watch": "vitest"
   },
   dependencies: {
-    "@vitejs/plugin-rsc": "^0.5.19",
+    "@vitejs/plugin-rsc": "^0.5.23",
     "magic-string": "^0.30.17",
     picomatch: "^4.0.3",
     "rsc-html-stream": "^0.0.7"
@@ -1907,7 +2026,7 @@ var package_default = {
   },
   peerDependencies: {
     "@cloudflare/vite-plugin": "^1.25.0",
-    "@vitejs/plugin-rsc": "^0.5.14",
+    "@vitejs/plugin-rsc": "^0.5.23",
     react: "^18.0.0 || ^19.0.0",
     vite: "^7.3.0"
   },
@@ -3141,6 +3260,8 @@ function createCjsToEsmPlugin() {
 import { createServer as createViteServer } from "vite";
 import { resolve as resolve8 } from "node:path";
 import { readFileSync as readFileSync6 } from "node:fs";
+import { createRequire } from "node:module";
+import { pathToFileURL } from "node:url";
 
 // src/vite/plugins/virtual-stub-plugin.ts
 function createVirtualStubPlugin() {
@@ -3324,8 +3445,8 @@ function createDiscoveryState(entryPath, opts) {
     perRouterManifestDataMap: /* @__PURE__ */ new Map(),
     prerenderManifestEntries: null,
     staticManifestEntries: null,
-    handlerChunkInfo: null,
-    staticHandlerChunkInfo: null,
+    handlerChunkInfoMap: /* @__PURE__ */ new Map(),
+    staticHandlerChunkInfoMap: /* @__PURE__ */ new Map(),
     rscEntryFileName: null,
     resolvedPrerenderModules: void 0,
     resolvedStaticModules: void 0,
@@ -3456,13 +3577,31 @@ function encodePathParam(value) {
 }
 function substituteRouteParams(pattern, params, encode = encodeURIComponent) {
   let result = pattern;
+  let hadOmittedOptional = false;
   for (const [key, value] of Object.entries(params)) {
     const escaped = escapeRegExp2(key);
-    result = result.replace(
-      new RegExp(`:${escaped}(\\([^)]*\\))?\\??`),
-      encode(value)
-    );
-    result = result.replace(`*${key}`, encode(value));
+    if (value === "") {
+      result = result.replace(
+        new RegExp(`:${escaped}(\\([^)]*\\))?(?!\\?)`),
+        ""
+      );
+      result = result.replace(`*${key}`, "");
+    } else {
+      result = result.replace(
+        new RegExp(`:${escaped}(\\([^)]*\\))?\\??`),
+        encode(value)
+      );
+      result = result.replace(`*${key}`, encode(value));
+    }
+  }
+  result = result.replace(/:([a-zA-Z_][a-zA-Z0-9_]*)(\([^)]*\))?\?/g, () => {
+    hadOmittedOptional = true;
+    return "";
+  });
+  if (hadOmittedOptional) {
+    const hadTrailingSlash = pattern.length > 1 && pattern.endsWith("/");
+    result = result.replace(/\/\/+/g, "/").replace(/\/+$/, "") || "/";
+    if (hadTrailingSlash && !result.endsWith("/")) result += "/";
   }
   return result;
 }
@@ -3572,84 +3711,126 @@ async function expandPrerenderRoutes(state, rscEnv, registry, allManifests) {
     if (!params) return pattern;
     return substituteRouteParams(pattern, params);
   };
+  let resolvedRoutes = 0;
+  let totalDynamic = 0;
   for (const { manifest } of allManifests) {
     if (!manifest.prerenderRoutes) continue;
-    const defs = manifest._prerenderDefs || {};
     for (const routeName of manifest.prerenderRoutes) {
       const pattern = manifest.routeManifest[routeName];
-      if (!pattern) continue;
-      const def = defs[routeName];
-      const isPassthroughRoute = !!def?.options?.passthrough;
-      const hasDynamic = pattern.includes(":") || pattern.includes("*");
-      if (!hasDynamic) {
-        entries.push({
-          urlPath: pattern.replace(/\/$/, "") || "/",
-          routeName,
-          concurrency: 1,
-          isPassthroughRoute
-        });
-      } else {
-        if (def?.getParams) {
-          try {
-            const buildVars = {};
-            const getParamsCtx = {
-              build: true,
-              set: ((keyOrVar, value) => {
-                contextSet(buildVars, keyOrVar, value);
-              }),
-              reverse: getParamsReverse
-            };
-            const paramsList = await def.getParams(getParamsCtx);
-            const concurrency = def.options?.concurrency ?? 1;
-            const hasBuildVars = Object.keys(buildVars).length > 0 || Object.getOwnPropertySymbols(buildVars).length > 0;
-            for (const params of paramsList) {
-              let url = substituteRouteParams(
-                pattern,
-                params,
-                encodePathParam
-              );
-              if (url.includes("*")) {
-                const wildcardValue = params["*"] ?? params.splat;
-                if (wildcardValue !== void 0) {
-                  url = url.replace(/\*[^/]*$/, encodePathParam(wildcardValue));
+      if (pattern && (pattern.includes(":") || pattern.includes("*"))) {
+        totalDynamic++;
+      }
+    }
+  }
+  const paramsStart = performance.now();
+  const progressInterval = totalDynamic > 0 ? setInterval(() => {
+    const elapsed = ((performance.now() - paramsStart) / 1e3).toFixed(1);
+    console.log(
+      `[rsc-router] Resolving prerender params... ${resolvedRoutes}/${totalDynamic} routes (${elapsed}s)`
+    );
+  }, 5e3) : void 0;
+  try {
+    for (const { manifest } of allManifests) {
+      if (!manifest.prerenderRoutes) continue;
+      const defs = manifest._prerenderDefs || {};
+      const passthroughSet = new Set(manifest.passthroughRoutes || []);
+      for (const routeName of manifest.prerenderRoutes) {
+        const pattern = manifest.routeManifest[routeName];
+        if (!pattern) continue;
+        const def = defs[routeName];
+        const isPassthroughRoute = passthroughSet.has(routeName);
+        const hasDynamic = pattern.includes(":") || pattern.includes("*");
+        if (!hasDynamic) {
+          entries.push({
+            urlPath: pattern.replace(/\/$/, "") || "/",
+            routeName,
+            concurrency: 1,
+            isPassthroughRoute
+          });
+        } else {
+          if (def?.getParams) {
+            try {
+              const buildVars = {};
+              const buildEnv = state.resolvedBuildEnv;
+              const getParamsCtx = {
+                build: true,
+                dev: !state.isBuildMode,
+                set: ((keyOrVar, value) => {
+                  contextSet(buildVars, keyOrVar, value);
+                }),
+                reverse: getParamsReverse,
+                get env() {
+                  if (buildEnv !== void 0) return buildEnv;
+                  throw new Error(
+                    "[rsc-router] ctx.env is not available during build-time getParams(). Configure buildEnv in your rango() plugin options to enable build-time env access."
+                  );
                 }
+              };
+              const paramsList = await def.getParams(getParamsCtx);
+              const concurrency = def.options?.concurrency ?? 1;
+              const hasBuildVars = Object.keys(buildVars).length > 0 || Object.getOwnPropertySymbols(buildVars).length > 0;
+              for (const params of paramsList) {
+                let url = substituteRouteParams(
+                  pattern,
+                  params,
+                  encodePathParam
+                );
+                if (url.includes("*")) {
+                  const wildcardValue = params["*"] ?? params.splat;
+                  if (wildcardValue !== void 0) {
+                    url = url.replace(
+                      /\*[^/]*$/,
+                      encodePathParam(wildcardValue)
+                    );
+                  }
+                }
+                entries.push({
+                  urlPath: url.replace(/\/$/, "") || "/",
+                  routeName,
+                  concurrency,
+                  ...hasBuildVars ? { buildVars } : {},
+                  isPassthroughRoute
+                });
               }
-              entries.push({
-                urlPath: url.replace(/\/$/, "") || "/",
-                routeName,
-                concurrency,
-                ...hasBuildVars ? { buildVars } : {},
-                isPassthroughRoute
-              });
-            }
-          } catch (err) {
-            if (err.name === "Skip") {
-              console.log(
-                `[rsc-router]   SKIP route "${routeName}" - ${err.message}`
-              );
-              notifyOnError(
-                registry,
-                err,
-                "prerender",
-                routeName,
-                void 0,
-                true
+              resolvedRoutes++;
+            } catch (err) {
+              resolvedRoutes++;
+              if (err.name === "Skip") {
+                console.log(
+                  `[rsc-router]   SKIP route "${routeName}" - ${err.message}`
+                );
+                notifyOnError(
+                  registry,
+                  err,
+                  "prerender",
+                  routeName,
+                  void 0,
+                  true
+                );
+                continue;
+              }
+              console.error(
+                `[rsc-router] Failed to get params for prerender route "${routeName}": ${err.message}`
               );
-              continue;
+              notifyOnError(registry, err, "prerender", routeName);
+              throw err;
             }
-            console.error(
-              `[rsc-router] Failed to get params for prerender route "${routeName}": ${err.message}`
+          } else {
+            console.warn(
+              `[rsc-router] Dynamic prerender route "${routeName}" has no getParams(), skipping`
             );
-            notifyOnError(registry, err, "prerender", routeName);
-            throw err;
           }
-        } else {
-          console.warn(
-            `[rsc-router] Dynamic prerender route "${routeName}" has no getParams(), skipping`
-          );
         }
       }
     }
+  } finally {
+    if (progressInterval) {
+      clearInterval(progressInterval);
+      const elapsed = ((performance.now() - paramsStart) / 1e3).toFixed(1);
+      console.log(
+        `[rsc-router] Resolved prerender params: ${resolvedRoutes}/${totalDynamic} routes (${elapsed}s)`
+      );
+    }
   }
   if (entries.length === 0) return;
   const maxConcurrency = Math.max(...entries.map((e) => e.concurrency));
@@ -3676,7 +3857,8 @@ async function expandPrerenderRoutes(state, rscEnv, registry, allManifests) {
               entry.urlPath,
               {},
               entry.buildVars,
-              entry.isPassthroughRoute
+              entry.isPassthroughRoute,
+              state.resolvedBuildEnv
             );
             if (!result) continue;
             if (result.passthrough) {
@@ -3800,7 +3982,9 @@ async function renderStaticHandlers(state, rscEnv, registry) {
           const result = await routerInstance.renderStaticSegment(
             def.handler,
             def.$$id,
-            def.$$routePrefix
+            def.$$routePrefix,
+            state.resolvedBuildEnv,
+            !state.isBuildMode
           );
           if (result) {
             const hasHandles = Object.keys(result.handles).length > 0;
@@ -4378,48 +4562,45 @@ function postprocessBundle(state) {
   );
   const evictionTargets = [
     {
-      info: state.handlerChunkInfo,
+      infos: state.handlerChunkInfoMap.values(),
       fnName: "Prerender",
       brand: "prerenderHandler",
       label: "handler code from RSC bundle"
     },
     {
-      info: state.staticHandlerChunkInfo,
+      infos: state.staticHandlerChunkInfoMap.values(),
       fnName: "Static",
       brand: "staticHandler",
       label: "static handler code"
     }
   ];
   for (const target of evictionTargets) {
-    if (!target.info) continue;
-    const chunkPath = resolve7(
-      state.projectRoot,
-      "dist/rsc",
-      target.info.fileName
-    );
-    try {
-      const code = readFileSync5(chunkPath, "utf-8");
-      const result = evictHandlerCode(
-        code,
-        target.info.exports,
-        target.fnName,
-        target.brand
-      );
-      if (result) {
-        writeFileSync4(chunkPath, result.code);
-        const savedKB = (result.savedBytes / 1024).toFixed(1);
-        console.log(
-          `[rsc-router] Evicted ${target.label} (${savedKB} KB saved): ${target.info.fileName}`
+    for (const info of target.infos) {
+      const chunkPath = resolve7(state.projectRoot, "dist/rsc", info.fileName);
+      try {
+        const code = readFileSync5(chunkPath, "utf-8");
+        const result = evictHandlerCode(
+          code,
+          info.exports,
+          target.fnName,
+          target.brand
+        );
+        if (result) {
+          writeFileSync4(chunkPath, result.code);
+          const savedKB = (result.savedBytes / 1024).toFixed(1);
+          console.log(
+            `[rsc-router] Evicted ${target.label} (${savedKB} KB saved): ${info.fileName}`
+          );
+        }
+      } catch (replaceErr) {
+        console.warn(
+          `[rsc-router] Failed to evict ${target.label}: ${replaceErr.message}`
         );
       }
-    } catch (replaceErr) {
-      console.warn(
-        `[rsc-router] Failed to evict ${target.label}: ${replaceErr.message}`
-      );
     }
   }
-  state.handlerChunkInfo = null;
-  state.staticHandlerChunkInfo = null;
+  state.handlerChunkInfoMap.clear();
+  state.staticHandlerChunkInfoMap.clear();
   if (hasPrerenderData && existsSync6(rscEntryPath)) {
     const rscCode = readFileSync5(rscEntryPath, "utf-8");
     if (!rscCode.includes("__prerender-manifest.js")) {
@@ -4462,7 +4643,7 @@ function postprocessBundle(state) {
   }
   if (hasStaticData && existsSync6(rscEntryPath)) {
     const rscCode = readFileSync5(rscEntryPath, "utf-8");
-    if (!rscCode.includes("__STATIC_MANIFEST")) {
+    if (!rscCode.includes("__static-manifest.js")) {
       try {
         const manifestEntries = [];
         let totalBytes = copyStagedBuildAssets(
@@ -4531,8 +4712,67 @@ async function createTempRscServer(state, options = {}) {
     ]
   });
 }
+async function resolveBuildEnv(option, factoryCtx) {
+  if (!option) return null;
+  if (option === "auto") {
+    if (factoryCtx.preset !== "cloudflare") {
+      throw new Error(
+        '[rsc-router] buildEnv: "auto" is only supported with preset: "cloudflare". Use a factory function or plain object for other presets.'
+      );
+    }
+    try {
+      const userRequire = createRequire(
+        resolve8(factoryCtx.root, "package.json")
+      );
+      const wranglerPath = userRequire.resolve("wrangler");
+      const { getPlatformProxy } = await import(pathToFileURL(wranglerPath).href);
+      const proxy = await getPlatformProxy();
+      return {
+        env: proxy.env,
+        dispose: proxy.dispose
+      };
+    } catch (err) {
+      throw new Error(
+        `[rsc-router] buildEnv: "auto" requires wrangler to be installed.
+Install it with: pnpm add -D wrangler
+${err.message}`
+      );
+    }
+  }
+  if (typeof option === "function") {
+    return await option(factoryCtx);
+  }
+  return { env: option };
+}
+async function acquireBuildEnv(s, command, mode) {
+  const option = s.opts?.buildEnv;
+  if (!option) return false;
+  const result = await resolveBuildEnv(option, {
+    root: s.projectRoot,
+    mode,
+    command,
+    preset: s.opts?.preset ?? "node"
+  });
+  if (!result) return false;
+  s.resolvedBuildEnv = result.env;
+  s.buildEnvDispose = result.dispose ?? null;
+  return true;
+}
+async function releaseBuildEnv(s) {
+  if (s.buildEnvDispose) {
+    try {
+      await s.buildEnvDispose();
+    } catch (err) {
+      console.warn(`[rsc-router] buildEnv dispose failed: ${err.message}`);
+    }
+    s.buildEnvDispose = null;
+  }
+  s.resolvedBuildEnv = void 0;
+}
 function createRouterDiscoveryPlugin(entryPath, opts) {
   const s = createDiscoveryState(entryPath, opts);
+  let viteCommand = "build";
+  let viteMode = "production";
   return {
     name: "@rangojs/router:discovery",
     config() {
@@ -4541,31 +4781,13 @@ function createRouterDiscoveryPlugin(entryPath, opts) {
           __RANGO_DEBUG__: JSON.stringify(!!process.env.INTERNAL_RANGO_DEBUG)
         }
       };
-      if (opts?.enableBuildPrerender) {
-        config.environments = {
-          rsc: {
-            build: {
-              rollupOptions: {
-                output: {
-                  manualChunks(id) {
-                    if (s.resolvedPrerenderModules?.has(id)) {
-                      return "__prerender-handlers";
-                    }
-                    if (s.resolvedStaticModules?.has(id)) {
-                      return "__static-handlers";
-                    }
-                  }
-                }
-              }
-            }
-          }
-        };
-      }
       return config;
     },
     configResolved(config) {
       s.projectRoot = config.root;
       s.isBuildMode = config.command === "build";
+      viteCommand = config.command;
+      viteMode = config.mode;
       s.userResolveAlias = config.resolve.alias;
       if (!s.resolvedEntryPath && opts?.routerPathRef?.path) {
         s.resolvedEntryPath = opts.routerPathRef.path;
@@ -4610,6 +4832,8 @@ function createRouterDiscoveryPlugin(entryPath, opts) {
           });
           prerenderTempServer = null;
         }
+        releaseBuildEnv(s).catch(() => {
+        });
       });
       async function getOrCreateTempServer() {
         if (prerenderNodeRegistry) {
@@ -4640,6 +4864,7 @@ function createRouterDiscoveryPlugin(entryPath, opts) {
         if (!rscEnv?.runner) {
           s.devServerOrigin = getDevServerOrigin();
           try {
+            await acquireBuildEnv(s, viteCommand, viteMode);
             const tempRscEnv = await getOrCreateTempServer();
             if (tempRscEnv) {
               await discoverRouters(s, tempRscEnv);
@@ -4655,6 +4880,7 @@ ${err.stack}`
           return;
         }
         try {
+          await acquireBuildEnv(s, viteCommand, viteMode);
           const serverMod = await rscEnv.runner.import(
             "@rangojs/router/server"
           );
@@ -4719,7 +4945,26 @@ ${err.stack}`
           res.end("Missing pathname");
           return;
         }
-        let registry = mainRegistry;
+        const rscEnv = server.environments?.rsc;
+        let registry = null;
+        if (rscEnv?.runner && s.resolvedEntryPath) {
+          try {
+            await rscEnv.runner.import(s.resolvedEntryPath);
+            const serverMod = await rscEnv.runner.import(
+              "@rangojs/router/server"
+            );
+            registry = serverMod.RouterRegistry ?? null;
+          } catch (err) {
+            console.warn(
+              `[rsc-router] Dev prerender module refresh failed: ${err.message}`
+            );
+            res.statusCode = 500;
+            res.end(`Prerender handler error: ${err.message}`);
+            return;
+          }
+        } else {
+          registry = mainRegistry;
+        }
         if (!registry) {
           if (!prerenderNodeRegistry) {
             await getOrCreateTempServer();
@@ -4741,7 +4986,10 @@ ${err.stack}`
               pathname,
               {},
               void 0,
-              wantPassthrough
+              wantPassthrough,
+              s.resolvedBuildEnv,
+              true
+              // devMode: check getParams for passthrough routes
             );
             if (!result) continue;
             if (result.passthrough) continue;
@@ -4877,6 +5125,7 @@ ${err.stack}`
       resetStagedBuildAssets(s.projectRoot);
       s.prerenderManifestEntries = null;
       s.staticManifestEntries = null;
+      await acquireBuildEnv(s, viteCommand, viteMode);
       let tempServer = null;
       globalThis.__rscRouterDiscoveryActive = true;
       try {
@@ -4916,6 +5165,7 @@ ${details}`
         if (tempServer) {
           await tempServer.close();
         }
+        await releaseBuildEnv(s);
       }
     },
     // Virtual module: provides the pre-generated route manifest as a JS module
@@ -4958,20 +5208,30 @@ ${details}`
       }
       if (!s.resolvedPrerenderModules?.size && !s.resolvedStaticModules?.size)
         return;
+      s.handlerChunkInfoMap.clear();
+      s.staticHandlerChunkInfoMap.clear();
       for (const [fileName, chunk] of Object.entries(bundle)) {
         if (chunk.type !== "chunk") continue;
-        if (fileName.includes("__prerender-handlers") && s.resolvedPrerenderModules?.size) {
+        if (s.resolvedPrerenderModules?.size) {
           const handlers = extractHandlerExportsFromChunk(
             chunk.code,
             s.resolvedPrerenderModules,
             "Prerender",
-            true
+            false
           );
           if (handlers.length > 0) {
-            s.handlerChunkInfo = { fileName, exports: handlers };
+            const existing = s.handlerChunkInfoMap.get(fileName);
+            if (existing) {
+              existing.exports.push(...handlers);
+            } else {
+              s.handlerChunkInfoMap.set(fileName, {
+                fileName,
+                exports: handlers
+              });
+            }
           }
         }
-        if (fileName.includes("__static-handlers") && s.resolvedStaticModules?.size) {
+        if (s.resolvedStaticModules?.size) {
           const handlers = extractHandlerExportsFromChunk(
             chunk.code,
             s.resolvedStaticModules,
@@ -4979,7 +5239,15 @@ ${details}`
             false
           );
           if (handlers.length > 0) {
-            s.staticHandlerChunkInfo = { fileName, exports: handlers };
+            const existing = s.staticHandlerChunkInfoMap.get(fileName);
+            if (existing) {
+              existing.exports.push(...handlers);
+            } else {
+              s.staticHandlerChunkInfoMap.set(fileName, {
+                fileName,
+                exports: handlers
+              });
+            }
           }
         }
       }
@@ -5271,7 +5539,8 @@ ${list}`);
     createRouterDiscoveryPlugin(discoveryEntryPath, {
       routerPathRef: discoveryRouterRef,
       enableBuildPrerender: prerenderEnabled,
-      staticRouteTypesGeneration: resolvedOptions.staticRouteTypesGeneration
+      buildEnv: options?.buildEnv,
+      preset
     })
   );
   return plugins;