fix(ci): invalidate worker importers on route refresh

Author: sonofmagic

packages/mokup/src/vite/plugin/refresh.ts

@@ -9,6 +9,29 @@ import { resolveDirs, toPosix } from '../../shared/utils'
 import { buildRouteSignature } from './routes'
 import { isViteDevServer } from './server'
 
+interface InvalidatableModuleNode {
+  importers?: Set<InvalidatableModuleNode>
+}
+
+function invalidateModuleChain(server: ViteDevServer, node: InvalidatableModuleNode) {
+  const visited = new Set<InvalidatableModuleNode>()
+  const stack = [node]
+
+  while (stack.length > 0) {
+    const current = stack.pop()
+    if (!current || visited.has(current)) {
+      continue
+    }
+    visited.add(current)
+    server.moduleGraph.invalidateModule(current as Parameters<typeof server.moduleGraph.invalidateModule>[0])
+    if (current.importers) {
+      for (const importer of current.importers) {
+        stack.push(importer)
+      }
+    }
+  }
+}
+
 function createRouteRefresher(params: {
   state: PluginState
   optionList: VitePluginOptions[]
@@ -157,7 +180,7 @@ function createRouteRefresher(params: {
           for (const id of virtualModuleIds) {
             const moduleNode = server.moduleGraph.getModuleById(id)
             if (moduleNode) {
-              server.moduleGraph.invalidateModule(moduleNode)
+              invalidateModuleChain(server, moduleNode as InvalidatableModuleNode)
             }
           }
         }

packages/mokup/test/vite-plugin-refresh.test.ts

@@ -160,6 +160,60 @@ describe('vite plugin route refresh', () => {
     expect(server.moduleGraph.invalidateModule).toHaveBeenCalledWith(moduleNode)
   })
 
+  it('invalidates importer chains for virtual modules when routes change', async () => {
+    const parsed = parseRouteTemplate('/ping')
+    mocks.scanRoutes.mockResolvedValueOnce([
+      {
+        file: '/root/mock/ping.get.json',
+        template: parsed.template,
+        method: 'GET',
+        tokens: parsed.tokens,
+        score: parsed.score,
+        handler: { ok: true },
+      },
+    ])
+
+    const state = {
+      routes: [],
+      serverRoutes: [],
+      swRoutes: [],
+      disabledRoutes: [],
+      ignoredRoutes: [],
+      configFiles: [],
+      disabledConfigFiles: [],
+      app: null,
+      lastSignature: 'old',
+      swModuleVersion: 0,
+    }
+
+    const workerImporter = { id: '/worker/index.ts', importers: new Set() }
+    const moduleNode = {
+      id: '\0virtual:mokup-bundle',
+      importers: new Set([workerImporter]),
+    }
+    const server = {
+      ws: { send: vi.fn() },
+      moduleGraph: {
+        getModuleById: vi.fn().mockReturnValue(moduleNode),
+        invalidateModule: vi.fn(),
+      },
+    }
+
+    const refresher = createRouteRefresher({
+      state: state as never,
+      optionList: [{ dir: '/root/mock', prefix: '/api' }],
+      root: () => '/root',
+      logger: { info: vi.fn(), warn: vi.fn(), error: vi.fn() },
+      enableViteMiddleware: false,
+      virtualModuleIds: ['\0virtual:mokup-bundle'],
+    })
+
+    await refresher(server as never)
+
+    expect(server.moduleGraph.invalidateModule).toHaveBeenCalledWith(moduleNode)
+    expect(server.moduleGraph.invalidateModule).toHaveBeenCalledWith(workerImporter)
+  })
+
   it('forces virtual module invalidation when signatures are unchanged', async () => {
     const parsed = parseRouteTemplate('/about')
     const route = {