Merge pull request #12 from haejinjo/getMultipleHooksNames

Fix bug in capturing hook names of components with multiple hooks

Author: haejinjo

src/backend/__tests__/helpers.test.ts

@@ -75,7 +75,7 @@ describe('AST Unit Tests', () => {
       expect(getHooksNames(elementType)).toEqual(['testCount', 'setTestCount']);
     });
 
-    it('Should output the right number of properties when taking in multiple function definitions', () => {
+    it('Should output the right number of properties when given multiple hooks', () => {
       const elementType = `function SingleUseFakeComponent() { 
                             const [testCount, setTestCount] = useState(0);
                             const [biscuitCount, setBiscuitCount] = useState(0);

src/backend/helpers.ts

@@ -91,9 +91,8 @@ export const getHooksNames = (elementType: string): Array<string> => {
             // * Works for useState hooks
             if (hook.id.type === 'ArrayPattern') {
               hook.id.elements.forEach((hook) => {
+                statements.push(`_useWildcard${tsCount}`);
                 statements.push(hook.name);
-                // * Unshift a wildcard name to achieve similar functionality as before
-                statements.unshift(`_useWildcard${tsCount}`);
                 tsCount += 1;
               });
             } else {
@@ -111,7 +110,7 @@ export const getHooksNames = (elementType: string): Array<string> => {
         }
       });
       statements.forEach((el, i) => {
-        if (el.match(/_use/)) hooksNames[el] = statements[i + 2];
+        if (el.match(/_use/)) hooksNames[el] = statements[i + 1];
       });
     });
   }