feat(@angular/build): utilize ssr.entry during prerendering to enable access to local API routes

The `ssr.entry` (server.ts file) is now utilized during prerendering, allowing access to locally defined API routes for improved data fetching and rendering.

Author: alan-agius4

packages/angular/build/src/tools/vite/middlewares/ssr-middleware.ts

@@ -8,13 +8,15 @@
 
 import type {
   AngularAppEngine as SSRAngularAppEngine,
-  createRequestHandler,
   ɵgetOrCreateAngularServerApp as getOrCreateAngularServerApp,
 } from '@angular/ssr';
-import type { createNodeRequestHandler } from '@angular/ssr/node';
 import type { ServerResponse } from 'node:http';
 import type { Connect, ViteDevServer } from 'vite';
 import { loadEsmModule } from '../../../utils/load-esm';
+import {
+  isSsrNodeRequestHandler,
+  isSsrRequestHandler,
+} from '../../../utils/server-rendering/utils';
 
 export function createAngularSsrInternalMiddleware(
   server: ViteDevServer,
@@ -136,13 +138,3 @@ export async function createAngularSsrExternalMiddleware(
     })().catch(next);
   };
 }
-
-function isSsrNodeRequestHandler(
-  value: unknown,
-): value is ReturnType<typeof createNodeRequestHandler> {
-  return typeof value === 'function' && '__ng_node_request_handler__' in value;
-}
-
-function isSsrRequestHandler(value: unknown): value is ReturnType<typeof createRequestHandler> {
-  return typeof value === 'function' && '__ng_request_handler__' in value;
-}

packages/angular/build/src/utils/server-rendering/fetch-patch.ts

@@ -21,7 +21,7 @@ const { assetFiles } = workerData as {
 const assetsCache: Map<string, { headers: undefined | Record<string, string>; content: Buffer }> =
   new Map();
 
-export function patchFetchToLoadInMemoryAssets(): void {
+export function patchFetchToLoadInMemoryAssets(baseURL: URL): void {
   const originalFetch = globalThis.fetch;
   const patchedFetch: typeof fetch = async (input, init) => {
     let url: URL;
@@ -38,7 +38,7 @@ export function patchFetchToLoadInMemoryAssets(): void {
     const { hostname } = url;
     const pathname = decodeURIComponent(url.pathname);
 
-    if (hostname !== 'local-angular-prerender' || !assetFiles[pathname]) {
+    if (hostname !== baseURL.hostname || !assetFiles[pathname]) {
       // Only handle relative requests or files that are in assets.
       return originalFetch(input, init);
     }

packages/angular/build/src/utils/server-rendering/launch-server.ts

@@ -0,0 +1,83 @@
+/**
+ * @license
+ * Copyright Google LLC All Rights Reserved.
+ *
+ * Use of this source code is governed by an MIT-style license that can be
+ * found in the LICENSE file at https://angular.dev/license
+ */
+
+import assert from 'node:assert';
+import { createServer } from 'node:http';
+import { workerData } from 'node:worker_threads';
+import { OutputMode } from '../../builders/application/schema';
+import { loadEsmModule } from '../load-esm';
+import { loadEsmModuleFromMemory } from './load-esm-from-memory';
+import { isSsrNodeRequestHandler, isSsrRequestHandler } from './utils';
+
+const DEFAULT_URL = new URL('http://ng-localhost/');
+
+/**
+ * This is passed as workerData when setting up the worker via the `piscina` package.
+ */
+const { outputMode } = workerData as {
+  outputMode: OutputMode | undefined;
+};
+
+/**
+ * Launches a server that handles requests based on the output mode and the provided server handler.
+ * If the output mode is not defined or the handler is invalid, it returns a default URL.
+ *
+ * @returns A promise that resolves to the URL of the running server.
+ */
+export async function launchServer(): Promise<URL> {
+  if (!outputMode) {
+    // Legacy mode
+    return DEFAULT_URL;
+  }
+
+  let handler: unknown;
+  try {
+    const serverEntry = await loadEsmModuleFromMemory('./server.mjs');
+    handler = serverEntry.default;
+  } catch {
+    return DEFAULT_URL;
+  }
+
+  const { createWebRequestFromNodeRequest, writeResponseToNodeResponse } =
+    await loadEsmModule<typeof import('@angular/ssr/node')>('@angular/ssr/node');
+
+  if (!isSsrNodeRequestHandler(handler) && !isSsrRequestHandler(handler)) {
+    return DEFAULT_URL;
+  }
+
+  // eslint-disable-next-line @typescript-eslint/no-misused-promises
+  const server = createServer(async (req, res) => {
+    // handle request
+    if (isSsrNodeRequestHandler(handler)) {
+      // eslint-disable-next-line no-console
+      await handler(req, res, (e) => {
+        throw e;
+      });
+    } else {
+      const webRes = await handler(createWebRequestFromNodeRequest(req));
+      if (webRes) {
+        await writeResponseToNodeResponse(webRes, res);
+      } else {
+        res.statusCode = 501;
+        res.end('Not Implemented.');
+      }
+    }
+  });
+
+  server.unref();
+
+  return new Promise<URL>((resolve) => {
+    server.listen(0, () => {
+      const serverAddress = server.address();
+      assert(serverAddress, 'Server address should be defined.');
+      assert(typeof serverAddress !== 'string', 'Server address should not be a string.');
+
+      resolve(new URL(`http://localhost:${serverAddress.port}/`));
+    });
+  });
+}

packages/angular/build/src/utils/server-rendering/load-esm-from-memory.ts

@@ -20,9 +20,20 @@ interface MainServerBundleExports {
   ɵgetOrCreateAngularServerApp: typeof ɵgetOrCreateAngularServerApp;
 }
 
+/**
+ * Represents the exports available from the server bundle.
+ */
+interface ServerBundleExports {
+  default: unknown;
+}
+
 export function loadEsmModuleFromMemory(
   path: './main.server.mjs',
-): Promise<MainServerBundleExports> {
+): Promise<MainServerBundleExports>;
+export function loadEsmModuleFromMemory(path: './server.mjs'): Promise<ServerBundleExports>;
+export function loadEsmModuleFromMemory(
+  path: './main.server.mjs' | './server.mjs',
+): Promise<MainServerBundleExports | ServerBundleExports> {
   return loadEsmModule(new URL(path, 'memory://')).catch((e) => {
     assertIsError(e);
 

packages/angular/build/src/utils/server-rendering/prerender.ts

@@ -160,6 +160,7 @@ export async function prerenderPages(
     outputFilesForWorker,
     assetsReversed,
     appShellOptions,
+    outputMode,
   );
 
   errors.push(...renderingErrors);
@@ -181,6 +182,7 @@ async function renderPages(
   outputFilesForWorker: Record<string, string>,
   assetFilesForWorker: Record<string, string>,
   appShellOptions: AppShellOptions | undefined,
+  outputMode: OutputMode | undefined,
 ): Promise<{
   output: PrerenderOutput;
   errors: string[];
@@ -205,6 +207,7 @@ async function renderPages(
       workspaceRoot,
       outputFiles: outputFilesForWorker,
       assetFiles: assetFilesForWorker,
+      outputMode,
     } as RenderWorkerData,
     execArgv: workerExecArgv,
   });
@@ -309,14 +312,15 @@ async function getAllRoutes(
       workspaceRoot,
       outputFiles: outputFilesForWorker,
       assetFiles: assetFilesForWorker,
+      outputMode,
     } as RoutesExtractorWorkerData,
     execArgv: workerExecArgv,
   });
 
   try {
-    const { serializedRouteTree, errors }: RoutersExtractorWorkerResult = await renderWorker.run({
-      outputMode,
-    });
+    const { serializedRouteTree, errors }: RoutersExtractorWorkerResult = await renderWorker.run(
+      {},
+    );
 
     return { errors, serializedRouteTree: [...routes, ...serializedRouteTree] };
   } catch (err) {

packages/angular/build/src/utils/server-rendering/render-worker.ts

@@ -6,18 +6,23 @@
  * found in the LICENSE file at https://angular.dev/license
  */
 
+import type { OutputMode } from '../../builders/application/schema';
 import type { ESMInMemoryFileLoaderWorkerData } from './esm-in-memory-loader/loader-hooks';
 import { patchFetchToLoadInMemoryAssets } from './fetch-patch';
+import { launchServer } from './launch-server';
 import { loadEsmModuleFromMemory } from './load-esm-from-memory';
 
 export interface RenderWorkerData extends ESMInMemoryFileLoaderWorkerData {
   assetFiles: Record</** Destination */ string, /** Source */ string>;
+  outputMode: OutputMode | undefined;
 }
 
 export interface RenderOptions {
   url: string;
 }
 
+let serverURL: URL;
+
 /**
  * Renders each route in routes and writes them to <outputPath>/<route>/index.html.
  */
@@ -26,15 +31,16 @@ async function renderPage({ url }: RenderOptions): Promise<string | null> {
     await loadEsmModuleFromMemory('./main.server.mjs');
   const angularServerApp = getOrCreateAngularServerApp();
   const response = await angularServerApp.renderStatic(
-    new URL(url, 'http://local-angular-prerender'),
+    new URL(url, serverURL),
     AbortSignal.timeout(30_000),
   );
 
   return response ? response.text() : null;
 }
 
-function initialize() {
-  patchFetchToLoadInMemoryAssets();
+async function initialize() {
+  serverURL = await launchServer();
+  patchFetchToLoadInMemoryAssets(serverURL);
 
   return renderPage;
 }

packages/angular/build/src/utils/server-rendering/routes-extractor-worker.ts

@@ -6,24 +6,34 @@
  * found in the LICENSE file at https://angular.dev/license
  */
 
+import { workerData } from 'worker_threads';
 import { OutputMode } from '../../builders/application/schema';
+import { ESMInMemoryFileLoaderWorkerData } from './esm-in-memory-loader/loader-hooks';
 import { patchFetchToLoadInMemoryAssets } from './fetch-patch';
+import { launchServer } from './launch-server';
 import { loadEsmModuleFromMemory } from './load-esm-from-memory';
 import { RoutersExtractorWorkerResult } from './models';
 
-export interface ExtractRoutesOptions {
-  outputMode?: OutputMode;
+export interface ExtractRoutesWorkerData extends ESMInMemoryFileLoaderWorkerData {
+  outputMode: OutputMode | undefined;
 }
 
+/**
+ * This is passed as workerData when setting up the worker via the `piscina` package.
+ */
+const { outputMode } = workerData as {
+  outputMode: OutputMode | undefined;
+};
+
+let serverURL: URL;
+
 /** Renders an application based on a provided options. */
-async function extractRoutes({
-  outputMode,
-}: ExtractRoutesOptions): Promise<RoutersExtractorWorkerResult> {
+async function extractRoutes(): Promise<RoutersExtractorWorkerResult> {
   const { ɵextractRoutesAndCreateRouteTree: extractRoutesAndCreateRouteTree } =
     await loadEsmModuleFromMemory('./main.server.mjs');
 
   const { routeTree, errors } = await extractRoutesAndCreateRouteTree(
-    new URL('http://local-angular-prerender/'),
+    serverURL,
     undefined /** manifest */,
     true /** invokeGetPrerenderParams */,
     outputMode === OutputMode.Server /** includePrerenderFallbackRoutes */,
@@ -35,8 +45,9 @@ async function extractRoutes({
   };
 }
 
-function initialize() {
-  patchFetchToLoadInMemoryAssets();
+async function initialize() {
+  serverURL = await launchServer();
+  patchFetchToLoadInMemoryAssets(serverURL);
 
   return extractRoutes;
 }

packages/angular/build/src/utils/server-rendering/utils.ts

@@ -0,0 +1,21 @@
+/**
+ * @license
+ * Copyright Google LLC All Rights Reserved.
+ *
+ * Use of this source code is governed by an MIT-style license that can be
+ * found in the LICENSE file at https://angular.dev/license
+ */
+
+import type { createRequestHandler } from '@angular/ssr';
+import type { createNodeRequestHandler } from '@angular/ssr/node';
+
+export function isSsrNodeRequestHandler(
+  value: unknown,
+): value is ReturnType<typeof createNodeRequestHandler> {
+  return typeof value === 'function' && '__ng_node_request_handler__' in value;
+}
+export function isSsrRequestHandler(
+  value: unknown,
+): value is ReturnType<typeof createRequestHandler> {
+  return typeof value === 'function' && '__ng_request_handler__' in value;
+}