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
|
use super::*;
#[derive(Debug)]
pub struct Tree {
pub namespace: &'static str,
pub nested: std::collections::BTreeMap<&'static str, Tree>,
}
impl Tree {
pub fn new(reader: &'static metadata::Reader) -> Self {
let mut tree = Tree::from_namespace("");
for ns in reader.namespaces() {
if reader.includes_namespace(ns) {
tree.insert_namespace(ns, 0);
}
}
tree
}
fn from_namespace(namespace: &'static str) -> Self {
Self {
namespace,
nested: std::collections::BTreeMap::new(),
}
}
fn insert_namespace(&mut self, namespace: &'static str, pos: usize) -> &mut Self {
if let Some(next) = namespace[pos..].find('.') {
let next = pos + next;
self.nested
.entry(&namespace[pos..next])
.or_insert_with(|| Self::from_namespace(&namespace[..next]))
.insert_namespace(namespace, next + 1)
} else {
self.nested
.entry(&namespace[pos..])
.or_insert_with(|| Self::from_namespace(namespace))
}
}
pub fn flatten(&self) -> Vec<&Self> {
let mut flatten = if self.namespace.is_empty() {
vec![]
} else {
vec![self]
};
flatten.extend(self.nested.values().flat_map(|tree| tree.flatten()));
flatten
}
}
|
