212 lines
8.8 KiB
Go
212 lines
8.8 KiB
Go
package service
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"photowall/internal/model"
|
|
"strconv"
|
|
"sync"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type ConfigService struct {
|
|
db *gorm.DB
|
|
cache map[string]string
|
|
mu sync.RWMutex
|
|
}
|
|
|
|
func NewConfigService(db *gorm.DB) *ConfigService {
|
|
return &ConfigService{
|
|
db: db,
|
|
cache: make(map[string]string),
|
|
}
|
|
}
|
|
|
|
// defaultConfigs 所有可配置项及说明
|
|
var defaultConfigs = []model.SystemConfig{
|
|
// ===== 基础配置 =====
|
|
{Key: "site_name", Value: "PhotoWall 毕业照墙", Label: "站点名称", Description: "显示在浏览器标题和登录页的站点名称", Category: "basic", Placeholder: "PhotoWall 毕业照墙", Sort: 1},
|
|
{Key: "jwt_secret", Value: "photowall-dev-secret-change-in-prod", Label: "JWT 密钥", Description: "用于签发和验证登录Token，生产环境务必修改为随机字符串", Category: "basic", Placeholder: "随机字符串至少32位", Sort: 2},
|
|
{Key: "jwt_expire_hours", Value: "3", Label: "登录有效期(小时)", Description: "用户登录后Token的有效时长，超时需重新登录", Category: "basic", Placeholder: "3", Sort: 3},
|
|
{Key: "audit_grace_days", Value: "30", Label: "审核补齐期限(天)", Description: "班级被打回后，超过该天数未补齐则自动删除", Category: "basic", Placeholder: "30", Sort: 4},
|
|
|
|
// ===== 极验行为验证 =====
|
|
{Key: "geetest_id", Value: "662e68b7ce24be211fa97a53975c5d96", Label: "极验 ID", Description: "极验行为验证的Captcha ID，用于防机器人和暴力破解", Category: "geetest", Placeholder: "如：64a####################", DocLink: "https://www.geetest.com 注册后在应用管理获取", Sort: 1},
|
|
{Key: "geetest_key", Value: "2789075bc2b25a04c0f7c5420bfa3d6f", Label: "极验 Key", Description: "极验行为验证的私钥，与ID配对使用", Category: "geetest", Placeholder: "如：64a####################", DocLink: "https://www.geetest.com 注册后在应用管理获取", Sort: 2},
|
|
|
|
// ===== 短信验证码 =====
|
|
{Key: "sms_provider", Value: "aliyun", Label: "短信服务商", Description: "选择短信服务商，目前支持阿里云和腾讯云", Category: "sms", Placeholder: "aliyun 或 tencent", DocLink: "阿里云: https://dysms.console.aliyun.com | 腾讯云: https://console.cloud.tencent.com/sms", Sort: 1},
|
|
{Key: "sms_access_key", Value: "", Label: "短信 AccessKey", Description: "短信服务商的访问密钥ID", Category: "sms", Placeholder: "如：LTAI5t################", DocLink: "阿里云: https://ram.console.aliyun.com | 腾讯云: https://console.cloud.tencent.com/cam", Sort: 2},
|
|
{Key: "sms_secret_key", Value: "", Label: "短信 SecretKey", Description: "短信服务商的访问密钥密码", Category: "sms", Placeholder: "如：########################", DocLink: "与AccessKey在同一页面获取", Sort: 3},
|
|
{Key: "sms_sign_name", Value: "", Label: "短信签名", Description: "短信签名，需在短信服务商审核通过，如：PhotoWall", Category: "sms", Placeholder: "如：PhotoWall", DocLink: "在短信服务商控制台的国内消息-签名管理中申请", Sort: 4},
|
|
{Key: "sms_template_code", Value: "", Label: "短信模板CODE", Description: "短信模板编码，需在短信服务商审核通过，模板内容需包含验证码变量", Category: "sms", Placeholder: "如：SMS_123456789", DocLink: "在短信服务商控制台的国内消息-模板管理中申请", Sort: 5},
|
|
|
|
// ===== 微信扫码登录 =====
|
|
{Key: "wechat_app_id", Value: "", Label: "微信 AppID", Description: "微信开放平台应用的AppID，用于扫码登录", Category: "wechat", Placeholder: "如：wx################", DocLink: "https://open.weixin.qq.com 注册开发者账号并创建网站应用", Sort: 1},
|
|
{Key: "wechat_app_secret", Value: "", Label: "微信 AppSecret", Description: "微信开放平台应用的密钥，与AppID配对", Category: "wechat", Placeholder: "如：########################", DocLink: "在微信开放平台应用详情中查看", Sort: 2},
|
|
{Key: "wechat_redirect_uri", Value: "", Label: "微信回调地址", Description: "微信授权后的回调URL，需与开放平台配置一致，需HTTPS", Category: "wechat", Placeholder: "如：https://yourdomain.com/api/auth/wechat/callback", DocLink: "在微信开放平台应用的网站应用-授权回调域中配置", Sort: 3},
|
|
|
|
// ===== 存储配置 =====
|
|
{Key: "upload_dir", Value: "uploads", Label: "上传目录", Description: "图片文件存储的相对路径", Category: "storage", Placeholder: "uploads", Sort: 1},
|
|
{Key: "max_file_size_mb", Value: "10", Label: "单文件大小上限(MB)", Description: "上传图片的最大大小限制", Category: "storage", Placeholder: "10", Sort: 2},
|
|
}
|
|
|
|
// WarmUpCache 从数据库全量加载配置到内存缓存（单次快速查询）
|
|
func (s *ConfigService) WarmUpCache() {
|
|
var list []model.SystemConfig
|
|
if err := s.db.Find(&list).Error; err == nil {
|
|
s.mu.Lock()
|
|
for _, item := range list {
|
|
s.cache[item.Key] = item.Value
|
|
}
|
|
s.mu.Unlock()
|
|
}
|
|
}
|
|
|
|
// InitDefaults 初始化默认配置（不存在则创建，已存在但为空且默认有初始值则补齐）
|
|
func (s *ConfigService) InitDefaults() error {
|
|
for _, c := range defaultConfigs {
|
|
var existing model.SystemConfig
|
|
err := s.db.Where("`key` = ?", c.Key).First(&existing).Error
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
if err := s.db.Create(&c).Error; err != nil {
|
|
return err
|
|
}
|
|
} else if err == nil && existing.Value == "" && c.Value != "" {
|
|
s.db.Model(&existing).Update("value", c.Value)
|
|
}
|
|
}
|
|
|
|
s.WarmUpCache()
|
|
return nil
|
|
}
|
|
|
|
// GetAll 获取所有配置（按分类和排序）
|
|
func (s *ConfigService) GetAll() ([]model.SystemConfig, error) {
|
|
var list []model.SystemConfig
|
|
err := s.db.Order("category, sort").Find(&list).Error
|
|
return list, err
|
|
}
|
|
|
|
// GetByCategory 按分类获取配置
|
|
func (s *ConfigService) GetByCategory(category string) ([]model.SystemConfig, error) {
|
|
var list []model.SystemConfig
|
|
err := s.db.Where("category = ?", category).Order("sort").Find(&list).Error
|
|
return list, err
|
|
}
|
|
|
|
// UpdateConfigReq 更新配置请求（支持 map 与 key-value 列表格式）
|
|
type UpdateConfigReq struct {
|
|
Configs map[string]string `json:"configs"`
|
|
}
|
|
|
|
func (r *UpdateConfigReq) UnmarshalJSON(data []byte) error {
|
|
// 格式1: {"configs": {"k1": "v1", "k2": "v2"}}
|
|
var auxMap struct {
|
|
Configs map[string]string `json:"configs"`
|
|
}
|
|
if err := json.Unmarshal(data, &auxMap); err == nil && auxMap.Configs != nil {
|
|
r.Configs = auxMap.Configs
|
|
return nil
|
|
}
|
|
|
|
// 格式2: {"configs": [{"key": "k1", "value": "v1"}]}
|
|
var auxList struct {
|
|
Configs []struct {
|
|
Key string `json:"key"`
|
|
Value string `json:"value"`
|
|
} `json:"configs"`
|
|
}
|
|
if err := json.Unmarshal(data, &auxList); err == nil && auxList.Configs != nil {
|
|
r.Configs = make(map[string]string, len(auxList.Configs))
|
|
for _, item := range auxList.Configs {
|
|
r.Configs[item.Key] = item.Value
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// 格式3: 直接传入 map: {"k1": "v1"}
|
|
var directMap map[string]string
|
|
if err := json.Unmarshal(data, &directMap); err == nil {
|
|
r.Configs = directMap
|
|
return nil
|
|
}
|
|
|
|
return errors.New("invalid configs format")
|
|
}
|
|
|
|
// Update 批量更新配置
|
|
func (s *ConfigService) Update(req *UpdateConfigReq) error {
|
|
err := s.db.Transaction(func(tx *gorm.DB) error {
|
|
for key, value := range req.Configs {
|
|
if err := tx.Model(&model.SystemConfig{}).Where("`key` = ?", key).Update("value", value).Error; err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
if err == nil {
|
|
s.mu.Lock()
|
|
for key, value := range req.Configs {
|
|
s.cache[key] = value
|
|
}
|
|
s.mu.Unlock()
|
|
}
|
|
return err
|
|
}
|
|
|
|
// GetValue 获取单个配置值（优先走内存缓存）
|
|
func (s *ConfigService) GetValue(key string) string {
|
|
s.mu.RLock()
|
|
val, ok := s.cache[key]
|
|
s.mu.RUnlock()
|
|
if ok {
|
|
return val
|
|
}
|
|
|
|
var c model.SystemConfig
|
|
if err := s.db.Where("`key` = ?", key).First(&c).Error; err != nil {
|
|
return ""
|
|
}
|
|
|
|
s.mu.Lock()
|
|
s.cache[key] = c.Value
|
|
s.mu.Unlock()
|
|
|
|
return c.Value
|
|
}
|
|
|
|
// GetInt 获取整数配置值，解析失败或不存在则返回 def
|
|
func (s *ConfigService) GetInt(key string, def int) int {
|
|
val := s.GetValue(key)
|
|
if val == "" {
|
|
return def
|
|
}
|
|
if n, err := strconv.Atoi(val); err == nil {
|
|
return n
|
|
}
|
|
return def
|
|
}
|
|
|
|
// GetGeetestConfig 获取极验验证配置
|
|
func (s *ConfigService) GetGeetestConfig() (id, key string) {
|
|
return s.GetValue("geetest_id"), s.GetValue("geetest_key")
|
|
}
|
|
|
|
// GetSmsConfig 获取短信配置
|
|
func (s *ConfigService) GetSmsConfig() (provider, accessKey, secretKey, signName, templateCode string) {
|
|
return s.GetValue("sms_provider"),
|
|
s.GetValue("sms_access_key"),
|
|
s.GetValue("sms_secret_key"),
|
|
s.GetValue("sms_sign_name"),
|
|
s.GetValue("sms_template_code")
|
|
}
|
|
|
|
// GetWechatConfig 获取微信登录配置
|
|
func (s *ConfigService) GetWechatConfig() (appID, appSecret, redirectURI string) {
|
|
return s.GetValue("wechat_app_id"),
|
|
s.GetValue("wechat_app_secret"),
|
|
s.GetValue("wechat_redirect_uri")
|
|
}
|