forked from freethenation/node-falafel
/
index.js
64 lines (53 loc) · 1.73 KB
/
index.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
var parse = require('esprima').parse;
module.exports = function (src, fn) {
var opts = {};
if (typeof src === 'object') {
opts = src;
src = opts.source;
delete opts.source;
}
opts.range = true;
if (typeof src !== 'string') src = String(src);
var ast = parse(src, opts);
var result = {
chunks : src.split(''),
toString : function () { return result.chunks.join('') },
inspect : function () { return result.toString() }
};
var index = 0;
function insertHelpers (node, parent) {
if (!node.range) return;
node.parent = parent;
node.source = function () {
return result.chunks.slice(
node.range[0], node.range[1] + 1
).join('');
};
node.update = function (s) {
result.chunks[node.range[0]] = s;
for (var i = node.range[0] + 1; i < node.range[1] + 1; i++) {
result.chunks[i] = '';
}
};
}
(function walk (node, parent) {
insertHelpers(node, parent);
Object.keys(node).forEach(function (key) {
if (key === 'parent') return;
var child = node[key];
if (Array.isArray(child)) {
child.forEach(function (c) {
if (c && typeof c === 'object' && c.type) {
walk(c, node);
}
});
}
else if (child && typeof child === 'object' && child.type) {
insertHelpers(child, node);
walk(child, node);
}
});
fn(node);
})(ast, undefined);
return result;
};