-
Notifications
You must be signed in to change notification settings - Fork 83
/
index.js
114 lines (97 loc) · 3 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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
'use strict';
var hasOwnProperty = Object.prototype.hasOwnProperty;
function each(processChunk, list) {
var cursor = list.head;
while (cursor !== null) {
this.generate(processChunk, cursor.data, cursor, list);
cursor = cursor.next;
}
}
function eachComma(processChunk, list) {
var cursor = list.head;
while (cursor !== null) {
if (cursor.prev) {
processChunk(',');
}
this.generate(processChunk, cursor.data, cursor, list);
cursor = cursor.next;
}
}
function createGenerator(types) {
var context = {
generate: function(processChunk, node, item, list) {
if (hasOwnProperty.call(types, node.type)) {
types[node.type].call(this, processChunk, node, item, list);
} else {
throw new Error('Unknown node type: ' + node.type);
}
},
each: each,
eachComma: eachComma
};
return function(node, fn) {
if (typeof fn !== 'function') {
var buffer = [];
context.generate(function() {
buffer.push.apply(buffer, arguments);
}, node);
return buffer.join('');
}
context.generate(fn, node);
};
}
function createMarkupGenerator(types) {
var context = {
generate: function(buffer, node, item, list) {
if (hasOwnProperty.call(types, node.type)) {
var nodeBuffer = [];
types[node.type].call(this, function() {
nodeBuffer.push.apply(nodeBuffer, arguments);
}, node, item, list);
buffer({
node: node,
value: nodeBuffer
});
} else {
throw new Error('Unknown node type: ' + node.type);
}
},
each: each,
eachComma: eachComma
};
return function(node, before, after) {
function walk(node, buffer) {
var value = node.value;
before(node.node, buffer, value);
if (typeof value === 'string') {
buffer += value;
} else {
for (var i = 0; i < value.length; i++) {
if (typeof value[i] === 'string') {
buffer += value[i];
} else {
buffer = walk(value[i], buffer);
}
}
}
after(node.node, buffer, value);
return buffer;
}
if (typeof before !== 'function') {
before = function() {};
}
if (typeof after !== 'function') {
after = function() {};
}
var buffer = [];
context.generate(function() {
buffer.push.apply(buffer, arguments);
}, node);
return walk(buffer[0], '');
};
}
module.exports = {
createGenerator: createGenerator,
createMarkupGenerator: createMarkupGenerator,
sourceMap: require('./sourceMap')
};