-
Notifications
You must be signed in to change notification settings - Fork 464
/
index.rs
69 lines (60 loc) · 2.24 KB
/
index.rs
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
use super::*;
use rust::cfg::*;
use serde::Serialize;
use std::collections::BTreeMap;
use windows_metadata::Item;
#[derive(Default, Serialize)]
struct Index {
namespace_map: Vec<String>,
feature_map: Vec<String>,
namespaces: BTreeMap<usize, Vec<IndexItem>>,
}
#[derive(Default, Serialize)]
struct IndexItem {
name: String,
features: Vec<usize>,
}
pub fn gen_index(writer: &Writer) -> String {
let mut feature_index = Index { ..Default::default() };
for namespace in writer.reader.namespaces() {
let namespace_idx = match feature_index.namespace_map.iter().position(|ns| ns == namespace) {
Some(idx) => idx,
None => {
feature_index.namespace_map.push(namespace.to_string());
feature_index.namespace_map.len() - 1
}
};
for item in writer.reader.namespace_items(namespace) {
let mut index_item = IndexItem { ..Default::default() };
let mut cfg = Cfg::default();
cfg.add_feature(namespace);
cfg = match item {
Item::Type(def) => {
index_item.name = def.name().to_string();
cfg.union(type_def_cfg(writer, def, &[]))
}
Item::Const(field) => {
index_item.name = field.name().to_string();
cfg.union(field_cfg(writer, field))
}
Item::Fn(method, _) => {
index_item.name = method.name().to_string();
cfg.union(signature_cfg(writer, method))
}
};
let cfg_features = cfg_features(&cfg);
index_item.features = cfg_features
.iter()
.map(|feature| match feature_index.feature_map.iter().position(|f| f == feature) {
Some(idx) => idx,
None => {
feature_index.feature_map.push(feature.to_string());
feature_index.feature_map.len() - 1
}
})
.collect();
feature_index.namespaces.entry(namespace_idx).or_default().push(index_item);
}
}
serde_json::to_string(&feature_index).unwrap()
}