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

use std::{fmt, fs};

use yarte_config::{read_config_file, Config};
pub use yarte_derive::*;
pub use yarte_helpers::{helpers::MarkupAsStr, Error, Result};

// TODO: document
pub trait Template: fmt::Display {
    fn call(&self) -> Result<String> {
        let mut buf = String::with_capacity(Self::size_hint());
        self.call_into(&mut buf)?;
        Ok(buf)
    }
    fn call_into(&self, writer: &mut fmt::Write) -> fmt::Result {
        write!(writer, "{}", self)
    }

    fn mime() -> &'static str
    where
        Self: Sized;

    fn size_hint() -> usize;
}

#[cfg(feature = "with-actix-web")]
pub mod actix_web {
    pub use actix_web::{
        error::ErrorInternalServerError, Error, HttpRequest, HttpResponse, Responder,
    };
}

// TODO:
pub fn rerun_if_templates_changed() {
    let file = read_config_file();
    let mut stack = Config::new(&file).dirs;
    loop {
        if let Some(dir) = stack.pop() {
            for entry in fs::read_dir(dir).expect("valid directory") {
                let entry = entry.expect("valid directory entry");
                let path = entry.path();
                if path.is_dir() {
                    stack.push(path);
                } else {
                    println!("cargo:rerun-if-changed={}", entry.path().to_str().unwrap());
                }
            }
        } else {
            break;
        }
    }
}