feat: Implement 'balance' command to fetch wallet balance

src/client/mod.rs

@@ -17,6 +17,8 @@ pub use batch::Batch;
 
 use self::query_me::WalletCurrency;
 
+use crate::types::Wallet;
+
 pub mod server;
 
 pub struct GaloyClient {
@@ -97,6 +99,57 @@ impl GaloyClient {
         Ok(me)
     }
 
+    pub fn fetch_balance(&self, wallet: Option<Wallet>) -> anyhow::Result<String> {
+        let me = self.me()?;
+
+        let default_wallet_id = me.default_account.default_wallet_id;
+
+        let wallets = &me.default_account.wallets;
+        let mut wallet_balances = String::new();
+
+        for wallet_info in wallets {
+            match &wallet {
+                Some(Wallet::Usd) if wallet_info.wallet_currency == WalletCurrency::USD => {
+                    return Ok(format!("USD wallet: {} cents", wallet_info.balance));
+                }
+                Some(Wallet::Btc) if wallet_info.wallet_currency == WalletCurrency::BTC => {
+                    return Ok(format!("BTC wallet: {} sats", wallet_info.balance));
+                }
+                None => {
+                    if wallet_info.wallet_currency == WalletCurrency::USD {
+                        wallet_balances += &format!(
+                            "USD wallet{}: {} cents\n",
+                            if wallet_info.id == default_wallet_id {
+                                " (default)"
+                            } else {
+                                ""
+                            },
+                            wallet_info.balance
+                        );
+                    }
+                    if wallet_info.wallet_currency == WalletCurrency::BTC {
+                        wallet_balances += &format!(
+                            "BTC wallet{}: {} sats\n",
+                            if wallet_info.id == default_wallet_id {
+                                " (default)"
+                            } else {
+                                ""
+                            },
+                            wallet_info.balance
+                        );
+                    }
+                }
+                _ => {}
+            }
+        }
+
+        if wallet_balances.is_empty() {
+            Err(anyhow::anyhow!("No matching wallet found"))
+        } else {
+            Ok(wallet_balances)
+        }
+    }
+
     pub fn request_phone_code(&self, phone: String, nocaptcha: bool) -> std::io::Result<()> {
         match nocaptcha {
             false => {

src/lib.rs

@@ -4,3 +4,5 @@
 mod client;
 
 pub use client::*;
+
+pub mod types;

src/main.rs

@@ -15,6 +15,8 @@ use std::fs::{self};
 mod constants;
 mod token;
 
+use galoy_cli::types::*;
+
 #[derive(Parser)]
 #[clap(author, version, about, long_about = None)]
 struct Cli {
@@ -65,6 +67,13 @@ enum Commands {
     },
     /// Execute Me query
     Me,
+    /// Fetch the balance of a wallet
+    Balance {
+        #[clap(long)]
+        btc: bool,
+        #[clap(long)]
+        usd: bool,
+    },
     /// Execute a Payment
     Pay {
         #[clap(short, long)]
@@ -82,12 +91,6 @@ enum Commands {
     Batch { filename: String, price: Decimal },
 }
 
-#[derive(Debug, Clone, clap::ValueEnum, PartialEq, Eq)]
-enum Wallet {
-    Btc,
-    Usd,
-}
-
 fn main() -> anyhow::Result<()> {
     log::set_max_level(LevelFilter::Warn);
 
@@ -128,6 +131,18 @@ fn main() -> anyhow::Result<()> {
                 serde_json::to_string_pretty(&result).expect("Can't serialize json")
             );
         }
+        Commands::Balance { btc, usd } => {
+            let wallet_type = match (btc, usd) {
+                (true, true) | (false, false) => None,
+                (true, false) => Some(Wallet::Btc),
+                (false, true) => Some(Wallet::Usd),
+            };
+
+            let balance = galoy_cli
+                .fetch_balance(wallet_type)
+                .context("can't fetch balance")?;
+            println!("{}", balance);
+        }
         Commands::Pay {
             username,
             wallet,

src/types.rs

@@ -0,0 +1,5 @@
+#[derive(Debug, Clone, clap::ValueEnum, PartialEq, Eq)]
+pub enum Wallet {
+    Btc,
+    Usd,
+}