From e332161a6df99db43a41e1ac65895623552d9ffe Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tom=C3=A1=C5=A1=20Zemanovi=C4=8D?= <tomas@heliax.dev>
Date: Wed, 5 Jun 2024 17:06:21 +0200
Subject: [PATCH] sdk: add a query for MASP total rewards

---
 crates/sdk/src/queries/vp/token.rs | 25 ++++++++++++++++++++++++-
 crates/sdk/src/rpc.rs              |  7 +++++++
 2 files changed, 31 insertions(+), 1 deletion(-)

diff --git a/crates/sdk/src/queries/vp/token.rs b/crates/sdk/src/queries/vp/token.rs
index 18e994ebc93..89d8a15e554 100644
--- a/crates/sdk/src/queries/vp/token.rs
+++ b/crates/sdk/src/queries/vp/token.rs
@@ -49,7 +49,7 @@ pub mod client_only_methods {
     use borsh::BorshDeserialize;
     use namada_core::address::Address;
     use namada_core::token;
-    use namada_token::storage_key::balance_key;
+    use namada_token::storage_key::{balance_key, masp_total_rewards};
 
     use super::Token;
     use crate::queries::{Client, RPC};
@@ -79,5 +79,28 @@ pub mod client_only_methods {
             };
             Ok(balance)
         }
+
+        /// Get the total rewards minted by MASP.
+        pub async fn masp_total_rewards<CLIENT>(
+            &self,
+            client: &CLIENT,
+        ) -> Result<token::Amount, <CLIENT as Client>::Error>
+        where
+            CLIENT: Client + Sync,
+        {
+            let total_rewards_key = masp_total_rewards();
+            let response = RPC
+                .shell()
+                .storage_value(client, None, None, false, &total_rewards_key)
+                .await?;
+
+            let tokens = if response.data.is_empty() {
+                token::Amount::zero()
+            } else {
+                token::Amount::try_from_slice(&response.data)
+                    .unwrap_or_default()
+            };
+            Ok(tokens)
+        }
     }
 }
diff --git a/crates/sdk/src/rpc.rs b/crates/sdk/src/rpc.rs
index 52f079c5add..5e674c86b7f 100644
--- a/crates/sdk/src/rpc.rs
+++ b/crates/sdk/src/rpc.rs
@@ -347,6 +347,13 @@ pub async fn query_conversions<C: crate::queries::Client + Sync>(
     convert_response::<C, _>(RPC.shell().read_conversions(client).await)
 }
 
+/// Query the total rewards minted by MASP
+pub async fn query_masp_total_rewards<C: crate::queries::Client + Sync>(
+    client: &C,
+) -> Result<token::Amount, error::Error> {
+    convert_response::<C, _>(RPC.vp().token().masp_total_rewards(client).await)
+}
+
 /// Query to read the tokens that earn masp rewards.
 pub async fn query_masp_reward_tokens<C: crate::queries::Client + Sync>(
     client: &C,