vite: typesafe server build

packages/remix-dev/index.ts

@@ -12,4 +12,8 @@ export type {
   ServerBundlesFunction,
   VitePluginConfig,
 } from "./vite";
-export { vitePlugin, cloudflareDevProxyVitePlugin } from "./vite";
+export {
+  vitePlugin,
+  cloudflareDevProxyVitePlugin,
+  getServerBuild,
+} from "./vite";

packages/remix-dev/vite/get-server-build.ts

@@ -0,0 +1,21 @@
+import type { ViteDevServer } from "vite";
+import type { ServerBuild } from "@remix-run/server-runtime";
+
+const devServerBuild = "virtual:remix/server-build";
+
+export async function getServerBuild(
+  path: string,
+  viteDevServer?: ViteDevServer
+): Promise<ServerBuild> {
+  if (viteDevServer) {
+    return viteDevServer.ssrLoadModule(devServerBuild) as Promise<ServerBuild>;
+  }
+  try {
+    return import(path);
+  } catch (error) {
+    console.error(
+      `Could not find server build at '${path}'. Did you forget to run 'remix vite:build' first?`
+    );
+    throw error;
+  }
+}

packages/remix-dev/vite/index.ts

@@ -15,4 +15,5 @@ export const vitePlugin: RemixVitePlugin = (...args) => {
   return remixVitePlugin(...args);
 };
 
+export { getServerBuild } from "./get-server-build";
 export { cloudflareDevProxyVitePlugin } from "./cloudflare-proxy-plugin";