1245f07a2d
新增功能： - 支持 Google Authenticator 等应用进行 TOTP 二次验证 - 用户可在个人设置中启用/禁用 2FA - 登录时支持 TOTP 验证流程 - 管理后台可全局开关 TOTP 功能 安全增强： - TOTP 密钥使用 AES-256-GCM 加密存储 - 添加 TOTP_ENCRYPTION_KEY 配置项，必须手动配置才能启用功能 - 防止服务重启导致加密密钥变更使用户无法登录 - 验证失败次数限制，防止暴力破解 配置说明： - Docker 部署：在 .env 中设置 TOTP_ENCRYPTION_KEY - 非 Docker 部署：在 config.yaml 中设置 totp.encryption_key - 生成密钥命令：openssl rand -hex 32
156 lines
4.7 KiB
Go
156 lines
4.7 KiB
Go
package repository
|
||
|
||
import (
|
||
"database/sql"
|
||
"errors"
|
||
|
||
entsql "entgo.io/ent/dialect/sql"
|
||
"github.com/Wei-Shaw/sub2api/ent"
|
||
"github.com/Wei-Shaw/sub2api/internal/config"
|
||
"github.com/Wei-Shaw/sub2api/internal/service"
|
||
"github.com/google/wire"
|
||
"github.com/redis/go-redis/v9"
|
||
)
|
||
|
||
// ProvideConcurrencyCache 创建并发控制缓存，从配置读取 TTL 参数
|
||
// 性能优化：TTL 可配置，支持长时间运行的 LLM 请求场景
|
||
func ProvideConcurrencyCache(rdb *redis.Client, cfg *config.Config) service.ConcurrencyCache {
|
||
waitTTLSeconds := int(cfg.Gateway.Scheduling.StickySessionWaitTimeout.Seconds())
|
||
if cfg.Gateway.Scheduling.FallbackWaitTimeout > cfg.Gateway.Scheduling.StickySessionWaitTimeout {
|
||
waitTTLSeconds = int(cfg.Gateway.Scheduling.FallbackWaitTimeout.Seconds())
|
||
}
|
||
if waitTTLSeconds <= 0 {
|
||
waitTTLSeconds = cfg.Gateway.ConcurrencySlotTTLMinutes * 60
|
||
}
|
||
return NewConcurrencyCache(rdb, cfg.Gateway.ConcurrencySlotTTLMinutes, waitTTLSeconds)
|
||
}
|
||
|
||
// ProvideGitHubReleaseClient 创建 GitHub Release 客户端
|
||
// 从配置中读取代理设置，支持国内服务器通过代理访问 GitHub
|
||
func ProvideGitHubReleaseClient(cfg *config.Config) service.GitHubReleaseClient {
|
||
return NewGitHubReleaseClient(cfg.Update.ProxyURL)
|
||
}
|
||
|
||
// ProvidePricingRemoteClient 创建定价数据远程客户端
|
||
// 从配置中读取代理设置，支持国内服务器通过代理访问 GitHub 上的定价数据
|
||
func ProvidePricingRemoteClient(cfg *config.Config) service.PricingRemoteClient {
|
||
return NewPricingRemoteClient(cfg.Update.ProxyURL)
|
||
}
|
||
|
||
// ProvideSessionLimitCache 创建会话限制缓存
|
||
// 用于 Anthropic OAuth/SetupToken 账号的并发会话数量控制
|
||
func ProvideSessionLimitCache(rdb *redis.Client, cfg *config.Config) service.SessionLimitCache {
|
||
defaultIdleTimeoutMinutes := 5 // 默认 5 分钟空闲超时
|
||
if cfg != nil && cfg.Gateway.SessionIdleTimeoutMinutes > 0 {
|
||
defaultIdleTimeoutMinutes = cfg.Gateway.SessionIdleTimeoutMinutes
|
||
}
|
||
return NewSessionLimitCache(rdb, defaultIdleTimeoutMinutes)
|
||
}
|
||
|
||
// ProviderSet is the Wire provider set for all repositories
|
||
var ProviderSet = wire.NewSet(
|
||
NewUserRepository,
|
||
NewAPIKeyRepository,
|
||
NewGroupRepository,
|
||
NewAccountRepository,
|
||
NewProxyRepository,
|
||
NewRedeemCodeRepository,
|
||
NewPromoCodeRepository,
|
||
NewUsageLogRepository,
|
||
NewUsageCleanupRepository,
|
||
NewDashboardAggregationRepository,
|
||
NewSettingRepository,
|
||
NewOpsRepository,
|
||
NewUserSubscriptionRepository,
|
||
NewUserAttributeDefinitionRepository,
|
||
NewUserAttributeValueRepository,
|
||
|
||
// Cache implementations
|
||
NewGatewayCache,
|
||
NewBillingCache,
|
||
NewAPIKeyCache,
|
||
NewTempUnschedCache,
|
||
NewTimeoutCounterCache,
|
||
ProvideConcurrencyCache,
|
||
ProvideSessionLimitCache,
|
||
NewDashboardCache,
|
||
NewEmailCache,
|
||
NewIdentityCache,
|
||
NewRedeemCache,
|
||
NewUpdateCache,
|
||
NewGeminiTokenCache,
|
||
NewSchedulerCache,
|
||
NewSchedulerOutboxRepository,
|
||
NewProxyLatencyCache,
|
||
NewTotpCache,
|
||
|
||
// Encryptors
|
||
NewAESEncryptor,
|
||
|
||
// HTTP service ports (DI Strategy A: return interface directly)
|
||
NewTurnstileVerifier,
|
||
ProvidePricingRemoteClient,
|
||
ProvideGitHubReleaseClient,
|
||
NewProxyExitInfoProber,
|
||
NewClaudeUsageFetcher,
|
||
NewClaudeOAuthClient,
|
||
NewHTTPUpstream,
|
||
NewOpenAIOAuthClient,
|
||
NewGeminiOAuthClient,
|
||
NewGeminiCliCodeAssistClient,
|
||
|
||
ProvideEnt,
|
||
ProvideSQLDB,
|
||
ProvideRedis,
|
||
)
|
||
|
||
// ProvideEnt 为依赖注入提供 Ent 客户端。
|
||
//
|
||
// 该函数是 InitEnt 的包装器，符合 Wire 的依赖提供函数签名要求。
|
||
// Wire 会在编译时分析依赖关系，自动生成初始化代码。
|
||
//
|
||
// 依赖：config.Config
|
||
// 提供：*ent.Client
|
||
func ProvideEnt(cfg *config.Config) (*ent.Client, error) {
|
||
client, _, err := InitEnt(cfg)
|
||
return client, err
|
||
}
|
||
|
||
// ProvideSQLDB 从 Ent 客户端提取底层的 *sql.DB 连接。
|
||
//
|
||
// 某些 Repository 需要直接执行原生 SQL（如复杂的批量更新、聚合查询），
|
||
// 此时需要访问底层的 sql.DB 而不是通过 Ent ORM。
|
||
//
|
||
// 设计说明：
|
||
// - Ent 底层使用 sql.DB，通过 Driver 接口可以访问
|
||
// - 这种设计允许在同一事务中混用 Ent 和原生 SQL
|
||
//
|
||
// 依赖：*ent.Client
|
||
// 提供：*sql.DB
|
||
func ProvideSQLDB(client *ent.Client) (*sql.DB, error) {
|
||
if client == nil {
|
||
return nil, errors.New("nil ent client")
|
||
}
|
||
// 从 Ent 客户端获取底层驱动
|
||
drv, ok := client.Driver().(*entsql.Driver)
|
||
if !ok {
|
||
return nil, errors.New("ent driver does not expose *sql.DB")
|
||
}
|
||
// 返回驱动持有的 sql.DB 实例
|
||
return drv.DB(), nil
|
||
}
|
||
|
||
// ProvideRedis 为依赖注入提供 Redis 客户端。
|
||
//
|
||
// Redis 用于：
|
||
// - 分布式锁（如并发控制）
|
||
// - 缓存（如用户会话、API 响应缓存）
|
||
// - 速率限制
|
||
// - 实时统计数据
|
||
//
|
||
// 依赖：config.Config
|
||
// 提供：*redis.Client
|
||
func ProvideRedis(cfg *config.Config) *redis.Client {
|
||
return InitRedis(cfg)
|
||
}
|