canic-core 0.26.3

Canic — a canister orchestration and management toolkit for the Internet Computer
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

use crate::{
    InternalError,
    dto::topology::SubnetDirectoryArgs,
    ops::storage::directory::mapper::SubnetDirectoryRecordMapper,
    ops::storage::directory::{ensure_required_roles, ensure_unique_roles},
    ops::{config::ConfigOps, prelude::*},
    storage::stable::directory::subnet::{SubnetDirectory, SubnetDirectoryRecord},
};

///
/// SubnetDirectoryOps
///

pub struct SubnetDirectoryOps;

impl SubnetDirectoryOps {
    // -------------------------------------------------------------
    // Getters
    // -------------------------------------------------------------

    #[must_use]
    pub fn get(role: &CanisterRole) -> Option<Principal> {
        // This is still an ops-level convenience, but it stays data-based
        // and does not leak DTOs.
        SubnetDirectory::export()
            .entries
            .iter()
            .find_map(|(r, pid)| (r == role).then_some(*pid))
    }

    // -------------------------------------------------------------
    // Snapshot
    // -------------------------------------------------------------

    #[must_use]
    pub fn data() -> SubnetDirectoryRecord {
        SubnetDirectory::export()
    }

    #[must_use]
    pub fn snapshot_args() -> SubnetDirectoryArgs {
        SubnetDirectoryRecordMapper::record_to_view(SubnetDirectory::export())
    }

    pub(crate) fn import_args_allow_incomplete(
        args: SubnetDirectoryArgs,
    ) -> Result<(), InternalError> {
        let data = SubnetDirectoryRecordMapper::dto_to_record(args);
        Self::import_allow_incomplete(data)
    }

    // -------------------------------------------------------------
    // Import
    // -------------------------------------------------------------

    /// Import data into stable storage.
    pub fn import(data: SubnetDirectoryRecord) -> Result<(), InternalError> {
        ensure_unique_roles(&data.entries, "subnet")?;
        let subnet_cfg = ConfigOps::current_subnet()?;
        ensure_required_roles(&data.entries, "subnet", &subnet_cfg.subnet_directory)?;
        SubnetDirectory::import(data);

        Ok(())
    }

    pub(crate) fn import_allow_incomplete(
        data: SubnetDirectoryRecord,
    ) -> Result<(), InternalError> {
        ensure_unique_roles(&data.entries, "subnet")?;
        SubnetDirectory::import(data);

        Ok(())
    }
}