From 28d76964080f5c6c36d5f6ed72828b679fe352b4 Mon Sep 17 00:00:00 2001
From: bott <mhpoin@gmail.com>
Date: Sun, 7 Oct 2018 17:13:52 +0200
Subject: Add Rust macro support at templates

---
 askama_derive/src/parser.rs | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

(limited to 'askama_derive/src/parser.rs')

diff --git a/askama_derive/src/parser.rs b/askama_derive/src/parser.rs
index 1c5e28e..742a7a2 100644
--- a/askama_derive/src/parser.rs
+++ b/askama_derive/src/parser.rs
@@ -21,6 +21,7 @@ pub enum Expr<'a> {
     Range(&'a str, Option<Box<Expr<'a>>>, Option<Box<Expr<'a>>>),
     Group(Box<Expr<'a>>),
     MethodCall(Box<Expr<'a>>, &'a str, Vec<Expr<'a>>),
+    RustMacro(&'a str, Vec<Expr<'a>>),
 }
 
 #[derive(Debug)]
@@ -427,6 +428,14 @@ named!(expr_unary<Input, Expr>, do_parse!(
     })
 ));
 
+named!(rust_macro<Input, Expr>, do_parse!(
+    mname: identifier >>
+    tag!("!") >>
+    args: arguments >>
+    (Expr::RustMacro(mname, args))
+));
+
+
 macro_rules! expr_prec_layer {
     ( $name:ident, $inner:ident, $( $op:expr ),* ) => {
         named!($name<Input, Expr>, do_parse!(
@@ -463,6 +472,7 @@ named!(range_right<Input, Expr>, do_parse!(
 
 named!(expr_any<Input, Expr>, alt!(
     range_right |
+    rust_macro |
     do_parse!(
         left: expr_or >>
         rest: range_right >> (match rest {
@@ -776,6 +786,7 @@ mod tests {
             }
         }
     }
+
     #[test]
     fn test_ws_splitter() {
         check_ws_split("", &("", "", ""));
@@ -784,6 +795,7 @@ mod tests {
         check_ws_split("b\n", &("", "b", "\n"));
         check_ws_split(" \t\r\n", &(" \t\r\n", "", ""));
     }
+
     #[test]
     #[should_panic]
     fn test_invalid_block() {
@@ -805,5 +817,4 @@ mod tests {
 
         super::parse("{~ strvar|e ~}", &syntax);
     }
-
 }
-- 
cgit