Merge pull request #124 from input-output-hk/whankinsiv/drep-blockfrost-endpoints

feat: Add /governance/dreps/ endpoint handlers to rest_blockfrost

Author: sandtreader

Cargo.lock

@@ -130,7 +130,7 @@ pub struct GovernanceProceduresMessage {
     pub proposal_procedures: Vec<ProposalProcedure>,
 
     /// Voting
-    pub voting_procedures: Vec<([u8; 32], VotingProcedures)>,
+    pub voting_procedures: Vec<(TxHash, VotingProcedures)>,
 
     /// Alonzo-compatible (from Shelley) and Babbage updates
     pub alonzo_babbage_updates: Vec<AlonzoBabbageUpdateProposal>,

common/src/messages.rs

@@ -18,6 +18,8 @@ pub enum AccountsStateQuery {
     GetAccountAssets { stake_key: Vec<u8> },
     GetAccountAssetsTotals { stake_key: Vec<u8> },
     GetAccountUTxOs { stake_key: Vec<u8> },
+    GetAccountsBalancesMap { stake_keys: Vec<Vec<u8>> },
+    GetAccountsBalancesSum { stake_keys: Vec<Vec<u8>> },
 
     // Pools related queries
     GetPoolsLiveStakes { pools_operators: Vec<Vec<u8>> },
@@ -39,6 +41,8 @@ pub enum AccountsStateQueryResponse {
     AccountAssets(AccountAssets),
     AccountAssetsTotals(AccountAssetsTotals),
     AccountUTxOs(AccountUTxOs),
+    AccountsBalancesMap(HashMap<Vec<u8>, u64>),
+    AccountsBalancesSum(u64),
 
     // Pools related responses
     PoolsLiveStakes(PoolsLiveStakes),

common/src/queries/accounts.rs

@@ -1,13 +1,19 @@
 use std::collections::HashMap;
 
 use crate::{
-    Anchor, DRepCredential, GovActionId, Lovelace, ProposalProcedure, Voter, VotingProcedure,
+    Anchor, Credential, DRepCredential, GovActionId, Lovelace, ProposalProcedure, TxHash, Vote,
+    Voter, VotingProcedure,
 };
 
+pub const DEFAULT_DREPS_QUERY_TOPIC: (&str, &str) =
+    ("drep-state-query-topic", "cardano.query.dreps");
+pub const DEFAULT_GOVERNANCE_QUERY_TOPIC: (&str, &str) =
+    ("governance-state-query-topic", "cardano.query.governance");
+
 #[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
 pub enum GovernanceStateQuery {
     GetDRepsList,
-    GetDRepInfo { drep_credential: DRepCredential },
+    GetDRepInfoWithDelegators { drep_credential: DRepCredential },
     GetDRepDelegators { drep_credential: DRepCredential },
     GetDRepMetadata { drep_credential: DRepCredential },
     GetDRepUpdates { drep_credential: DRepCredential },
@@ -23,9 +29,9 @@ pub enum GovernanceStateQuery {
 #[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
 pub enum GovernanceStateQueryResponse {
     DRepsList(DRepsList),
-    DRepInfo(DRepInfo),
+    DRepInfoWithDelegators(DRepInfoWithDelegators),
     DRepDelegators(DRepDelegatorAddresses),
-    DRepMetadata(DRepMetadata),
+    DRepMetadata(Option<Option<Anchor>>),
     DRepUpdates(DRepUpdates),
     DRepVotes(DRepVotes),
     ProposalsList(ProposalsList),
@@ -46,20 +52,53 @@ pub struct DRepsList {
 #[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
 pub struct DRepInfo {
     pub deposit: Lovelace,
-    pub anchor: Option<Anchor>,
+    pub retired: bool,
+    pub expired: bool,
+    pub active_epoch: Option<u64>,
+    pub last_active_epoch: u64,
 }
 
 #[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
-pub struct DRepDelegatorAddresses {}
+pub struct DRepInfoWithDelegators {
+    pub info: DRepInfo,
+    pub delegators: Vec<Credential>,
+}
 
 #[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
-pub struct DRepMetadata {}
+pub struct DRepDelegatorAddresses {
+    pub addresses: Vec<Credential>,
+}
 
 #[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
-pub struct DRepUpdates {}
+pub struct DRepUpdates {
+    pub updates: Vec<DRepUpdateEvent>,
+}
 
 #[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
-pub struct DRepVotes {}
+pub struct DRepUpdateEvent {
+    pub tx_hash: TxHash,
+    pub cert_index: u64,
+    pub action: DRepActionUpdate,
+}
+
+#[derive(serde::Serialize, serde::Deserialize, Clone, Debug)]
+pub enum DRepActionUpdate {
+    Registered,
+    Updated,
+    Deregistered,
+}
+
+#[derive(Clone, serde::Serialize, serde::Deserialize, Debug)]
+pub struct DRepVotes {
+    pub votes: Vec<VoteRecord>,
+}
+
+#[derive(Clone, serde::Serialize, serde::Deserialize, Debug)]
+pub struct VoteRecord {
+    pub tx_hash: TxHash,
+    pub vote_index: u32,
+    pub vote: Vote,
+}
 
 #[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
 pub struct ProposalsList {
@@ -79,8 +118,19 @@ pub struct ProposalWithdrawals {}
 
 #[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
 pub struct ProposalVotes {
-    pub votes: HashMap<Voter, (Vec<u8>, VotingProcedure)>,
+    pub votes: HashMap<Voter, (TxHash, VotingProcedure)>,
 }
 
 #[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
 pub struct ProposalMetadata {}
+
+pub fn handle_governance_query_result<T>(
+    result: anyhow::Result<Option<T>>,
+    mapper: impl FnOnce(T) -> GovernanceStateQueryResponse,
+) -> GovernanceStateQueryResponse {
+    match result {
+        Ok(Some(val)) => mapper(val),
+        Ok(None) => GovernanceStateQueryResponse::NotFound,
+        Err(e) => GovernanceStateQueryResponse::Error(e.to_string()),
+    }
+}