From ca91797c7fa93df58a8e2b582ef954f5203061db Mon Sep 17 00:00:00 2001
From: Evan You <yyx990803@gmail.com>
Date: Tue, 13 Nov 2018 01:07:13 -0500
Subject: [PATCH] fix: fix functional updates

---
 packages/runtime-core/src/createRenderer.ts | 20 ++++++++++++++------
 1 file changed, 14 insertions(+), 6 deletions(-)

diff --git a/packages/runtime-core/src/createRenderer.ts b/packages/runtime-core/src/createRenderer.ts
index 6051d40a..27a008e9 100644
--- a/packages/runtime-core/src/createRenderer.ts
+++ b/packages/runtime-core/src/createRenderer.ts
@@ -79,6 +79,7 @@ export interface FunctionalHandle {
   prev: VNode
   next: VNode
   update: Autorun
+  container: RenderNode | null
 }
 
 handleSchedulerError(err => handleError(err, null, ErrorTypes.SCHEDULER))
@@ -266,7 +267,8 @@ export function createRenderer(options: RendererOptions) {
     const handle: FunctionalHandle = (vnode.handle = {
       prev: vnode,
       next: null as any,
-      update: null as any
+      update: null as any,
+      container
     })
 
     const doMount = () => {
@@ -282,6 +284,7 @@ export function createRenderer(options: RendererOptions) {
               vnode.el = subTree.el as RenderNode
             })
             mount(subTree, container, vnode as MountedVNode, isSVG, endNode)
+            handle.next = vnode
             if (__DEV__) {
               popWarningContext()
             }
@@ -310,19 +313,19 @@ export function createRenderer(options: RendererOptions) {
   }
 
   function updateFunctionalComponent(handle: FunctionalHandle, isSVG: boolean) {
-    // mounted
     const { prev, next } = handle
     if (__DEV__) {
       pushWarningContext(next)
     }
+    const prevTree = prev.children as MountedVNode
     const nextTree = (next.children = renderFunctionalRoot(next))
     queueEffect(() => {
       next.el = nextTree.el
     })
     patch(
-      prev.children as MountedVNode,
+      prevTree,
       nextTree,
-      platformParentNode(prev.el),
+      handle.container as RenderNode,
       next as MountedVNode,
       isSVG
     )
@@ -567,7 +570,7 @@ export function createRenderer(options: RendererOptions) {
     } else if (flags & VNodeFlags.COMPONENT_STATEFUL) {
       patchStatefulComponent(prevVNode, nextVNode)
     } else {
-      patchFunctionalComponent(prevVNode, nextVNode)
+      patchFunctionalComponent(prevVNode, nextVNode, container)
     }
   }
 
@@ -605,11 +608,16 @@ export function createRenderer(options: RendererOptions) {
     nextVNode.el = instance.$vnode.el
   }
 
-  function patchFunctionalComponent(prevVNode: MountedVNode, nextVNode: VNode) {
+  function patchFunctionalComponent(
+    prevVNode: MountedVNode,
+    nextVNode: VNode,
+    container: RenderNode
+  ) {
     const prevTree = prevVNode.children as VNode
     const handle = (nextVNode.handle = prevVNode.handle as FunctionalHandle)
     handle.prev = prevVNode
     handle.next = nextVNode
+    handle.container = container
 
     if (shouldUpdateComponent(prevVNode, nextVNode)) {
       queueJob(handle.update)