From f57a1a9a0e2cfbf988b48a72218ae3aced0d3195 Mon Sep 17 00:00:00 2001
From: brentstone <brentstone01@gmail.com>
Date: Tue, 27 Aug 2024 17:01:04 -0700
Subject: [PATCH] better error handling for sdk query in client

---
 crates/apps_lib/src/client/rpc.rs | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/crates/apps_lib/src/client/rpc.rs b/crates/apps_lib/src/client/rpc.rs
index ab280fb4ff..a642bd4856 100644
--- a/crates/apps_lib/src/client/rpc.rs
+++ b/crates/apps_lib/src/client/rpc.rs
@@ -1183,11 +1183,8 @@ pub async fn query_total_supply<N: Namada>(
     args: args::QueryTotalSupply,
 ) {
     let token = args.token;
-    let supply = unwrap_client_response::<N::Client, token::Amount>(
-        RPC.vp()
-            .token()
-            .total_supply(context.client(), &token)
-            .await,
+    let supply = unwrap_sdk_result(
+        rpc::get_token_total_supply(context.client(), &token).await,
     );
     let amount_str = format_denominated_amount(
         context.client(),
@@ -2046,6 +2043,14 @@ fn unwrap_client_response<C: namada_sdk::queries::Client, T>(
     })
 }
 
+/// A helper to unwrap an SDK query result. Will shut down process on error.
+fn unwrap_sdk_result<T>(response: Result<T, namada_sdk::error::Error>) -> T {
+    response.unwrap_or_else(|err| {
+        eprintln!("Error in the query: {:?}", err);
+        cli::safe_exit(1)
+    })
+}
+
 pub async fn compute_proposal_votes<C: namada_sdk::queries::Client + Sync>(
     client: &C,
     proposal_id: u64,