TimeJump now works with useState

Author: davidchai717

package/linkFiber.js

@@ -52,6 +52,7 @@ module.exports = (snap, mode) => {
 
     // redefining the dispatch function so we can inject our code
     component.queue.dispatch = function (fiber, queue, action) {
+      console.log('mode', mode);
       if (mode.locked && !mode.jumping) return;
       oldDispatch(fiber, queue, action);
       setTimeout(() => {
@@ -104,7 +105,6 @@ module.exports = (snap, mode) => {
       memoizedState.traversed = traverseHooks(memoizedState);
       nextTree = tree.appendChild(memoizedState);
     }
-
     // iterate through siblings
     createTree(sibling, tree);
     // iterate through children

package/timeJump.js

@@ -23,12 +23,20 @@ module.exports = (origin, mode) => {
         });
       });
     }
-    // else {
-    //   // if component uses hooks
-    //   // traverse down the memoize tree again
-    //   console.log('component', originNode.component);
-    //   originNode.component.queue.dispatch(target.state);
-    // }
+    else {
+      // if component uses hooks
+      // variable for current location
+      let currLocation = originNode.component;
+      // state no
+      let stateNum = 1;
+      console.log('component', originNode.component);
+      // while loop through the memoize tree
+      while (currLocation) {
+        currLocation.queue.dispatch(target.state[`state${stateNum}`]);
+        currLocation = currLocation.next;
+        stateNum += 1;
+      }
+    }
   }
 
   return target => {