itsscb/askama
1
0
Fork 0
mirror of https://github.com/askama-rs/askama.git synced 2025-09-29 05:51:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move impl code generator into separate module

Browse Source
This commit is contained in:
Dirkjan Ochtman 2016-12-24 10:12:59 +01:00
parent 8f0aae265c
commit 99a83551f5
2 changed files with 25 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
askama_codegen/src/generator.rs Normal file
View File

@ -0,0 +1,23 @@
use std::str;
pub fn generate(ctx_name: &str, tokens: &Vec<&[u8]>) -> String {
let mut code = String::new();
code.push_str("impl askama::Template for ");
code.push_str(ctx_name);
code.push_str(" {\n");
code.push_str(" fn render(&self) -> String {\n");
code.push_str(" let mut buf = String::new();\n");
code.push_str(" buf.push_str(\"");
code.push_str(str::from_utf8(tokens[0]).unwrap());
code.push_str("\");\n");
code.push_str(" buf.push_str(&self.");
code.push_str(str::from_utf8(tokens[1]).unwrap());
code.push_str(");\n");
code.push_str(" buf.push_str(\"");
code.push_str(str::from_utf8(tokens[2]).unwrap());
code.push_str("\");\n");
code.push_str(" buf");
code.push_str(" }\n");
code.push_str("}\n\n");
code
}

23
askama_codegen/src/lib.rs
View File

@ -7,13 +7,13 @@ extern crate proc_macro;
extern crate quote;
extern crate syn;
mod generator;
mod parser;
use proc_macro::TokenStream;
use std::fs::File;
use std::io::Read;
use std::path::{Path, PathBuf};
use std::str;
fn get_path_from_attrs(attrs: &Vec<syn::Attribute>) -> String {
for attr in attrs {
@ -68,24 +68,5 @@ pub fn derive_template(input: TokenStream) -> TokenStream {
let path = get_path_from_attrs(&ast.attrs);
let src = get_template_source(&path);
let tokens = parser::parse(&src);
let mut code = String::new();
code.push_str("impl askama::Template for ");
code.push_str(name.as_ref());
code.push_str(" {\n");
code.push_str(" fn render(&self) -> String {\n");
code.push_str(" let mut buf = String::new();\n");
code.push_str(" buf.push_str(\"");
code.push_str(str::from_utf8(tokens[0]).unwrap());
code.push_str("\");\n");
code.push_str(" buf.push_str(&self.");
code.push_str(str::from_utf8(tokens[1]).unwrap());
code.push_str(");\n");
code.push_str(" buf.push_str(\"");
code.push_str(str::from_utf8(tokens[2]).unwrap());
code.push_str("\");\n");
code.push_str(" buf");
code.push_str(" }\n");
code.push_str("}\n\n");
code.parse().unwrap()
generator::generate(name.as_ref(), &tokens).parse().unwrap()
}