1.0

internal/handlers/admin_payment_handler.go

@@ -0,0 +1,86 @@
+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))
+}