itsscb/askama
1
0
Fork 0
mirror of https://github.com/askama-rs/askama.git synced 2025-10-02 15:25:19 +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; 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(); let attr = attrs.iter().find(|a| a.name() == "template").unwrap();
if let syn::MetaItem::List(_, ref inner) = attr.value { if let syn::MetaItem::List(_, ref inner) = attr.value {
if let syn::NestedMetaItem::MetaItem(ref item) = inner[0] { for nm_item in inner {
if let &syn::MetaItem::NameValue(ref key, ref val) = item { if let &syn::NestedMetaItem::MetaItem(ref item) = nm_item {
assert_eq!(key.as_ref(), "path"); if let &syn::MetaItem::NameValue(ref key, ref val) = item {
if let &syn::Lit::Str(ref s, _) = val { match key.as_ref() {
return s.clone(); "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))] #[proc_macro_derive(Template, attributes(template))]
@ -26,6 +40,6 @@ pub fn derive_template(input: TokenStream) -> TokenStream {
syn::Body::Struct(ref data) => data, syn::Body::Struct(ref data) => data,
_ => panic!("#[derive(Template)] can only be used with structs"), _ => panic!("#[derive(Template)] can only be used with structs"),
}; };
let path = get_path_from_attrs(&ast.attrs); let meta = get_path_from_attrs(&ast.attrs);
askama::build_template(&path, &ast).parse().unwrap() askama::build_template(&meta.path, &ast).parse().unwrap()
} }