Let build_template() own the template source

1 files changed, 9 insertions, 2 deletions

diff --git a/askama_derive/src/lib.rs b/askama_derive/src/lib.rs
index 0e372ed..a2b4ace 100644
--- a/askama_derive/src/lib.rs
+++ b/askama_derive/src/lib.rs
@@ -11,8 +11,9 @@ mod input;
 mod generator;
 mod parser;
 
-use input::Print;
+use input::{Print, Source};
 use proc_macro::TokenStream;
+use shared::path;
 
 #[proc_macro_derive(Template, attributes(template))]
 pub fn derive_template(input: TokenStream) -> TokenStream {
@@ -33,10 +34,16 @@ pub fn derive_template(input: TokenStream) -> TokenStream {
 /// value as passed to the `template()` attribute.
 fn build_template(ast: &syn::DeriveInput) -> String {
     let input = input::TemplateInput::new(ast);
-    let nodes = parser::parse(input.source.as_ref());
+    let source = match input.meta.source {
+        Source::Source(ref s) => s.clone(),
+        Source::Path(_) => path::get_template_source(&input.path)
+    };
+
+    let nodes = parser::parse(&source);
     if input.meta.print == Print::Ast || input.meta.print == Print::All {
         println!("{:?}", nodes);
     }
+
     let code = generator::generate(&input, &nodes);
     if input.meta.print == Print::Code || input.meta.print == Print::All {
         println!("{}", code);