diff options
Diffstat (limited to 'askama_codegen/src/lib.rs')
-rw-r--r-- | askama_codegen/src/lib.rs | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/askama_codegen/src/lib.rs b/askama_codegen/src/lib.rs index 4a17b18..b325bae 100644 --- a/askama_codegen/src/lib.rs +++ b/askama_codegen/src/lib.rs @@ -10,6 +10,7 @@ extern crate syn; use proc_macro::TokenStream; use std::fs::File; use std::io::Read; +use std::path::{Path, PathBuf}; fn get_path_from_attrs(attrs: &Vec<syn::Attribute>) -> String { for attr in attrs { @@ -39,7 +40,11 @@ fn get_path_from_attrs(attrs: &Vec<syn::Attribute>) -> String { panic!("template annotation not found"); } -fn get_template_source(path: &str) -> String { +fn get_template_source(tpl_file: &str) -> String { + let root = ::std::env::var("CARGO_MANIFEST_DIR").unwrap(); + let mut path = PathBuf::from(root); + path.push("templates"); + path.push(Path::new(tpl_file)); let mut f = File::open(path).unwrap(); let mut s = String::new(); f.read_to_string(&mut s).unwrap(); @@ -64,7 +69,7 @@ pub fn derive_template(input: TokenStream) -> TokenStream { let expanded = quote! { impl #impl_generics askama::Template for #name #ty_generics #where_clause { fn render(&self) -> String { - src + #src.into() } } }; |