ic-query 0.29.6

Internet Computer query library for NNS, SNS, ICRC, system canisters, and public network metadata
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109

//! Module: nns::proposals::report::source::live
//!
//! Responsibility: query live NNS governance proposal APIs.
//! Does not own: report DTO assembly, cache publication, or text rendering.
//! Boundary: adapts source trait calls to candid queries against mainnet governance.

use crate::{
    nns::{
        LiveNnsSource, NnsSourceRequest,
        governance_query::query_nns_governance,
        proposals::report::{
            NnsProposalHostError, enforce_mainnet_network,
            model::{NnsProposalRewardStatusFilter, NnsProposalRow, NnsProposalStatusFilter},
            source::{NnsProposalSource, nns_proposal_row_from_info},
            wire::{
                NnsListProposalInfoRequest, NnsListProposalInfoResponse, NnsProposalId,
                NnsProposalInfo,
            },
        },
    },
    runtime::block_on_current_thread,
};

impl NnsProposalSource for LiveNnsSource {
    fn fetch_proposals(
        &self,
        request: &NnsSourceRequest,
        limit: u32,
        before_proposal_id: Option<u64>,
        status: NnsProposalStatusFilter,
        reward_status: NnsProposalRewardStatusFilter,
    ) -> Result<Vec<NnsProposalRow>, NnsProposalHostError> {
        enforce_mainnet_network(&request.network)?;
        let include_status = status
            .governance_status_code()
            .into_iter()
            .collect::<Vec<_>>();
        let include_reward_status = reward_status
            .governance_reward_status_code()
            .into_iter()
            .collect::<Vec<_>>();
        let proposals = block_on_current_thread(fetch_nns_proposal_list_async(
            request,
            limit,
            before_proposal_id,
            &include_status,
            &include_reward_status,
        ))
        .map_err(NnsProposalHostError::Runtime)??;
        Ok(proposals
            .into_iter()
            .map(nns_proposal_row_from_info)
            .collect())
    }

    fn fetch_proposal(
        &self,
        request: &NnsSourceRequest,
        proposal_id: u64,
    ) -> Result<NnsProposalRow, NnsProposalHostError> {
        enforce_mainnet_network(&request.network)?;
        Ok(nns_proposal_row_from_info(
            block_on_current_thread(fetch_nns_proposal_async(request, proposal_id))
                .map_err(NnsProposalHostError::Runtime)??,
        ))
    }
}

async fn fetch_nns_proposal_list_async(
    request: &NnsSourceRequest,
    limit: u32,
    before_proposal_id: Option<u64>,
    include_status: &[i32],
    include_reward_status: &[i32],
) -> Result<Vec<NnsProposalInfo>, NnsProposalHostError> {
    let response: NnsListProposalInfoResponse = query_nns_governance(
        request,
        "list_proposals",
        "ListProposalInfoRequest",
        "ListProposalInfoResponse",
        &NnsListProposalInfoRequest {
            include_reward_status: include_reward_status.to_vec(),
            omit_large_fields: Some(false),
            before_proposal: before_proposal_id.map(|id| NnsProposalId { id }),
            limit,
            exclude_topic: Vec::new(),
            include_all_manage_neuron_proposals: Some(true),
            include_status: include_status.to_vec(),
            return_self_describing_action: Some(false),
        },
    )
    .await?;
    Ok(response.proposal_info)
}

async fn fetch_nns_proposal_async(
    request: &NnsSourceRequest,
    proposal_id: u64,
) -> Result<NnsProposalInfo, NnsProposalHostError> {
    let proposal: Option<NnsProposalInfo> = query_nns_governance(
        request,
        "get_proposal_info",
        "ProposalId",
        "ProposalInfo",
        &proposal_id,
    )
    .await?;
    proposal.ok_or(NnsProposalHostError::ProposalNotFound { proposal_id })
}