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
107
108
109
110
use std::io::{BufReader, BufRead};
use serde_json::{json, Value};
use nom::{named, exact, none_of, many0_count, many_till, recognize, eof, pair, char};
named!(
parse_line<&str, (usize, &str)>,
exact!(pair!(
many0_count!(char!(' ')),
recognize!(many_till!(none_of!(""), eof!()))
))
);
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() {
if let Ok(line) = line {
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)
}