changes reflect updates to support useEffect and context API

Author: chriswillsflannery

package/astParser.js

@@ -11,28 +11,34 @@ module.exports = elementType => {
   // All module exports will always start off as a single 'FunctionDeclaration' type
   while (Object.hasOwnProperty.call(ast, 'body')) {
     // Traverse down .body once before invoking parsing logic and will loop through any .body after
-    ast = ast.body;
+    // depending on the type of state mgmt, ast will have varying levels where
+    // body array appears in the hierarchy. console.log(ast) on various apps to see this.
+    if (ast.body[0].expression) {
+      ast = ast.body[0].expression.body.body;
+    } else {
+      ast = ast.body[0].body.body;
+    }
     // Iterate through AST of every function declaration
     // Check within each function declaration if there are hook declarations
-    ast.forEach(functionDec => {
-      const { body } = functionDec.body;
-      const statements = [];
-      // Traverse through the function's funcDecs and Expression Statements
-      body.forEach(program => {
-        // Hook Declarations will only be under 'VariableDeclaration' type
-        if (program.type === 'VariableDeclaration') {
-          program.declarations.forEach(dec => {
-            statements.push(dec.id.name);
-          });
-        }
-      });
-      // Iterate through the array and determine getter/setters based on pattern
-      for (let i = 0; i < statements.length; i += 1) {
-        if (statements[i].match(/_use/)) {
-          hookState[statements[i]] = statements[i + 2];
-        }
+    // ast.forEach(functionDec => { // removed 11/6 to account for change on line 16
+    //   const { body } = functionDec.body; // removed 11/6 to account for change on line 16
+    const statements = [];
+    // Traverse through the function's funcDecs and Expression Statements
+    ast.forEach(program => { // changed from body to ast 11/6 to account for change on line 16
+      // Hook Declarations will only be under 'VariableDeclaration' type
+      if (program.type === 'VariableDeclaration') {
+        program.declarations.forEach(dec => {
+          statements.push(dec.id.name);
+        });
       }
     });
+    // Iterate through the array and determine getter/setters based on pattern
+    for (let i = 0; i < statements.length; i += 1) {
+      if (statements[i].match(/_use/)) {
+        hookState[statements[i]] = statements[i + 2];
+      }
+    }
+    // });
   }
   return hookState;
 };

package/linkFiber.js

@@ -71,7 +71,7 @@ module.exports = (snap, mode) => {
     astHooks = Object.values(astHooks);
     // while memoizedState is truthy, save the value to the object
     while (memoizedState && memoizedState.queue) { // prevents useEffect from crashing on load
-      if (memoizedState.next.queue === null) { // prevents double pushing snapshot updates
+      if (memoizedState.next && memoizedState.next.queue === null) { // prevents double pushing snapshot updates
         changeUseState(memoizedState);
       }
       // memoized[astHooks[index]] = memoizedState.memoizedState;