60 lines
1.5 KiB
Go
60 lines
1.5 KiB
Go
package api
|
|
|
|
import (
|
|
"errors"
|
|
"git.solsynth.dev/hypernet/wallet/pkg/internal/server/exts"
|
|
|
|
"git.solsynth.dev/hypernet/nexus/pkg/nex/sec"
|
|
"git.solsynth.dev/hypernet/wallet/pkg/internal/database"
|
|
"git.solsynth.dev/hypernet/wallet/pkg/internal/models"
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/shopspring/decimal"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func createWallet(c *fiber.Ctx) error {
|
|
if err := sec.EnsureGrantedPerm(c, "CreateWallet", true); err != nil {
|
|
return err
|
|
}
|
|
user := c.Locals("user").(*sec.UserInfo)
|
|
|
|
var data struct {
|
|
Password string `json:"password" validate:"min=4"`
|
|
}
|
|
|
|
if err := exts.BindAndValidate(c, &data); err != nil {
|
|
return err
|
|
}
|
|
|
|
var wallet models.Wallet
|
|
if err := database.C.Where("account_id = ?", user.ID).
|
|
First(&wallet).Error; err == nil || errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return fiber.NewError(fiber.StatusConflict, "wallet already exists")
|
|
}
|
|
|
|
wallet = models.Wallet{
|
|
Balance: decimal.NewFromInt(0),
|
|
AccountID: user.ID,
|
|
}
|
|
|
|
if err := database.C.Create(&wallet).Error; err != nil {
|
|
return fiber.NewError(fiber.StatusInternalServerError, err.Error())
|
|
}
|
|
|
|
return c.JSON(wallet)
|
|
}
|
|
|
|
func getMyWallet(c *fiber.Ctx) error {
|
|
if err := sec.EnsureAuthenticated(c); err != nil {
|
|
return err
|
|
}
|
|
user := c.Locals("user").(*sec.UserInfo)
|
|
|
|
var wallet models.Wallet
|
|
if err := database.C.Where("account_id = ?", user.ID).First(&wallet).Error; err != nil {
|
|
return fiber.NewError(fiber.StatusNotFound, err.Error())
|
|
}
|
|
|
|
return c.JSON(wallet)
|
|
}
|