github.com/ChainSafe/chainbridge-core@v1.4.2/chains/evm/cli/erc20/balance.go (about)

     1  package erc20
     2  
     3  import (
     4  	"fmt"
     5  
     6  	"github.com/ChainSafe/chainbridge-core/chains/evm/calls/contracts/erc20"
     7  	"github.com/ChainSafe/chainbridge-core/chains/evm/calls/evmtransaction"
     8  	"github.com/ChainSafe/chainbridge-core/chains/evm/cli/initialize"
     9  	"github.com/ChainSafe/chainbridge-core/util"
    10  
    11  	"github.com/ChainSafe/chainbridge-core/chains/evm/cli/flags"
    12  	"github.com/ChainSafe/chainbridge-core/chains/evm/cli/logger"
    13  	"github.com/ethereum/go-ethereum/common"
    14  	"github.com/rs/zerolog/log"
    15  	"github.com/spf13/cobra"
    16  )
    17  
    18  var balanceCmd = &cobra.Command{
    19  	Use:   "balance",
    20  	Short: "Query an ERC20 token balance",
    21  	Long:  "The balance subcommand queries the balance of an account in an ERC20 contract",
    22  	PreRun: func(cmd *cobra.Command, args []string) {
    23  		logger.LoggerMetadata(cmd.Name(), cmd.Flags())
    24  	},
    25  	PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
    26  		return util.CallPersistentPreRun(cmd, args)
    27  	},
    28  	RunE: func(cmd *cobra.Command, args []string) error {
    29  		c, err := initialize.InitializeClient(url, senderKeyPair)
    30  		if err != nil {
    31  			return err
    32  		}
    33  		t, err := initialize.InitializeTransactor(gasPrice, evmtransaction.NewTransaction, c, prepare)
    34  		if err != nil {
    35  			return err
    36  		}
    37  		return BalanceCmd(cmd, args, erc20.NewERC20Contract(c, Erc20Addr, t))
    38  	},
    39  	Args: func(cmd *cobra.Command, args []string) error {
    40  		err := ValidateBalanceFlags(cmd, args)
    41  		if err != nil {
    42  			return err
    43  		}
    44  
    45  		ProcessBalanceFlags(cmd, args)
    46  		return nil
    47  	},
    48  }
    49  
    50  func BindBalanceFlags(cmd *cobra.Command) {
    51  	cmd.Flags().StringVar(&Erc20Address, "contract", "", "ERC20 contract address")
    52  	cmd.Flags().StringVar(&AccountAddress, "address", "", "Address to receive balance of")
    53  	flags.MarkFlagsAsRequired(cmd, "contract", "address")
    54  }
    55  
    56  func init() {
    57  	BindBalanceFlags(balanceCmd)
    58  }
    59  
    60  var accountAddr common.Address
    61  
    62  func ValidateBalanceFlags(cmd *cobra.Command, args []string) error {
    63  	if !common.IsHexAddress(Erc20Address) {
    64  		return fmt.Errorf("invalid recipient address %s", Recipient)
    65  	}
    66  	if !common.IsHexAddress(AccountAddress) {
    67  		return fmt.Errorf("invalid account address %s", AccountAddress)
    68  	}
    69  	return nil
    70  }
    71  
    72  func ProcessBalanceFlags(cmd *cobra.Command, args []string) {
    73  	Erc20Addr = common.HexToAddress(Erc20Address)
    74  	accountAddr = common.HexToAddress(AccountAddress)
    75  }
    76  
    77  func BalanceCmd(cmd *cobra.Command, args []string, contract *erc20.ERC20Contract) error {
    78  	balance, err := contract.GetBalance(accountAddr)
    79  	if err != nil {
    80  		log.Error().Err(fmt.Errorf("failed contract call error: %v", err))
    81  		return err
    82  	}
    83  
    84  	log.Info().Msgf("balance of %s is %s", accountAddr.String(), balance.String())
    85  	return nil
    86  }