itsscb/askama
1
0
Fork 0
mirror of https://github.com/askama-rs/askama.git synced 2025-09-30 14:31:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Create new TemplateInput abstraction

Browse Source
This commit is contained in:
Dirkjan Ochtman 2017-09-05 19:54:15 +02:00
parent 41a41579ac
commit cf360ade77
1 changed files with 27 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
askama_shared/src/lib.rs
View File

@ -31,26 +31,40 @@ use std::path::PathBuf;
/// 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 = TemplateMeta::new(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 {
let data = TemplateInput::new(ast);
let nodes = parser::parse(data.source.as_ref());
if data.meta.print == Print::Ast || data.meta.print == Print::All {
println!("{:?}", nodes);
}
let code = generator::generate(ast, &path, &nodes);
if meta.print == Print::Code || meta.print == Print::All {
let code = generator::generate(data.ast, &data.path, &nodes);
if data.meta.print == Print::Code || data.meta.print == Print::All {
println!("{}", code);
}
code
}
struct TemplateInput<'a> {
pub ast: &'a syn::DeriveInput,
pub meta: TemplateMeta<'a>,
pub path: PathBuf,
pub source: Cow<'a, str>,
}
impl<'a> TemplateInput<'a> {
fn new(ast: &'a syn::DeriveInput) -> TemplateInput<'a> {
let meta = TemplateMeta::new(ast);
let (path, source) = 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))
},
};
TemplateInput { ast, meta, path, source }
}
}
mod errors {
error_chain! {
foreign_links {