You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 

86 lines
2.5 KiB

package jwt
import (
"encoding/json"
"git.diulo.com/mogfee/kit/internal/xuuid"
"github.com/golang-jwt/jwt/v5"
"time"
)
// iss : jwt签发者
// sub : 主题名称
// aud : 面向的用户,一般都是通过ip或者域名控制
// exp : jwt的有效时间(过期时间),这个有效时间必须要大于签发时间,对于交互接口来说,建议是预设5秒
// nbf : 在什么时候jwt开始生效(在此之前不可用)
// iat : jwt的签发时间
// jti : 唯一标识,主要用来回避被重复使用攻击
type UserInfo struct {
UserId string
UserName string
UserType string
Permissions []string
UniqueId string
}
func GetToken(key string, info *UserInfo) (token string, uniqId string, err error) {
uniqId = xuuid.UUID()
info.UniqueId = uniqId
var tokenStr string
ctime := time.Now().Unix()
tokenStr, err = jwt.NewWithClaims(jwt.SigningMethodHS256, &jwt.MapClaims{
//jwt签发者
//"iss": "auth.diulo.com",
//主题名称
//"sub": "www.diulo.com",
// 面向的用户,一般都是通过ip或者域名控制
//"aud": "api.diulo.com",
//jwt的有效时间(过期时间),这个有效时间必须要大于签发时间,对于交互接口来说,建议是预设5秒
"exp": ctime + 7200,
//在什么时候jwt开始生效(在此之前不可用)
"nbf": ctime,
//jwt的签发时间
"iat": ctime,
//唯一标识,主要用来回避被重复使用攻击
"jti": uniqId,
"info": info,
}).SignedString([]byte(key))
if err != nil {
return
}
token, err = Encrypt(tokenStr, []byte(key), key)
return
}
func Parse(key string, tokenStr string) (*UserInfo, error) {
if tokenStr == "" {
return &UserInfo{}, nil
}
str, err := Decrypt(tokenStr, []byte(key), key)
if err != nil {
return nil, errorx.Unauthorized("TOKEN_ERROR", err.Error())
}
token, err := jwt.Parse(str, func(token *jwt.Token) (interface{}, error) {
return []byte(key), nil
})
if err != nil {
if errorx.Is(err, jwt.ErrTokenExpired) {
return nil, errorx.Unauthorized("TOKEN_EXPIRED", "")
}
return nil, errorx.Unauthorized("TOKEN_ERROR", err.Error())
}
if token.Valid {
row := struct {
Info *UserInfo
}{}
b, _ := json.Marshal(token.Claims)
if err = json.Unmarshal(b, &row); err != nil {
return nil, errorx.Unauthorized("TOKEN_ERROR", err.Error())
}
return row.Info, nil
} else if errorx.Is(err, jwt.ErrTokenExpired) || errorx.Is(err, jwt.ErrTokenNotValidYet) {
return nil, errorx.Unauthorized("TOKEN_EXPIRED", "")
}
return nil, errorx.Unauthorized("TOKEN_ERROR", "")
}