canic_backup/discovery/
mod.rs

1use crate::{
2    manifest::{FleetMember, FleetSection, IdentityMode, SourceSnapshot, VerificationCheck},
3    topology::{TopologyHasher, TopologyRecord},
4};
5use std::collections::BTreeSet;
6use thiserror::Error as ThisError;
7
8///
9/// DiscoveredFleet
10///
11
12#[derive(Clone, Debug)]
13pub struct DiscoveredFleet {
14    pub topology_records: Vec<TopologyRecord>,
15    pub members: Vec<DiscoveredMember>,
16}
17
18impl DiscoveredFleet {
19    /// Convert discovered topology and member policy into a manifest fleet section.
20    pub fn into_fleet_section(self) -> Result<FleetSection, DiscoveryError> {
21        validate_discovered_members(&self.members)?;
22
23        let topology_hash = TopologyHasher::hash(&self.topology_records);
24        let members = self
25            .members
26            .into_iter()
27            .map(DiscoveredMember::into_fleet_member)
28            .collect();
29
30        Ok(FleetSection {
31            topology_hash_algorithm: topology_hash.algorithm,
32            topology_hash_input: topology_hash.input,
33            discovery_topology_hash: topology_hash.hash.clone(),
34            pre_snapshot_topology_hash: topology_hash.hash.clone(),
35            topology_hash: topology_hash.hash,
36            members,
37        })
38    }
39}
40
41///
42/// DiscoveredMember
43///
44
45#[derive(Clone, Debug)]
46pub struct DiscoveredMember {
47    pub role: String,
48    pub canister_id: String,
49    pub parent_canister_id: Option<String>,
50    pub subnet_canister_id: Option<String>,
51    pub controller_hint: Option<String>,
52    pub identity_mode: IdentityMode,
53    pub restore_group: u16,
54    pub verification_class: String,
55    pub verification_checks: Vec<VerificationCheck>,
56    pub snapshot_plan: SnapshotPlan,
57}
58
59impl DiscoveredMember {
60    /// Project this discovery member into the manifest restore contract.
61    fn into_fleet_member(self) -> FleetMember {
62        FleetMember {
63            role: self.role,
64            canister_id: self.canister_id,
65            parent_canister_id: self.parent_canister_id,
66            subnet_canister_id: self.subnet_canister_id,
67            controller_hint: self.controller_hint,
68            identity_mode: self.identity_mode,
69            restore_group: self.restore_group,
70            verification_class: self.verification_class,
71            verification_checks: self.verification_checks,
72            source_snapshot: SourceSnapshot {
73                snapshot_id: self.snapshot_plan.snapshot_id,
74                module_hash: self.snapshot_plan.module_hash,
75                wasm_hash: self.snapshot_plan.wasm_hash,
76                code_version: self.snapshot_plan.code_version,
77                artifact_path: self.snapshot_plan.artifact_path,
78                checksum_algorithm: self.snapshot_plan.checksum_algorithm,
79                checksum: self.snapshot_plan.checksum,
80            },
81        }
82    }
83}
84
85///
86/// SnapshotPlan
87///
88
89#[derive(Clone, Debug)]
90pub struct SnapshotPlan {
91    pub snapshot_id: String,
92    pub module_hash: Option<String>,
93    pub wasm_hash: Option<String>,
94    pub code_version: Option<String>,
95    pub artifact_path: String,
96    pub checksum_algorithm: String,
97    pub checksum: Option<String>,
98}
99
100///
101/// DiscoveryError
102///
103
104#[derive(Debug, ThisError)]
105pub enum DiscoveryError {
106    #[error("discovered fleet has no members")]
107    EmptyFleet,
108
109    #[error("duplicate discovered canister id {0}")]
110    DuplicateCanisterId(String),
111
112    #[error("discovered member {0} has no verification checks")]
113    MissingVerificationChecks(String),
114}
115
116// Validate discovery output before building a manifest projection.
117fn validate_discovered_members(members: &[DiscoveredMember]) -> Result<(), DiscoveryError> {
118    if members.is_empty() {
119        return Err(DiscoveryError::EmptyFleet);
120    }
121
122    let mut canister_ids = BTreeSet::new();
123    for member in members {
124        if !canister_ids.insert(member.canister_id.clone()) {
125            return Err(DiscoveryError::DuplicateCanisterId(
126                member.canister_id.clone(),
127            ));
128        }
129        if member.verification_checks.is_empty() {
130            return Err(DiscoveryError::MissingVerificationChecks(
131                member.canister_id.clone(),
132            ));
133        }
134    }
135
136    Ok(())
137}
138
139#[cfg(test)]
140mod tests;
canic_backup/discovery/mod.rs

canic_backup/discovery/
mod.rs
