fix(hmr): should reload if relies file changed after re-render

sxzz · sxzz · commit da26e525d9a0 · 2024-11-26T22:53:45.000+08:00
ref: vitejs/vite-plugin-vue@62b17f3
diff --git a/src/core/index.ts b/src/core/index.ts
@@ -227,6 +227,12 @@ export const plugin = createUnplugin<Options | undefined, false>(
       vite: {
         api,
         handleHotUpdate(ctx) {
+          ctx.server.ws.send({
+            type: 'custom',
+            event: 'file-changed',
+            data: { file: normalizePath(ctx.file) },
+          })
+
           if (options.value.compiler.invalidateTypeCache) {
             options.value.compiler.invalidateTypeCache(ctx.file)
           }
diff --git a/src/core/main.ts b/src/core/main.ts
@@ -149,10 +149,19 @@ export async function transformMain(
       `_sfc_main.__hmrId = ${JSON.stringify(descriptor.id)}`,
       `typeof __VUE_HMR_RUNTIME__ !== 'undefined' && ` +
         `__VUE_HMR_RUNTIME__.createRecord(_sfc_main.__hmrId, _sfc_main)`,
+      `import.meta.hot.on('file-changed', ({ file }) => {`,
+      `  __VUE_HMR_RUNTIME__.CHANGED_FILE = file`,
+      `})`,
     )
     // check if the template is the only thing that changed
     if (prevDescriptor && isOnlyTemplateChanged(prevDescriptor, descriptor)) {
-      output.push(`export const _rerender_only = true`)
+      // #7 only consider re-render if the HMR is triggered by the current component,
+      // otherwise reload. Due to vite will cache the transform result. If the HMR
+      // is triggered by other files that the current component relies on, a reload
+      // is required.
+      output.push(
+        `export const _rerender_only = __VUE_HMR_RUNTIME__.CHANGED_FILE === ${JSON.stringify(normalizePath(filename))}`,
+      )
     }
     output.push(
       `import.meta.hot.accept(mod => {`,
