from typing import cast
from web3 import Web3

from autotx.utils.ethereum import get_erc20_balance, get_native_balance
from autotx.utils.ethereum.constants import NATIVE_TOKEN_ADDRESS
from autotx.utils.ethereum.helpers.get_native_token_symbol import (
    get_native_token_symbol,
)
from autotx.utils.ethereum.networks import SUPPORTED_NETWORKS_CONFIGURATION_MAP, ChainId, NetworkConfiguration
from autotx.eth_address import ETHAddress


def show_address_balances(web3: Web3, network: ChainId, address: ETHAddress) -> None:
    native_token_symbol = get_native_token_symbol(network)
    native_token_balance = get_native_balance(web3, address)
    print(f"{native_token_symbol.upper()} balance: {native_token_balance}")

    current_network = cast(NetworkConfiguration, SUPPORTED_NETWORKS_CONFIGURATION_MAP.get(network))
    for token in current_network.default_tokens:
        if current_network.default_tokens[token] == NATIVE_TOKEN_ADDRESS:
            continue
        token_address = ETHAddress(current_network.default_tokens[token])
        balance = get_erc20_balance(web3, token_address, address)

        if balance > 0:
            print(f"{token.upper()} balance: {balance}")
