yaak-mountain-loop/src-tauri/templates/src/renderer.rs

use std::collections::HashMap;

use crate::{Parser, Token, Val};

type TemplateCallback = fn(name: &str, args: Vec<String>) -> String;

pub fn parse_and_render(
    template: &str,
    vars: HashMap<&str, &str>,
    cb: Option<TemplateCallback>,
) -> String {
    let mut p = Parser::new(template);
    let tokens = p.parse();
    render(tokens, vars, cb)
}

pub fn render(
    tokens: Vec<Token>,
    vars: HashMap<&str, &str>,
    cb: Option<TemplateCallback>,
) -> String {
    let mut doc_str: Vec<String> = Vec::new();

    for t in tokens {
        match t {
            Token::Raw(s) => doc_str.push(s),
            Token::Tag(val) => doc_str.push(render_tag(val, vars.clone(), cb)),
            Token::Eof => {}
        }
    }

    return doc_str.join("");
}

fn render_tag<'s>(
    val: Val,
    vars: HashMap<&'s str, &'s str>,
    cb: Option<TemplateCallback>,
) -> String {
    match val {
        Val::Str(s) => s.into(),
        Val::Var(name) => match vars.get(name.as_str()) {
            Some(v) => v.to_string(),
            None => "".into(),
        },
        Val::Fn { name, args } => {
            let empty = &"";
            let resolved_args = args
                .iter()
                .map(|a| match a {
                    Val::Str(s) => s.to_string(),
                    Val::Var(i) => vars.get(i.as_str()).unwrap_or(empty).to_string(),
                    val => render_tag(val.clone(), vars.clone(), cb),
                })
                .collect::<Vec<String>>();
            match cb {
                Some(cb) => cb(name.as_str(), resolved_args),
                None => "".into(),
            }
        }
    }
}

#[cfg(test)]
mod tests {
    use std::collections::HashMap;

    use crate::*;

    #[test]
    fn render_empty() {
        let template = "";
        let vars = HashMap::new();
        let result = "";
        assert_eq!(parse_and_render(template, vars, None), result.to_string());
    }

    #[test]
    fn render_text_only() {
        let template = "Hello World!";
        let vars = HashMap::new();
        let result = "Hello World!";
        assert_eq!(parse_and_render(template, vars, None), result.to_string());
    }

    #[test]
    fn render_simple() {
        let template = "${[ foo ]}";
        let vars = HashMap::from([("foo", "bar")]);
        let result = "bar";
        assert_eq!(parse_and_render(template, vars, None), result.to_string());
    }

    #[test]
    fn render_surrounded() {
        let template = "hello ${[ word ]} world!";
        let vars = HashMap::from([("word", "cruel")]);
        let result = "hello cruel world!";
        assert_eq!(parse_and_render(template, vars, None), result.to_string());
    }

    #[test]
    fn render_valid_fn() {
        let vars = HashMap::new();
        let template = r#"${[ say_hello("John", "Kate") ]}"#;
        let result = r#"say_hello: ["John", "Kate"]"#;

        fn cb(name: &str, args: Vec<String>) -> String {
            format!("{name}: {:?}", args)
        }
        assert_eq!(parse_and_render(template, vars, Some(cb)), result);
    }

    #[test]
    fn render_nested_fn() {
        let vars = HashMap::new();
        let template = r#"${[ upper(secret()) ]}"#;
        let result = r#"ABC"#;
        fn cb(name: &str, args: Vec<String>) -> String {
            match name {
                "secret" => "abc".to_string(),
                "upper" => args[0].to_string().to_uppercase(),
                _ => "".to_string(),
            }
        }

        assert_eq!(
            parse_and_render(template, vars, Some(cb)),
            result.to_string()
        );
    }
}