kra-new/internal/data/system/runtime.go

// Package system contains data-layer implementations for system runtime
// settings, authentication tokens, and bootstrap definitions.
package system

import (
	"errors"
	"kra/internal/biz/system"
	"time"

	"kra/internal/config"

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

type runtimeSettings struct{ runtime *config.Store }

func NewRuntimeSettings(runtime *config.Store) system.RuntimeSettings {
	return &runtimeSettings{runtime: runtime}
}

func (s *runtimeSettings) RouterPrefix() string {
	config := s.runtime.Admin()
	if config == nil {
		return ""
	}
	return config.RouterPrefix
}

func (s *runtimeSettings) JWTSettings() system.JWTSettings {
	value := system.JWTSettings{Issuer: "kra", Expires: 7 * 24 * time.Hour, Buffer: 24 * time.Hour}
	config := s.runtime.Admin()
	if config == nil || config.JWT == nil {
		return value
	}
	value.SigningKey = config.JWT.SigningKey
	if config.JWT.Issuer != "" {
		value.Issuer = config.JWT.Issuer
	}
	if config.JWT.ExpiresTime > 0 {
		value.Expires = config.JWT.ExpiresTime
	}
	if config.JWT.BufferTime > 0 {
		value.Buffer = config.JWT.BufferTime
	}
	return value
}

func (s *runtimeSettings) CaptchaSettings() system.CaptchaSettings {
	value := system.CaptchaSettings{KeyLong: 6, ImageWidth: 240, ImageHeight: 80, StoreExpiration: 3 * time.Minute}
	config := s.runtime.Admin()
	if config == nil || config.Captcha == nil {
		return value
	}
	if config.Captcha.KeyLong > 0 {
		value.KeyLong = config.Captcha.KeyLong
	}
	if config.Captcha.ImgWidth > 0 {
		value.ImageWidth = config.Captcha.ImgWidth
	}
	if config.Captcha.ImgHeight > 0 {
		value.ImageHeight = config.Captcha.ImgHeight
	}
	if config.Captcha.StoreExpiration > 0 {
		value.StoreExpiration = config.Captcha.StoreExpiration
	}
	return value
}

func (s *runtimeSettings) MediaSettings() system.MediaSettings {
	config := s.runtime.Admin()
	if config == nil || config.Media == nil {
		return system.MediaSettings{}
	}
	return system.MediaSettings{SessionTTL: config.Media.SessionTTL, MaxFileSize: config.Media.MaxFileSize, ChunkDir: config.Media.ChunkDir}
}

func (s *runtimeSettings) UseMultipoint() bool {
	config := s.runtime.Admin()
	return config != nil && config.System != nil && config.System.UseMultipoint
}

type tokenIssuer struct{ settings system.RuntimeSettings }

func NewTokenIssuer(settings system.RuntimeSettings) system.TokenIssuer {
	return &tokenIssuer{settings: settings}
}

func (i *tokenIssuer) IssueToken(user *system.User, authorityID uint, mustChangePassword bool, expires time.Duration) (*system.IssuedToken, error) {
	settings := i.settings.JWTSettings()
	if expires <= 0 {
		expires = settings.Expires
	}
	passwordVersion := int64(0)
	if user.PasswordUpdatedAt != nil {
		passwordVersion = user.PasswordUpdatedAt.UnixNano()
	}
	token, claims, err := generateToken(settings.SigningKey, settings.Issuer, expires, settings.Buffer, user.ID, authorityID, user.UUID, user.Username, user.NickName, mustChangePassword, passwordVersion)
	if err != nil {
		return nil, err
	}
	return &system.IssuedToken{Value: token, ExpiresAt: claims.ExpiresAt.Time, TTL: expires}, nil
}

func (i *tokenIssuer) ReissueToken(source *system.AuthClaims, authorityID uint) (*system.IssuedToken, error) {
	if source == nil {
		return nil, errors.New("nil JWT claims")
	}
	settings := i.settings.JWTSettings()
	claims := &tokenClaims{
		UUID: source.UUID, ID: source.ID, Username: source.Username, NickName: source.NickName,
		AuthorityID: authorityID, BufferTime: int64(source.BufferTime / time.Second), UserType: source.UserType,
		MustChangePwd: source.MustChangePwd, PasswordVersion: source.PasswordVersion,
		RegisteredClaims: jwt.RegisteredClaims{
			Audience: jwt.ClaimStrings{tokenAudience}, Issuer: settings.Issuer,
			IssuedAt: jwt.NewNumericDate(source.IssuedAt), NotBefore: jwt.NewNumericDate(source.NotBefore), ExpiresAt: jwt.NewNumericDate(source.ExpiresAt),
		},
	}
	token, err := signToken(settings.SigningKey, claims)
	if err != nil {
		return nil, err
	}
	return &system.IssuedToken{Value: token, ExpiresAt: source.ExpiresAt, TTL: time.Until(source.ExpiresAt)}, nil
}

func (i *tokenIssuer) ParseToken(token string) (*system.AuthClaims, error) {
	settings := i.settings.JWTSettings()
	claims, err := parseTokenWithIssuer(token, settings.SigningKey, settings.Issuer)
	if err != nil {
		return nil, err
	}
	audience := append([]string(nil), claims.Audience...)
	issuedAt := time.Time{}
	if claims.IssuedAt != nil {
		issuedAt = claims.IssuedAt.Time
	}
	notBefore, expiresAt := time.Time{}, time.Time{}
	if claims.NotBefore != nil {
		notBefore = claims.NotBefore.Time
	}
	if claims.ExpiresAt != nil {
		expiresAt = claims.ExpiresAt.Time
	}
	return &system.AuthClaims{UUID: claims.UUID, ID: claims.ID, Username: claims.Username, NickName: claims.NickName, AuthorityID: claims.AuthorityID, UserType: claims.UserType, BufferTime: time.Duration(claims.BufferTime) * time.Second, MustChangePwd: claims.MustChangePwd, PasswordVersion: claims.PasswordVersion, Issuer: claims.Issuer, Audience: audience, IssuedAt: issuedAt, NotBefore: notBefore, ExpiresAt: expiresAt}, nil
}