fix up formatting in traverse

This commit is contained in:
Sebastian McKenzie 2015-01-18 02:35:19 +11:00
parent 621e66e9ac
commit d0dc972840

View File

@ -104,17 +104,18 @@ TraversalContext.prototype.visitNode = function (obj, key, opts, scope, parent,
var node = obj[key];
// type is blacklisted
if (opts.blacklist && opts.blacklist.indexOf(node.type) > -1)
return;
if (opts.blacklist && opts.blacklist.indexOf(node.type) > -1) return;
var ourScope = scope;
if (t.isScope(node))
if (t.isScope(node)) {
ourScope = new Scope(node, scope);
}
node = this.enterNode(obj, key, node, opts.enter, parent, ourScope, state);
if (this.didSkip)
if (this.didSkip) {
return this.didStop;
}
traverseNode(node, opts, ourScope, state);
this.exitNode(obj, key, node, opts.exit, parent, ourScope, state);
@ -155,10 +156,11 @@ function traverseNode(node, opts, scope, state) {
var context = new TraversalContext();
for (var j = 0; j < keys.length; j++) {
if (context.visit(node, keys[j], opts, scope, state))
if (context.visit(node, keys[j], opts, scope, state)) {
return;
}
}
}
function traverse(parent, opts, scope, state) {
// falsy node
@ -169,14 +171,13 @@ function traverse(parent, opts, scope, state) {
if (!opts.exit) opts.exit = noop;
// array of nodes
if (!Array.isArray(parent)) {
traverseNode(parent, opts, scope, state);
return;
}
if (Array.isArray(parent)) {
for (var i = 0; i < parent.length; i++) {
traverseNode(parent[i], opts, scope, state);
}
} else {
traverseNode(parent, opts, scope, state);
}
}
traverse.removeProperties = function (tree) {