use crate::sns::report::{
MAINNET_SNS_WASM_CANISTER_ID, SNS_SWAP_QUERY_COUNT, SNS_SWAP_REPORT_SCHEMA_VERSION,
SnsHostError, SnsLookupRequest, SnsSwapReport, SnsSwapSource, live::LiveSnsSource,
lookup::resolve_sns_lookup, source::canonicalize_mainnet_sns_swap,
};
pub async fn build_sns_swap_report(
request: &SnsLookupRequest,
) -> Result<SnsSwapReport, SnsHostError> {
build_sns_swap_report_with_source(request, &LiveSnsSource::default()).await
}
pub async fn build_sns_swap_report_with_source(
request: &SnsLookupRequest,
source: &dyn SnsSwapSource,
) -> Result<SnsSwapReport, SnsHostError> {
let lookup = resolve_sns_lookup(request, source).await?;
let mut swap = source
.fetch_sns_swap(&lookup.fetch_request, &lookup.sns)
.await?;
canonicalize_mainnet_sns_swap(&mut swap, &lookup.sns.canisters.swap_canister_id)?;
let component_gap_count = swap.gaps.len();
Ok(SnsSwapReport {
schema_version: SNS_SWAP_REPORT_SCHEMA_VERSION,
network: lookup.fetch_request.network,
sns_wasm_canister_id: MAINNET_SNS_WASM_CANISTER_ID.to_string(),
fetched_at: lookup.fetch_request.fetched_at,
source_endpoint: lookup.fetch_request.endpoint,
fetched_by: lookup.fetch_request.fetched_by,
id: lookup.sns.id,
name: lookup.sns.name,
root_canister_id: lookup.sns.canisters.root_canister_id,
component_query_count: SNS_SWAP_QUERY_COUNT,
successful_component_query_count: SNS_SWAP_QUERY_COUNT - component_gap_count,
component_gap_count,
swap,
})
}