cerxes/babel
3
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

optimise traverse, improves traversal speed by 50%

Browse Source
This commit is contained in:
Sebastian McKenzie 2014-11-16 18:50:07 +11:00
parent fa3b24e5b4
commit f1183505b1
1 changed files with 17 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
lib/6to5/traverse/index.js
View File

@ -18,7 +18,7 @@ function traverse(parent, callbacks, opts) {
// unknown node type to traverse
var keys = t.VISITOR_KEYS[parent.type];
if (!keys) return;
if (!keys) throw new Error("unknown node " + parent.type);
opts = opts || {};
if (_.isArray(opts)) opts = { blacklist: opts };
@ -29,25 +29,32 @@ function traverse(parent, callbacks, opts) {
// normalise callbacks
if (_.isFunction(callbacks)) callbacks = { enter: callbacks };
_.each(keys, function (key) {
for (var i in keys) {
var key = keys[i];
var nodes = parent[key];
if (!nodes) return;
if (!nodes) continue;
var updated = false;
var handle = function (obj, key) {
var node = obj[key];
if (!node) return;
// type is blacklisted
if (_.contains(blacklistTypes, node.type)) return;
if (blacklistTypes.indexOf(node.type) > -1) return;
// replace node
var maybeReplace = function (result) {
if (result === false) return;
if (result != null) node = obj[key] = result;
if (result != null) {
updated = true;
node = obj[key] = result;
}
};
//
var opts2 = _.clone(opts);
var opts2 = { scope: opts.scope, blacklist: opts.blacklist };
if (t.isScope(node)) opts2.scope = new Scope(opts.scope, node);
// enter
@ -69,15 +76,15 @@ function traverse(parent, callbacks, opts) {
};
if (_.isArray(nodes)) {
_.each(nodes, function (node, i) {
for (var i in nodes) {
handle(nodes, i);
});
}
parent[key] = _.flatten(parent[key]);
if (updated) parent[key] = _.flatten(parent[key]);
} else {
handle(parent, key);
}
});
}
}
traverse.removeProperties = function (tree) {