feat(server-runtime): pass AppLoadContext to handleRequest

remix-run · Mar 17, 2023 · 6c4146c · 6c4146c
1 parent 12e5aa6
commit 6c4146c
Show file tree

Hide file tree

Showing 5 changed files with 59 additions and 5 deletions.
diff --git a/docs/file-conventions/entry.server.md b/docs/file-conventions/entry.server.md
@@ -16,6 +16,7 @@ Here's a basic example:
 ```tsx
 import { renderToString } from "react-dom/server";
 import type {
+  AppLoadContext,
   EntryContext,
   HandleDataRequestFunction,
 } from "@remix-run/node"; // or cloudflare/deno
@@ -25,7 +26,8 @@ export default function handleRequest(
   request: Request,
   responseStatusCode: number,
   responseHeaders: Headers,
-  remixContext: EntryContext
+  remixContext: EntryContext,
+  loadContext: AppLoadContext
 ) {
   const markup = renderToString(
     <RemixServer context={remixContext} url={request.url} />

diff --git a/packages/remix-server-runtime/__tests__/server-test.ts b/packages/remix-server-runtime/__tests__/server-test.ts
@@ -1721,5 +1721,47 @@ describe("shared server runtime", () => {
       expect(calls.length).toBe(2 * DATA_CALL_MULTIPIER);
       expect(spy.console.mock.calls.length).toBe(1 * DATA_CALL_MULTIPIER);
     });
+
+    test("provides load context to server entrypoint", async () => {
+      let rootLoader = jest.fn(() => {
+        return "root";
+      });
+      let indexLoader = jest.fn(() => {
+        return "index";
+      });
+      let build = mockServerBuild({
+        root: {
+          default: {},
+          loader: rootLoader,
+          ErrorBoundary: {},
+        },
+        "routes/index": {
+          parentId: "root",
+          default: {},
+          loader: indexLoader,
+        },
+      });
+
+      build.entry.module.default = jest.fn(
+        async (
+          request,
+          responseStatusCode,
+          responseHeaders,
+          entryContext,
+          loadContext
+        ) =>
+          new Response(JSON.stringify(loadContext), {
+            status: responseStatusCode,
+            headers: responseHeaders,
+          })
+      );
+
+      let handler = createRequestHandler(build, ServerMode.Development);
+      let request = new Request(`${baseUrl}/`, { method: "get" });
+      let loadContext = { "load-context": "load-value" };
+
+      let result = await handler(request, loadContext);
+      expect(await result.text()).toBe(JSON.stringify(loadContext));
+    });
   });
 });
diff --git a/packages/remix-server-runtime/__tests__/utils.ts b/packages/remix-server-runtime/__tests__/utils.ts
@@ -48,7 +48,13 @@ export function mockServerBuild(
     entry: {
       module: {
         default: jest.fn(
-          async (request, responseStatusCode, responseHeaders, entryContext) =>
+          async (
+            request,
+            responseStatusCode,
+            responseHeaders,
+            entryContext,
+            loadContext
+          ) =>
             new Response(null, {
               status: responseStatusCode,
               headers: responseHeaders,

diff --git a/packages/remix-server-runtime/build.ts b/packages/remix-server-runtime/build.ts
@@ -1,6 +1,7 @@
 import type { DataFunctionArgs } from "./routeModules";
 import type { AssetsManifest, EntryContext, FutureConfig } from "./entry";
 import type { ServerRouteManifest } from "./routes";
+import type { AppLoadContext } from "./data";
 
 /**
  * The output of the compiler for the server build.
@@ -22,7 +23,8 @@ export interface HandleDocumentRequestFunction {
     request: Request,
     responseStatusCode: number,
     responseHeaders: Headers,
-    context: EntryContext
+    context: EntryContext,
+    loadContext: AppLoadContext
   ): Promise<Response> | Response;
 }
 

diff --git a/packages/remix-server-runtime/server.ts b/packages/remix-server-runtime/server.ts
@@ -294,7 +294,8 @@ async function handleDocumentRequestRR(
       request,
       context.statusCode,
       headers,
-      entryContext
+      entryContext,
+      loadContext
     );
   } catch (error: unknown) {
     // Get a new StaticHandlerContext that contains the error at the right boundary
@@ -328,7 +329,8 @@ async function handleDocumentRequestRR(
         request,
         context.statusCode,
         headers,
-        entryContext
+        entryContext,
+        loadContext
       );
     } catch (error: any) {
       logServerErrorIfNotAborted(error, request, serverMode);