Sha256: e65109ada005c66140d1b4452d36c4a57b794f0bdc713c02eca0be65e752dfd0
Contents?: true
Size: 1.05 KB
Versions: 5
Compression:
Stored size: 1.05 KB
Contents
export function walk ( ast, { enter, leave }) {
visit( ast, null, enter, leave );
}
const context = {
skip: () => context.shouldSkip = true,
shouldSkip: false
};
let childKeys = {};
const toString = Object.prototype.toString;
function isArray ( thing ) {
return toString.call( thing ) === '[object Array]';
}
function visit ( node, parent, enter, leave, prop, index ) {
if ( !node ) return;
if ( enter ) {
context.shouldSkip = false;
enter.call( context, node, parent, prop, index );
if ( context.shouldSkip ) return;
}
const keys = childKeys[ node.type ] || (
childKeys[ node.type ] = Object.keys( node ).filter( key => typeof node[ key ] === 'object' )
);
for ( let i = 0; i < keys.length; i += 1 ) {
const key = keys[i];
const value = node[ key ];
if ( isArray( value ) ) {
for ( let j = 0; j < value.length; j += 1 ) {
visit( value[j], node, enter, leave, key, j );
}
}
else if ( value && value.type ) {
visit( value, node, enter, leave, key, null );
}
}
if ( leave ) {
leave( node, parent, prop, index );
}
}
Version data entries
5 entries across 5 versions & 3 rubygems