package utils

import (
	"Lee-WineList/config"
	"time"

	"github.com/golang-jwt/jwt"
)

type JWT struct {
	claims jwt.MapClaims
}

// GenerateToken 根据UserId生成并返回Token
func GenerateToken(userId uint64) string {
	jwtConfig := config.Scd.JWT
	now := time.Now().Unix()
	claims := make(jwt.MapClaims)
	claims["exp"] = now + jwtConfig.AccessExpire
	claims["iat"] = now
	claims["rft"] = now + jwtConfig.RefreshAfter
	claims["userId"] = userId

	token := jwt.New(jwt.SigningMethodHS256)
	token.Claims = claims
	tokenStr, _ := token.SignedString([]byte(jwtConfig.AccessSecret))

	return tokenStr
}

// ParseToken 解析Token
func ParseToken(tokenStr string) (*JWT, error) {
	token, err := jwt.Parse(tokenStr, func(token *jwt.Token) (interface{}, error) {
		return []byte(config.Scd.JWT.AccessSecret), nil
	})
	if err != nil {
		return nil, err
	}

	return &JWT{claims: token.Claims.(jwt.MapClaims)}, nil
}

// Valid 验证token是否有效
func (jwt *JWT) Valid() bool {
	if err := jwt.claims.Valid(); err != nil {
		return false
	}
	return true
}

// CheckRefresh 检查是否可以刷新
func (jwt *JWT) CheckRefresh() bool {
	if time.Now().Unix() > jwt.GetRefreshTime() {
		return true
	}
	return false
}

func (jwt *JWT) RefreshToken() string {
	return GenerateToken(jwt.GetUserId())
}

// GetUserId 从Token中解析userId
func (jwt *JWT) GetUserId() uint64 {
	return uint64(jwt.claims["userId"].(float64))
}

// GetRefreshTime 从Token中解析refreshTime
func (jwt *JWT) GetRefreshTime() int64 {
	return int64(jwt.claims["rft"].(float64))
}