129 lines
3.2 KiB
Go
129 lines
3.2 KiB
Go
package adminauth
|
|
|
|
import (
|
|
"errors"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"hfb_sys/backend/internal/middleware"
|
|
"hfb_sys/backend/pkg/response"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type Handler struct {
|
|
service *Service
|
|
}
|
|
|
|
func NewHandler(service *Service) *Handler {
|
|
return &Handler{service: service}
|
|
}
|
|
|
|
func (h *Handler) Captcha(c *gin.Context) {
|
|
item, err := h.service.Captcha(c.Request.Context())
|
|
if err != nil {
|
|
writeAdminAuthError(c, err)
|
|
return
|
|
}
|
|
response.OK(c, item)
|
|
}
|
|
|
|
func (h *Handler) Login(c *gin.Context) {
|
|
var req LoginRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
response.BadRequest(c, "用户名、密码和验证码不能为空")
|
|
return
|
|
}
|
|
result, err := h.service.Login(c.Request.Context(), strings.TrimSpace(req.Username), req.Password, strings.TrimSpace(req.CaptchaID), strings.TrimSpace(req.CaptchaCode))
|
|
if err != nil {
|
|
writeAdminAuthError(c, err)
|
|
return
|
|
}
|
|
response.OK(c, result)
|
|
}
|
|
|
|
func (h *Handler) Me(c *gin.Context) {
|
|
value, ok := c.Get(middleware.ContextAdminID)
|
|
if !ok {
|
|
response.Unauthorized(c, "缺少管理员上下文")
|
|
return
|
|
}
|
|
adminID, ok := value.(uint64)
|
|
if !ok {
|
|
response.Unauthorized(c, "管理员上下文无效")
|
|
return
|
|
}
|
|
admin, err := h.service.Me(c.Request.Context(), adminID)
|
|
if err != nil {
|
|
writeAdminAuthError(c, err)
|
|
return
|
|
}
|
|
response.OK(c, admin)
|
|
}
|
|
|
|
func (h *Handler) Logout(c *gin.Context) {
|
|
response.OK(c, gin.H{"logged_out": true})
|
|
}
|
|
|
|
func (h *Handler) UpdateSupportStatus(c *gin.Context) {
|
|
adminID, ok := currentAdminID(c)
|
|
if !ok {
|
|
response.Unauthorized(c, "缺少管理员上下文")
|
|
return
|
|
}
|
|
var req UpdateSupportStatusRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
response.BadRequest(c, "状态值无效，必须是 online、offline 或 busy")
|
|
return
|
|
}
|
|
if err := h.service.UpdateSupportStatus(c.Request.Context(), adminID, req.Status); err != nil {
|
|
writeAdminAuthError(c, err)
|
|
return
|
|
}
|
|
response.OK(c, gin.H{"updated": true, "support_status": req.Status})
|
|
}
|
|
|
|
func currentAdminID(c *gin.Context) (uint64, bool) {
|
|
value, ok := c.Get(middleware.ContextAdminID)
|
|
if !ok {
|
|
return 0, false
|
|
}
|
|
adminID, ok := value.(uint64)
|
|
return adminID, ok
|
|
}
|
|
|
|
type AdminRefreshRequest struct {
|
|
RefreshToken string `json:"refresh_token" binding:"required"`
|
|
}
|
|
|
|
func (h *Handler) Refresh(c *gin.Context) {
|
|
var req AdminRefreshRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
response.BadRequest(c, "refresh_token 不能为空")
|
|
return
|
|
}
|
|
tokens, err := h.service.Refresh(c.Request.Context(), req.RefreshToken)
|
|
if err != nil {
|
|
writeAdminAuthError(c, err)
|
|
return
|
|
}
|
|
response.OK(c, tokens)
|
|
}
|
|
|
|
func writeAdminAuthError(c *gin.Context, err error) {
|
|
switch {
|
|
case errors.Is(err, ErrDependencyUnavailable):
|
|
response.ServiceUnavailable(c, "数据库未连接")
|
|
case errors.Is(err, ErrInvalidRefreshToken):
|
|
response.Unauthorized(c, "刷新令牌无效或已过期")
|
|
case errors.Is(err, ErrInvalidCredential):
|
|
response.BadRequest(c, "用户名或密码错误")
|
|
case errors.Is(err, ErrCaptchaInvalid):
|
|
response.BadRequest(c, "验证码错误或已过期")
|
|
case errors.Is(err, ErrAdminDisabled):
|
|
response.Error(c, http.StatusForbidden, "admin_disabled", "管理员已禁用")
|
|
default:
|
|
response.Error(c, http.StatusInternalServerError, "internal_error", "后台认证服务暂时不可用")
|
|
}
|
|
}
|