wip: reused compiled script on template hmr

Author: yyx990803

src/handleHotUpdate.ts

@@ -2,6 +2,7 @@ import fs from 'fs'
 import _debug from 'debug'
 import { parse, SFCBlock } from '@vue/compiler-sfc'
 import { getDescriptor, setDescriptor } from './utils/descriptorCache'
+import { getResolvedScript, setResolvedScript } from './script'
 
 const debug = _debug('vite:hmr')
 
@@ -50,6 +51,11 @@ export async function handleHotUpdate(file: string, modules: any[]) {
   }
 
   if (!isEqualBlock(descriptor.template, prevDescriptor.template)) {
+    // when a <script setup> component's template changes, it will need correct
+    // binding metadata. However, when reloading the template alone the binding
+    // metadata will not be available since the script part isn't loaded.
+    // in this case, reuse the compiled script from previous descriptor.
+    setResolvedScript(descriptor, getResolvedScript(prevDescriptor)!)
     needRerender = true
   }
 
@@ -75,13 +81,6 @@ export async function handleHotUpdate(file: string, modules: any[]) {
     const prev = prevStyles[i]
     const next = nextStyles[i]
     if (!prev || !isEqualBlock(prev, next)) {
-      // css modules update causes a reload because the $style object is changed
-      // and it may be used in JS.
-      // if (prev.module != null || next.module != null) {
-      //   return modules.filter(
-      //     (m) => !/type=/.test(m.id) || /type=script/.test(m.id)
-      //   )
-      // }
       didUpdateStyle = true
       filteredModules.push(modules.find((m) => m.id.includes(`index=${i}`)))
     }

src/script.ts

@@ -10,11 +10,19 @@ const serverCache = new WeakMap<SFCDescriptor, SFCScriptBlock | null>()
 
 export function getResolvedScript(
   descriptor: SFCDescriptor,
-  isServer: boolean
+  isServer = false
 ): SFCScriptBlock | null | undefined {
   return (isServer ? serverCache : clientCache).get(descriptor)
 }
 
+export function setResolvedScript(
+  descriptor: SFCDescriptor,
+  script: SFCScriptBlock,
+  isServer = false
+) {
+  ;(isServer ? serverCache : clientCache).set(descriptor, script)
+}
+
 export function resolveScript(
   descriptor: SFCDescriptor,
   scopeId: string,

src/utils/descriptorCache.ts

@@ -10,6 +10,5 @@ export function getDescriptor(id: string) {
   if (cache.has(id)) {
     return cache.get(id)!
   }
-
   throw new Error(`${id} is not parsed yet`)
 }