nparse 0.0.10

Parser in rust for specific use-cases
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106

use std::io::{BufRead, BufReader};

use nom::{
    character::complete::{anychar, char},
    combinator::{eof, recognize},
    multi::{many0_count, many_till},
    sequence::pair,
    IResult,
};
use serde_json::{json, Value};
fn parse_line(i: &str) -> IResult<&str, (usize, &str)> {
    pair(many0_count(char(' ')), recognize(many_till(anychar, eof)))(i)
}

pub fn parse_multiline_kv_str(s: &str) -> Result<Value, String> {
    let mut obj = json!({});

    let mut keys = vec![];

    for line in BufReader::new(s.as_bytes()).lines().flatten() {
        if let Ok((_, (sp, content))) = parse_line(&line) {
            loop {
                match keys.last() {
                    Some(&(a, _)) if a >= sp => {
                        keys.pop().unwrap();
                    }
                    _ => break,
                }
            }

            let mut colon_index = 0;
            let mut content = content.to_string();

            for c in content.chars() {
                if c == ':' {
                    break;
                }
                colon_index += 1;
            }

            if colon_index < content.len() {
                let mut key = content.clone();
                content = key.split_off(colon_index + 1);
                keys.push((sp, key));
            }

            let mut splited_keys = keys.iter().flat_map(|x| x.1.split(':').next());

            let mut entry = obj
                .as_object_mut()
                .unwrap()
                .entry(splited_keys.next().unwrap());

            for key in splited_keys {
                let obj = entry.or_insert_with(|| json!({}));
                match obj {
                    Value::Array(ar) => {
                        if ar.first().map(|x| x.is_object()) != Some(true) {
                            ar.insert(0, json!({}));
                        }
                        entry = ar.first_mut().unwrap().as_object_mut().unwrap().entry(key);
                    }
                    Value::Object(map) => {
                        entry = map.entry(key);
                    }
                    Value::String(_) => {
                        *obj = json!([{}, obj]);
                        entry = obj
                            .as_array_mut()
                            .unwrap()
                            .first_mut()
                            .unwrap()
                            .as_object_mut()
                            .unwrap()
                            .entry(key);
                    }
                    _ => {
                        unreachable!()
                    }
                }
            }

            let obj = entry.or_insert_with(|| Value::Null);

            match obj.clone() {
                Value::Array(mut ar) => {
                    ar.push(json!(content.trim()));
                }
                Value::Object(map) => {
                    *obj = json!([map, content.trim()]);
                }
                Value::String(st) => {
                    *obj = json!([st, content.trim()]);
                }
                Value::Null => {
                    *obj = json!(content.trim());
                }
                _ => {
                    unreachable!()
                }
            }
        }
    }

    Ok(obj)
}