mrps 0.1.0

Mini reverse proxy server
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

mod file;
mod parse;
mod fetch;
mod format;
mod command;

use file::IO;
use parse::parse;
use std::error::Error;
use command::command;
use std::path::PathBuf;
use format::{format, bytes};
use minijinja::{Environment, path_loader, Value};
use fetch::{get, delete, head, options, post, put, patch};

pub fn new(dir: PathBuf, data: Option<PathBuf>) -> Result<Environment<'static>, Box<dyn Error>> {
    let mut env = Environment::new();
    env.set_loader(path_loader(dir));
    env.add_filter("parse", parse);
    env.add_filter("bytes", bytes);
    env.add_filter("format", format);
    env.add_function("get", get);
    env.add_function("post", post);
    env.add_function("put", put);
    env.add_function("patch", patch);
    env.add_function("head", head);
    env.add_function("options", options);
    env.add_function("delete", delete);
    env.add_function("command", command);
    env.add_function("log", |message: &str| -> () {
        println!("{}", message);
        ()
    });
    if let Some(data) = data {
        let io1 = IO::new(data)?;
        let io2 = io1.clone();
        let io3 = io1.clone();
        env.add_function("read", move |entry: &str| -> Option<Value> {
            io1.read(entry)
        });
        env.add_function("write", move |
            file: &str,
            data: &Vec<u8>
        | -> Option<String> {
            io2.write(file, data)
        });
        env.add_function("remove", move |entry: &str| -> Option<String> {
            io3.remove(entry)
        });
    }

    Ok(env)
}