add balance command

src/subcommand/wallet.rs

@@ -1,5 +1,6 @@
 use {super::*, transaction_builder::TransactionBuilder};
 
+mod balance;
 mod create;
 mod inscribe;
 mod inscriptions;
@@ -12,6 +13,8 @@ mod utxos;
 
 #[derive(Debug, Parser)]
 pub(crate) enum Wallet {
+  #[clap(about = "Get wallet balance")]
+  Balance(balance::Balance),
   #[clap(about = "Create a new wallet")]
   Create(create::Create),
   #[clap(about = "Create an inscription")]
@@ -33,6 +36,7 @@ pub(crate) enum Wallet {
 impl Wallet {
   pub(crate) fn run(self, options: Options) -> Result {
     match self {
+      Self::Balance(balance) => balance.run(options),
       Self::Create(create) => create.run(options),
       Self::Inscribe(inscribe) => inscribe.run(options),
       Self::Inscriptions(inscriptions) => inscriptions.run(options),

src/subcommand/wallet/balance.rs

@@ -0,0 +1,20 @@
+use super::*;
+
+#[derive(Debug, Parser)]
+pub(crate) struct Balance {}
+
+impl Balance {
+  pub(crate) fn run(self, options: Options) -> Result {
+    println!(
+      "{}",
+      options
+        .bitcoin_rpc_client_for_wallet_command("ord wallet balance")?
+        .get_balances()?
+        .mine
+        .trusted
+        .to_sat()
+    );
+
+    Ok(())
+  }
+}

tests/wallet.rs

@@ -688,3 +688,21 @@ fn inscribe_gif() {
     ),
   )
 }
+
+#[test]
+fn wallet_balance() {
+  let rpc_server = test_bitcoincore_rpc::spawn_with(Network::Regtest, "ord");
+
+  CommandBuilder::new("--regtest wallet balance")
+    .rpc_server(&rpc_server)
+    .expected_stdout("0\n")
+    .run();
+
+  let coinbase_tx = &rpc_server.mine_blocks_with_subsidy(1, 1_000_000)[0].txdata[0];
+  let amount = coinbase_tx.output[0].value;
+
+  CommandBuilder::new("--regtest wallet balance")
+    .rpc_server(&rpc_server)
+    .expected_stdout(format!("{amount}\n"))
+    .run();
+}