120 lines
3.2 KiB
Go
120 lines
3.2 KiB
Go
package qualityowner
|
|
|
|
import (
|
|
"errors"
|
|
"net/http"
|
|
"strconv"
|
|
|
|
"hfb_sys/backend/internal/middleware"
|
|
"hfb_sys/backend/pkg/response"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type Handler struct {
|
|
service *Service
|
|
}
|
|
|
|
func NewHandler(service *Service) *Handler {
|
|
return &Handler{service: service}
|
|
}
|
|
|
|
func (h *Handler) Status(c *gin.Context) {
|
|
userID, ok := currentUserID(c)
|
|
if !ok {
|
|
response.Unauthorized(c, "缺少用户上下文")
|
|
return
|
|
}
|
|
item, err := h.service.Status(c.Request.Context(), userID)
|
|
if err != nil {
|
|
writeError(c, err)
|
|
return
|
|
}
|
|
response.OK(c, item)
|
|
}
|
|
|
|
func (h *Handler) Activate(c *gin.Context) {
|
|
userID, ok := currentUserID(c)
|
|
if !ok {
|
|
response.Unauthorized(c, "缺少用户上下文")
|
|
return
|
|
}
|
|
item, err := h.service.Activate(c.Request.Context(), userID)
|
|
if err != nil {
|
|
writeError(c, err)
|
|
return
|
|
}
|
|
response.OK(c, item)
|
|
}
|
|
|
|
func (h *Handler) AdminSet(c *gin.Context) {
|
|
adminID, ok := currentAdminID(c)
|
|
if !ok {
|
|
response.Unauthorized(c, "缺少管理员上下文")
|
|
return
|
|
}
|
|
userID, err := strconv.ParseUint(c.Param("id"), 10, 64)
|
|
if err != nil || userID == 0 {
|
|
response.BadRequest(c, "用户 ID 不正确")
|
|
return
|
|
}
|
|
var req AdminSetRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
response.BadRequest(c, "请填写操作原因")
|
|
return
|
|
}
|
|
item, err := h.service.AdminSet(c.Request.Context(), adminID, userID, req, AuditMeta{
|
|
IP: c.ClientIP(),
|
|
UserAgent: c.GetHeader("User-Agent"),
|
|
RequestID: middleware.GetRequestID(c),
|
|
})
|
|
if err != nil {
|
|
writeError(c, err)
|
|
return
|
|
}
|
|
response.OK(c, item)
|
|
}
|
|
|
|
func currentUserID(c *gin.Context) (uint64, bool) {
|
|
value, ok := c.Get(middleware.ContextUserID)
|
|
if !ok {
|
|
return 0, false
|
|
}
|
|
id, ok := value.(uint64)
|
|
return id, ok
|
|
}
|
|
|
|
func currentAdminID(c *gin.Context) (uint64, bool) {
|
|
value, ok := c.Get(middleware.ContextAdminID)
|
|
if !ok {
|
|
return 0, false
|
|
}
|
|
id, ok := value.(uint64)
|
|
return id, ok
|
|
}
|
|
|
|
func writeError(c *gin.Context, err error) {
|
|
response.RecordError(c, err)
|
|
switch {
|
|
case errors.Is(err, ErrDependencyUnavailable):
|
|
response.ServiceUnavailable(c, "数据库未连接")
|
|
case errors.Is(err, ErrInvalidUser):
|
|
response.BadRequest(c, "用户不符合开通条件，请先完成实名认证")
|
|
case errors.Is(err, ErrDepositInsufficient):
|
|
response.Error(c, http.StatusConflict, "insufficient_balance", "余额不足，请先确保可用余额足够缴纳保证金")
|
|
case errors.Is(err, ErrAlreadyActive):
|
|
response.Error(c, http.StatusConflict, "quality_owner_active", "当前已经是优质号主")
|
|
case errors.Is(err, ErrPaymentPending):
|
|
response.Error(c, http.StatusConflict, "quality_owner_payment_pending", "已有待完成的保证金支付，请先完成支付或稍后重试")
|
|
case errors.Is(err, ErrInactive):
|
|
response.Error(c, http.StatusConflict, "quality_owner_inactive", "当前不是优质号主")
|
|
case errors.Is(err, ErrInvalidReason):
|
|
response.BadRequest(c, "请填写 2-200 字的操作原因")
|
|
case errors.Is(err, gorm.ErrRecordNotFound):
|
|
response.Error(c, http.StatusNotFound, "not_found", "用户不存在")
|
|
default:
|
|
response.Error(c, http.StatusInternalServerError, "internal_error", "优质号主服务暂时不可用")
|
|
}
|
|
}
|