fix(overmind-vue): track state from connect

Author: christianalfoni

packages/node_modules/overmind-vue/src/index.ts

@@ -5,7 +5,7 @@ const IS_PRODUCTION = process.env.NODE_ENV === 'production'
 
 let nextComponentId = 0
 
-function createMixin(overmind, propsCallback) {
+function createMixin(overmind, propsCallback, trackPropsCallback = false) {
   const componentId = nextComponentId++
   let componentInstanceId = 0
 
@@ -48,6 +48,8 @@ function createMixin(overmind, propsCallback) {
           reaction: overmind.reaction,
         }
 
+        this[OVERMIND].tree.track(this[OVERMIND].onUpdate)
+
         if (propsCallback) {
           Object.assign(
             this,
@@ -58,13 +60,23 @@ function createMixin(overmind, propsCallback) {
             })
           )
         }
-        this[OVERMIND].tree.track(this[OVERMIND].onUpdate)
       }
     },
     beforeUpdate(this: any) {
       if (overmind.mode.mode === MODE_SSR) return
 
       this[OVERMIND].tree.track(this[OVERMIND].onUpdate)
+
+      if (propsCallback && trackPropsCallback) {
+        Object.assign(
+          this,
+          propsCallback({
+            state: this[OVERMIND].tree.state,
+            actions: overmind.actions,
+            effects: overmind.effects,
+          })
+        )
+      }
     },
     ...(IS_PRODUCTION
       ? {
@@ -144,7 +156,7 @@ export const createConnect = (overmind) => (...args) => {
   return {
     ...options,
     mixins: (options.mixins ? options.mixins : []).concat(
-      createMixin(overmind, propsCallback)
+      createMixin(overmind, propsCallback, true)
     ),
     overmind,
   } as any