wallet/pkg/rest/tron_v1_balance.go
Anton Nesterov f1bc53ce2a
[init]
2024-08-31 17:00:14 +02:00

86 lines
2 KiB
Go

package rest
import (
"net/http"
store "custodial/pkg/store"
trc "custodial/pkg/tron"
gin "github.com/gin-gonic/gin"
)
type TronAddressBalanceRequest struct {
Address string `json:"address"`
}
type TronAccountBalanceRequest struct {
AccountID string `json:"account_id"`
}
// @BasePath /api/v1
//
// AddressBalance godoc
//
// @Summary Get address balance
// @Schemes
// @Description Get address balance
// @Tags Tron
// @Accept json
// @Produce json
// @Security ApiKeyAuth
// @Param message body TronAddressBalanceRequest true "Account Info"
// @Success 200 {object} trc.TronAccountBalance
// @Failure 400 {object} ErrorResponse
// @Router /tron/address-balance [post]
func TronAddressBalanceV1(c *gin.Context) {
var req TronAddressBalanceRequest
if err := c.ShouldBindJSON(&req); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
balance, err := trc.TronBalance(req.Address)
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
c.JSON(http.StatusOK, balance)
}
// @BasePath /api/v1
//
// AddressBalance godoc
//
// @Summary Get address balance
// @Schemes
// @Description Get address balance
// @Tags Tron
// @Accept json
// @Produce json
// @Security ApiKeyAuth
// @Param message body TronAccountBalanceRequest true "Account Info"
// @Success 200 {object} trc.TronAccountBalance
// @Failure 400 {object} ErrorResponse
// @Router /tron/account-balance [post]
func TronAccountBalanceV1(c *gin.Context) {
var req TronAccountBalanceRequest
if err := c.ShouldBindJSON(&req); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
storedAccount, err := store.GetAccount(req.AccountID, store.Tron)
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
balance, err := trc.TronBalance(storedAccount.GetAddress())
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
c.JSON(http.StatusOK, balance)
}