feat: 添加简单模式功能

新增简单模式设置,适合个人使用场景:
- 隐藏多用户管理相关菜单(用户管理、兑换码等)
- 自动关闭用户注册功能
- 管理员并发数自动设为无限制(99999)
- 侧边栏根据模式动态调整菜单项

同时优化分组页面的"专属分组"功能,添加帮助提示说明使用场景
This commit is contained in:
shaw
2025-12-28 22:19:18 +08:00
parent fb9d087838
commit 7d4b7deea9
16 changed files with 378 additions and 90 deletions

View File

@@ -99,7 +99,7 @@ func initializeApplication(buildInfo handler.BuildInfo) (*Application, error) {
geminiOAuthHandler := admin.NewGeminiOAuthHandler(geminiOAuthService)
proxyHandler := admin.NewProxyHandler(adminService)
adminRedeemHandler := admin.NewRedeemHandler(adminService)
settingHandler := admin.NewSettingHandler(settingService, emailService)
settingHandler := admin.NewSettingHandler(settingService, emailService, userService)
updateCache := repository.NewUpdateCache(client)
gitHubReleaseClient := repository.NewGitHubReleaseClient()
serviceBuildInfo := provideServiceBuildInfo(buildInfo)

View File

@@ -12,13 +12,15 @@ import (
type SettingHandler struct {
settingService *service.SettingService
emailService *service.EmailService
userService *service.UserService
}
// NewSettingHandler 创建系统设置处理器
func NewSettingHandler(settingService *service.SettingService, emailService *service.EmailService) *SettingHandler {
func NewSettingHandler(settingService *service.SettingService, emailService *service.EmailService, userService *service.UserService) *SettingHandler {
return &SettingHandler{
settingService: settingService,
emailService: emailService,
userService: userService,
}
}
@@ -52,6 +54,7 @@ func (h *SettingHandler) GetSettings(c *gin.Context) {
DocUrl: settings.DocUrl,
DefaultConcurrency: settings.DefaultConcurrency,
DefaultBalance: settings.DefaultBalance,
SimpleMode: settings.SimpleMode,
})
}
@@ -86,6 +89,9 @@ type UpdateSettingsRequest struct {
// 默认配置
DefaultConcurrency int `json:"default_concurrency"`
DefaultBalance float64 `json:"default_balance"`
// 使用模式
SimpleMode bool `json:"simple_mode"`
}
// UpdateSettings 更新系统设置
@@ -108,8 +114,14 @@ func (h *SettingHandler) UpdateSettings(c *gin.Context) {
req.SmtpPort = 587
}
// 简单模式下自动关闭开放注册
registrationEnabled := req.RegistrationEnabled
if req.SimpleMode {
registrationEnabled = false
}
settings := &service.SystemSettings{
RegistrationEnabled: req.RegistrationEnabled,
RegistrationEnabled: registrationEnabled,
EmailVerifyEnabled: req.EmailVerifyEnabled,
SmtpHost: req.SmtpHost,
SmtpPort: req.SmtpPort,
@@ -129,6 +141,7 @@ func (h *SettingHandler) UpdateSettings(c *gin.Context) {
DocUrl: req.DocUrl,
DefaultConcurrency: req.DefaultConcurrency,
DefaultBalance: req.DefaultBalance,
SimpleMode: req.SimpleMode,
}
if err := h.settingService.UpdateSettings(c.Request.Context(), settings); err != nil {
@@ -136,6 +149,14 @@ func (h *SettingHandler) UpdateSettings(c *gin.Context) {
return
}
// 如果切换到简单模式,自动将管理员并发数设为 99999
if req.SimpleMode {
admin, err := h.userService.GetFirstAdmin(c.Request.Context())
if err == nil && admin != nil {
_ = h.userService.UpdateConcurrency(c.Request.Context(), admin.ID, 99999)
}
}
// 重新获取设置返回
updatedSettings, err := h.settingService.GetAllSettings(c.Request.Context())
if err != nil {
@@ -164,6 +185,7 @@ func (h *SettingHandler) UpdateSettings(c *gin.Context) {
DocUrl: updatedSettings.DocUrl,
DefaultConcurrency: updatedSettings.DefaultConcurrency,
DefaultBalance: updatedSettings.DefaultBalance,
SimpleMode: updatedSettings.SimpleMode,
})
}

View File

@@ -26,6 +26,8 @@ type SystemSettings struct {
DefaultConcurrency int `json:"default_concurrency"`
DefaultBalance float64 `json:"default_balance"`
SimpleMode bool `json:"simple_mode"` // 简单模式
}
type PublicSettings struct {
@@ -40,4 +42,5 @@ type PublicSettings struct {
ContactInfo string `json:"contact_info"`
DocUrl string `json:"doc_url"`
Version string `json:"version"`
SimpleMode bool `json:"simple_mode"` // 简单模式
}

View File

@@ -43,5 +43,6 @@ func (h *SettingHandler) GetPublicSettings(c *gin.Context) {
ContactInfo: settings.ContactInfo,
DocUrl: settings.DocUrl,
Version: h.version,
SimpleMode: settings.SimpleMode,
})
}

View File

@@ -90,6 +90,9 @@ const (
// 管理员 API Key
SettingKeyAdminApiKey = "admin_api_key" // 全局管理员 API Key用于外部系统集成
// 使用模式
SettingKeySimpleMode = "simple_mode" // 简单模式(隐藏多用户管理功能)
)
// Admin API Key prefix (distinct from user "sk-" keys)

View File

@@ -64,6 +64,7 @@ func (s *SettingService) GetPublicSettings(ctx context.Context) (*PublicSettings
SettingKeyApiBaseUrl,
SettingKeyContactInfo,
SettingKeyDocUrl,
SettingKeySimpleMode,
}
settings, err := s.settingRepo.GetMultiple(ctx, keys)
@@ -82,6 +83,7 @@ func (s *SettingService) GetPublicSettings(ctx context.Context) (*PublicSettings
ApiBaseUrl: settings[SettingKeyApiBaseUrl],
ContactInfo: settings[SettingKeyContactInfo],
DocUrl: settings[SettingKeyDocUrl],
SimpleMode: settings[SettingKeySimpleMode] == "true",
}, nil
}
@@ -123,6 +125,9 @@ func (s *SettingService) UpdateSettings(ctx context.Context, settings *SystemSet
updates[SettingKeyDefaultConcurrency] = strconv.Itoa(settings.DefaultConcurrency)
updates[SettingKeyDefaultBalance] = strconv.FormatFloat(settings.DefaultBalance, 'f', 8, 64)
// 使用模式
updates[SettingKeySimpleMode] = strconv.FormatBool(settings.SimpleMode)
return s.settingRepo.SetMultiple(ctx, updates)
}
@@ -223,6 +228,7 @@ func (s *SettingService) parseSettings(settings map[string]string) *SystemSettin
ApiBaseUrl: settings[SettingKeyApiBaseUrl],
ContactInfo: settings[SettingKeyContactInfo],
DocUrl: settings[SettingKeyDocUrl],
SimpleMode: settings[SettingKeySimpleMode] == "true",
}
// 解析整数类型

View File

@@ -25,6 +25,8 @@ type SystemSettings struct {
DefaultConcurrency int
DefaultBalance float64
SimpleMode bool // 简单模式
}
type PublicSettings struct {
@@ -39,4 +41,5 @@ type PublicSettings struct {
ContactInfo string
DocUrl string
Version string
SimpleMode bool // 简单模式
}

View File

@@ -164,6 +164,14 @@ func (s *UserService) UpdateBalance(ctx context.Context, userID int64, amount fl
return nil
}
// UpdateConcurrency 更新用户并发数(管理员功能)
func (s *UserService) UpdateConcurrency(ctx context.Context, userID int64, concurrency int) error {
if err := s.userRepo.UpdateConcurrency(ctx, userID, concurrency); err != nil {
return fmt.Errorf("update concurrency: %w", err)
}
return nil
}
// UpdateStatus 更新用户状态(管理员功能)
func (s *UserService) UpdateStatus(ctx context.Context, userID int64, status string) error {
user, err := s.userRepo.GetByID(ctx, userID)