Merge pull request #29 from the-go-dragons/develop

Added wallet balance endpoint
the-go-dragons · Jul 14, 2023 · 063dafe · 063dafe
2 parents 9d64b4a + 66f8a40
commit 063dafe
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/internal/app/app.go b/internal/app/app.go
@@ -99,6 +99,7 @@ func routing(e *echo.Echo) {
 
 	e.POST("/wallets/charge-request", walletHandler.CharageRequest, customeMiddleware.RequireAuth)
 	e.POST("/wallets/finalize-charge", walletHandler.FinalizeCharge, customeMiddleware.RequireAuth)
+	e.GET("/wallet/balance", walletHandler.GetUserBalance, customeMiddleware.RequireAuth)
 
 	e.GET("/numbers", numberHandler.GetAvailables)
 	e.GET("/numbers/user", numberHandler.GetUserNumbers, customeMiddleware.RequireAuth)

diff --git a/internal/interfaces/http/wallet_handler.go b/internal/interfaces/http/wallet_handler.go
@@ -2,6 +2,7 @@ package http
 
 import (
 	"net/http"
+	"strconv"
 
 	"github.com/labstack/echo/v4"
 	"github.com/labstack/gommon/log"
@@ -86,3 +87,14 @@ func (wh WalletHandler) FinalizeCharge(c echo.Context) error {
 	return c.JSON(http.StatusOK, WalletFinalizeCharageResponse{walletId})
 
 }
+
+func (wh WalletHandler) GetUserBalance(c echo.Context) error {
+	user := c.Get("user").(domain.User)
+
+	wallet, err := wh.walletService.GetByUserId(user.ID)
+	if err != nil || wallet.ID == 0 {
+		return c.JSON(http.StatusBadRequest, Response{Message: "Can't get the wallet"})
+	}
+
+	return c.JSON(http.StatusOK, Response{Message: strconv.Itoa(int(wallet.Balance))})
+}