fix(overmind-react): fix production

Author: christianalfoni

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

@@ -108,16 +108,33 @@ const useState =  <Config extends IConfiguration>(): Overmind<Config>['state'] =
   }
 
   if (IS_PRODUCTION) {
-    react.useEffect(
-      () => () => {
-        ;(overmind as any).proxyStateTree.disposeTree(trackingRef.current.tree)
-      },
-      []
-    )
+    react.useLayoutEffect(() => {
+      trackingRef.current.mounted = true
+      
 
-    react.useLayoutEffect(() => trackingRef.current.tree.stopTracking())
+      if (trackingRef.current.hasUpdatedBeforeCommit) {
+        forceRerender()
+      }
 
-    trackingRef.current.tree.track(forceRerender)
+      return () => {
+        ;(overmind as any).proxyStateTree.disposeTree(trackingRef.current.tree)
+      }
+    }, [])
+
+    react.useLayoutEffect(() => {
+      if (trackingRef.current.stopTrackingTask) {
+        unstable_cancelCallback(trackingRef.current.stopTrackingTask)
+        trackingRef.current.stopTrackingTask = null
+      }
+      trackingRef.current.tree.stopTracking()
+    })
+    trackingRef.current.tree.track(() => {
+      if (trackingRef.current.mounted) {
+        forceRerender()
+      } else {
+        trackingRef.current.hasUpdatedBeforeCommit = true
+      }
+    })
   } else {
     const component = useCurrentComponent()
     const name = component.name