solosw
86 lines
2.3 KiB
Go
86 lines
2.3 KiB
Go
package handlers
|
|
|
|
import (
|
|
"awesomeProject/internal/common"
|
|
"awesomeProject/internal/models"
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// AdminPaymentHandler 管理员付款方式处理器
|
|
type AdminPaymentHandler struct{}
|
|
|
|
// GetPaymentInfo 获取管理员付款方式信息
|
|
func (h *AdminPaymentHandler) GetPaymentInfo(c *gin.Context) {
|
|
db := common.GetDB()
|
|
|
|
// 查询系统角色为管理员的用户的付款信息
|
|
var admin models.User
|
|
err := db.Where("system_role = ?", 0).Select("main_payment_qr_image, sub_payment_qr_image").First(&admin).Error
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, common.Error(500, "查询管理员付款信息失败"))
|
|
return
|
|
}
|
|
|
|
// 返回管理员付款方式信息
|
|
paymentInfo := map[string]interface{}{
|
|
"main_payment_qr_image": admin.MainPaymentQrImage,
|
|
"sub_payment_qr_image": admin.SubPaymentQrImage,
|
|
}
|
|
|
|
c.JSON(http.StatusOK, common.Success(paymentInfo))
|
|
}
|
|
|
|
// UpdatePaymentInfo 更新管理员付款方式信息
|
|
func (h *AdminPaymentHandler) UpdatePaymentInfo(c *gin.Context) {
|
|
currentUser, _ := c.Get("current_user")
|
|
user := currentUser.(models.User)
|
|
|
|
// 验证是否为管理员
|
|
if user.SystemRole != 0 {
|
|
c.JSON(http.StatusOK, common.Error(403, "权限不足"))
|
|
return
|
|
}
|
|
|
|
// 解析请求数据
|
|
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))
|
|
} |