adding recursive traversal

vercel · Jan 13, 2020 · f94eb3e · f94eb3e
1 parent d197ae8
commit f94eb3e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/packages/next/build/webpack/plugins/webpack-conformance-plugin/index.ts b/packages/next/build/webpack/plugins/webpack-conformance-plugin/index.ts
@@ -6,8 +6,6 @@ import {
   IGetAstNodeResult,
   NodeInspector,
 } from './TestInterface'
-// import { Node } from './ASTInterface';
-//import {NodePath, ASTNode} from 'ast-types';
 import { NodePath } from 'ast-types/lib/node-path'
 import { visit } from 'recast'
 
@@ -115,17 +113,19 @@ export default class WebpackConformancePlugin {
         .tap(this.constructor.name, parser => {
           parser.hooks.program.tap(this.constructor.name, (ast: any) => {
             const visitors: VisitorMap = {}
+            const that = this
             for (const visitorKey of collectedVisitors.keys()) {
-              visitors[visitorKey] = (path: NodePath) => {
+              visitors[visitorKey] = function(path: NodePath) {
                 const callbacks = collectedVisitors.get(visitorKey) || []
                 callbacks.forEach(cb => {
                   if (!cb) {
                     return
                   }
                   const { request } = parser.state.module
                   const outcome = cb(path, { request })
-                  this.gatherResults([outcome])
+                  that.gatherResults([outcome])
                 })
+                this.traverse(path)
                 return false
               }
             }