@reckona/mreact-router 0.0.182 → 0.0.183

package/src/vite.ts

@@ -265,7 +265,11 @@ export function createAppRouterVitePlugin(options: AppRouterVitePluginOptions):
 
       devServerModuleCacheVersion += 1;
       const timestamp = Date.now();
-      const updates = Array.from(context.server.moduleGraph.idToModuleMap.values())
+      const hotModules =
+        context.modules.length > 0
+          ? context.modules
+          : Array.from(context.server.moduleGraph.idToModuleMap.values());
+      const updates = hotModules
         .filter((moduleNode) => isMreactClientDevModuleId(moduleNode.id))
         .map((moduleNode) => {
           context.server.moduleGraph.invalidateModule(moduleNode);
@@ -584,6 +588,9 @@ async function handleAppRouterViteRequest(
         clientStylesByFile,
         clientRouteInferenceCache: options.clientRouteInferenceCache,
         navigationScripts,
+        onRenderError(error) {
+          sendViteErrorOverlay(options.viteDevServer, error);
+        },
         request,
         logger: options.logger,
         routeCache: options.routeCache,
@@ -595,10 +602,37 @@ async function handleAppRouterViteRequest(
       }),
     );
   } catch (error) {
+    sendViteErrorOverlay(options.viteDevServer, error);
     next(error);
   }
 }
 
+function sendViteErrorOverlay(server: ViteDevServer | undefined, error: unknown): void {
+  if (server === undefined) {
+    return;
+  }
+
+  if (error instanceof Error) {
+    server.ssrFixStacktrace(error);
+  }
+  server.ws.send({
+    type: "error",
+    err: viteErrorPayload(error),
+  });
+}
+
+function viteErrorPayload(error: unknown): { message: string; stack: string } {
+  if (error instanceof Error) {
+    return {
+      message: error.message,
+      stack: error.stack ?? error.message,
+    };
+  }
+
+  const message = String(error);
+  return { message, stack: message };
+}
+
 /**
  * Renders a development client asset generated by the app-router Vite integration.
  */