aboutsummaryrefslogtreecommitdiffstats
path: root/askama_derive
diff options
context:
space:
mode:
Diffstat (limited to 'askama_derive')
-rw-r--r--askama_derive/src/generator.rs2
-rw-r--r--askama_derive/src/input.rs43
-rw-r--r--askama_derive/src/lib.rs6
3 files changed, 24 insertions, 27 deletions
diff --git a/askama_derive/src/generator.rs b/askama_derive/src/generator.rs
index a6e74bc..bfff265 100644
--- a/askama_derive/src/generator.rs
+++ b/askama_derive/src/generator.rs
@@ -589,7 +589,7 @@ impl<'a> Generator<'a> {
use self::DisplayWrap::*;
use super::input::EscapeMode::*;
self.write("writer.write_fmt(format_args!(\"{}\", ");
- self.write(match (wrapped, &state.input.meta.escaping) {
+ self.write(match (wrapped, &state.input.escaping) {
(Wrapped, &Html) |
(Wrapped, &None) |
(Unwrapped, &None) => "askama_expr",
diff --git a/askama_derive/src/input.rs b/askama_derive/src/input.rs
index 62c1072..6c631be 100644
--- a/askama_derive/src/input.rs
+++ b/askama_derive/src/input.rs
@@ -7,34 +7,15 @@ use syn;
pub struct TemplateInput<'a> {
pub ast: &'a syn::DeriveInput,
- pub meta: TemplateMeta,
- pub path: PathBuf,
-}
-
-impl<'a> TemplateInput<'a> {
- pub fn new(ast: &'a syn::DeriveInput) -> TemplateInput<'a> {
- let meta = TemplateMeta::new(ast);
- let path = match meta.source {
- Source::Source(_) => match meta.ext {
- Some(ref v) => PathBuf::from(format!("{}.{}", ast.ident, v)),
- None => PathBuf::new(),
- },
- Source::Path(ref s) => path::find_template_from_path(s, None),
- };
- TemplateInput { ast, meta, path }
- }
-}
-
-// Holds metadata for the template, based on the `template()` attribute.
-pub struct TemplateMeta {
pub source: Source,
pub print: Print,
pub escaping: EscapeMode,
pub ext: Option<String>,
+ pub path: PathBuf,
}
-impl TemplateMeta {
- fn new(ast: &syn::DeriveInput) -> TemplateMeta {
+impl<'a> TemplateInput<'a> {
+ pub fn new(ast: &'a syn::DeriveInput) -> TemplateInput<'a> {
let attr = ast.attrs
.iter()
.find(|a| a.interpret_meta().unwrap().name() == "template");
@@ -115,7 +96,23 @@ impl TemplateMeta {
}
},
};
- TemplateMeta { source, print, escaping, ext }
+
+ let path = match source {
+ Source::Source(_) => match ext {
+ Some(ref v) => PathBuf::from(format!("{}.{}", ast.ident, v)),
+ None => PathBuf::new(),
+ },
+ Source::Path(ref s) => path::find_template_from_path(s, None),
+ };
+
+ TemplateInput {
+ ast,
+ source,
+ print,
+ escaping,
+ ext,
+ path,
+ }
}
}
diff --git a/askama_derive/src/lib.rs b/askama_derive/src/lib.rs
index a2b4ace..dcae4d3 100644
--- a/askama_derive/src/lib.rs
+++ b/askama_derive/src/lib.rs
@@ -34,18 +34,18 @@ 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 source = match input.meta.source {
+ let source = match input.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 {
+ if input.print == Print::Ast || input.print == Print::All {
println!("{:?}", nodes);
}
let code = generator::generate(&input, &nodes);
- if input.meta.print == Print::Code || input.meta.print == Print::All {
+ if input.print == Print::Code || input.print == Print::All {
println!("{}", code);
}
code