Add sncast balance command (#3817)

<!-- Reference any GitHub issues resolved by this PR -->

Towards #2319 

**Stack**:
- #3818
- #3817 ⬅

## Introduced changes

Add `sncast balance` command

## Checklist

<!-- Make sure all of these are complete -->

- [x] Linked relevant issue
- [ ] Updated relevant documentation
- [x] Added relevant tests
- [x] Performed self-review of the code
- [ ] Added changes to `CHANGELOG.md`

Author: franciszekjob

Cargo.lock

@@ -93,7 +93,7 @@ snapbox = "0.4.17"
 scarb-ui = "0.1.7"
 semver = "1.0.27"
 bimap = "0.6.3"
-primitive-types = "0.13.1"
+primitive-types = {  version =  "0.13.1", features = ["serde"] }
 shellexpand = "3.1.0"
 toml = "0.8.20"
 rpassword = "7.3.1"

Cargo.toml

@@ -61,6 +61,9 @@ universal-sierra-compiler-api = { path = "../universal-sierra-compiler-api" }
 tracing-subscriber.workspace = true
 tracing.workspace = true
 tracing-log.workspace = true
+strum.workspace = true
+strum_macros.workspace = true
+primitive-types.workspace = true
 
 [dev-dependencies]
 ctor.workspace = true

crates/sncast/Cargo.toml

@@ -11,3 +11,4 @@ pub mod interactive;
 pub mod output_format;
 pub mod rpc;
 pub mod scarb_utils;
+pub mod token;

crates/sncast/src/helpers/mod.rs

@@ -0,0 +1,31 @@
+use clap::ValueEnum;
+use serde::Serialize;
+use starknet::macros::felt;
+use starknet_types_core::felt::Felt;
+
+const STRK_CONTRACT_ADDRESS: Felt =
+    felt!("0x04718f5a0fc34cc1af16a1cdee98ffb20c31f5cd61d6ab07201858f4287c938d");
+
+const ETH_CONTRACT_ADDRESS: Felt =
+    felt!("0x049d36570d4e46f48e99674bd3fcc84644ddd6b96f7c741b1562b82f9e004dc7");
+
+#[derive(Default, Serialize, Clone, Copy, Debug, ValueEnum, strum_macros::Display)]
+// Both serde and strum enums need to have proper
+// casing configuration in clap and for serialization.
+#[strum(serialize_all = "lowercase")]
+#[serde(rename_all = "lowercase")]
+pub enum Token {
+    #[default]
+    Strk,
+    Eth,
+}
+
+impl Token {
+    #[must_use]
+    pub fn contract_address(&self) -> Felt {
+        match self {
+            Token::Strk => STRK_CONTRACT_ADDRESS,
+            Token::Eth => ETH_CONTRACT_ADDRESS,
+        }
+    }
+}

crates/sncast/src/helpers/token.rs

@@ -1,12 +1,14 @@
+use crate::starknet_commands::balance::Balance;
 use crate::starknet_commands::declare::declare;
 use crate::starknet_commands::declare_from::DeclareFrom;
 use crate::starknet_commands::deploy::DeployArguments;
 use crate::starknet_commands::multicall;
 use crate::starknet_commands::script::run_script_command;
 use crate::starknet_commands::utils::{self, Utils};
 use crate::starknet_commands::{
-    account, account::Account, call::Call, declare::Declare, deploy::Deploy, invoke::Invoke,
-    multicall::Multicall, script::Script, show_config::ShowConfig, tx_status::TxStatus,
+    account, account::Account as AccountCommand, call::Call, declare::Declare, deploy::Deploy,
+    invoke::Invoke, multicall::Multicall, script::Script, show_config::ShowConfig,
+    tx_status::TxStatus,
 };
 use anyhow::{Context, Result, bail};
 use camino::Utf8PathBuf;
@@ -37,6 +39,7 @@ use sncast::{
     ValidatedWaitParams, WaitForTx, get_account, get_block_id, get_class_hash_by_address,
     get_contract_class,
 };
+use starknet::accounts::Account;
 use starknet::core::types::ContractClass;
 use starknet::core::types::contract::{AbiEntry, SierraClass};
 use starknet::core::utils::get_selector_from_name;
@@ -139,7 +142,7 @@ enum Commands {
     Multicall(Multicall),
 
     /// Create and deploy an account
-    Account(Account),
+    Account(AccountCommand),
 
     /// Show current configuration being used
     ShowConfig(ShowConfig),
@@ -160,6 +163,9 @@ enum Commands {
 
     /// Utility commands
     Utils(Utils),
+
+    /// Fetch balance of the account for specified token
+    Balance(Balance),
 }
 
 #[derive(Debug, Clone, clap::Args)]
@@ -679,6 +685,25 @@ async fn run_async_command(cli: Cli, config: CastConfig, ui: &UI) -> Result<()>
             Ok(())
         }
 
+        Commands::Balance(balance) => {
+            let provider = balance.rpc.get_provider(&config, ui).await?;
+            let account = get_account(
+                &config,
+                &provider,
+                &balance.rpc,
+                config.keystore.as_ref(),
+                ui,
+            )
+            .await?;
+
+            let result =
+                starknet_commands::balance::balance(account.address(), &provider, &balance).await?;
+
+            process_command_result("balance", Ok(result), ui, None);
+
+            Ok(())
+        }
+
         Commands::Script(_) => unreachable!(),
     }
 }

crates/sncast/src/main.rs

@@ -0,0 +1,46 @@
+use crate::{
+    helpers::token::Token,
+    response::{
+        cast_message::{SncastCommandMessage, SncastMessage},
+        command::CommandResponse,
+    },
+};
+use foundry_ui::styling;
+use primitive_types::U256;
+use serde::ser::{Serialize, SerializeStruct, Serializer};
+
+#[derive(Debug)]
+pub struct BalanceResponse {
+    pub balance: U256,
+    pub token: Option<Token>,
+}
+
+impl CommandResponse for BalanceResponse {}
+
+impl SncastCommandMessage for SncastMessage<BalanceResponse> {
+    fn text(&self) -> String {
+        let balance_str = if let Some(token) = self.command_response.token {
+            format!("{} {}", self.command_response.balance, token)
+        } else {
+            self.command_response.balance.to_string()
+        };
+
+        styling::OutputBuilder::new()
+            .field("Balance", &balance_str)
+            .build()
+    }
+}
+
+// We need custom serialization because U256's default serialization is hex string
+impl Serialize for BalanceResponse {
+    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
+    where
+        S: Serializer,
+    {
+        let mut s = serializer.serialize_struct("BalanceResponse", 2)?;
+        // Default U256 serialization uses hex string, we want decimal string
+        s.serialize_field("balance", &self.balance.to_string())?;
+        s.serialize_field("token", &self.token)?;
+        s.end()
+    }
+}

crates/sncast/src/response/balance.rs

@@ -1,4 +1,5 @@
 pub mod account;
+pub mod balance;
 pub mod call;
 pub mod cast_message;
 pub mod command;