ic-query 0.59.0

Internet Computer query library for CloudEngine, NNS, SNS, ICRC, system canisters, and public network metadata
Documentation
use super::fetch_topology_inventory_reports;
use crate::nns::{
    LiveNnsSource,
    topology::report::{
        NnsTopologyCapacityReport, NnsTopologyGapsReport, NnsTopologyHostError,
        NnsTopologyProvidersReport, NnsTopologyReadRequest, NnsTopologyRegionsReport,
        NnsTopologySource, capacity::topology_capacity_report_from_report, enforce_mainnet_network,
        gaps::topology_gaps_report_from_reports, providers::topology_providers_report_from_reports,
        regions::topology_regions_report_from_report,
    },
};

pub async fn build_nns_topology_gaps_report(
    request: &NnsTopologyReadRequest,
) -> Result<NnsTopologyGapsReport, NnsTopologyHostError> {
    build_nns_topology_gaps_report_with_source(request, &LiveNnsSource::default()).await
}

pub async fn build_nns_topology_gaps_report_with_source(
    request: &NnsTopologyReadRequest,
    source: &dyn NnsTopologySource,
) -> Result<NnsTopologyGapsReport, NnsTopologyHostError> {
    enforce_mainnet_network(request.network.as_str())?;

    let reports = fetch_topology_inventory_reports(request, source).await?;

    Ok(topology_gaps_report_from_reports(
        request.network.clone(),
        request.source_endpoint.clone(),
        reports.node,
        reports.node_provider,
        reports.node_operator,
        reports.data_center,
    ))
}

pub async fn build_nns_topology_capacity_report(
    request: &NnsTopologyReadRequest,
) -> Result<NnsTopologyCapacityReport, NnsTopologyHostError> {
    build_nns_topology_capacity_report_with_source(request, &LiveNnsSource::default()).await
}

pub async fn build_nns_topology_capacity_report_with_source(
    request: &NnsTopologyReadRequest,
    source: &dyn NnsTopologySource,
) -> Result<NnsTopologyCapacityReport, NnsTopologyHostError> {
    enforce_mainnet_network(request.network.as_str())?;

    let node_operator_report = source.fetch_node_operator_list_report(request).await?;

    Ok(topology_capacity_report_from_report(
        request.network.clone(),
        request.source_endpoint.clone(),
        node_operator_report,
    ))
}

pub async fn build_nns_topology_regions_report(
    request: &NnsTopologyReadRequest,
) -> Result<NnsTopologyRegionsReport, NnsTopologyHostError> {
    build_nns_topology_regions_report_with_source(request, &LiveNnsSource::default()).await
}

pub async fn build_nns_topology_regions_report_with_source(
    request: &NnsTopologyReadRequest,
    source: &dyn NnsTopologySource,
) -> Result<NnsTopologyRegionsReport, NnsTopologyHostError> {
    enforce_mainnet_network(request.network.as_str())?;

    let data_center_report = source.fetch_data_center_list_report(request).await?;

    Ok(topology_regions_report_from_report(
        request.network.clone(),
        request.source_endpoint.clone(),
        data_center_report,
    ))
}

pub async fn build_nns_topology_providers_report(
    request: &NnsTopologyReadRequest,
) -> Result<NnsTopologyProvidersReport, NnsTopologyHostError> {
    build_nns_topology_providers_report_with_source(request, &LiveNnsSource::default()).await
}

pub async fn build_nns_topology_providers_report_with_source(
    request: &NnsTopologyReadRequest,
    source: &dyn NnsTopologySource,
) -> Result<NnsTopologyProvidersReport, NnsTopologyHostError> {
    enforce_mainnet_network(request.network.as_str())?;

    let reports = fetch_topology_inventory_reports(request, source).await?;

    Ok(topology_providers_report_from_reports(
        request.network.clone(),
        request.source_endpoint.clone(),
        reports.node,
        reports.node_provider,
        reports.node_operator,
        reports.data_center,
    ))
}