Attempt 2

Author: JoviDeCroock

packages/core/src/runtime/vnode.js

@@ -20,8 +20,20 @@ const isBuiltIn = type => {
   return BUILT_IN_COMPONENTS.includes(type.name);
 };
 
-const oldVnode = options.__b;
-options.__b = vnode => {
+const oldVnode = options.vnode;
+options.vnode = vnode => {
+  if (vnode && typeof vnode.type === 'function' && !isBuiltIn(vnode.type)) {
+    const vnodes = vnodesForComponent.get(vnode.type);
+    if (!vnodes) {
+      vnodesForComponent.set(vnode.type, [vnode]);
+    }
+  }
+
+  if (oldVnode) oldVnode(vnode);
+};
+
+const oldDiff = options._diff;
+options._diff = vnode => {
   if (vnode && typeof vnode.type === 'function' && !isBuiltIn(vnode.type)) {
     const vnodes = vnodesForComponent.get(vnode.type);
     if (!vnodes) {
@@ -50,7 +62,7 @@ options.__b = vnode => {
     }
   }
 
-  if (oldVnode) oldVnode(vnode);
+  if (oldDiff) oldDiff(vnode);
 };
 
 const oldDiffed = options.diffed;