@rangojs/router 0.0.0-experimental.dc0369d7 → 0.0.0-experimental.dcbea258

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.dc0369d7",
+  version: "0.0.0-experimental.dcbea258",
   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"
   },
@@ -3461,11 +3580,19 @@ function substituteRouteParams(pattern, params, encode = encodeURIComponent) {
   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;
@@ -3584,93 +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 || {};
-    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."
+      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
                 );
-              }
-            };
-            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 (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));
@@ -4785,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();