toolcraft 0.1.3

A modular Rust toolkit
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
42
43
44
45
46
47
48
49
50
51

use serde_json::{Map, Value, json};

use crate::jinjia2::parser::JinjaNode;

pub(crate) fn ast_to_json(ast: &[JinjaNode]) -> Value {
    let mut root = Map::new();

    for node in ast {
        merge_value(&mut root, node);
    }

    Value::Object(root)
}

fn merge_value(target: &mut Map<String, Value>, node: &JinjaNode) {
    match node {
        JinjaNode::Variable { path } => {
            insert_path(target, path);
        }
        JinjaNode::ForLoop { iterable, body, .. } => {
            let mut item_obj = Map::new();
            for item in body {
                merge_value(&mut item_obj, item);
            }
            target.insert(
                iterable.clone(),
                Value::Array(vec![Value::Object(item_obj)]),
            );
        }
    }
}

fn insert_path(target: &mut Map<String, Value>, path: &[String]) {
    if path.is_empty() {
        return;
    }

    let mut current = target;

    for (i, key) in path.iter().enumerate() {
        if i == path.len() - 1 {
            current.insert(key.clone(), json!(""));
        } else {
            current = current
                .entry(key)
                .or_insert_with(|| Value::Object(Map::new()))
                .as_object_mut()
                .unwrap();
        }
    }
}