ic-query 0.78.0

//! Module: sns::report::live::query
//!
//! Responsibility: low-level live SNS query helpers.
//! Does not own: report assembly, Candid wire type definitions, or rendering.
//! Boundary: builds agents, parses principals, and maps query failures to typed errors.

use super::SnsHostError;
use crate::{
    agent::MAX_IC_AGENT_RESPONSE_BODY_BYTES,
    candid_call::encode_request,
    sns::report::{SnsSourceRequest, enforce_mainnet_network},
};
use ic_platform::agent::Agent;
use ic_platform::candid::contract::{CandidType, Principal, serde::Deserialize};

crate::candid_call::impl_candid_call_error!(SnsHostError);

/// Call one Candid canister ingress method and wait for its response.
pub(super) async fn update_canister<Arg, Response>(
    agent: &Agent,
    canister: &Principal,
    method: &'static str,
    request_message: &'static str,
    response_message: &'static str,
    arg: &Arg,
) -> Result<Response, SnsHostError>
where
    Arg: CandidType + Sync,
    Response: for<'de> Deserialize<'de> + CandidType,
{
    let arg = encode_request::<_, SnsHostError>((arg,), request_message)?;
    let bytes = agent
        .update(canister, method)
        .with_arg(arg)
        .call_and_wait()
        .await
        .map_err(|err| SnsHostError::AgentUpdateCall {
            method,
            reason: err.to_string(),
        })?;
    crate::candid_decode::decode_reply(&bytes).map_err(|err| SnsHostError::CandidDecode {
        message: response_message,
        reason: err.to_string(),
    })
}

/// Build an IC agent after enforcing one explicit SNS source request.
pub(super) fn sns_agent<'a>(
    source: &'a crate::sns::LiveSnsSource,
    request: &SnsSourceRequest,
) -> Result<&'a Agent, SnsHostError> {
    enforce_mainnet_network(&request.network)?;
    crate::agent::endpoint_agent(
        &source.agent,
        &request.endpoint,
        MAX_IC_AGENT_RESPONSE_BODY_BYTES,
        |reason| SnsHostError::AgentBuild {
            endpoint: request.endpoint.clone(),
            reason,
        },
    )
}

/// Parse a principal text field into a typed principal or host error.
pub(super) fn principal_from_text(
    value: &str,
    field: &'static str,
) -> Result<Principal, SnsHostError> {
    Principal::from_text(value).map_err(|err| SnsHostError::InvalidPrincipal {
        field,
        reason: err.to_string(),
    })
}