package jwtoken

import (
	"time"

	"mini-chat/internal/proposal"

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

var _ Token = (*token)(nil)

type Token interface {
	i()
	Sign(jwtInfo proposal.SessionUserInfo, expireDuration time.Duration) (tokenString string, err error)
	Parse(tokenString string) (*claims, error)
	Refresh(tokenString string) (refreshTokenString string, err error)
}

type token struct {
	secret string
}

type claims struct {
	proposal.SessionUserInfo
	jwt.RegisteredClaims
}

func New(secret string) Token {
	return &token{
		secret: secret,
	}
}

func (t *token) i() {}

func (t *token) Sign(sessionUserInfo proposal.SessionUserInfo, expireDuration time.Duration) (tokenString string, err error) {
	claims := claims{
		sessionUserInfo,
		jwt.RegisteredClaims{
			NotBefore: jwt.NewNumericDate(time.Now()),
			IssuedAt:  jwt.NewNumericDate(time.Now()),
			ExpiresAt: jwt.NewNumericDate(time.Now().Add(expireDuration)),
		},
	}

	tokenString, err = jwt.NewWithClaims(jwt.SigningMethodHS256, claims).SignedString([]byte(t.secret))
	return
}

func (t *token) Parse(tokenString string) (*claims, error) {
	tokenClaims, err := jwt.ParseWithClaims(tokenString, &claims{}, func(token *jwt.Token) (interface{}, error) {
		return []byte(t.secret), nil
	})

	if tokenClaims != nil {
		if claims, ok := tokenClaims.Claims.(*claims); ok && tokenClaims.Valid {
			return claims, nil
		}
	}

	return nil, err
}

func (t *token) Refresh(tokenString string) (refreshTokenString string, err error) {
	tokenClaims, err := jwt.ParseWithClaims(tokenString, &claims{}, func(token *jwt.Token) (interface{}, error) {
		return []byte(t.secret), nil
	})

	if tokenClaims != nil {
		if claims, ok := tokenClaims.Claims.(*claims); ok && tokenClaims.Valid {
			claims.IssuedAt = jwt.NewNumericDate(time.Now())
			claims.ExpiresAt = jwt.NewNumericDate(time.Now().Add(time.Hour * 24))
			refreshTokenString, err = jwt.NewWithClaims(jwt.SigningMethodHS256, claims).SignedString([]byte(t.secret))
			return
		}
	}

	return refreshTokenString, err
}