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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
// For tests
#[allow(unused_imports)]
extern crate bytes;
#[allow(unused_imports)]
#[macro_use] extern crate fuel_line_derive;
#[allow(unused_imports)]
extern crate uuid;

pub trait Render {
  fn render(&self) -> String;
}

#[macro_export]
macro_rules! templatify {
  ( $head_template:expr $(;$key:expr; $template:expr)* ) => {
    {
      let mut total_length = 0;
      total_length = total_length + $head_template.len();

      $(
        total_length = total_length + $key.len() + $template.len();
      )*

      let mut output_string = String::with_capacity(total_length);
      output_string.push_str($head_template);

      $(
        output_string.push_str($key);
        output_string.push_str($template);
      )*

      output_string
    }
  }
}

#[macro_export]
macro_rules! templatify_buffer {
  ( $buffer:ident, $head_template:expr $(;$key:expr; $template:expr)* ) => {
    {
      let mut total_length = 0;
      total_length = total_length + $head_template.len();

      $(
        total_length = total_length + $key.len() + $template.len();
      )*

      $buffer.reserve(total_length);
      $buffer.put($head_template);

      $(
        $buffer.put($key);
        $buffer.put($template);
      )*
    }
  }
}

#[cfg(test)]
mod tests {
  use bytes::{BytesMut, BufMut};
  use Render;
  use uuid::Uuid;

  #[test]
  fn templatify_should_work() {
    let world = "world";
    let results: String = templatify! { "hello, "; world ;"!" };
    assert!(results == "hello, world!");
  }

  #[test]
  fn templatify_buffer_should_work() {
    let mut buf = BytesMut::new();

    let world = "world";
    templatify_buffer! { buf, "hello, "; world ;"!" };
    assert!(buf == "hello, world!");
  }

  #[test]
  fn render_derive_should_work() {

    #[derive(Render)]
    #[TemplateName = "./fuel_line/test_data/test.html"]
    struct TestStruct {
      a: String,
      b: String
    };

    let t = TestStruct {
      a: "a_value".to_owned(),
      b: "b_value".to_owned()
    };

    assert!(t.render() == "<h1>b_value</h1>\n<p>a_value</p>\n");
  }
}