Fix HMR for files imported into routes

.changeset/popular-bikes-brush.md

@@ -0,0 +1,5 @@
+---
+'@shopify/mini-oxygen': patch
+---
+
+Fix HMR when changing files imported into routes.

packages/mini-oxygen/src/vite/worker-entry.ts

@@ -94,25 +94,26 @@ function fetchEntryModule(publicUrl: URL, env: ViteEnv) {
       .then((hmrWs) => {
         hmrReady = !!hmrWs;
         hmrWs?.addEventListener('message', (message) => {
-          if (onHmrRecieve) {
-            if (!message.data) return;
-            const data: HMRPayload = JSON.parse(message.data.toString());
-
-            if (!data) return;
-
-            if (data.type === 'update') {
-              // Invalidate cache synchronously without revalidating the
-              // module to avoid hanging promises in workerd
-              for (const update of data.updates) {
-                runtime.moduleCache.invalidateDepTree([update.path]);
-              }
-            } else if (data.type !== 'custom') {
-              // Custom events are only used in browser HMR, so ignore them.
-              // This type is wrong in ViteRuntime:
-              (onHmrRecieve(data) as unknown as Promise<unknown>)?.catch(
-                (error) => console.error('During SSR HMR:', error),
-              );
+          if (!message.data) return;
+          const data: HMRPayload = JSON.parse(message.data.toString());
+
+          if (!data) return;
+
+          if (data.type === 'update') {
+            // Invalidate cache synchronously without revalidating the
+            // module to avoid hanging promises in workerd
+            for (const update of data.updates) {
+              runtime.moduleCache.invalidateDepTree([
+                update.path,
+                ...(update.ssrInvalidates ?? []),
+              ]);
             }
+          } else if (data.type !== 'custom' && onHmrRecieve) {
+            // Custom events are only used in browser HMR, so ignore them.
+            // This type is wrong in ViteRuntime:
+            (onHmrRecieve(data) as unknown as Promise<unknown>)?.catch(
+              (error) => console.error('During SSR HMR:', error),
+            );
           }
         });
       })