freefire
6da4309453
此提交完成了全站管理员后台核心功能开发： 1. 新增AdminUserService与管理接口，支持用户列表查询、角色变更与封禁解封 2. 重构认证链路，统一账号封禁错误码与处理逻辑 3. 新增全局强制下线机制，账号被封禁时自动弹出告知弹窗并清理登录态 4. 重构管理员后台路由与布局，统一权限校验逻辑 5. 优化通知轮询逻辑，页面切回前台时立即刷新未读计数 6. 合并原公告、外观管理页到统一后台面板
125 lines
3.2 KiB
Go
125 lines
3.2 KiB
Go
package middleware
|
||
|
||
import (
|
||
"errors"
|
||
"net/http"
|
||
"strings"
|
||
|
||
"github.com/freefire/jiang13-bbs/service"
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// AccountBannedKey 写入 gin.Context 的标记：当前凭据所属账号已被封禁。
|
||
// parseToken 失败原因对 handler 不可见，通过 context 显式传递，
|
||
// 以便 /me 等 OptionalAuth 接口也能告知前端"被封禁"而非"未登录"。
|
||
const AccountBannedKey = "account_banned"
|
||
|
||
// 封禁响应体：code 供前端机器识别，error 供直接展示
|
||
func bannedJSON(c *gin.Context) {
|
||
c.AbortWithStatusJSON(http.StatusForbidden, gin.H{
|
||
"error": "账号已被封禁",
|
||
"code": "account_banned",
|
||
})
|
||
}
|
||
|
||
// AuthMiddleware 认证中间件
|
||
type AuthMiddleware struct {
|
||
auth *service.AuthService
|
||
}
|
||
|
||
func NewAuthMiddleware(auth *service.AuthService) *AuthMiddleware {
|
||
return &AuthMiddleware{auth: auth}
|
||
}
|
||
|
||
// OptionalAuth 可选登录：解析 token，失败不阻断
|
||
func (m *AuthMiddleware) OptionalAuth() gin.HandlerFunc {
|
||
return func(c *gin.Context) {
|
||
user, ok := m.parseToken(c)
|
||
if ok {
|
||
c.Set("user", user)
|
||
}
|
||
c.Next()
|
||
}
|
||
}
|
||
|
||
// RequireAuth 必须登录
|
||
func (m *AuthMiddleware) RequireAuth() gin.HandlerFunc {
|
||
return func(c *gin.Context) {
|
||
user, ok := m.parseToken(c)
|
||
if !ok {
|
||
if c.GetBool(AccountBannedKey) {
|
||
bannedJSON(c)
|
||
return
|
||
}
|
||
c.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{"error": "未登录"})
|
||
return
|
||
}
|
||
c.Set("user", user)
|
||
c.Next()
|
||
}
|
||
}
|
||
|
||
// RequireAdmin 必须管理员
|
||
func (m *AuthMiddleware) RequireAdmin() gin.HandlerFunc {
|
||
return func(c *gin.Context) {
|
||
user, ok := m.parseToken(c)
|
||
if !ok {
|
||
if c.GetBool(AccountBannedKey) {
|
||
bannedJSON(c)
|
||
return
|
||
}
|
||
c.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{"error": "未登录"})
|
||
return
|
||
}
|
||
if user.Role != service.RoleAdmin {
|
||
c.AbortWithStatusJSON(http.StatusForbidden, gin.H{"error": "需要管理员权限"})
|
||
return
|
||
}
|
||
c.Set("user", user)
|
||
c.Next()
|
||
}
|
||
}
|
||
|
||
// parseToken 解析并校验 token：
|
||
// 1. 优先从 HttpOnly cookie 读取，回退 Authorization header
|
||
// 2. 校验 JWT 签名和过期
|
||
// 3. 查 DB 实时校验 token_version 和 banned 状态
|
||
func (m *AuthMiddleware) parseToken(c *gin.Context) (*service.UserClaims, bool) {
|
||
tokenStr, err := c.Cookie(service.CookieName)
|
||
if err != nil || tokenStr == "" {
|
||
auth := c.GetHeader("Authorization")
|
||
if auth == "" {
|
||
return nil, false
|
||
}
|
||
tokenStr = strings.TrimPrefix(auth, "Bearer ")
|
||
if tokenStr == auth {
|
||
return nil, false
|
||
}
|
||
}
|
||
claims, err := m.auth.ParseToken(tokenStr)
|
||
if err != nil {
|
||
return nil, false
|
||
}
|
||
// 实时校验：token_version 匹配 + 未封禁（防止旧 JWT 在封禁/改密码后仍有效）
|
||
if _, err := m.auth.ValidateClaims(claims); err != nil {
|
||
// 封禁原因写入 context：401 与 403 的区分由上层中间件/handler 完成
|
||
if errors.Is(err, service.ErrAccountBanned) {
|
||
c.Set(AccountBannedKey, true)
|
||
}
|
||
return nil, false
|
||
}
|
||
// 异步刷新在线心跳（SQL 每 60s 限频一次），不阻塞请求
|
||
go m.auth.TouchLastSeen(claims.ID)
|
||
return claims, true
|
||
}
|
||
|
||
// CurrentUser 从 context 获取当前用户
|
||
func CurrentUser(c *gin.Context) *service.UserClaims {
|
||
if v, ok := c.Get("user"); ok {
|
||
if u, ok := v.(*service.UserClaims); ok {
|
||
return u
|
||
}
|
||
}
|
||
return nil
|
||
}
|