fix(hmr): update hmrRerender to accept dirtyInstance for proper unmounting

Author: edison1105

packages/runtime-core/src/component.ts

@@ -548,7 +548,7 @@ export interface GenericComponentInstance {
   /**
    * @internal vapor only
    */
-  hmrRerender?: () => void
+  hmrRerender?: (dirtyInstance?: any) => void
   /**
    * @internal vapor only
    */

packages/runtime-core/src/hmr.ts

@@ -172,9 +172,7 @@ function reload(id: string, newComp: HMRComponent): void {
           isHmrUpdating = true
           const parent = instance.parent! as ComponentInternalInstance
           if (parent.vapor) {
-            // ensure unregisterHMR is called to avoid stale instances
-            instance.scope.stop()
-            parent.hmrRerender!()
+            parent.hmrRerender!(instance)
           } else {
             if (!(parent.effect.flags! & EffectFlags.STOP)) {
               parent.renderCache = []

packages/runtime-vapor/src/hmr.ts

@@ -15,10 +15,16 @@ import {
 } from './component'
 import { isArray } from '@vue/shared'
 
-export function hmrRerender(instance: VaporComponentInstance): void {
+export function hmrRerender(
+  instance: VaporComponentInstance,
+  dirtyInstance?: VaporComponentInstance,
+): void {
   const normalized = normalizeBlock(instance.block)
   const parent = normalized[0].parentNode!
   const anchor = normalized[normalized.length - 1].nextSibling
+  // When dirty instance is provided, instance is its parent component
+  // and dirty instance needs to be unmount to ensure unregister HMR
+  if (dirtyInstance) unmountComponent(dirtyInstance)
   remove(instance.block, parent)
   const prev = setCurrentInstance(instance)
   if (instance.renderEffects) {
@@ -38,8 +44,7 @@ export function hmrReload(
 ): void {
   // if parent is KeepAlive, we need to rerender it
   if (instance.parent && isKeepAlive(instance.parent)) {
-    instance.scope.stop()
-    instance.parent.hmrRerender!()
+    instance.parent.hmrRerender!(instance)
     return
   }
   const normalized = normalizeBlock(instance.block)