feat: split runtime/build server entries to exclude static routes from runtime bundle

packages/waku/src/lib/utils/managed.ts

@@ -1,9 +1,113 @@
+import fs from 'node:fs';
+import { readdir, readFile } from 'node:fs/promises';
+import path from 'node:path';
 import { EXTENSIONS, SRC_MIDDLEWARE, SRC_PAGES } from '../constants.js';
 
-export const getManagedServerEntry = (srcDir: string) => {
+type ManagedServerEntryMode = 'runtime' | 'build';
+
+const toPosix = (p: string) => p.split(path.sep).join('/');
+
+const GET_CONFIG_EXPORT_PATTERNS = [
+  /export\s+(?:async\s+)?function\s+getConfig\b/m,
+  /export\s+(?:const|let|var)\s+getConfig\b/m,
+  /export\s*\{[^}]*\bgetConfig\b[^}]*\}/m,
+];
+
+const HAS_DYNAMIC_RENDER_LITERAL_PATTERN = /render\s*:\s*['"]dynamic['"]/m;
+const HAS_STATIC_RENDER_LITERAL_PATTERN = /render\s*:\s*['"]static['"]/m;
+
+const hasGetConfigExport = (code: string) =>
+  GET_CONFIG_EXPORT_PATTERNS.some((pattern) => pattern.test(code));
+
+const collectPageFiles = async (
+  dir: string,
+  baseDir: string,
+): Promise<string[]> => {
+  const entries = await readdir(dir, { withFileTypes: true });
+  const files: string[] = [];
+  for (const entry of entries) {
+    const fullPath = path.join(dir, entry.name);
+    if (entry.isDirectory()) {
+      files.push(...(await collectPageFiles(fullPath, baseDir)));
+      continue;
+    }
+    if (!EXTENSIONS.some((ext) => entry.name.endsWith(ext))) {
+      continue;
+    }
+    files.push(toPosix(path.relative(baseDir, fullPath)));
+  }
+  return files;
+};
+
+const getRuntimeExcludedPageFiles = async ({
+  rootDir,
+  srcDir,
+  apiDir = '_api',
+  slicesDir = '_slices',
+}: {
+  rootDir: string;
+  srcDir: string;
+  apiDir?: string;
+  slicesDir?: string;
+}) => {
+  const pagesDir = path.join(rootDir, srcDir, SRC_PAGES);
+  if (!fs.existsSync(pagesDir)) {
+    return [] as string[];
+  }
+  const files = await collectPageFiles(pagesDir, pagesDir);
+  const excluded: string[] = [];
+  for (const file of files) {
+    const parts = file.split('/');
+    const stem = parts[parts.length - 1]!.replace(/\.[^.]+$/, '');
+    if (
+      parts[0] === apiDir ||
+      parts[0] === slicesDir ||
+      stem === '_layout' ||
+      stem === '_root'
+    ) {
+      continue;
+    }
+    const code = await readFile(path.join(pagesDir, file), 'utf8');
+    if (!hasGetConfigExport(code)) {
+      excluded.push(file);
+      continue;
+    }
+    if (HAS_DYNAMIC_RENDER_LITERAL_PATTERN.test(code)) {
+      continue;
+    }
+    if (HAS_STATIC_RENDER_LITERAL_PATTERN.test(code)) {
+      excluded.push(file);
+      continue;
+    }
+  }
+  return excluded;
+};
+
+export const getManagedServerEntry = async ({
+  srcDir,
+  rootDir,
+  mode,
+}: {
+  srcDir: string;
+  rootDir: string;
+  mode: ManagedServerEntryMode;
+}) => {
   const globBase = `/${srcDir}/${SRC_PAGES}`;
   const exts = EXTENSIONS.map((ext) => ext.slice(1)).join(',');
-  const globPattern = `${globBase}/**/*.{${exts}}`;
+  const fullPagesGlobPattern = `${globBase}/**/*.{${exts}}`;
+  let pagesGlob: string | string[] = fullPagesGlobPattern;
+  if (mode === 'runtime') {
+    const excludedPageFiles = await getRuntimeExcludedPageFiles({
+      rootDir,
+      srcDir,
+    });
+    if (excludedPageFiles.length) {
+      pagesGlob = [
+        fullPagesGlobPattern,
+        ...excludedPageFiles.map((file) => `!${globBase}/${toPosix(file)}`),
+      ];
+    }
+  }
   const middlewareGlob = [
     `/${srcDir}/${SRC_MIDDLEWARE}/*.{${exts}}`,
     `!/${srcDir}/${SRC_MIDDLEWARE}/*.{test,spec}.{${exts}}`,
@@ -15,7 +119,7 @@ import adapter from 'waku/adapters/default';
 export default adapter(
   fsRouter(
     import.meta.glob(
-      ${JSON.stringify(globPattern)},
+      ${JSON.stringify(pagesGlob)},
       { base: ${JSON.stringify(globBase)} }
     )
   ),

packages/waku/src/lib/vite-entries/entry.build.ts

@@ -1,6 +1,6 @@
 import { createRequire } from 'node:module';
 import { pathToFileURL } from 'node:url';
-import serverEntry from 'virtual:vite-rsc-waku/server-entry';
+import serverEntry from 'virtual:vite-rsc-waku/server-entry-build';
 import { INTERNAL_setAllEnv } from '../../server.js';
 import type { Unstable_EmitFile } from '../types.js';
 import { joinPath } from '../utils/path.js';

packages/waku/src/lib/vite-entries/entry.server.tsx

@@ -1,4 +1,4 @@
-import serverEntry from 'virtual:vite-rsc-waku/server-entry';
+import serverEntry from 'virtual:vite-rsc-waku/server-entry-runtime';
 import { INTERNAL_setAllEnv } from '../../server.js';
 
 export { serverEntry as unstable_serverEntry };

packages/waku/src/lib/vite-plugins/user-entries.ts

@@ -6,14 +6,33 @@ import {
 } from '../utils/managed.js';
 
 export function userEntriesPlugin({ srcDir }: { srcDir: string }): Plugin {
+  let rootDir = '';
+
   return {
     name: 'waku:vite-plugins:user-entries',
+    configResolved(config) {
+      rootDir = config.root;
+    },
     // resolve user entries and fallbacks to "managed mode" if not found.
     async resolveId(source, _importer, options) {
       if (source === 'virtual:vite-rsc-waku/server-entry') {
+        return '\0virtual:vite-rsc-waku/server-entry-runtime';
+      }
+      if (source === 'virtual:vite-rsc-waku/server-entry-runtime') {
+        return '\0' + source;
+      }
+      if (source === 'virtual:vite-rsc-waku/server-entry-build') {
         return '\0' + source;
       }
-      if (source === 'virtual:vite-rsc-waku/server-entry-inner') {
+      if (source === 'virtual:vite-rsc-waku/server-entry-runtime-inner') {
+        const resolved = await this.resolve(
+          `/${srcDir}/${SRC_SERVER_ENTRY}`,
+          undefined,
+          options,
+        );
+        return resolved ? resolved : '\0' + source;
+      }
+      if (source === 'virtual:vite-rsc-waku/server-entry-build-inner') {
         const resolved = await this.resolve(
           `/${srcDir}/${SRC_SERVER_ENTRY}`,
           undefined,
@@ -30,17 +49,36 @@ export function userEntriesPlugin({ srcDir }: { srcDir: string }): Plugin {
         return resolved ? resolved : '\0' + source;
       }
     },
-    load(id) {
-      if (id === '\0virtual:vite-rsc-waku/server-entry') {
+    async load(id) {
+      if (id === '\0virtual:vite-rsc-waku/server-entry-runtime') {
         return `\
-export { default } from 'virtual:vite-rsc-waku/server-entry-inner';
+export { default } from 'virtual:vite-rsc-waku/server-entry-runtime-inner';
 if (import.meta.hot) {
   import.meta.hot.accept()
 }
 `;
       }
-      if (id === '\0virtual:vite-rsc-waku/server-entry-inner') {
-        return getManagedServerEntry(srcDir);
+      if (id === '\0virtual:vite-rsc-waku/server-entry-build') {
+        return `\
+export { default } from 'virtual:vite-rsc-waku/server-entry-build-inner';
+if (import.meta.hot) {
+  import.meta.hot.accept()
+}
+`;
+      }
+      if (id === '\0virtual:vite-rsc-waku/server-entry-runtime-inner') {
+        return getManagedServerEntry({
+          srcDir,
+          rootDir,
+          mode: 'runtime',
+        });
+      }
+      if (id === '\0virtual:vite-rsc-waku/server-entry-build-inner') {
+        return getManagedServerEntry({
+          srcDir,
+          rootDir,
+          mode: 'build',
+        });
       }
       if (id === '\0virtual:vite-rsc-waku/client-entry') {
         return getManagedClientEntry();

packages/waku/src/lib/vite-types.d.ts

@@ -6,6 +6,16 @@ declare module 'virtual:vite-rsc-waku/server-entry' {
   export default default_;
 }
 
+declare module 'virtual:vite-rsc-waku/server-entry-runtime' {
+  const default_: import('./types.ts').Unstable_ServerEntry;
+  export default default_;
+}
+
+declare module 'virtual:vite-rsc-waku/server-entry-build' {
+  const default_: import('./types.ts').Unstable_ServerEntry;
+  export default default_;
+}
+
 declare module 'virtual:vite-rsc-waku/client-entry' {}
 
 declare module 'virtual:vite-rsc-waku/build-metadata' {

packages/waku/src/router/create-pages.tsx

@@ -641,7 +641,8 @@ export const createPages = <
   };
 
   const definedRouter = unstable_defineRouter({
-    getConfigs: async () => {
+    getConfigs: async (context?: { mode?: 'runtime' | 'build' }) => {
+      const mode = context?.mode || 'runtime';
       await configure();
       type ElementSpec = {
         isStatic: boolean;
@@ -893,7 +894,17 @@ export const createPages = <
       const pathConfigs = [...routeConfigs, ...apiConfigs]
         // Sort routes by priority: "standard routes" -> api routes -> api wildcard routes -> standard wildcard routes
         .sort((configA, configB) => routePriorityComparator(configA, configB));
-      return [...pathConfigs, ...sliceConfigs];
+      const filteredPathConfigs =
+        mode === 'runtime'
+          ? pathConfigs.filter(
+              (config) =>
+                !(
+                  (config.type === 'route' || config.type === 'api') &&
+                  config.isStatic
+                ),
+            )
+          : pathConfigs;
+      return [...filteredPathConfigs, ...sliceConfigs];
     },
   });