45 lines
1.0 KiB
Go
45 lines
1.0 KiB
Go
package middleware
|
||
|
||
import (
|
||
"net/http"
|
||
|
||
"soul-api/internal/auth"
|
||
"soul-api/internal/config"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
const adminClaimsKey = "admin_claims"
|
||
|
||
// AdminAuth 管理端鉴权：校验 JWT（Authorization: Bearer 或 Cookie admin_session），未登录返回 401；通过则设置 admin_claims 到 context
|
||
func AdminAuth() gin.HandlerFunc {
|
||
return func(c *gin.Context) {
|
||
cfg := config.Get()
|
||
if cfg == nil {
|
||
c.Next()
|
||
return
|
||
}
|
||
token := auth.GetAdminJWTFromRequest(c.Request)
|
||
claims, ok := auth.ParseAdminJWT(token, cfg.AdminSessionSecret)
|
||
if !ok {
|
||
c.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{"success": false, "error": "未授权访问，请先登录"})
|
||
return
|
||
}
|
||
c.Set(adminClaimsKey, claims)
|
||
c.Next()
|
||
}
|
||
}
|
||
|
||
// GetAdminClaims 从 context 获取 admin claims（需在 AdminAuth 之后调用）
|
||
func GetAdminClaims(c *gin.Context) *auth.AdminClaims {
|
||
v, ok := c.Get(adminClaimsKey)
|
||
if !ok || v == nil {
|
||
return nil
|
||
}
|
||
claims, ok := v.(*auth.AdminClaims)
|
||
if !ok {
|
||
return nil
|
||
}
|
||
return claims
|
||
}
|