vite: typesafe server build

Author: pcattori

integration/helpers/vite.ts

@@ -42,7 +42,7 @@ export const EXPRESS_SERVER = (args: {
 }) =>
   String.raw`
     import { createRequestHandler } from "@remix-run/express";
-    import { installGlobals } from "@remix-run/node";
+    import { installGlobals, getServerBuild } from "@remix-run/node";
     import express from "express";
 
     installGlobals();
@@ -71,9 +71,7 @@ export const EXPRESS_SERVER = (args: {
     app.all(
       "*",
       createRequestHandler({
-        build: viteDevServer
-          ? () => viteDevServer.ssrLoadModule("virtual:remix/server-build")
-          : await import("./build/index.js"),
+        build: await getServerBuild("./build/server/index.js", viteDevServer),
         getLoadContext: () => (${JSON.stringify(args.loadContext ?? {})}),
       })
     );

integration/vite-basename-test.ts

@@ -95,7 +95,7 @@ const customServerFile = ({
 
   return String.raw`
     import { createRequestHandler } from "@remix-run/express";
-    import { installGlobals } from "@remix-run/node";
+    import { installGlobals, getServerBuild } from "@remix-run/node";
     import express from "express";
     installGlobals();
 
@@ -115,9 +115,7 @@ const customServerFile = ({
     app.all(
       "${basename}*",
       createRequestHandler({
-        build: viteDevServer
-          ? () => viteDevServer.ssrLoadModule("virtual:remix/server-build")
-          : await import("./build/server/index.js"),
+        build: await getServerBuild("./build/server/index.js", viteDevServer),
       })
     );
     app.get("*", (_req, res) => {

packages/remix-cloudflare/index.ts

@@ -15,6 +15,7 @@ export {
   defer,
   broadcastDevReady,
   logDevReady,
+  getServerBuild,
   isCookie,
   isSession,
   json,

packages/remix-deno/index.ts

@@ -17,6 +17,7 @@ export {
   broadcastDevReady,
   createSession,
   defer,
+  getServerBuild,
   isCookie,
   isSession,
   json,

packages/remix-node/index.ts

@@ -27,6 +27,7 @@ export {
   defer,
   broadcastDevReady,
   logDevReady,
+  getServerBuild,
   isCookie,
   isSession,
   json,

packages/remix-server-runtime/__tests__/build-test.ts

@@ -0,0 +1,15 @@
+import { getServerBuild } from "../build";
+
+it("getServerBuild throws when path is relative", async () => {
+  await expect(() => getServerBuild("./build/server/index.js")).rejects.toThrow(
+    "Server build path must be absolute, but received relative path: ./build/server/index.js"
+  );
+});
+
+it("getServerBuild throws when build does not exist", async () => {
+  await expect(() =>
+    getServerBuild("/this/path/doesnt/exist.js")
+  ).rejects.toThrow(
+    "Could not import server build from '/this/path/doesnt/exist.js'. Did you forget to run 'remix vite:build' first?"
+  );
+});

packages/remix-server-runtime/build.ts

@@ -1,8 +1,41 @@
+import type { ViteDevServer } from "vite";
+import path from "pathe";
+
 import type { ActionFunctionArgs, LoaderFunctionArgs } from "./routeModules";
 import type { AssetsManifest, EntryContext, FutureConfig } from "./entry";
 import type { ServerRouteManifest } from "./routes";
 import type { AppLoadContext } from "./data";
 
+export async function getServerBuild(
+  buildPath: string,
+  {
+    viteDevServer,
+  }: {
+    viteDevServer?: ViteDevServer;
+  } = {}
+): Promise<ServerBuild | (() => Promise<ServerBuild>)> {
+  if (viteDevServer) {
+    return () =>
+      viteDevServer.ssrLoadModule(
+        "virtual:remix/server-build"
+      ) as Promise<ServerBuild>;
+  }
+
+  if (!path.isAbsolute(buildPath)) {
+    throw new Error(
+      `Server build path must be absolute, but received relative path: ${buildPath}`
+    );
+  }
+
+  // Convert file path meant for `import` to URL for Windows compatibility
+  let buildURL = "file:///" + encodeURI(buildPath);
+  return import(buildURL).catch(() => {
+    throw Error(
+      `Could not import server build from '${buildPath}'. Did you forget to run 'remix vite:build' first?`
+    );
+  });
+}
+
 // NOTE: IF you modify `ServerBuild`, be sure to modify the
 // `remix-dev/server-build.ts` file to reflect the new field as well
 

packages/remix-server-runtime/index.ts

@@ -80,3 +80,4 @@ export type {
   UploadHandler,
   UploadHandlerPart,
 } from "./reexport";
+export { getServerBuild } from "./reexport";

packages/remix-server-runtime/package.json

@@ -20,19 +20,25 @@
     "@types/cookie": "^0.6.0",
     "@web3-storage/multipart-parser": "^1.0.0",
     "cookie": "^0.6.0",
+    "pathe": "^1.1.2",
     "set-cookie-parser": "^2.4.8",
     "source-map": "^0.7.3"
   },
   "devDependencies": {
     "@types/set-cookie-parser": "^2.4.1",
-    "typescript": "^5.1.6"
+    "typescript": "^5.1.6",
+    "vite": "^5.1.0"
   },
   "peerDependencies": {
-    "typescript": "^5.1.0"
+    "typescript": "^5.1.0",
+    "vite": "^5.1.0"
   },
   "peerDependenciesMeta": {
     "typescript": {
       "optional": true
+    },
+    "vite": {
+      "optional": true
     }
   },
   "engines": {

packages/remix-server-runtime/reexport.ts

@@ -7,6 +7,7 @@ export type {
   ServerBuild,
   ServerEntryModule,
 } from "./build";
+export { getServerBuild } from "./build";
 
 export type { UploadHandlerPart, UploadHandler } from "./formData";
 export type {