35 lines
760 B
Go
35 lines
760 B
Go
package middleware
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"soda-api/backend/internal/utils"
|
|
)
|
|
|
|
type AuthMiddleware struct {
|
|
jwtManager *utils.JWTManager
|
|
}
|
|
|
|
func NewAuthMiddleware(jwtManager *utils.JWTManager) gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
authHeader := c.GetHeader("Authorization")
|
|
if authHeader == "" || !strings.HasPrefix(authHeader, "Bearer ") {
|
|
utils.JSONError(c, 401, "缺少或无效的认证信息")
|
|
c.Abort()
|
|
return
|
|
}
|
|
tokenString := strings.TrimPrefix(authHeader, "Bearer ")
|
|
claims, err := jwtManager.Parse(tokenString)
|
|
if err != nil {
|
|
utils.JSONError(c, 401, "认证失败: "+err.Error())
|
|
c.Abort()
|
|
return
|
|
}
|
|
c.Set("claims", claims)
|
|
c.Set("tokenString", tokenString)
|
|
c.Next()
|
|
}
|
|
}
|