itsscb/askama
1
0
Fork 0
mirror of https://github.com/askama-rs/askama.git synced 2025-09-30 06:21:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Define struct to represent template metadata

Browse Source
This commit is contained in:
Dirkjan Ochtman 2017-02-17 15:03:42 +01:00
parent b84a1242c4
commit 210ed06bdb
1 changed files with 23 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
askama_derive/src/lib.rs
View File

@ -4,19 +4,33 @@ extern crate syn;
use proc_macro::TokenStream;
fn get_path_from_attrs(attrs: &[syn::Attribute]) -> String {
struct TemplateMeta {
path: String,
}
fn get_path_from_attrs(attrs: &[syn::Attribute]) -> TemplateMeta {
let mut path = None;
let attr = attrs.iter().find(|a| a.name() == "template").unwrap();
if let syn::MetaItem::List(_, ref inner) = attr.value {
if let syn::NestedMetaItem::MetaItem(ref item) = inner[0] {
if let &syn::MetaItem::NameValue(ref key, ref val) = item {
assert_eq!(key.as_ref(), "path");
if let &syn::Lit::Str(ref s, _) = val {
return s.clone();
for nm_item in inner {
if let &syn::NestedMetaItem::MetaItem(ref item) = nm_item {
if let &syn::MetaItem::NameValue(ref key, ref val) = item {
match key.as_ref() {
"path" => if let &syn::Lit::Str(ref s, _) = val {
path = Some(s.clone());
} else {
panic!("template path must be string literal");
},
_ => { panic!("unsupported annotation key found") }
}
}
}
}
}
panic!("template path not found in struct attributes");
if path.is_none() {
panic!("template path not found in struct attributes");
}
TemplateMeta { path: path.unwrap() }
}
#[proc_macro_derive(Template, attributes(template))]
@ -26,6 +40,6 @@ pub fn derive_template(input: TokenStream) -> TokenStream {
syn::Body::Struct(ref data) => data,
_ => panic!("#[derive(Template)] can only be used with structs"),
};
let path = get_path_from_attrs(&ast.attrs);
askama::build_template(&path, &ast).parse().unwrap()
let meta = get_path_from_attrs(&ast.attrs);
askama::build_template(&meta.path, &ast).parse().unwrap()
}