query VFBC

Author: 0xbcdev

cmd/query/balance.go

@@ -66,12 +66,15 @@ func GetQueryBalanceCommand() *cobra.Command {
 				erc20TokenPairs, err := fetchErc20ModuleTokenPairsFromRest(restApiEndpoint)
 				if err != nil {
 					utils.PrintlnStdErr("ERR:", err)
-					return
 				} else {
 					for _, erc20TokenPair := range erc20TokenPairs {
+						if !erc20TokenPair.Enabled {
+							continue
+						}
+
 						contractAddr := common.HexToAddress(erc20TokenPair.Erc20Address)
 
-						tokenBalance, tokenBalanceDisplay, contractSymbol, contractDecimals, balancePartHigh, balancePartLow, err := fetchBalanceForErc20Contract(contractAddr, contextHeight, ethClient8545, accountAddr, "contract")
+						tokenBalance, tokenBalanceDisplay, contractSymbol, contractDecimals, balancePartHigh, balancePartLow, err := fetchBalanceForErc20Contract(contractAddr, contextHeight, ethClient8545, accountAddr, "x/erc20 contract")
 						if err != nil {
 							continue
 						}
@@ -83,6 +86,30 @@ func GetQueryBalanceCommand() *cobra.Command {
 						printRow("x/erc20", contractAddr.String(), contractSymbol, tokenBalanceDisplay, tokenBalance.String(), contractDecimals.String(), balancePartHigh.String(), balancePartLow.String(), erc20TokenPair.Denom)
 					}
 				}
+
+				vfcbPairs, err := fetchVirtualFrontierBankContractPairsFromRest(restApiEndpoint)
+				if err != nil {
+					utils.PrintlnStdErr("ERR:", err)
+				} else {
+					for _, vfbcPair := range vfcbPairs {
+						if !vfbcPair.Enabled {
+							continue
+						}
+
+						contractAddr := common.HexToAddress(vfbcPair.ContractAddress)
+
+						tokenBalance, tokenBalanceDisplay, contractSymbol, contractDecimals, balancePartHigh, balancePartLow, err := fetchBalanceForErc20Contract(contractAddr, contextHeight, ethClient8545, accountAddr, "VFBC")
+						if err != nil {
+							continue
+						}
+
+						if tokenBalance.Sign() == 0 {
+							continue
+						}
+
+						printRow("vfbc", contractAddr.String(), contractSymbol, tokenBalanceDisplay, tokenBalance.String(), contractDecimals.String(), balancePartHigh.String(), balancePartLow.String(), vfbcPair.MinDenom)
+					}
+				}
 			}
 		},
 	}

cmd/query/cosmos.go

@@ -82,3 +82,44 @@ type Erc20ModuleTokenPair struct {
 	Denom        string `json:"denom"`
 	Enabled      bool   `json:"enabled"`
 }
+
+func fetchVirtualFrontierBankContractPairsFromRest(rest string) (vfbcPairs []VfbcTokenPair, err error) {
+	var resp *http.Response
+	resp, err = http.Get(rest + "/ethermint/evm/v1/virtual_frontier_bank_contracts")
+	if err != nil {
+		err = errors.Wrap(err, "failed to fetch VFBC pairs")
+		return
+	}
+
+	if resp.StatusCode != http.StatusOK {
+		return nil, fmt.Errorf("failed to fetch VFBC pairs! Status code: %d", resp.StatusCode)
+	}
+
+	type responseStruct struct {
+		Pairs []VfbcTokenPair `json:"pairs"`
+	}
+
+	defer func() {
+		_ = resp.Body.Close()
+	}()
+
+	bz, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return nil, errors.Wrap(err, "failed to read response body of VFBC pairs")
+	}
+
+	var response responseStruct
+	err = json.Unmarshal(bz, &response)
+	if err != nil {
+		return nil, errors.Wrap(err, "failed to unmarshal response body of VFBC pairs")
+	}
+
+	vfbcPairs = response.Pairs
+	return
+}
+
+type VfbcTokenPair struct {
+	ContractAddress string `json:"contract_address"`
+	MinDenom        string `json:"min_denom"`
+	Enabled         bool   `json:"enabled"`
+}