优化

internal/data/runtime_settings.go

@@ -0,0 +1,94 @@
+package data
+
+import (
+	"time"
+
+	"kra/internal/biz"
+	"kra/internal/conf"
+	"kra/pkg/adminauth"
+)
+
+type runtimeSettings struct{ runtime *conf.Runtime }
+
+func NewRuntimeSettings(runtime *conf.Runtime) biz.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() biz.JWTSettings {
+	value := biz.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 != nil {
+		value.Expires = config.Jwt.ExpiresTime.AsDuration()
+	}
+	if config.Jwt.BufferTime != nil {
+		value.Buffer = config.Jwt.BufferTime.AsDuration()
+	}
+	return value
+}
+
+func (s *runtimeSettings) CaptchaSettings() biz.CaptchaSettings {
+	value := biz.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 = int(config.Captcha.KeyLong)
+	}
+	if config.Captcha.ImgWidth > 0 {
+		value.ImageWidth = int(config.Captcha.ImgWidth)
+	}
+	if config.Captcha.ImgHeight > 0 {
+		value.ImageHeight = int(config.Captcha.ImgHeight)
+	}
+	if config.Captcha.StoreExpiration != nil {
+		value.StoreExpiration = config.Captcha.StoreExpiration.AsDuration()
+	}
+	return value
+}
+
+func (s *runtimeSettings) MediaSettings() biz.MediaSettings {
+	config := s.runtime.Admin()
+	if config == nil || config.Media == nil {
+		return biz.MediaSettings{}
+	}
+	return biz.MediaSettings{SessionTTL: int(config.Media.SessionTtl), MaxFileSize: config.Media.MaxFileSize}
+}
+
+func (s *runtimeSettings) UseMultipoint() bool {
+	config := s.runtime.Admin()
+	return config != nil && config.System != nil && config.System.UseMultipoint
+}
+
+type tokenIssuer struct{ settings biz.RuntimeSettings }
+
+func NewTokenIssuer(settings biz.RuntimeSettings) biz.TokenIssuer {
+	return &tokenIssuer{settings: settings}
+}
+
+func (i *tokenIssuer) IssueToken(user *biz.User, authorityID uint, mustChangePassword bool, expires time.Duration) (*biz.IssuedToken, error) {
+	settings := i.settings.JWTSettings()
+	if expires <= 0 {
+		expires = settings.Expires
+	}
+	token, claims, err := adminauth.Generate(settings.SigningKey, settings.Issuer, expires, settings.Buffer, user.ID, authorityID, user.UUID, user.Username, user.NickName, mustChangePassword)
+	if err != nil {
+		return nil, err
+	}
+	return &biz.IssuedToken{Value: token, ExpiresAt: claims.ExpiresAt.Time, TTL: expires}, nil
+}