canic-core 0.25.2

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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

use super::{AttestationPublicKeyRecord, DelegationStateRecord, ShardPublicKeyRecord};
use crate::storage::prelude::*;

// Resolve the current root public key, if one is stored.
pub(super) fn get_root_public_key(data: &DelegationStateRecord) -> Option<Vec<u8>> {
    data.root_public_key.clone()
}

// Persist the current root public key.
pub(super) fn set_root_public_key(data: &mut DelegationStateRecord, public_key_sec1: Vec<u8>) {
    data.root_public_key = Some(public_key_sec1);
}

// Resolve one shard public key by shard principal.
pub(super) fn get_shard_public_key(
    data: &DelegationStateRecord,
    shard_pid: Principal,
) -> Option<Vec<u8>> {
    data.shard_public_keys
        .iter()
        .find(|entry| entry.shard_pid == shard_pid)
        .map(|entry| entry.public_key_sec1.clone())
}

// Persist or replace one shard public key by shard principal.
pub(super) fn set_shard_public_key(
    data: &mut DelegationStateRecord,
    shard_pid: Principal,
    public_key_sec1: Vec<u8>,
) {
    upsert_shard_public_key_record(&mut data.shard_public_keys, shard_pid, public_key_sec1);
}

// Resolve one attestation public key by key id.
pub(super) fn get_attestation_public_key(
    data: &DelegationStateRecord,
    key_id: u32,
) -> Option<AttestationPublicKeyRecord> {
    data.attestation_public_keys
        .iter()
        .find(|entry| entry.key_id == key_id)
        .cloned()
}

// Resolve the full attestation public key set.
pub(super) fn get_attestation_public_keys(
    data: &DelegationStateRecord,
) -> Vec<AttestationPublicKeyRecord> {
    data.attestation_public_keys.clone()
}

// Replace the attestation public key set.
pub(super) fn set_attestation_public_keys(
    data: &mut DelegationStateRecord,
    keys: Vec<AttestationPublicKeyRecord>,
) {
    data.attestation_public_keys = keys;
}

// Upsert one attestation public key by key id.
pub(super) fn upsert_attestation_public_key(
    data: &mut DelegationStateRecord,
    key: AttestationPublicKeyRecord,
) {
    if let Some(existing) = data
        .attestation_public_keys
        .iter_mut()
        .find(|entry| entry.key_id == key.key_id)
    {
        *existing = key;
    } else {
        data.attestation_public_keys.push(key);
    }
}

// Upsert one shard public key record by shard principal.
fn upsert_shard_public_key_record(
    entries: &mut Vec<ShardPublicKeyRecord>,
    shard_pid: Principal,
    public_key_sec1: Vec<u8>,
) {
    if let Some(entry) = entries
        .iter_mut()
        .find(|entry| entry.shard_pid == shard_pid)
    {
        entry.public_key_sec1 = public_key_sec1;
    } else {
        entries.push(ShardPublicKeyRecord {
            shard_pid,
            public_key_sec1,
        });
    }
}