package contextdata

import (
	"github.com/golang-jwt/jwt/v4"
	"time"
)

type UserToken struct {
	UserId     int64
	CompanyId  int64
	EmployeeId int64
}

func (tk UserToken) GenerateToken(secret string, expire int64) (string, error) {
	claims := make(jwt.MapClaims)
	claims["exp"] = time.Now().Unix() + expire
	claims["iat"] = time.Now().Unix()
	claims[CtxKeyJwtUserId] = tk.UserId
	claims[CtxKeyJwtCompanyId] = tk.CompanyId
	claims[CtxKeyJwtEmployeeId] = tk.EmployeeId

	token := jwt.New(jwt.SigningMethodHS256)
	token.Claims = claims

	return token.SignedString([]byte(secret))
}