99 lines
2.0 KiB
Go
99 lines
2.0 KiB
Go
|
package middleware
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"dianshang/pkg/jwt"
|
|||
|
|
"dianshang/pkg/response"
|
|||
|
|
"strings"
|
|||
|
|
|
|||
|
|
"github.com/gin-gonic/gin"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// AuthMiddleware JWT认证中间件
|
|||
|
|
func AuthMiddleware() gin.HandlerFunc {
|
|||
|
|
return func(c *gin.Context) {
|
|||
|
|
token := c.GetHeader("Authorization")
|
|||
|
|
if token == "" {
|
|||
|
|
response.Unauthorized(c)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 移除 "Bearer " 前缀
|
|||
|
|
if strings.HasPrefix(token, "Bearer ") {
|
|||
|
|
token = token[7:]
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
claims, err := jwt.ParseToken(token)
|
|||
|
|
if err != nil {
|
|||
|
|
response.Unauthorized(c)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 将用户信息存储到上下文中
|
|||
|
|
c.Set("user_id", claims.UserID)
|
|||
|
|
c.Set("user_type", claims.UserType)
|
|||
|
|
c.Next()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// AdminAuthMiddleware 管理员认证中间件
|
|||
|
|
func AdminAuthMiddleware() gin.HandlerFunc {
|
|||
|
|
return func(c *gin.Context) {
|
|||
|
|
token := c.GetHeader("Authorization")
|
|||
|
|
if token == "" {
|
|||
|
|
response.Unauthorized(c)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 移除 "Bearer " 前缀
|
|||
|
|
if strings.HasPrefix(token, "Bearer ") {
|
|||
|
|
token = token[7:]
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 开发环境：支持模拟token
|
|||
|
|
if strings.HasPrefix(token, "mock_admin_token_") {
|
|||
|
|
// 模拟管理员用户信息
|
|||
|
|
c.Set("user_id", uint(1))
|
|||
|
|
c.Set("user_type", "admin")
|
|||
|
|
c.Next()
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
claims, err := jwt.ParseToken(token)
|
|||
|
|
if err != nil {
|
|||
|
|
response.Unauthorized(c)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 检查用户类型是否为管理员
|
|||
|
|
if claims.UserType != "admin" {
|
|||
|
|
response.Forbidden(c)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 将用户信息存储到上下文中
|
|||
|
|
c.Set("user_id", claims.UserID)
|
|||
|
|
c.Set("user_type", claims.UserType)
|
|||
|
|
c.Next()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// OptionalAuthMiddleware 可选认证中间件（不强制要求登录）
|
|||
|
|
func OptionalAuthMiddleware() gin.HandlerFunc {
|
|||
|
|
return func(c *gin.Context) {
|
|||
|
|
token := c.GetHeader("Authorization")
|
|||
|
|
if token != "" {
|
|||
|
|
// 移除 "Bearer " 前缀
|
|||
|
|
if strings.HasPrefix(token, "Bearer ") {
|
|||
|
|
token = token[7:]
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
claims, err := jwt.ParseToken(token)
|
|||
|
|
if err == nil {
|
|||
|
|
// 将用户信息存储到上下文中
|
|||
|
|
c.Set("user_id", claims.UserID)
|
|||
|
|
c.Set("user_type", claims.UserType)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
c.Next()
|
|||
|
|
}
|
|||
|
|
}
|