fix(plugin-vue): handle custom element when hot update

vitejs · Jan 6, 2024 · 6ffee6d · 6ffee6d
1 parent 8c694f6
commit 6ffee6d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/packages/plugin-vue/src/handleHotUpdate.ts b/packages/plugin-vue/src/handleHotUpdate.ts
@@ -30,6 +30,7 @@ const directRequestRE = /(?:\?|&)direct\b/
 export async function handleHotUpdate(
   { file, modules, read }: HmrContext,
   options: ResolvedOptions,
+  customElement: boolean,
 ): Promise<ModuleNode[] | void> {
   const prevDescriptor = getDescriptor(file, options, false, true)
   if (!prevDescriptor) {
@@ -46,7 +47,7 @@ export async function handleHotUpdate(
   const templateModule = modules.find((m) => /type=template/.test(m.url))
 
   // trigger resolveScript for descriptor so that we'll have the AST ready
-  resolveScript(descriptor, options, false)
+  resolveScript(descriptor, options, false, customElement)
   const scriptChanged = hasScriptChanged(prevDescriptor, descriptor)
   if (scriptChanged) {
     affectedModules.add(getScriptModule(modules) || mainModule)

diff --git a/packages/plugin-vue/src/index.ts b/packages/plugin-vue/src/index.ts
@@ -137,7 +137,11 @@ export default function vuePlugin(rawOptions: Options = {}): Plugin<Api> {
         return handleTypeDepChange(typeDepToSFCMap.get(ctx.file)!, ctx)
       }
       if (filter.value(ctx.file)) {
-        return handleHotUpdate(ctx, options.value)
+        return handleHotUpdate(
+          ctx,
+          options.value,
+          customElementFilter.value(ctx.file),
+        )
       }
     },