Query for IBC tokens

.changelog/unreleased/improvements/1946-ibc-balance-query.md

@@ -0,0 +1,2 @@
+- Query also IBC token balances
+  ([\#1946](https://github.com/anoma/namada/issues/1946))

apps/src/lib/cli/context.rs

@@ -6,11 +6,13 @@ use std::path::{Path, PathBuf};
 use std::str::FromStr;
 
 use color_eyre::eyre::Result;
+use namada::ledger::ibc::storage::ibc_token;
 use namada::sdk::masp::ShieldedContext;
 use namada::sdk::wallet::Wallet;
 use namada::types::address::{Address, InternalAddress};
 use namada::types::chain::ChainId;
 use namada::types::ethereum_events::EthAddress;
+use namada::types::ibc::is_ibc_denom;
 use namada::types::io::Io;
 use namada::types::key::*;
 use namada::types::masp::*;
@@ -367,6 +369,20 @@ impl ArgFromContext for Address {
                     })
                     .unwrap_or(Err(Skip))
             })
+            // An IBC token
+            .or_else(|_| {
+                is_ibc_denom(raw)
+                    .map(|(trace_path, base_denom)| {
+                        let base_token = ctx
+                            .wallet
+                            .find_address(&base_denom)
+                            .map(|addr| addr.to_string())
+                            .unwrap_or(base_denom);
+                        let ibc_denom = format!("{trace_path}/{base_token}");
+                        ibc_token(ibc_denom)
+                    })
+                    .ok_or(Skip)
+            })
             // Or it can be an alias that may be found in the wallet
             .or_else(|_| ctx.wallet.find_address(raw).cloned().ok_or(Skip))
             .map_err(|_| format!("Unknown address {raw}"))