wip: save

Author: edison1105

packages/runtime-vapor/src/components/Transition.ts

@@ -27,7 +27,7 @@ import {
 } from '../component'
 import { extend, isArray } from '@vue/shared'
 import { renderEffect } from '../renderEffect'
-import { type VaporFragment, isFragment } from '../fragment'
+import { isFragment } from '../fragment'
 import {
   currentHydrationNode,
   isHydrating,
@@ -211,15 +211,21 @@ export function applyTransitionHooks(
   hooks: VaporTransitionHooks,
   fallthroughAttrs: boolean = true,
 ): VaporTransitionHooks {
+  // filter out comment nodes
+  if (isArray(block)) {
+    block = block.filter(b => !(b instanceof Comment))
+    if (block.length === 1) {
+      block = block[0]
+    } else if (block.length === 0) {
+      return hooks
+    }
+  }
+
   const isFrag = isFragment(block)
-  const child = findTransitionBlock(
-    block,
-    // set transition hooks on fragment for reusing during it's updating
-    frag => setTransitionHooksOnFragment(frag, hooks),
-    isFrag,
-  )
+  const child = findTransitionBlock(block, isFrag)
   if (!child) {
-    // if (isFrag) setTransitionHooksOnFragment(block, hooks)
+    // set transition hooks on fragment for reusing during it's updating
+    if (isFrag) setTransitionHooksOnFragment(block, hooks)
     return hooks
   }
 
@@ -297,7 +303,6 @@ export function applyTransitionLeaveHooks(
 const transitionBlockCache = new WeakMap<Block, TransitionBlock>()
 export function findTransitionBlock(
   block: Block,
-  processFragment?: (frag: VaporFragment) => void,
   inFragment: boolean = false,
 ): TransitionBlock | undefined {
   if (transitionBlockCache.has(block)) {
@@ -311,7 +316,7 @@ export function findTransitionBlock(
   } else if (isVaporComponent(block)) {
     // stop searching if encountering nested Transition component
     if (getComponentName(block.type) === displayName) return undefined
-    child = findTransitionBlock(block.block, processFragment, inFragment)
+    child = findTransitionBlock(block.block, inFragment)
     // use component id as key
     if (child && child.$key === undefined) child.$key = block.uid
   } else if (isArray(block)) {
@@ -320,7 +325,7 @@ export function findTransitionBlock(
       if (c instanceof Comment) continue
       // check if the child is a fragment to suppress warnings
       if (isFragment(c)) inFragment = true
-      const item = findTransitionBlock(c, processFragment, inFragment)
+      const item = findTransitionBlock(c, inFragment)
       if (__DEV__ && hasFound) {
         // warn more than one non-comment child
         warn(
@@ -339,9 +344,7 @@ export function findTransitionBlock(
     if (block.insert) {
       child = block
     } else {
-      processFragment && processFragment(block)
-      // once we encounter a fragment, we are inside a fragment
-      child = findTransitionBlock(block.nodes, processFragment, true)
+      child = findTransitionBlock(block.nodes, true)
     }
   }