esprima-walk
Version:
Walk the primary nodes of an esprima parse tree - fast.
84 lines (42 loc) • 1.13 kB
JavaScript
'use strict'
function walk( ast, fn ) {
var stack = [ ast ], i, j, key, len, node, child
for ( i = 0; i < stack.length; i += 1 ) {
node = stack[ i ]
fn( node )
for ( key in node ) {
child = node[ key ]
if ( child instanceof Array ) {
for ( j = 0, len = child.length; j < len; j += 1 ) {
stack.push( child[ j ] )
}
} else if ( child != void 0 && typeof child.type === 'string' ) {
stack.push( child )
}
}
}
}
walk.walk = walk
walk.walkAddParent = function ( ast, fn ) {
var stack = [ ast ], i, j, key, len, node, child, subchild
for ( i = 0; i < stack.length; i += 1 ) {
node = stack[ i ]
fn( node )
for ( key in node ) {
if ( key !== 'parent' ) {
child = node[ key ]
if ( child instanceof Array ) {
for ( j = 0, len = child.length; j < len; j += 1 ) {
subchild = child[ j ]
subchild.parent = node
stack.push( subchild )
}
} else if ( child != void 0 && typeof child.type === 'string' ) {
child.parent = node
stack.push( child )
}
}
}
}
}
exports = module.exports = walk