platform/web/handlers/user.go

package handlers

import (
	"errors"
	"platform/web/auth"
	q "platform/web/queries"
	s "platform/web/services"
	"strconv"
	"time"

	"github.com/gofiber/fiber/v2"
	"gorm.io/gorm"
)

// region GetUserByToken

type GetUserByTokenReq struct {
	Token string `json:"token" validate:"required"`
}

// GetUserByToken 通过token获取用户信息
func GetUserByToken(c *fiber.Ctx) error {

	// 解析请求参数
	req := new(GetUserByTokenReq)
	if err := c.BodyParser(req); err != nil {
		return err
	}

	// 查询会话信息
	session, err := s.Session.Find(c.Context(), req.Token)
	if err != nil {
		return err
	}

	// 查询用户信息
	user, err := q.User.Debug().
		Omit(q.User.Password, q.User.DeletedAt).
		Where(q.User.ID.Eq(session.Payload.Id)).
		Take()
	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return fiber.NewError(fiber.StatusNotFound, "用户不存在")
		}
		return err
	}

	// 返回用户信息
	return c.JSON(user)
}

// endregion

// region recharge

type RechargePrepareReq struct {
	Amount float64 `json:"amount" validate:"required,min=0.01"`
}

type RechargePrepareResp struct {
	TradeNo string `json:"trade_no"`
	PayURL  string `json:"pay_url"`
}

type RechargeConfirmReq struct {
	TradeNo string `json:"trade_no" validate:"required"`
}

type RechargeConfirmResp struct {
	Status string `json:"status"`
}

// RechargePrepareAlipay 通过支付宝充值
func RechargePrepareAlipay(c *fiber.Ctx) error {
	// 检查权限
	authContext, err := auth.Protect(c, []s.PayloadType{s.PayloadUser}, []string{})
	if err != nil {
		return err
	}

	// 解析请求参数
	req := new(RechargePrepareReq)
	if err := c.BodyParser(req); err != nil {
		return err
	}

	// 保存交易信息
	var result *s.TransactionPrepareResult
	err = q.Q.Transaction(func(tx *q.Query) error {
		result, err = s.Transaction.PrepareTransaction(c.Context(), tx, authContext.Payload.Id, &s.TransactionPrepareData{
			Subject:  "账户充值 - " + strconv.FormatFloat(req.Amount, 'f', 2, 64) + "元",
			Amount:   req.Amount,
			ExpireAt: time.Now().Add(30 * time.Minute),
			Type:     s.TransactionTypeRecharge,
			Method:   s.TransactionMethodAlipay,
		})
		return err
	})
	if err != nil {
		return err
	}

	// 返回结果
	return c.JSON(RechargePrepareResp{
		TradeNo: result.TradeNo,
		PayURL:  result.PayURL,
	})
}

func RechargeConfirmAlipay(c *fiber.Ctx) error {
	// 检查权限
	_, err := auth.Protect(c, []s.PayloadType{s.PayloadUser}, []string{})
	if err != nil {
		return err
	}

	// 解析请求参数
	req := new(RechargeConfirmReq)
	if err := c.BodyParser(req); err != nil {
		return err
	}

	// 验证支付结果
	result, err := s.Transaction.VerifyTransaction(c.Context(), &s.TransactionVerifyData{
		TradeNo: req.TradeNo,
		Method:  s.TransactionMethodAlipay,
	})
	if err != nil {
		return err
	}

	// 更新数据库
	err = s.User.RechargeConfirm(c.Context(), req.TradeNo, result)
	if err != nil {
		return err
	}

	return c.JSON(fiber.Map{"status": "success"})
}

func RechargePrepareWechat(c *fiber.Ctx) error {
	// 检查权限
	authContext, err := auth.Protect(c, []s.PayloadType{s.PayloadUser}, []string{})
	if err != nil {
		return err
	}

	// 解析请求参数
	req := new(RechargePrepareReq)
	if err := c.BodyParser(req); err != nil {
		return err
	}

	// 保存交易信息
	var result *s.TransactionPrepareResult
	err = q.Q.Transaction(func(tx *q.Query) error {
		result, err = s.Transaction.PrepareTransaction(c.Context(), tx, authContext.Payload.Id, &s.TransactionPrepareData{
			Subject:  "账户充值 - " + strconv.FormatFloat(req.Amount, 'f', 2, 64) + "元",
			Amount:   req.Amount,
			ExpireAt: time.Now().Add(30 * time.Minute),
			Type:     s.TransactionTypeRecharge,
			Method:   s.TransactionMethodWeChat,
		})
		return err
	})
	if err != nil {
		return err
	}

	// 返回结果
	return c.JSON(RechargePrepareResp{
		TradeNo: result.TradeNo,
		PayURL:  result.PayURL,
	})
}

func RechargeConfirmWechat(c *fiber.Ctx) error {
	// 检查权限
	_, err := auth.Protect(c, []s.PayloadType{s.PayloadUser}, []string{})
	if err != nil {
		return err
	}

	// 解析请求参数
	req := new(struct {
		TradeNo string `json:"trade_no" validate:"required"`
	})
	if err := c.BodyParser(req); err != nil {
		return err
	}

	// 验证支付结果
	result, err := s.Transaction.VerifyTransaction(c.Context(), &s.TransactionVerifyData{
		TradeNo: req.TradeNo,
	})
	if err != nil {
		return err
	}

	// 更新数据库
	err = s.User.RechargeConfirm(c.Context(), req.TradeNo, result)
	if err != nil {
		return err
	}

	return c.JSON(fiber.Map{"status": "success"})
}

// endregion