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

use std::sync::{Arc, RwLock};

#[derive(Clone)]
pub struct Template {
    template_dir: String,
    tera: Arc<RwLock<tera::Tera>>,
}

impl Template {
    #[must_use]
    pub fn new(template_dir: &str) -> Self {
        let path = format!("{template_dir}/**/*");
        let tera = match tera::Tera::new(&path) {
            Ok(tera) => tera,
            Err(err) => panic!("Parsing error(s): {err}"),
        };

        Self {
            tera: Arc::new(RwLock::new(tera)),
            template_dir: template_dir.to_string(),
        }
    }

    pub fn render(&self, template: &str, context: &tera::Context) -> tera::Result<String> {
        let tera = self.tera.read().unwrap();

        tera.render(template, context)
    }

    #[cfg(debug_assertions)]
    pub fn watch(self) {
        use notify::Watcher;

        let mut watcher = notify::recommended_watcher(move |res| {
            match res {
                Ok(event) => {
                    log::info!("reloading templates: {event:?}");

                    let mut tera = self.tera.write().unwrap();

                    match tera.full_reload() {
                        Ok(_) => log::info!("templates reloaded"),
                        Err(e) => log::error!("failed to reload templates: {e}"),
                    }
                }
                Err(e) => log::warn!("watch error: {e:?}"),
            }
        }).unwrap();

        log::debug!("watching {} for changes", self.template_dir);

        watcher
            .watch(&std::path::PathBuf::from(&self.template_dir), notify::RecursiveMode::Recursive)
            .unwrap();
    }

    pub fn register_function<F: tera::Function + 'static>(&mut self, name: &str, function: F) {
        let mut tera = self.tera.write().unwrap();

        tera.register_function(name, function);
    }

    pub fn register_filter<F: tera::Filter + 'static>(&mut self, name: &str, filter: F) {
        let mut tera = self.tera.write().unwrap();

        tera.register_filter(name, filter);
    }

    pub fn register_tester<T: tera::Test + 'static>(&mut self, name: &str, tester: T) {
        let mut tera = self.tera.write().unwrap();

        tera.register_tester(name, tester);
    }

    #[cfg(not(debug_assertions))]
    pub fn watch(self) {}
}