mce/internal/config/site_settings.go

package config

import (
	"strconv"
	"sync"

	"metazone.cc/mce/internal/model"

	"gorm.io/gorm"
	"gorm.io/gorm/clause"
)

// SiteSettings 站点设置管理器（DB 持久化 + 内存缓存，不重启生效）
// 启动时从 DB 加载全量到内存，运行时读内存不查 DB，写入同步写 DB 和内存
type SiteSettings struct {
	mu   sync.RWMutex
	data map[string]string
	db   *gorm.DB
}

// NewSiteSettings 创建站点设置管理器并加载 DB 中已有配置
func NewSiteSettings(db *gorm.DB) (*SiteSettings, error) {
	ss := &SiteSettings{
		data: make(map[string]string),
		db:   db,
	}
	if err := ss.reload(); err != nil {
		return nil, err
	}
	return ss, nil
}

// reload 从 DB 加载全量设置到内存
func (s *SiteSettings) reload() error {
	var rows []model.SiteSetting
	if err := s.db.Find(&rows).Error; err != nil {
		return err
	}
	s.mu.Lock()
	defer s.mu.Unlock()
	for _, row := range rows {
		s.data[row.Key] = row.Value
	}
	return nil
}

// Get 读取一个字符串值，不存在返回默认值
func (s *SiteSettings) Get(key, defaultVal string) string {
	s.mu.RLock()
	defer s.mu.RUnlock()
	val, ok := s.data[key]
	if !ok {
		return defaultVal
	}
	return val
}

// GetBool 读取布尔值，不存在返回默认值（仅 "true" 字符串视为 true）
func (s *SiteSettings) GetBool(key string, defaultVal bool) bool {
	s.mu.RLock()
	defer s.mu.RUnlock()
	val, ok := s.data[key]
	if !ok {
		return defaultVal
	}
	b, err := strconv.ParseBool(val)
	if err != nil {
		return defaultVal
	}
	return b
}

// Set 写入设置（同步写 DB + 更新内存，使用 UPSERT 兼容首次保存）
func (s *SiteSettings) Set(key, value string) error {
	entry := model.SiteSetting{Key: key, Value: value}
	if err := s.db.Clauses(clause.OnConflict{
		Columns:   []clause.Column{{Name: "key"}},
		DoUpdates: clause.AssignmentColumns([]string{"value"}),
	}).Create(&entry).Error; err != nil {
		return err
	}
	s.mu.Lock()
	s.data[key] = value
	s.mu.Unlock()
	return nil
}

// SetBool 写入布尔设置
func (s *SiteSettings) SetBool(key string, value bool) error {
	return s.Set(key, strconv.FormatBool(value))
}

// All 返回所有设置的快照（用于前端展示）
func (s *SiteSettings) All() map[string]string {
	s.mu.RLock()
	defer s.mu.RUnlock()
	cp := make(map[string]string, len(s.data))
	for k, v := range s.data {
		cp[k] = v
	}
	return cp
}

// IsAuditEnabled 审核总开关是否开启
// 优先查站点设置 DB，若无则取 config.yaml 默认值
func (s *SiteSettings) IsAuditEnabled() bool {
	return s.GetBool("audit.enabled", App.Audit.Enabled)
}

// IsAuditTypeEnabled 某类型审核是否开启
func (s *SiteSettings) IsAuditTypeEnabled(auditType string, defaultVal bool) bool {
	return s.GetBool("audit."+auditType, defaultVal)
}

// --- 站点信息设置 ---

// SiteInfoDefaults 站点展示信息的默认值
type SiteInfoDefaults struct {
	SiteName        string // 站点名称，默认 "MetaLab"
	SiteTagline     string // 站点标语，默认 "下一代开发者社区"
	ContactEmail    string // 联系/申诉邮箱，默认 "metazone@foxmail.com"
	Framework       string // 附加标识，默认 "Framework: MetaGenesis"
	CopyrightStart  string // 版权起始年份，默认 "2024"
	OperatorName    string // 运营主体
	ICPNumber       string // ICP 备案（HTML，含链接）
	ICPLicense      string // ICP 经营许可证（HTML）
	PoliceNumber    string // 公安备案（HTML）
	WenWangWen      string // 文网文（HTML）
	AlgorithmFiling string // 算法备案（HTML）
	PrivacyURL      string // 隐私政策（HTML）
	TermsURL        string // 服务条款（HTML）
}

// SiteInfo 获取站点展示信息（优先 DB 设置，fallback 默认值）
func (s *SiteSettings) SiteInfo() SiteInfoDefaults {
	return SiteInfoDefaults{
		SiteName:        s.Get("site.name", "MetaLab"),
		SiteTagline:     s.Get("site.tagline", "下一代开发者社区"),
		ContactEmail:    s.Get("site.contact_email", "metazone@foxmail.com"),
		Framework:       s.Get("site.framework", "Framework: MetaGenesis"),
		CopyrightStart:  s.Get("site.copyright_start", "2024"),
		OperatorName:    s.Get("site.operator_name", ""),
		ICPNumber:       s.Get("site.icp_number", ""),
		ICPLicense:      s.Get("site.icp_license", ""),
		PoliceNumber:    s.Get("site.police_number", ""),
		WenWangWen:      s.Get("site.wenwangwen", ""),
		AlgorithmFiling: s.Get("site.algorithm_filing", ""),
		PrivacyURL:      s.Get("site.privacy_url", ""),
		TermsURL:        s.Get("site.terms_url", ""),
	}
}

// CookieSecure 返回 cookie_secure 配置值（优先站点设置，fallback config.yaml）
func (s *SiteSettings) CookieSecure() bool {
	return s.GetBool("site.cookie_secure", App.Server.CookieSecure)
}

// IsRegistrationEnabled 是否允许新用户注册
func (s *SiteSettings) IsRegistrationEnabled() bool {
	return s.GetBool("registration.enabled", true)
}

// ShowGuidelinesModal 注册时是否弹窗显示社区准则
func (s *SiteSettings) ShowGuidelinesModal() bool {
	return s.GetBool("registration.show_guidelines", true)
}

// ForceGuidelines 是否强制阅读准则（需要倒计时+滚动到底部）
func (s *SiteSettings) ForceGuidelines() bool {
	return s.GetBool("registration.force_guidelines", true)
}

// PasswordStrength 密码强度级别
func (s *SiteSettings) PasswordStrength() string {
	level := s.Get("password.strength", "medium")
	switch level {
	case "low", "medium", "high", "very_high":
		return level
	}
	return "medium"
}

// GuidelinesTimer 强制阅读倒计时秒数
func (s *SiteSettings) GuidelinesTimer() int {
	val := s.Get("registration.guidelines_timer", "60")
	n, err := strconv.Atoi(val)
	if err != nil || n < 1 {
		return 60
	}
	return n
}

// IsMaintenanceEnabled 是否处于维护模式（维护期间仅站长可登录和访问）
func (s *SiteSettings) IsMaintenanceEnabled() bool {
	return s.GetBool("maintenance.enabled", false)
}

// ThemeName 当前选用的主题目录名，默认 "MetaLab-2026"
func (s *SiteSettings) ThemeName() string {
	return s.Get("site.theme", "MetaLab-2026")
}