添加支付宝和微信充值功能，重构交易处理逻辑，优化资源创建与支付链接生成

web/handlers/user.go

@@ -2,8 +2,11 @@ package handlers
 
 import (
 	"errors"
+	"platform/web/auth"
 	q "platform/web/queries"
-	"platform/web/services"
+	s "platform/web/services"
+	"strconv"
+	"time"
 
 	"github.com/gofiber/fiber/v2"
 	"gorm.io/gorm"
@@ -25,7 +28,7 @@ func GetUserByToken(c *fiber.Ctx) error {
 	}
 
 	// 查询会话信息
-	session, err := services.Session.Find(c.Context(), req.Token)
+	session, err := s.Session.Find(c.Context(), req.Token)
 	if err != nil {
 		return err
 	}
@@ -47,3 +50,160 @@ func GetUserByToken(c *fiber.Ctx) error {
 }
 
 // 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