From 2d5c2d2f1098471dfbb9e20aeb02f72126659c71 Mon Sep 17 00:00:00 2001
From: Dirkjan Ochtman <dirkjan@ochtman.nl>
Date: Tue, 7 Mar 2017 21:34:09 +0100
Subject: Extract writing of impl headers in generator
---
askama_derive/src/generator.rs | 24 ++++++++++++++----------
1 file changed, 14 insertions(+), 10 deletions(-)
(limited to 'askama_derive/src')
diff --git a/askama_derive/src/generator.rs b/askama_derive/src/generator.rs
index 2457c26..5b7026d 100644
--- a/askama_derive/src/generator.rs
+++ b/askama_derive/src/generator.rs
@@ -319,11 +319,20 @@ impl<'a> Generator<'a> {
}
}
- fn impl_template(&mut self, ast: &syn::DeriveInput, nodes: &'a [Node]) {
+ fn write_header(&mut self, ast: &syn::DeriveInput,
+ trait_suffix: Option<&str>) {
let anno = annotations(&ast.generics);
- self.writeln(&format!("impl{} askama::Template for {}{} {{",
- anno, ast.ident.as_ref(), anno));
+ let name = if trait_suffix.is_some() {
+ format!("TraitFrom{}", trait_suffix.unwrap())
+ } else {
+ "askama::Template".to_string()
+ };
+ self.writeln(&format!("impl{} {} for {}{} {{",
+ anno, &name, ast.ident.as_ref(), anno));
+ }
+ fn impl_template(&mut self, ast: &syn::DeriveInput, nodes: &'a [Node]) {
+ self.write_header(ast, None);
self.writeln("fn render_to(&self, writer: &mut std::fmt::Write) {");
self.handle(nodes);
self.flush_ws(&WS(false, false));
@@ -333,10 +342,7 @@ impl<'a> Generator<'a> {
fn impl_trait(&mut self, ast: &syn::DeriveInput, base: &str,
blocks: &'a [Node], nodes: Option<&'a [Node]>) {
- let anno = annotations(&ast.generics);
- self.writeln(&format!("impl{} TraitFrom{} for {}{} {{",
- anno, path_as_identifier(base),
- ast.ident.as_ref(), anno));
+ self.write_header(ast, Some(&path_as_identifier(base)));
self.handle(blocks);
self.writeln("#[allow(unused_variables)]");
@@ -358,9 +364,7 @@ impl<'a> Generator<'a> {
}
fn impl_template_for_trait(&mut self, ast: &syn::DeriveInput, derived: bool) {
- let anno = annotations(&ast.generics);
- self.writeln(&format!("impl{} askama::Template for {}{} {{",
- anno, ast.ident.as_ref(), anno));
+ self.write_header(ast, None);
self.writeln("fn render_to(&self, writer: &mut std::fmt::Write) {");
if derived {
self.writeln("self._parent.render_trait_to(self, writer);");
--
cgit