aboutsummaryrefslogtreecommitdiffstats
path: root/askama_codegen/src
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--askama_codegen/src/generator.rs12
-rw-r--r--askama_codegen/src/parser.rs12
2 files changed, 18 insertions, 6 deletions
diff --git a/askama_codegen/src/generator.rs b/askama_codegen/src/generator.rs
index 34c9757..b1babce 100644
--- a/askama_codegen/src/generator.rs
+++ b/askama_codegen/src/generator.rs
@@ -1,4 +1,4 @@
-use parser::Node;
+use parser::{Expr, Node};
use std::str;
struct Generator {
@@ -56,18 +56,24 @@ impl Generator {
self.writeln(");");
}
- fn visit_expr(&mut self, s: &[u8]) {
+ fn visit_var(&mut self, s: &[u8]) {
let var_name = str::from_utf8(s).unwrap();
let code = format!("std::fmt::Write::write_fmt(\
&mut buf, format_args!(\"{{}}\", self.{})).unwrap();", var_name);
self.writeln(&code);
}
+ fn visit_expr(&mut self, expr: &Expr) {
+ match expr {
+ &Expr::Var(s) => self.visit_var(s),
+ }
+ }
+
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); },
+ &Node::Expr(ref val) => { self.visit_expr(&val); },
}
}
}
diff --git a/askama_codegen/src/parser.rs b/askama_codegen/src/parser.rs
index b7a712b..5f3f8c5 100644
--- a/askama_codegen/src/parser.rs
+++ b/askama_codegen/src/parser.rs
@@ -1,8 +1,12 @@
use nom::{self, IResult};
+pub enum Expr<'a> {
+ Var(&'a [u8]),
+}
+
pub enum Node<'a> {
Lit(&'a [u8]),
- Expr(&'a [u8]),
+ Expr(Expr<'a>),
}
fn take_content(i: &[u8]) -> IResult<&[u8], Node> {
@@ -23,9 +27,11 @@ fn take_content(i: &[u8]) -> IResult<&[u8], Node> {
IResult::Done(&i[..0], Node::Lit(&i[..]))
}
-named!(expr_str, delimited!(tag!("{{"), take_until!("}}"), tag!("}}")));
+named!(expr_var<Expr>, map!(ws!(nom::alphanumeric), Expr::Var));
+
+named!(any_expr<Expr>, delimited!(tag!("{{"), expr_var, tag!("}}")));
-named!(expr_node<Node>, map!(expr_str, Node::Expr));
+named!(expr_node<Node>, map!(any_expr, Node::Expr));
named!(parse_template< Vec<Node> >, many1!(alt!(take_content | expr_node)));