arcway 0.1.20 → 0.1.22

package/server/pages/handler.js

@@ -6,6 +6,7 @@ import { compilePattern, sortBySpecificity, matchPattern } from '../router/route
 import { parseCookies, resolveSession, flattenHeaders } from '../session/helpers.js';
 import {
   buildCssLinkTag,
+  buildRouteCssLinkTags,
   buildScriptTags,
   loadComponent,
   renderErrorPage,
@@ -13,6 +14,7 @@ import {
   syncReactInternals,
 } from './ssr.js';
 import { MIME_TYPES } from './static.js';
+import { createArcwayDevEndpoint } from './arcway-endpoint.js';
 function createPagesHandler(options) {
   const rootDir = options.rootDir;
   const outDir = path.resolve(rootDir, options.outDir ?? '.build/pages');
@@ -21,7 +23,10 @@ function createPagesHandler(options) {
   const appContext = options.appContext ?? null;
   const mode = options.mode ?? 'production';
   const devMode = mode === 'development';
-  if (!fs.existsSync(manifestPath)) {
+  const lazyContext = options.lazyContext ?? null;
+  // Dev with a lazy context drives the handler off an in-memory manifest; no
+  // disk `pages-manifest.json` is produced. Prod still reads from disk.
+  if (!lazyContext && !fs.existsSync(manifestPath)) {
     return null;
   }
   const projectRequire = createRequire(path.join(options.rootDir, 'package.json'));
@@ -31,16 +36,47 @@ function createPagesHandler(options) {
     renderToPipeableStream: projectRequire('react-dom/server').renderToPipeableStream,
   };
   syncReactInternals(projectReactModule);
-  let manifest = JSON.parse(fs.readFileSync(manifestPath, 'utf-8'));
+  let manifest = lazyContext
+    ? lazyContext.manifest
+    : JSON.parse(fs.readFileSync(manifestPath, 'utf-8'));
   let routes = compileRoutes(manifest);
   let clientManifestJson = buildClientManifestJson(manifest);
+  let lastSeenVersion = lazyContext ? manifest.version : 0;
   const componentCache = new Map();
   let cacheVersion = 0;
   const reloadEmitter = devMode ? new EventEmitter() : null;
-  if (manifest.entries.length === 0 && !manifest.notFoundBundle && !manifest.errorBundle) {
+  if (
+    !lazyContext &&
+    manifest.entries.length === 0 &&
+    !manifest.notFoundBundle &&
+    !manifest.errorBundle
+  ) {
     return null;
   }
+  // Keep `routes` / `clientManifestJson` in sync with the lazy context's
+  // mutable manifest. Every `built` / `invalidate` event bumps the version,
+  // which is our cheap change signal — recompile only when it moves.
+  function refreshFromLazy() {
+    if (!lazyContext) return;
+    if (manifest.version === lastSeenVersion) return;
+    lastSeenVersion = manifest.version;
+    routes = compileRoutes(manifest);
+    clientManifestJson = buildClientManifestJson(manifest);
+    componentCache.clear();
+    cacheVersion++;
+  }
+  if (lazyContext) {
+    lazyContext.on('update', (event) => {
+      refreshFromLazy();
+      if (reloadEmitter) reloadEmitter.emit('update', event);
+    });
+  }
+  const arcwayEndpoint = lazyContext ? createArcwayDevEndpoint(lazyContext) : null;
   function reload() {
+    if (lazyContext) {
+      refreshFromLazy();
+      return;
+    }
     if (!fs.existsSync(manifestPath)) return;
     manifest = JSON.parse(fs.readFileSync(manifestPath, 'utf-8'));
     routes = compileRoutes(manifest);
@@ -57,6 +93,9 @@ function createPagesHandler(options) {
     const url = req.url ?? '/';
     const pathname = url.split('?')[0];
     if (method !== 'GET') return false;
+    // Internal `/_arcway/*` endpoints (dev only) resolve before the route
+    // matcher so they can never be shadowed by a user-defined page.
+    if (arcwayEndpoint && arcwayEndpoint(req, res)) return true;
     if (reloadEmitter && pathname === '/__livereload') {
       res.writeHead(200, {
         'Content-Type': 'text/event-stream',
@@ -73,6 +112,9 @@ function createPagesHandler(options) {
       req.on('close', () => reloadEmitter.off('update', onUpdate));
       return true;
     }
+    // Pick up any invalidations or completed builds that landed since the
+    // last request before we resolve the route.
+    refreshFromLazy();
     const matched = matchPageRoute(routes, pathname);
     if (!matched) {
       if (manifest.notFoundBundle) {
@@ -91,6 +133,28 @@ function createPagesHandler(options) {
       }
       return false;
     }
+    // Lazy dev: trigger on-demand build of this page's server bundle +
+    // every ancestor layout/middleware before we attempt to load any
+    // component. ensurePageBuilt() coalesces concurrent calls internally.
+    if (lazyContext) {
+      try {
+        await lazyContext.ensurePageBuilt(matched.route.pattern);
+      } catch (err) {
+        await renderDevBuildError(
+          err,
+          res,
+          manifest,
+          outDir,
+          componentCache,
+          cacheVersion,
+          projectReact,
+        );
+        return true;
+      }
+      refreshFromLazy();
+      const rematched = matchPageRoute(routes, pathname);
+      if (rematched) matched.route = rematched.route;
+    }
     if (matched.route.middlewareServerBundles.length > 0) {
       const middlewareResult = await runPageMiddleware(
         matched.route,
@@ -174,6 +238,43 @@ function createPagesHandler(options) {
       : {}),
   };
 }
+// Render a dev-only build-error page. Preferred path: use the project's
+// `_error.jsx` bundle so devs see the same shell they'd see for runtime errors,
+// with the esbuild message in the `error` prop. Fallback: plain-text 500 when
+// no errorBundle is available or the response has already started.
+async function renderDevBuildError(
+  err,
+  res,
+  manifest,
+  outDir,
+  componentCache,
+  cacheVersion,
+  projectReact,
+) {
+  const errorMessage = err instanceof Error ? err.message : String(err);
+  if (manifest.errorBundle && !res.headersSent) {
+    await renderErrorPage(
+      manifest.errorBundle,
+      500,
+      { error: errorMessage },
+      outDir,
+      res,
+      componentCache,
+      manifest,
+      cacheVersion,
+      projectReact,
+    );
+    return;
+  }
+  if (!res.headersSent) {
+    const escaped = String(errorMessage)
+      .replace(/&/g, '&')
+      .replace(/</g, '&lt;')
+      .replace(/>/g, '&gt;');
+    res.writeHead(500, { 'Content-Type': 'text/html; charset=utf-8' });
+    res.end(`<h1>500 - Build Error</h1><pre>${escaped}</pre>`);
+  }
+}
 function compileRoutes(manifest) {
   const routes = manifest.entries.map((entry) => {
     const { regex, paramNames, catchAllParam } = compilePattern(entry.pattern);
@@ -186,6 +287,8 @@ function compileRoutes(manifest) {
       clientBundle: entry.clientBundle,
       layoutServerBundles: entry.layoutServerBundles ?? [],
       middlewareServerBundles: entry.middlewareServerBundles ?? [],
+      sharedChunks: entry.sharedChunks ?? [],
+      sharedCssChunks: entry.sharedCssChunks ?? [],
     };
   });
   sortBySpecificity(routes);
@@ -269,6 +372,7 @@ export {
   MIME_TYPES,
   buildClientManifestJson,
   buildCssLinkTag,
+  buildRouteCssLinkTags,
   buildScriptTags,
   createPagesHandler,
   matchPageRoute,