add rewards query to SDK

anoma · Jan 1, 2025 · aadc4b6 · aadc4b6
1 parent b579d49
commit aadc4b6
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/crates/apps_lib/src/client/rpc.rs b/crates/apps_lib/src/client/rpc.rs
@@ -1482,9 +1482,7 @@ pub async fn query_rewards<C: Client + Sync>(
     source: &Option<Address>,
     validator: &Address,
 ) -> token::Amount {
-    unwrap_client_response::<C, token::Amount>(
-        RPC.vp().pos().rewards(client, validator, source).await,
-    )
+    unwrap_sdk_result(rpc::query_rewards(client, source, validator).await)
 }
 
 /// Query token total supply.

diff --git a/crates/sdk/src/rpc.rs b/crates/sdk/src/rpc.rs
@@ -858,6 +858,17 @@ pub async fn get_validator_state<C: namada_io::Client + Sync>(
     )
 }
 
+/// Query and return the available reward tokens corresponding to the bond
+pub async fn query_rewards<C: namada_io::Client + Sync>(
+    client: &C,
+    source: &Option<Address>,
+    validator: &Address,
+) -> Result<token::Amount, error::Error> {
+    convert_response::<C, token::Amount>(
+        RPC.vp().pos().rewards(client, validator, source).await,
+    )
+}
+
 /// Get the validators to which a delegator is bonded at a certain epoch
 pub async fn get_delegation_validators<C: namada_io::Client + Sync>(
     client: &C,