make traversal code simpler

2014-12-27 17:01:47 +11:00
parent 13a2d469a3
commit e96d02c972
11 changed files with 178 additions and 165 deletions
--- a/lib/6to5/util.js
+++ b/lib/6to5/util.js
@@ -150,9 +150,11 @@ exports.template = function (name, nodes, keepExpression) {
  template = _.cloneDeep(template);

  if (!_.isEmpty(nodes)) {
-    traverse(template, function (node) {
-      if (t.isIdentifier(node) && _.has(nodes, node.name)) {
-        return nodes[node.name];
+    traverse(template, {
+      enter: function (node) {
+        if (t.isIdentifier(node) && _.has(nodes, node.name)) {
+          return nodes[node.name];
+        }
      }
    });
  }