141 lines
3.6 KiB
Go
141 lines
3.6 KiB
Go
package middleware
|
||
|
||
import (
|
||
"crypto/subtle"
|
||
"errors"
|
||
"strings"
|
||
|
||
"github.com/Wei-Shaw/sub2api/internal/service"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// NewAdminAuthMiddleware 创建管理员认证中间件
|
||
func NewAdminAuthMiddleware(
|
||
authService *service.AuthService,
|
||
userService *service.UserService,
|
||
settingService *service.SettingService,
|
||
) AdminAuthMiddleware {
|
||
return AdminAuthMiddleware(adminAuth(authService, userService, settingService))
|
||
}
|
||
|
||
// adminAuth 管理员认证中间件实现
|
||
// 支持两种认证方式（通过不同的 header 区分）：
|
||
// 1. Admin API Key: x-api-key: <admin-api-key>
|
||
// 2. JWT Token: Authorization: Bearer <jwt-token> (需要管理员角色)
|
||
func adminAuth(
|
||
authService *service.AuthService,
|
||
userService *service.UserService,
|
||
settingService *service.SettingService,
|
||
) gin.HandlerFunc {
|
||
return func(c *gin.Context) {
|
||
// 检查 x-api-key header（Admin API Key 认证）
|
||
apiKey := c.GetHeader("x-api-key")
|
||
if apiKey != "" {
|
||
if !validateAdminApiKey(c, apiKey, settingService, userService) {
|
||
return
|
||
}
|
||
c.Next()
|
||
return
|
||
}
|
||
|
||
// 检查 Authorization header（JWT 认证）
|
||
authHeader := c.GetHeader("Authorization")
|
||
if authHeader != "" {
|
||
parts := strings.SplitN(authHeader, " ", 2)
|
||
if len(parts) == 2 && parts[0] == "Bearer" {
|
||
if !validateJWTForAdmin(c, parts[1], authService, userService) {
|
||
return
|
||
}
|
||
c.Next()
|
||
return
|
||
}
|
||
}
|
||
|
||
// 无有效认证信息
|
||
AbortWithError(c, 401, "UNAUTHORIZED", "Authorization required")
|
||
}
|
||
}
|
||
|
||
// validateAdminApiKey 验证管理员 API Key
|
||
func validateAdminApiKey(
|
||
c *gin.Context,
|
||
key string,
|
||
settingService *service.SettingService,
|
||
userService *service.UserService,
|
||
) bool {
|
||
storedKey, err := settingService.GetAdminApiKey(c.Request.Context())
|
||
if err != nil {
|
||
AbortWithError(c, 500, "INTERNAL_ERROR", "Internal server error")
|
||
return false
|
||
}
|
||
|
||
// 未配置或不匹配，统一返回相同错误（避免信息泄露）
|
||
if storedKey == "" || subtle.ConstantTimeCompare([]byte(key), []byte(storedKey)) != 1 {
|
||
AbortWithError(c, 401, "INVALID_ADMIN_KEY", "Invalid admin API key")
|
||
return false
|
||
}
|
||
|
||
// 获取真实的管理员用户
|
||
admin, err := userService.GetFirstAdmin(c.Request.Context())
|
||
if err != nil {
|
||
AbortWithError(c, 500, "INTERNAL_ERROR", "No admin user found")
|
||
return false
|
||
}
|
||
|
||
c.Set(string(ContextKeyUser), AuthSubject{
|
||
UserID: admin.ID,
|
||
Concurrency: admin.Concurrency,
|
||
})
|
||
c.Set(string(ContextKeyUserRole), admin.Role)
|
||
c.Set("auth_method", "admin_api_key")
|
||
return true
|
||
}
|
||
|
||
// validateJWTForAdmin 验证 JWT 并检查管理员权限
|
||
func validateJWTForAdmin(
|
||
c *gin.Context,
|
||
token string,
|
||
authService *service.AuthService,
|
||
userService *service.UserService,
|
||
) bool {
|
||
// 验证 JWT token
|
||
claims, err := authService.ValidateToken(token)
|
||
if err != nil {
|
||
if errors.Is(err, service.ErrTokenExpired) {
|
||
AbortWithError(c, 401, "TOKEN_EXPIRED", "Token has expired")
|
||
return false
|
||
}
|
||
AbortWithError(c, 401, "INVALID_TOKEN", "Invalid token")
|
||
return false
|
||
}
|
||
|
||
// 从数据库获取用户
|
||
user, err := userService.GetByID(c.Request.Context(), claims.UserID)
|
||
if err != nil {
|
||
AbortWithError(c, 401, "USER_NOT_FOUND", "User not found")
|
||
return false
|
||
}
|
||
|
||
// 检查用户状态
|
||
if !user.IsActive() {
|
||
AbortWithError(c, 401, "USER_INACTIVE", "User account is not active")
|
||
return false
|
||
}
|
||
|
||
// 检查管理员权限
|
||
if !user.IsAdmin() {
|
||
AbortWithError(c, 403, "FORBIDDEN", "Admin access required")
|
||
return false
|
||
}
|
||
|
||
c.Set(string(ContextKeyUser), AuthSubject{
|
||
UserID: user.ID,
|
||
Concurrency: user.Concurrency,
|
||
})
|
||
c.Set(string(ContextKeyUserRole), user.Role)
|
||
c.Set("auth_method", "jwt")
|
||
|
||
return true
|
||
}
|