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

149 lines
3.7 KiB
Go

package rest
import (
"net/http"
store "custodial/pkg/store"
trc "custodial/pkg/tron"
gin "github.com/gin-gonic/gin"
)
type TRXAprroveUSDTRequest struct {
AccountID string `json:"account_id"`
}
type TRXAprroveContractRequest struct {
AccountID string `json:"account_id"`
ConractAddress string `json:"contract_address"`
}
type TRXApproveUSDTResponse struct {
Status string `json:"status"`
TXID string `json:"txid"`
}
// @BasePath /api/v1
//
// TronApproveUsdtV1 godoc
//
// @Summary Approve USDT Spender
// @Schemes
// @Description Approve USDT spender for account
// @Tags Tron
// @Accept json
// @Produce json
// @Security ApiKeyAuth
// @Param message body TRXAprroveUSDTRequest true "Account Info"
// @Success 200 {object} TRXApproveUSDTResponse
// @Failure 400 {object} ErrorResponse
// @Router /tron/approve-usdt-spender [post]
func TronApproveUsdtSpenderV1(c *gin.Context) {
var req TRXAprroveUSDTRequest
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
}
// if storedAccount.GetSpender() != "" {
// c.JSON(http.StatusBadRequest, gin.H{"error": "Spender already approved"})
// return
// }
trcSpender, _ := trc.GetSpender()
approveAmount := "1000000000000"
hdIndex := storedAccount.GetHDIndex()
account, err := trc.DeriveAccount(hdIndex)
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
balance, err := trc.TronBalance(account.Address())
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
}
if balance.TRXRaw < 1000000 {
c.JSON(http.StatusBadRequest, gin.H{"error": "Not enough TRX to pay for transaction"})
return
}
txid, err := account.ApproveUSDTSpender(trcSpender.Address(), approveAmount)
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
storedAccount.(*store.TronAccount).Spender = trcSpender.Address()
store.UpdateAccount(storedAccount)
c.JSON(http.StatusOK, gin.H{"txid": txid, "status": "ok"})
}
// @BasePath /api/v1
//
// TronApproveContractV1 godoc
//
// @Summary Approve USDT Spender Contract
// @Schemes
// @Description Approve USDT spender for contract
// @Tags Tron
// @Accept json
// @Produce json
// @Security ApiKeyAuth
// @Param message body TRXAprroveContractRequest true "Account Info"
// @Success 200 {object} TRXApproveUSDTResponse
// @Failure 400 {object} ErrorResponse
// @Router /tron/approve-usdt-contract [post]
func TronApproveUsdtContractV1(c *gin.Context) {
var req TRXAprroveContractRequest
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
}
approveAmount := "1000000000000"
hdIndex := storedAccount.GetHDIndex()
account, err := trc.DeriveAccount(hdIndex)
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
balance, err := trc.TronBalance(account.Address())
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
}
if balance.TRXRaw < 1000000 {
c.JSON(http.StatusBadRequest, gin.H{"error": "Not enough TRX to pay for transaction"})
return
}
txid, err := account.ApproveUSDTSpender(req.ConractAddress, approveAmount)
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
storedAccount.(*store.TronAccount).Spender = req.ConractAddress
store.UpdateAccount(storedAccount)
c.JSON(http.StatusOK, gin.H{"txid": txid, "status": "ok"})
}