vite: typesafe server build

Author: pcattori

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

@@ -1,6 +1,7 @@
 // This file allows us to dynamically require the plugin so non-Vite consumers
 // don't need to have Vite installed as a peer dependency. Only types should
 // be imported at the top level, or code that doesn't import Vite.
+
 import type { RemixVitePlugin } from "./plugin";
 export type {
   BuildManifest,
@@ -15,4 +16,5 @@ export const vitePlugin: RemixVitePlugin = (...args) => {
   return remixVitePlugin(...args);
 };
 
+export { getServerBuild } from "./get-server-build";
 export { cloudflareDevProxyVitePlugin } from "./cloudflare-proxy-plugin";