mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-28 05:46:36 +08:00
ryan
455e2f8be5
Public config now comes from admin as a flat visibility=1 map instead of a cross-plugin query that compared an integer column to "visible". Login and register resolve CaptchaService per request so CAP is not skipped when user applies before cap.
57 lines
1.3 KiB
Go
57 lines
1.3 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package cap
|
|
|
|
import (
|
|
"Wavelet/core"
|
|
"Wavelet/core/contracts"
|
|
"context"
|
|
"sync"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
var (
|
|
dbMu sync.RWMutex
|
|
dbSvc contracts.DBService
|
|
)
|
|
|
|
// setDBService caches the DBService contract used by the persistence layer.
|
|
func setDBService(s contracts.DBService) {
|
|
dbMu.Lock()
|
|
defer dbMu.Unlock()
|
|
dbSvc = s
|
|
}
|
|
|
|
// getDB resolves a GORM handle from the request/app context, then the Bind fallback.
|
|
func getDB(ctx context.Context) *gorm.DB {
|
|
if s, err := core.InjectFrom[contracts.DBService](ctx); err == nil && s != nil {
|
|
return s.DB(ctx)
|
|
}
|
|
dbMu.RLock()
|
|
s := dbSvc
|
|
dbMu.RUnlock()
|
|
if s != nil {
|
|
return s.DB(ctx)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// loadRuntimeSettings reads the CAPTCHA owned rows from the system config table.
|
|
func loadRuntimeSettings(ctx context.Context) (RuntimeSettings, error) {
|
|
var records []configRecord
|
|
db := getDB(ctx)
|
|
if db == nil {
|
|
return parseRuntimeSettings(nil), nil
|
|
}
|
|
if err := db.Table("w_system_configs").Where("key IN ?", runtimeConfigKeys).Find(&records).Error; err != nil {
|
|
return RuntimeSettings{}, err
|
|
}
|
|
configs := make(map[string]string, len(records))
|
|
for _, r := range records {
|
|
configs[r.Key] = r.Value
|
|
}
|
|
return parseRuntimeSettings(configs), nil
|
|
}
|