feature: add check balance function to cli

Author: vid277

bin/cli/src/main.rs

@@ -16,7 +16,7 @@ use base64::{engine::general_purpose::STANDARD as BASE64, Engine as _};
 use clap::{Parser, Subcommand};
 use directories::ProjectDirs;
 use libp2p::identity::Keypair;
-use rpc_client::{rpc_create_deposit_intent, rpc_spend, rpc_start_signing};
+use rpc_client::{rpc_check_balance, rpc_create_deposit_intent, rpc_spend, rpc_start_signing};
 use std::{fs, path::PathBuf};
 
 use crate::{
@@ -173,6 +173,11 @@ enum Commands {
         #[arg(short, long)]
         endpoint: Option<String>,
     },
+    CheckBalance {
+        #[arg(short, long)]
+        endpoint: Option<String>,
+        address: String,
+    },
 }
 
 #[tokio::main]
@@ -244,6 +249,11 @@ async fn main() -> Result<(), CliError> {
                 .await
                 .map_err(CliError::RpcError)?;
         }
+        Commands::CheckBalance { endpoint, address } => {
+            rpc_check_balance(endpoint, address)
+                .await
+                .map_err(CliError::RpcError)?;
+        }
     }
 
     Ok(())

bin/cli/src/rpc_client.rs

@@ -1,6 +1,7 @@
 use node::grpc::grpc_handler::node_proto::{
-    self, node_control_client::NodeControlClient, CreateDepositIntentResponse,
-    GetPendingDepositIntentsResponse, SpendFundsResponse, StartSigningResponse,
+    self, node_control_client::NodeControlClient, CheckBalanceResponse,
+    CreateDepositIntentResponse, GetPendingDepositIntentsResponse, SpendFundsResponse,
+    StartSigningResponse,
 };
 use tonic::Status;
 
@@ -86,3 +87,21 @@ pub async fn rpc_get_pending_deposit_intents(
 
     Ok(get_pending_deposit_intents_response.into_inner())
 }
+
+pub async fn rpc_check_balance(
+    endpoint: Option<String>,
+    address: String,
+) -> Result<CheckBalanceResponse, Status> {
+    let mut client =
+        NodeControlClient::connect(endpoint.unwrap_or("http://[::1]:50051".to_string()))
+            .await
+            .expect("Failed to connect");
+
+    let check_balance_response = client
+        .check_balance(tonic::Request::new(node_proto::CheckBalanceRequest {
+            address,
+        }))
+        .await?;
+
+    Ok(check_balance_response.into_inner())
+}