ic-query 0.80.0

Internet Computer query library for CloudEngine, NNS, SNS, ICRC, system canisters, and public network metadata
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

use super::{ResolveAs, ResolvedSubnet};
use crate::identifier_match::{IdentifierMatch, match_identifier};
use crate::subnet_catalog::{CatalogError, RawSubnetCatalog, parse_principal};
use std::collections::BTreeMap;

impl RawSubnetCatalog {
    /// Resolve an exact principal or a unique cached subnet principal prefix.
    pub fn resolve_principal_or_prefix(
        &self,
        input: &str,
        forced: Option<ResolveAs>,
    ) -> Result<ResolvedSubnet, CatalogError> {
        if let Ok(principal) = parse_principal(input, "principal") {
            return self.resolve_parsed_principal(&principal, forced);
        }
        self.resolve_principal_prefix(input, forced)
    }

    /// Resolve a unique prefix of a cached subnet principal.
    pub fn resolve_principal_prefix(
        &self,
        input: &str,
        forced: Option<ResolveAs>,
    ) -> Result<ResolvedSubnet, CatalogError> {
        let prefix = input.trim().to_ascii_lowercase();
        if prefix.is_empty() {
            return Err(CatalogError::PrincipalPrefixNotFound { prefix });
        }

        let mut identifiers = BTreeMap::new();
        if forced != Some(ResolveAs::Canister) {
            for subnet in &self.subnets {
                identifiers
                    .entry(subnet.subnet_principal.as_str())
                    .or_insert(subnet);
            }
        }
        let matched = match match_identifier(
            identifiers.into_values(),
            |subnet| subnet.subnet_principal.as_str(),
            None,
            &prefix,
        ) {
            IdentifierMatch::Prefix(subnet) => subnet,
            IdentifierMatch::Missing => {
                return Err(CatalogError::PrincipalPrefixNotFound { prefix });
            }
            IdentifierMatch::Ambiguous(matches) => {
                return Err(CatalogError::AmbiguousPrincipalPrefix {
                    prefix,
                    matches: matches
                        .into_iter()
                        .map(|subnet| format!("subnet:{}", subnet.subnet_principal))
                        .collect(),
                });
            }
            IdentifierMatch::Exact(_) => {
                unreachable!("prefix lookup supplies no exact identifier")
            }
        };
        let mut resolved = self.resolved_subnet(&matched.subnet_principal, matched);
        resolved.input_principal = input.to_string();
        resolved.resolved_from = "subnet_principal_prefix".to_string();
        Ok(resolved)
    }
}