package middleware
|
|
import (
|
"errors"
|
"github.com/gin-gonic/gin"
|
"github.com/golang-jwt/jwt/v4"
|
"wms/request"
|
)
|
|
type JWT struct {
|
SigningKey []byte
|
}
|
|
var (
|
TokenExpired = errors.New("Token is expired")
|
TokenNotValidYet = errors.New("Token not active yet")
|
TokenMalformed = errors.New("That's not even a token")
|
TokenInvalid = errors.New("Couldn't handle this token:")
|
)
|
|
func NewJWT() *JWT {
|
return &JWT{
|
[]byte("327a9457-899a-481e-8b30-58cc97e5b808"),
|
}
|
}
|
|
// CreateToken 创建一个token
|
func (j *JWT) CreateToken(claims request.CustomClaims) (string, error) {
|
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
|
return token.SignedString(j.SigningKey)
|
}
|
|
// ParseToken 解析token
|
func (j *JWT) ParseToken(tokenString string) (*request.CustomClaims, error) {
|
token, err := jwt.ParseWithClaims(tokenString, &request.CustomClaims{}, func(token *jwt.Token) (i interface{}, e error) {
|
return j.SigningKey, nil
|
})
|
if err != nil {
|
if ve, ok := err.(*jwt.ValidationError); ok {
|
if ve.Errors&jwt.ValidationErrorMalformed != 0 {
|
return nil, TokenMalformed
|
} else if ve.Errors&jwt.ValidationErrorExpired != 0 {
|
// Token is expired
|
return nil, TokenExpired
|
} else if ve.Errors&jwt.ValidationErrorNotValidYet != 0 {
|
return nil, TokenNotValidYet
|
} else {
|
return nil, TokenInvalid
|
}
|
}
|
}
|
if token != nil {
|
if claims, ok := token.Claims.(*request.CustomClaims); ok && token.Valid {
|
return claims, nil
|
}
|
return nil, TokenInvalid
|
|
} else {
|
return nil, TokenInvalid
|
}
|
}
|
|
func GetUserInfo(c *gin.Context) *request.CustomClaims {
|
if claims, exists := c.Get("claims"); exists {
|
waitUse := claims.(*request.CustomClaims)
|
return waitUse
|
}
|
return nil
|
}
|
