provenance-cli 0.2.1

Command-line interface for Provenance requirements traceability
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

use super::references::node_type_name;
use provenance_core::{NodeType, ScopeId, StableId};
use std::collections::BTreeSet;

#[derive(Default)]
pub(super) struct CheckIndex {
    global_nodes: BTreeSet<(String, String)>,
    scoped_nodes: BTreeSet<(String, String, String)>,
}

impl CheckIndex {
    pub(super) fn add_node(&mut self, scope_id: &ScopeId, node_type: &str, id: &StableId) {
        let node_type = node_type.to_string();
        let id = id.as_str().to_string();
        self.global_nodes.insert((node_type.clone(), id.clone()));
        self.scoped_nodes
            .insert((scope_id.as_str().to_string(), node_type, id));
    }

    pub(super) fn has_global_node(&self, node_type: NodeType, id: &StableId) -> bool {
        self.global_nodes.contains(&(
            node_type_name(node_type).to_string(),
            id.as_str().to_string(),
        ))
    }

    pub(super) fn has_scoped_node(
        &self,
        scope_id: &ScopeId,
        node_type: &str,
        id: &StableId,
    ) -> bool {
        self.scoped_nodes.contains(&(
            scope_id.as_str().to_string(),
            node_type.to_string(),
            id.as_str().to_string(),
        ))
    }
}