Find the node with router information recursively

Author: rajeebbanstola

package/index.js

@@ -9,17 +9,27 @@ const mode = {
 const linkFiber = require('./linkFiber')(snapShot, mode);
 const timeJump = require('./timeJump')(snapShot, mode);
 
-window.addEventListener('message', ({ data: { action, payload } }) => {
-  //runs automatically twice per second with inspectedElement
+let url = '/';
+function getRouteURL(node) {
+  if (node.name === 'Router') {
+    url = node.state.location.pathname;
+  }
+  if (node.children.length >= 1) {
+    const tempNode = node.children;
+    for (let index = 0; index < tempNode.length; index += 1) {
+      getRouteURL(tempNode[index]);
+    }
+  }
+}
+
+window.addEventListener('message', ({ data: { action, payload } }) => { // runs automatically twice per second with inspectedElement
   switch (action) {
     case 'jumpToSnap':
       timeJump(payload);
+      getRouteURL(payload);
       // Get the pathname from payload and add new entry to browser history
       // MORE: https://developer.mozilla.org/en-US/docs/Web/API/History/pushState
-      if (payload.children[0].state && payload.children[0].state.location) {
-        const route = payload.children[0].state.location.pathname;
-        window.history.pushState('', '', route);
-      }
+      window.history.pushState('', '', url);
       break;
     case 'setLock':
       mode.locked = payload;