fix: hmr doesn't work when modifying the code of jsx in sfc（#88） (#137)

underfin · Oct 15, 2021 · 66d6485 · 66d6485
1 parent 7a6f9e9
commit 66d6485
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/hmr.ts b/src/hmr.ts
@@ -36,7 +36,14 @@ export async function handleHotUpdate(
   const templateModule = modules.find((m) => /type=template/.test(m.url))
 
   if (!isEqualBlock(descriptor.script, prevDescriptor.script)) {
-    affectedModules.add(mainModule)
+    let scriptModule: ModuleNode | undefined
+    if (descriptor.script?.lang && !descriptor.script.src) {
+      const scriptModuleRE = new RegExp(
+        `type=script.*&lang\.${descriptor.script.lang}$`
+      )
+      scriptModule = modules.find((m) => scriptModuleRE.test(m.url))
+    }
+    affectedModules.add(scriptModule || mainModule)
   }
 
   if (!isEqualBlock(descriptor.template, prevDescriptor.template)) {