nitr 0.0.0-beta.0

A dynamic web server written in Rust with Lua scripting support.
Documentation 
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

use minijinja::{path_loader, Environment};
use mlua::{AnyUserData, ExternalResult, Lua, LuaSerdeExt, Table, UserData, UserDataMethods};
use std::sync::Arc;

use crate::error::{Context, Result};

pub(crate) struct LuaTemplate<'a>(Arc<Environment<'a>>);

impl UserData for LuaTemplate<'_> {
    fn add_methods<'lua, M: UserDataMethods<Self>>(methods: &mut M) {
        methods.add_method_mut("render", |lua, templ, args: (String, Option<Table>)| {
            let file_path = args.0;
            let data = args.1;
            let templ_store = templ.0.clone();
            let templ = templ_store
                .get_template(file_path.as_str())
                .into_lua_err()?;
            let content = templ
                .render(data)
                .with_context(|| "template_fn: error rendering template")
                .into_lua_err()?;
            lua.to_value(&content)
        });
    }
}

/// Templating function support.
pub(crate) fn create_template_fn(lua: &Lua) -> Result<AnyUserData> {
    let mut env = Environment::new();
    // TODO: add custom config for templates directory
    env.set_loader(path_loader("scripts/templates"));

    let env = Arc::new(env);
    lua.create_userdata(LuaTemplate(env))
        .with_context(|| "template_fn: failed to initialize templates")
}