-
Notifications
You must be signed in to change notification settings - Fork 17
/
template-compiler.js
103 lines (80 loc) · 1.99 KB
/
template-compiler.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
let compileFlags = null;
let tEnv = null
function transform(ast) {
let walker = new tEnv.syntax.Walker();
walker.visit(ast, function(node) {
if (!validate(node)) {
return;
}
processParams(node);
});
// return ast;
}
function ConditionalTemplateCompiler(env) {
tEnv = env;
return {
name: 'condititional-compiler',
visitor: {
Program(ast) {
return transform(ast);
}
}
}
}
function mungeNode(node, flag, unless) {
let b = tEnv.syntax.builders;
if (!unless) {
node.path = b.path('if');
} else {
node.path = b.path('unless');
}
let flagEnabled = compileFlags[flag];
node.params = [b.boolean(flagEnabled)];
if (unless) {
flagEnabled = !flagEnabled;
}
if (flagEnabled && node.inverse) {
node.inverse = b.program(false, false, node.inverse.loc);
} else if (!flagEnabled && node.program) {
node.program = b.program(false, false, node.program.loc);
}
}
function mungePlainHelperNode(node, unless) {
let compileFlag = node.params[0]['original'];
if (!(compileFlag in compileFlags)) {
throw 'No compile flag found for ' + compileFlag;
}
mungeNode(node, compileFlag, unless);
}
function processParams(node) {
for (let flag in compileFlags) {
if (node.path.original === 'if-flag-' + flag) {
mungeNode(node, flag, false);
}
if (node.path.original === 'unless-flag-' + flag) {
mungeNode(node, flag, true);
}
}
if (node.path.original === 'if-flag') {
mungePlainHelperNode(node, false);
}
if (node.path.original === 'unless-flag') {
mungePlainHelperNode(node, true);
}
}
function validate(node) {
let nodeType = (node.type === 'BlockStatement' || node.type === 'MustacheStatement');
if (!nodeType) {
return false;
}
return (
node.path && (
node.path.original.startsWith('if-flag') ||
node.path.original.startsWith('unless-flag')
)
);
}
module.exports = function(flags) {
compileFlags = flags;
return ConditionalTemplateCompiler;
};