canic-host 0.100.78

//! Module: subnet_registry
//!
//! Responsibility: query one root canister's maintained subnet registry.
//! Does not own: registry state, endpoint DTOs, or deployment discovery policy.
//! Boundary: selects one transport and returns validated canonical host entries.

use crate::{
    icp::{IcpCli, IcpCommandError},
    registry::{RegistryEntry, RegistryParseError, parse_registry_entries},
    replica_query::{self, ReplicaQueryError},
};
use std::path::Path;
use thiserror::Error as ThisError;

const CANIC_SUBNET_REGISTRY_METHOD: &str = "canic_subnet_registry";
const ICP_JSON_OUTPUT: &str = "json";

///
/// SubnetRegistryQueryError
///

#[derive(Debug, ThisError)]
pub enum SubnetRegistryQueryError {
    #[error(transparent)]
    Icp(#[from] IcpCommandError),

    #[error(transparent)]
    Registry(#[from] RegistryParseError),

    #[error(transparent)]
    Replica(#[from] ReplicaQueryError),
}

/// Query `canic_subnet_registry`, using the local replica API for local targets.
pub fn query_subnet_registry(
    icp: &IcpCli,
    root: &str,
    environment: &str,
    icp_root: Option<&Path>,
    candid_path: Option<&Path>,
) -> Result<Vec<RegistryEntry>, SubnetRegistryQueryError> {
    if replica_query::should_use_local_replica_query(Some(environment)) {
        return query_local_subnet_registry(root, environment, icp_root);
    }

    let output = icp.canister_query_output_with_candid(
        root,
        CANIC_SUBNET_REGISTRY_METHOD,
        Some(ICP_JSON_OUTPUT),
        candid_path,
    )?;
    Ok(parse_registry_entries(&output)?)
}

fn query_local_subnet_registry(
    root: &str,
    environment: &str,
    icp_root: Option<&Path>,
) -> Result<Vec<RegistryEntry>, SubnetRegistryQueryError> {
    Ok(replica_query::query_subnet_registry_entries(
        Some(environment),
        root,
        icp_root,
    )?)
}