fix(plugin-vue): hmr not working when updating script+template at the same time with a template preprocessor

ref: vitejs/vite-plugin-vue@93c444c

Author: sxzz

src/core/index.ts

@@ -292,13 +292,14 @@ export default createUnplugin<Options | undefined, false>(
           return
         }
 
+        const context = Object.assign({}, this, meta)
         if (!query.vue) {
           // main request
           return transformMain(
             code,
             filename,
             options,
-            Object.assign({}, this, meta),
+            context,
             ssr,
             customElementFilter(filename)
           )
@@ -313,7 +314,7 @@ export default createUnplugin<Options | undefined, false>(
               code,
               descriptor,
               options,
-              this,
+              context,
               ssr
             )
           } else if (query.type === 'style') {

src/core/script.ts

@@ -1,7 +1,8 @@
 import { type SFCDescriptor, type SFCScriptBlock } from 'vue/compiler-sfc'
+import { type UnpluginContextMeta } from 'unplugin'
 import { resolveTemplateCompilerOptions } from './template'
 import { cache as descriptorCache } from './utils/descriptorCache'
-import { type Context, type ResolvedOptions } from '.'
+import { type ResolvedOptions } from '.'
 
 // ssr and non ssr builds would output different script content
 const clientCache = new WeakMap<SFCDescriptor, SFCScriptBlock | null>()
@@ -50,7 +51,7 @@ export function isUseInlineTemplate(
 export const scriptIdentifier = `_sfc_main`
 
 export function resolveScript(
-  pluginContext: Context,
+  pluginContext: UnpluginContextMeta,
   descriptor: SFCDescriptor,
   options: ResolvedOptions,
   ssr: boolean
@@ -104,7 +105,7 @@ export function resolveScript(
 // If the script is js/ts and has no external src, it can be directly placed
 // in the main module. Skip for build
 export function canInlineMain(
-  pluginContext: Context,
+  pluginContext: UnpluginContextMeta,
   descriptor: SFCDescriptor,
   options: ResolvedOptions
 ): boolean {

src/core/template.ts

@@ -6,17 +6,16 @@ import {
   type SFCTemplateCompileOptions,
   type SFCTemplateCompileResults,
 } from 'vue/compiler-sfc'
-import { type UnpluginContext } from 'unplugin'
-import { getResolvedScript } from './script'
+import { getResolvedScript, resolveScript } from './script'
 import { createRollupError } from './utils/error'
-import { type ResolvedOptions } from '.'
+import { type Context, type ResolvedOptions } from '.'
 
 // eslint-disable-next-line require-await
 export async function transformTemplateAsModule(
   code: string,
   descriptor: SFCDescriptor,
   options: ResolvedOptions,
-  pluginContext: UnpluginContext,
+  pluginContext: Context,
   ssr: boolean
 ): Promise<{
   code: string
@@ -49,7 +48,7 @@ export function transformTemplateInMain(
   code: string,
   descriptor: SFCDescriptor,
   options: ResolvedOptions,
-  pluginContext: UnpluginContext,
+  pluginContext: Context,
   ssr: boolean
 ): SFCTemplateCompileResults {
   const result = compile(code, descriptor, options, pluginContext, ssr)
@@ -62,15 +61,15 @@ export function transformTemplateInMain(
   }
 }
 
-// eslint-disable-next-line @typescript-eslint/explicit-module-boundary-types
 export function compile(
   code: string,
   descriptor: SFCDescriptor,
   options: ResolvedOptions,
-  pluginContext: UnpluginContext,
+  pluginContext: Context,
   ssr: boolean
 ) {
   const filename = descriptor.filename
+  resolveScript(pluginContext, descriptor, options, ssr)
   const result = options.compiler.compileTemplate({
     ...resolveTemplateCompilerOptions(descriptor, options, ssr)!,
     source: code,