aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--askama_derive/src/lib.rs112
-rw-r--r--askama_shared/src/lib.rs113
2 files changed, 112 insertions, 113 deletions
diff --git a/askama_derive/src/lib.rs b/askama_derive/src/lib.rs
index 0333c06..012ad65 100644
--- a/askama_derive/src/lib.rs
+++ b/askama_derive/src/lib.rs
@@ -4,9 +4,6 @@ extern crate syn;
use proc_macro::TokenStream;
-use std::borrow::Cow;
-use std::path::PathBuf;
-
#[proc_macro_derive(Template, attributes(template))]
pub fn derive_template(input: TokenStream) -> TokenStream {
let ast = syn::parse_derive_input(&input.to_string()).unwrap();
@@ -14,112 +11,5 @@ pub fn derive_template(input: TokenStream) -> TokenStream {
syn::Body::Struct(ref data) => data,
_ => panic!("#[derive(Template)] can only be used with structs"),
};
- build_template(&ast).parse().unwrap()
-}
-
-/// Takes a `syn::DeriveInput` and generates source code for it
-///
-/// Reads the metadata from the `template()` attribute to get the template
-/// metadata, then fetches the source from the filesystem. The source is
-/// parsed, and the parse tree is fed to the code generator. Will print
-/// the parse tree and/or generated source according to the `print` key's
-/// value as passed to the `template()` attribute.
-fn build_template(ast: &syn::DeriveInput) -> String {
- let meta = get_template_meta(ast);
- let (path, src) = match meta.source {
- Source::Source(s) => (PathBuf::new(), Cow::Borrowed(s)),
- Source::Path(s) => {
- let path = shared::path::find_template_from_path(&s, None);
- let src = shared::path::get_template_source(&path);
- (path, Cow::Owned(src))
- },
- };
- let nodes = shared::parse(src.as_ref());
- if meta.print == Print::Ast || meta.print == Print::All {
- println!("{:?}", nodes);
- }
- let code = shared::generate(ast, &path, nodes);
- if meta.print == Print::Code || meta.print == Print::All {
- println!("{}", code);
- }
- code
-}
-
-// Returns a `TemplateMeta` based on the `template()` attribute data found
-// in the parsed struct or enum. Will panic if it does not find the required
-// template path, or if the `print` key has an unexpected value.
-fn get_template_meta<'a>(ast: &'a syn::DeriveInput) -> TemplateMeta<'a> {
- let attr = ast.attrs.iter().find(|a| a.name() == "template");
- if attr.is_none() {
- let msg = format!("'template' attribute not found on struct '{}'",
- ast.ident.as_ref());
- panic!(msg);
- }
-
- let attr = attr.unwrap();
- let mut source = None;
- let mut print = Print::None;
- if let syn::MetaItem::List(_, ref inner) = attr.value {
- 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 {
- source = Some(Source::Path(s.as_ref()));
- } else {
- panic!("template path must be string literal");
- },
- "source" => if let syn::Lit::Str(ref s, _) = *val {
- source = Some(Source::Source(s.as_ref()));
- } else {
- panic!("template source must be string literal");
- },
- "print" => if let syn::Lit::Str(ref s, _) = *val {
- print = s.into();
- } else {
- panic!("print value must be string literal");
- },
- _ => { panic!("unsupported annotation key found") }
- }
- }
- }
- }
- }
-
- match source {
- Some(s) => TemplateMeta { source: s, print },
- None => panic!("template path or source not found in struct attributes"),
- }
-}
-
-// Holds metadata for the template, based on the `template()` attribute.
-struct TemplateMeta<'a> {
- source: Source<'a>,
- print: Print,
-}
-
-enum Source<'a> {
- Path(&'a str),
- Source(&'a str),
-}
-
-#[derive(PartialEq)]
-enum Print {
- All,
- Ast,
- Code,
- None,
-}
-
-impl<'a> From<&'a String> for Print {
- fn from(s: &'a String) -> Print {
- use Print::*;
- match s.as_ref() {
- "all" => All,
- "ast" => Ast,
- "code" => Code,
- "none" => None,
- v => panic!("invalid value for print option: {}", v),
- }
- }
+ shared::build_template(&ast).parse().unwrap()
}
diff --git a/askama_shared/src/lib.rs b/askama_shared/src/lib.rs
index b5bc9af..60a263f 100644
--- a/askama_shared/src/lib.rs
+++ b/askama_shared/src/lib.rs
@@ -14,13 +14,90 @@ pub use escaping::MarkupDisplay;
pub use errors::{Error, Result};
pub mod filters;
pub mod path;
-pub use parser::parse;
-pub use generator::generate;
mod escaping;
mod generator;
mod parser;
+use std::borrow::Cow;
+use std::path::PathBuf;
+
+
+/// Takes a `syn::DeriveInput` and generates source code for it
+///
+/// Reads the metadata from the `template()` attribute to get the template
+/// metadata, then fetches the source from the filesystem. The source is
+/// parsed, and the parse tree is fed to the code generator. Will print
+/// the parse tree and/or generated source according to the `print` key's
+/// value as passed to the `template()` attribute.
+pub fn build_template(ast: &syn::DeriveInput) -> String {
+ let meta = get_template_meta(ast);
+ let (path, src) = match meta.source {
+ Source::Source(s) => (PathBuf::new(), Cow::Borrowed(s)),
+ Source::Path(s) => {
+ let path = path::find_template_from_path(&s, None);
+ let src = path::get_template_source(&path);
+ (path, Cow::Owned(src))
+ },
+ };
+ let nodes = parser::parse(src.as_ref());
+ if meta.print == Print::Ast || meta.print == Print::All {
+ println!("{:?}", nodes);
+ }
+ let code = generator::generate(ast, &path, nodes);
+ if meta.print == Print::Code || meta.print == Print::All {
+ println!("{}", code);
+ }
+ code
+}
+
+// Returns a `TemplateMeta` based on the `template()` attribute data found
+// in the parsed struct or enum. Will panic if it does not find the required
+// template path, or if the `print` key has an unexpected value.
+fn get_template_meta<'a>(ast: &'a syn::DeriveInput) -> TemplateMeta<'a> {
+ let attr = ast.attrs.iter().find(|a| a.name() == "template");
+ if attr.is_none() {
+ let msg = format!("'template' attribute not found on struct '{}'",
+ ast.ident.as_ref());
+ panic!(msg);
+ }
+
+ let attr = attr.unwrap();
+ let mut source = None;
+ let mut print = Print::None;
+ if let syn::MetaItem::List(_, ref inner) = attr.value {
+ 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 {
+ source = Some(Source::Path(s.as_ref()));
+ } else {
+ panic!("template path must be string literal");
+ },
+ "source" => if let syn::Lit::Str(ref s, _) = *val {
+ source = Some(Source::Source(s.as_ref()));
+ } else {
+ panic!("template source must be string literal");
+ },
+ "print" => if let syn::Lit::Str(ref s, _) = *val {
+ print = s.into();
+ } else {
+ panic!("print value must be string literal");
+ },
+ _ => { panic!("unsupported annotation key found") }
+ }
+ }
+ }
+ }
+ }
+
+ match source {
+ Some(s) => TemplateMeta { source: s, print },
+ None => panic!("template path or source not found in struct attributes"),
+ }
+}
+
mod errors {
error_chain! {
foreign_links {
@@ -29,3 +106,35 @@ mod errors {
}
}
}
+
+// Holds metadata for the template, based on the `template()` attribute.
+struct TemplateMeta<'a> {
+ source: Source<'a>,
+ print: Print,
+}
+
+enum Source<'a> {
+ Path(&'a str),
+ Source(&'a str),
+}
+
+#[derive(PartialEq)]
+enum Print {
+ All,
+ Ast,
+ Code,
+ None,
+}
+
+impl<'a> From<&'a String> for Print {
+ fn from(s: &'a String) -> Print {
+ use Print::*;
+ match s.as_ref() {
+ "all" => All,
+ "ast" => Ast,
+ "code" => Code,
+ "none" => None,
+ v => panic!("invalid value for print option: {}", v),
+ }
+ }
+}