1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66

use proc_macro2::{Ident, TokenStream};
use quote::{format_ident, quote};

use yarte_hir::{Struct, HIR};

use crate::CodeGen;

pub struct FmtCodeGen<'a, T: CodeGen> {
    codegen: T,
    s: &'a Struct<'a>,
    parent: Ident,
}

impl<'a, T: CodeGen> FmtCodeGen<'a, T> {
    pub fn new<'n>(codegen: T, s: &'n Struct, parent: &'static str) -> FmtCodeGen<'n, T> {
        FmtCodeGen {
            codegen,
            s,
            parent: format_ident!("{}", parent),
        }
    }

    #[inline]
    fn template(&self, size_hint: usize, tokens: &mut TokenStream) {
        tokens.extend(self.s.implement_head(
            quote!(yarte::TemplateTrait),
            &quote!(
            fn size_hint() -> usize {
                #size_hint
            }),
        ));
    }

    fn display(&mut self, nodes: Vec<HIR>, tokens: &mut TokenStream) -> usize {
        let nodes = self.codegen.gen(nodes);
        // heuristic based on https://github.com/lfairy/maud
        let size_hint = nodes.to_string().len();
        let parent = &self.parent;
        let func = quote!(
            fn fmt(&self, _fmt: &mut std::fmt::Formatter) -> std::fmt::Result {
                #[allow(unused_imports)]
                use std::fmt::Display;
                #[allow(unused_imports)]
                use #parent::*;
                #nodes
                Ok(())
            }
        );

        tokens.extend(self.s.implement_head(quote!(std::fmt::Display), &func));

        size_hint
    }
}

// TODO: Create new trait for decouple concepts
impl<'a, T: CodeGen> CodeGen for FmtCodeGen<'a, T> {
    fn gen(&mut self, v: Vec<HIR>) -> TokenStream {
        let mut tokens = TokenStream::new();

        let size_hint = self.display(v, &mut tokens);
        self.template(size_hint, &mut tokens);

        tokens
    }
}