fixed timeJump and added helpers to save the state obj

Author: yujinkay

package/linkFiber.js

@@ -4,11 +4,12 @@
 // links component state tree to library
 // changes the setState method to also update our snapshot
 const Tree = require('./tree');
-const astParser = require('./astParser.js');
+const astParser = require('./astParser'); 
+const { saveState } = require('./masterState'); 
 
 module.exports = (snap, mode) => {
   let fiberRoot = null;
-  let astHooks; 
+  let astHooks;
 
   function sendSnapshot() {
     // don't send messages while jumping or while paused
@@ -61,16 +62,16 @@ module.exports = (snap, mode) => {
   }
 
   // Helper function to traverse through the memoized state
-  // TODO: WE NEED TO CLEAN IT UP A BIT
   function traverseHooks(memoizedState) {
     // Declare variables and assigned to 0th index and an empty object, respectively
     const memoized = {};
     let index = 0; 
     astHooks = Object.values(astHooks); 
     // while memoizedState is truthy, save the value to the object
-    while (memoizedState && astHooks) {
+    while (memoizedState) {
       changeUseState(memoizedState);
-      memoized[astHooks[index]] = memoizedState.memoizedState;
+      //memoized[astHooks[index]] = memoizedState.memoizedState;
+      memoized[astHooks[index]] = memoizedState.memoizedState; 
       // Reassign memoizedState to its next value
       memoizedState = memoizedState.next;
       // Increment the index by 2
@@ -125,8 +126,11 @@ module.exports = (snap, mode) => {
     // only assign internal rootp if it actually exists
     fiberRoot = _internalRoot || _reactRootContainer;
     // If hooks are implemented, traverse through the source code 
-    if (entryFile) astHooks = astParser(entryFile);
-
+    // Save the getter/setter combo for timeJump
+    if (entryFile) {
+      astHooks = astParser(entryFile);
+      saveState(astHooks); 
+    }
     updateSnapShotTree();
     // send the initial snapshot once the content script has started up
     window.addEventListener('message', ({ data: { action } }) => {

package/masterState.js

@@ -0,0 +1,14 @@
+// Export two functions that either saves the AST state object into an array
+// or returns the array for use
+const masterState = [];
+
+module.exports = {
+  saveState: state => {
+    for (const key in state) {
+      masterState.push(state[key]);
+    }
+  },
+  returnState: () => {
+    return masterState; 
+  },
+};

package/timeJump.js

@@ -1,5 +1,7 @@
 /* eslint-disable no-param-reassign */
 // traverses given tree by accessing children through coords array
+const { saveState, returnState } = require('./masterState'); 
+
 function traverseTree(tree, coords) {
   let curr = tree;
   coords.forEach(coord => {
@@ -21,13 +23,16 @@ module.exports = (origin, mode) => {
         });
       });
     } else {
-      // if component uses hooks
+      // if component uses hooks, traverse through the memoize tree
       let current = originNode.component;
-      let index = 1;
-      // Iterate through the memoized tree
+      let index = 0;
+      const hooks = returnState(); 
+      // while loop through the memoize tree
       while (current) {
-        current.queue.dispatch(target.state[`state${index++}`]);
+        current.queue.dispatch(target.state[hooks[index]]);
+        // Reassign the current value
         current = current.next;
+        index += 2;
       }
     }
   }