You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
1.3 KiB
JavaScript
53 lines
1.3 KiB
JavaScript
7 years ago
|
var parse = require('acorn').parse;
|
||
|
|
||
|
module.exports = function (src, opts) {
|
||
|
if (!opts) opts = {}
|
||
|
var ast = src;
|
||
|
if (typeof src === 'string') {
|
||
|
try {
|
||
|
ast = parse(src, {
|
||
|
ecmaVersion: opts.ecmaVersion || 8,
|
||
|
allowReturnOutsideFunction: true
|
||
|
})
|
||
|
}
|
||
|
catch (err) { ast = parse('(' + src + ')') }
|
||
|
}
|
||
|
return function (cb) {
|
||
|
walk(ast, undefined, cb);
|
||
|
};
|
||
|
};
|
||
|
|
||
|
function walk (node, parent, cb) {
|
||
|
var keys = objectKeys(node);
|
||
|
for (var i = 0; i < keys.length; i++) {
|
||
|
var key = keys[i];
|
||
|
if (key === 'parent') continue;
|
||
|
|
||
|
var child = node[key];
|
||
|
if (isArray(child)) {
|
||
|
for (var j = 0; j < child.length; j++) {
|
||
|
var c = child[j];
|
||
|
if (c && typeof c.type === 'string') {
|
||
|
c.parent = node;
|
||
|
walk(c, node, cb);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
else if (child && typeof child.type === 'string') {
|
||
|
child.parent = node;
|
||
|
walk(child, node, cb);
|
||
|
}
|
||
|
}
|
||
|
cb(node);
|
||
|
}
|
||
|
|
||
|
var isArray = Array.isArray || function (xs) {
|
||
|
return Object.prototype.toString.call(xs) === '[object Array]';
|
||
|
};
|
||
|
|
||
|
var objectKeys = Object.keys || function (obj) {
|
||
|
var keys = [];
|
||
|
for (var key in obj) keys.push(key);
|
||
|
return keys;
|
||
|
};
|