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
|
use parser::Node;
use std::str;
struct Generator {
buf: String,
indent: u8,
start: bool,
}
impl Generator {
fn new() -> Generator {
Generator { buf: String::new(), indent: 0, start: true }
}
fn init(&mut self, name: &str) {
self.write("impl askama::Template for ");
self.write(name);
self.writeln(" {");
self.indent();
self.writeln("fn render(&self) -> String {");
self.indent();
self.writeln("let mut buf = String::new();");
}
fn indent(&mut self) {
self.indent += 1;
}
fn dedent(&mut self) {
self.indent -= 1;
}
fn write(&mut self, s: &str) {
if self.start {
for _ in 0..(self.indent * 4) {
self.buf.push(' ');
}
self.start = false;
}
self.buf.push_str(s);
}
fn writeln(&mut self, s: &str) {
if s.is_empty() {
return;
}
self.write(s);
self.buf.push('\n');
self.start = true;
}
fn visit_lit(&mut self, s: &[u8]) {
self.write("buf.push_str(");
self.write(&format!("{:#?}", str::from_utf8(s).unwrap()));
self.writeln(");");
}
fn visit_expr(&mut self, s: &[u8]) {
self.write("buf.push_str(");
self.write(&format!("&self.{}", str::from_utf8(s).unwrap()));
self.writeln(");");
}
fn handle(&mut self, tokens: &Vec<Node>) {
for n in tokens {
match n {
&Node::Lit(val) => { self.visit_lit(val); },
&Node::Expr(val) => { self.visit_expr(val); },
}
}
}
fn finalize(&mut self) {
self.writeln("buf");
self.dedent();
self.writeln("}");
self.dedent();
self.writeln("}");
}
fn result(self) -> String {
self.buf
}
}
pub fn generate(ctx_name: &str, tokens: &Vec<Node>) -> String {
let mut gen = Generator::new();
gen.init(ctx_name);
gen.handle(tokens);
gen.finalize();
gen.result()
}
|
