dge-gen 0.2.0

Crate for building a computational graph to be executed distributely
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
37
38
39
40
41

use askama::Template;
use heck::CamelCase;

use crate::Result;

struct Command {
    module: String,
    variant: String,
}

#[derive(Template)]
#[template(path = "main.rs", escape = "none")]
struct MainTemplate {
    commands: Vec<Command>,
    setup_logger: String,
}

pub(crate) fn generate<S: AsRef<str>>(
    modules: Vec<String>,
    setup_logger: S,
) -> Result<String> {
    let setup_logger = setup_logger.as_ref();

    let mut commands = Vec::new();
    for module in modules {
        let variant = &module.to_camel_case();
        commands.push(Command {
            module,
            variant: variant.clone()
        })
    }

    let template = MainTemplate {
        commands,
        setup_logger: String::from(setup_logger),
    };

    let generated = template.render()?;

    Ok(generated)
}