Revert "useState showing on the tree"

This reverts commit 6870edd.

Author: davidchai717

package/index.js

@@ -7,7 +7,6 @@ const mode = {
 };
 
 const linkFiber = require('./linkFiber')(snapShot, mode);
-
 const timeJump = require('./timeJump')(snapShot, mode);
 
 window.addEventListener('message', ({ data: { action, payload } }) => {

package/linkFiber.js

@@ -1,4 +1,3 @@
-/* eslint-disable func-names */
 /* eslint-disable no-use-before-define */
 /* eslint-disable no-param-reassign */
 // links component state tree to library
@@ -13,6 +12,7 @@ module.exports = (snap, mode) => {
     // DEV: So that when we are jumping to an old snapshot it wouldn't think we want to create new snapshots
     if (mode.jumping || mode.paused) return;
     const payload = snap.tree.getCopy();
+    console.log(payload);
     // console.log('payload', payload);
     window.postMessage({
       action: 'recordSnap',
@@ -103,55 +103,20 @@ module.exports = (snap, mode) => {
     snap.tree = createTree(current);
   }
 
-  // return container => {
-  //   // console.log('Container', container);
-  //   const {
-  //     _reactRootContainer: { _internalRoot },
-  //     _reactRootContainer,
-  //   } = container;
-  //   // console.log('Root container', _reactRootContainer);
-  //   // only assign internal root if it actually exists
-  //   fiberRoot = _internalRoot || _reactRootContainer;
-  //   updateSnapShotTree();
-
-  //   // send the initial snapshot once the content script has started up
-  //   window.addEventListener('message', ({ data: { action } }) => {
-  //     if (action === 'contentScriptStarted') sendSnapshot();
-  //   });
-  // };
+  return container => {
+    // console.log('Container', container);
+    const {
+      _reactRootContainer: { _internalRoot },
+      _reactRootContainer,
+    } = container;
+    // console.log('Root container', _reactRootContainer);
+    // only assign internal root if it actually exists
+    fiberRoot = _internalRoot || _reactRootContainer;
+    updateSnapShotTree();
 
-  return {
-    main(container) {
-      // console.log('Container', container);
-      const {
-        _reactRootContainer: { _internalRoot },
-        _reactRootContainer,
-      } = container;
-      // console.log('Root container', _reactRootContainer);
-      // only assign internal root if it actually exists
-      fiberRoot = _internalRoot || _reactRootContainer;
-      updateSnapShotTree();
-  
-      // send the initial snapshot once the content script has started up
-      window.addEventListener('message', ({ data: { action } }) => {
-        if (action === 'contentScriptStarted') sendSnapshot();
-      });
-    },
-    changeUseState(useState) {
-      return function (initial) {
-        // running the original useState and storing its result (state and dispatch function)
-        const toReturn = useState(initial);
-        console.log(toReturn);
-        // storing the original dispatch function definition somewhere
-        const oldDispatch = toReturn[1];
-        // redefining the dispatch function so we can inject our code
-        toReturn[1] = function (callback) {
-          oldDispatch(callback);
-          updateSnapShotTree();
-          sendSnapshot();
-        };
-        return toReturn;
-      };
-    },
+    // send the initial snapshot once the content script has started up
+    window.addEventListener('message', ({ data: { action } }) => {
+      if (action === 'contentScriptStarted') sendSnapshot();
+    });
   };
-};
+};