fix(injector): don't visit FunctionDeclarations more than once

merceyz · Jul 9, 2019 · 236276b · 236276b
1 parent 3a40065
commit 236276b
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/injector.ts b/src/injector.ts
@@ -103,10 +103,8 @@ function plugin(propTypes: t.ProgramNode, options: InjectOptions = {}): babel.Pl
         needImport = true;
 
         // Prevent visiting again
-        if ((node as any).hasBeenVisited) {
-          path.skip();
-          return;
-        }
+        (node as any).hasBeenVisited = true;
+        path.skip();
 
         injectPropTypes({
           nodeName: node.id.name,