use super::{
error::NnsRegistryHostError,
model::{
NNS_REGISTRY_VERSION_REPORT_SCHEMA_VERSION, NnsRegistryVersionReport,
NnsRegistryVersionRequest,
},
source::{LiveNnsRegistrySource, NnsRegistrySource},
};
use crate::{
ic_registry::{MainnetRegistryFetchRequest, MainnetRegistryVersion},
subnet_catalog::format_utc_timestamp_secs,
};
impl_nns_mainnet_network_enforcer!(NnsRegistryHostError);
pub fn build_nns_registry_version_report(
request: &NnsRegistryVersionRequest,
) -> Result<NnsRegistryVersionReport, NnsRegistryHostError> {
build_nns_registry_version_report_with_source(request, &LiveNnsRegistrySource)
}
pub(super) fn build_nns_registry_version_report_with_source(
request: &NnsRegistryVersionRequest,
source: &dyn NnsRegistrySource,
) -> Result<NnsRegistryVersionReport, NnsRegistryHostError> {
enforce_mainnet_network(&request.network)?;
let fetched_at = format_utc_timestamp_secs(request.now_unix_secs);
let mut fetch_request = MainnetRegistryFetchRequest::new(fetched_at);
fetch_request.endpoint.clone_from(&request.source_endpoint);
let version = source.fetch_registry_version(&fetch_request)?;
Ok(registry_version_report_from_version(version))
}
fn registry_version_report_from_version(
version: MainnetRegistryVersion,
) -> NnsRegistryVersionReport {
NnsRegistryVersionReport {
schema_version: NNS_REGISTRY_VERSION_REPORT_SCHEMA_VERSION,
network: version.network,
registry_canister_id: version.registry_canister_id,
registry_version: version.registry_version,
fetched_at: version.fetched_at,
source_endpoint: version.source_endpoint,
fetched_by: version.fetched_by,
}
}