beetry-editor-types 0.2.0

Internal beetry crate. For the public API, check the beetry crate.
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
107
108
109
110
111
112
113
114
115
116
117
118

use std::collections::BTreeMap;

use getset::{CopyGetters, Getters};
use indexmap::IndexSet;
use schemars::JsonSchema;
use serde::{Deserialize, Serialize};

use crate::{
    id::{NodeId, NodeSpecId},
    spec::node::NodeSpecKey,
};

#[derive(Debug, Serialize, Deserialize, JsonSchema)]
pub struct Store {
    pub specs: SpecStore,
    pub nodes: RecordStore,
}

// The remaining parts of spec are to be loaded by the appropriate plugin.
#[derive(Debug, Serialize, Deserialize, JsonSchema)]
pub struct SpecStore {
    store: BTreeMap<NodeSpecId, NodeSpecKey>,
}

impl FromIterator<(NodeSpecId, NodeSpecKey)> for SpecStore {
    fn from_iter<T: IntoIterator<Item = (NodeSpecId, NodeSpecKey)>>(iter: T) -> Self {
        Self {
            store: iter.into_iter().collect(),
        }
    }
}

impl SpecStore {
    pub fn get(&self, id: &NodeSpecId) -> Option<&NodeSpecKey> {
        self.store.get(id)
    }

    pub fn iter(&self) -> impl Iterator<Item = (&NodeSpecId, &NodeSpecKey)> {
        self.store.iter()
    }

    pub fn values(&self) -> impl Iterator<Item = &NodeSpecKey> {
        self.store.values()
    }
}

#[derive(Debug, Serialize, Deserialize, JsonSchema)]
pub struct RecordStore {
    store: BTreeMap<NodeId, RecordValue>,
}

impl FromIterator<(NodeId, RecordValue)> for RecordStore {
    fn from_iter<T: IntoIterator<Item = (NodeId, RecordValue)>>(iter: T) -> Self {
        Self {
            store: iter.into_iter().collect(),
        }
    }
}

impl RecordStore {
    pub fn get(&self, id: &NodeId) -> Option<&RecordValue> {
        self.store.get(id)
    }

    pub fn remove(&mut self, id: &NodeId) -> Option<RecordValue> {
        self.store.remove(id)
    }

    pub fn iter(&self) -> impl Iterator<Item = RecordView<'_>> {
        self.store
            .iter()
            .map(|(id, value)| RecordView { id, value })
    }

    pub fn values(&self) -> impl Iterator<Item = &RecordValue> {
        self.store.values()
    }

    pub fn into_records(self) -> impl Iterator<Item = Record> {
        self.store
            .into_iter()
            .map(|(id, value)| Record { id, value })
    }
}

pub struct Record {
    pub id: NodeId,
    pub value: RecordValue,
}

#[derive(Debug, Getters, CopyGetters)]
pub struct RecordView<'a> {
    #[getset(get_copy = "pub")]
    pub id: &'a NodeId,
    #[getset(get_copy = "pub")]
    pub value: &'a RecordValue,
}

#[derive(Debug, Clone, Getters, CopyGetters, Serialize, Deserialize, JsonSchema)]
pub struct RecordValue {
    #[getset(get_copy = "pub")]
    spec_id: NodeSpecId,
    #[schemars(with = "Vec<NodeId>")]
    children: IndexSet<NodeId>,
}

impl RecordValue {
    pub fn new(spec_id: NodeSpecId, children: impl IntoIterator<Item = NodeId>) -> Self {
        Self {
            spec_id,
            children: children.into_iter().collect(),
        }
    }

    pub fn children(&self) -> impl Iterator<Item = &NodeId> {
        self.children.iter()
    }
}