getBalance.go

package cmd

import (
	"context"
	"fmt"
	"os"

	"github.com/spf13/cobra"
	"github.com/spf13/viper"
)

var getBalanceCmd = &cobra.Command{
	Use:   "balance [account]",
	Short: "Retrieve currency balance for an account",
	Args:  cobra.ExactArgs(1),
	Run: func(cmd *cobra.Command, args []string) {
		api := getAPI()

		account := toAccount(args[0], "account")
		symbol := viper.GetString("get-balance-cmd-symbol")
		tokenAccount := toAccount(viper.GetString("get-balance-cmd-contract"), "--contract")

		balances, err := api.GetCurrencyBalance(context.Background(), account, symbol, tokenAccount)
		if err != nil {
			fmt.Printf("Error: get balance: %s\n", err)
			os.Exit(1)
		}

		for _, asset := range balances {
			fmt.Printf("%s\n", asset)
		}
	},
}

func init() {
	getCmd.AddCommand(getBalanceCmd)

	getBalanceCmd.Flags().StringP("contract", "", "eosio.token", "Account managing the token")
	getBalanceCmd.Flags().StringP("symbol", "", "", "Only query this symbol. Try EOS")
}