solosw
108 lines
2.9 KiB
Go
108 lines
2.9 KiB
Go
package handlers
|
|
|
|
import (
|
|
"awesomeProject/internal/common"
|
|
"awesomeProject/internal/models"
|
|
"net/http"
|
|
"strconv"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// UserPaymentHandler 用户收款方式处理器
|
|
type UserPaymentHandler struct{}
|
|
|
|
// GetPaymentInfo 获取用户收款方式信息
|
|
func (h *UserPaymentHandler) GetPaymentInfo(c *gin.Context) {
|
|
currentUser, _ := c.Get("current_user")
|
|
user := currentUser.(models.User)
|
|
|
|
// 返回收款方式信息
|
|
paymentInfo := map[string]interface{}{
|
|
"main_payment_qr_image": user.MainPaymentQrImage,
|
|
"sub_payment_qr_image": user.SubPaymentQrImage,
|
|
}
|
|
|
|
c.JSON(http.StatusOK, common.Success(paymentInfo))
|
|
}
|
|
|
|
// UpdatePaymentInfo 更新用户收款方式信息
|
|
func (h *UserPaymentHandler) UpdatePaymentInfo(c *gin.Context) {
|
|
currentUser, _ := c.Get("current_user")
|
|
user := currentUser.(models.User)
|
|
|
|
// 解析请求数据
|
|
var req struct {
|
|
MainPaymentQrImage string `json:"main_payment_qr_image"`
|
|
SubPaymentQrImage string `json:"sub_payment_qr_image"`
|
|
}
|
|
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
c.JSON(http.StatusOK, common.Error(400, "请求参数错误"))
|
|
return
|
|
}
|
|
|
|
// 更新用户收款方式信息
|
|
updates := map[string]interface{}{}
|
|
if req.MainPaymentQrImage != "" {
|
|
updates["main_payment_qr_image"] = req.MainPaymentQrImage
|
|
}
|
|
if req.SubPaymentQrImage != "" {
|
|
updates["sub_payment_qr_image"] = req.SubPaymentQrImage
|
|
}
|
|
|
|
if len(updates) == 0 {
|
|
c.JSON(http.StatusOK, common.Error(400, "没有要更新的信息"))
|
|
return
|
|
}
|
|
|
|
if err := common.GetDB().Model(&user).Updates(updates).Error; err != nil {
|
|
c.JSON(http.StatusOK, common.Error(500, "更新收款方式失败"))
|
|
return
|
|
}
|
|
|
|
// 返回更新后的信息
|
|
var updatedUser models.User
|
|
common.GetDB().First(&updatedUser, user.ID)
|
|
|
|
paymentInfo := map[string]interface{}{
|
|
"main_payment_qr_image": updatedUser.MainPaymentQrImage,
|
|
"sub_payment_qr_image": updatedUser.SubPaymentQrImage,
|
|
}
|
|
|
|
c.JSON(http.StatusOK, common.Success(paymentInfo))
|
|
}
|
|
|
|
// GetSellerPaymentInfo 获取卖家收款方式信息
|
|
func (h *UserPaymentHandler) GetSellerPaymentInfo(c *gin.Context) {
|
|
sellerID, err := strconv.ParseUint(c.Param("id"), 10, 32)
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, common.Error(400, "卖家ID无效"))
|
|
return
|
|
}
|
|
|
|
var seller models.User
|
|
err = common.GetDB().Select("main_payment_qr_image, sub_payment_qr_image").First(&seller, sellerID).Error
|
|
if err != nil {
|
|
if err == gorm.ErrRecordNotFound {
|
|
c.JSON(http.StatusOK, common.Error(404, "卖家不存在"))
|
|
} else {
|
|
c.JSON(http.StatusOK, common.Error(500, "查询卖家信息失败"))
|
|
}
|
|
return
|
|
}
|
|
|
|
// 返回卖家收款方式信息
|
|
paymentInfo := map[string]interface{}{
|
|
"main_payment_qr_image": seller.MainPaymentQrImage,
|
|
"sub_payment_qr_image": seller.SubPaymentQrImage,
|
|
}
|
|
|
|
c.JSON(http.StatusOK, common.Success(paymentInfo))
|
|
}
|
|
|
|
func (h *UserPaymentHandler) GetSellerInfo(c *gin.Context) {
|
|
|
|
}
|