From 3fe1231edd9924bac9611dd224df3824e8607220 Mon Sep 17 00:00:00 2001
From: Luis Carvalho <luis@volume.finance>
Date: Mon, 2 Sep 2024 10:51:11 +0100
Subject: [PATCH] fix: get-balance response parsing

---
 faucet.go | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/faucet.go b/faucet.go
index 6110edd..7605e18 100644
--- a/faucet.go
+++ b/faucet.go
@@ -79,14 +79,15 @@ type BalanceResponse struct {
 	Balance CoreCoin `json:"balance"`
 }
 
-func getBalance(address string) (amount int64) {
+func getBalance(address, denom string) (amount int64) {
 	cmd := exec.Command(
 		palomad,
 		"--node", rpcUrl,
-		"q", "bank", "balances",
+		"q", "bank", "balance",
 		"--output", "json",
 		"--chain-id", chainID,
 		address,
+		denom,
 	)
 	res, err := cmd.CombinedOutput()
 
@@ -103,6 +104,11 @@ func getBalance(address string) (amount int64) {
 	if err != nil {
 		panic(err)
 	}
+
+	if balance.Amount == "" {
+		return 0
+	}
+
 	amount, err = strconv.ParseInt(balance.Amount, 10, 64)
 	if err != nil {
 		panic(err)
@@ -154,7 +160,7 @@ func (requestLog *RequestLog) dripCoin(denom string) error {
 func checkAndUpdateLimit(db *leveldb.DB, account []byte, denom string) error {
 	address, _ := bech32.ConvertAndEncode("paloma", account)
 
-	if getBalance(address) >= amountTable[denom]*2 {
+	if getBalance(address, denom) >= amountTable[denom]*2 {
 		return errors.New("amount limit exceeded")
 	}
 
@@ -202,6 +208,7 @@ func createGetCoinsHandler(db *leveldb.DB) http.HandlerFunc {
 	return func(w http.ResponseWriter, request *http.Request) {
 		defer func() {
 			if err := recover(); err != nil {
+				fmt.Printf("req error: %v\n", err)
 				http.Error(w, err.(error).Error(), 400)
 			}
 		}()