Add onEnd handler into traverse. Closes estools#73

a-x- · Jul 4, 2016 · 95e27f9 · 95e27f9
1 parent 6f6a4e9
commit 95e27f9
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/estraverse.js b/estraverse.js
@@ -468,7 +468,8 @@
             current2,
             candidates,
             candidate,
-            sentinel;
+            sentinel,
+            onEnd;
 
         this.__initialize(root, visitor);
 
@@ -482,6 +483,8 @@
         worklist.push(new Element(root, null, null, null));
         leavelist.push(new Element(null, null, null, null));
 
+        onEnd = visitor.onEnd;
+
         while (worklist.length) {
             element = worklist.pop();
 
@@ -551,6 +554,8 @@
                 }
             }
         }
+
+        onEnd && onEnd();
     };
 
     Controller.prototype.replace = function replace(root, visitor) {