mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-30 22:26:38 +08:00
ryan
528240026d
- make format 现在与 code-check 使用同一格式化器（golangci-lint fmt），消除 goimports -local 与 gofumpt 的格式拉锯 - .golangci.yml 关闭默认 50/3 截断，完整上报所有问题（只增强不弱化） - 全库 gofumpt 规范化（203 files, 纯格式无行为变更）
125 lines
3.3 KiB
Go
125 lines
3.3 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package auth
|
|
|
|
import (
|
|
"Wavelet/core"
|
|
"Wavelet/core/contracts"
|
|
"context"
|
|
"sync"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
var (
|
|
dbMu sync.RWMutex
|
|
dbSvc contracts.DBService
|
|
cacheMu sync.RWMutex
|
|
cacheSvc contracts.CacheService
|
|
)
|
|
|
|
func setDBService(s contracts.DBService) {
|
|
dbMu.Lock()
|
|
defer dbMu.Unlock()
|
|
dbSvc = s
|
|
}
|
|
|
|
func setCacheService(s contracts.CacheService) {
|
|
cacheMu.Lock()
|
|
defer cacheMu.Unlock()
|
|
cacheSvc = s
|
|
}
|
|
|
|
func getDB(ctx context.Context) *gorm.DB {
|
|
if c, ok := ctx.(*core.Context); ok && c != nil {
|
|
if s, err := core.Inject[contracts.DBService](c); err == nil && s != nil {
|
|
return s.DB(ctx)
|
|
}
|
|
}
|
|
dbMu.RLock()
|
|
s := dbSvc
|
|
dbMu.RUnlock()
|
|
if s != nil {
|
|
return s.DB(ctx)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func getCache(ctx context.Context) contracts.CacheService {
|
|
if c, ok := ctx.(*core.Context); ok && c != nil {
|
|
if s, err := core.Inject[contracts.CacheService](c); err == nil && s != nil {
|
|
return s
|
|
}
|
|
}
|
|
cacheMu.RLock()
|
|
s := cacheSvc
|
|
cacheMu.RUnlock()
|
|
return s
|
|
}
|
|
|
|
// GetAuthSourceByID 根据 ID 获取认证源
|
|
func GetAuthSourceByID(ctx context.Context, id uint64) (*AuthSource, error) {
|
|
var src AuthSource
|
|
if err := getDB(ctx).First(&src, id).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &src, nil
|
|
}
|
|
|
|
// GetAuthSourceByName 根据名称获取认证源
|
|
func GetAuthSourceByName(ctx context.Context, name string) (*AuthSource, error) {
|
|
var src AuthSource
|
|
if err := getDB(ctx).Where("name = ?", name).First(&src).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &src, nil
|
|
}
|
|
|
|
// ListActiveAuthSources 获取所有启用的认证源
|
|
func ListActiveAuthSources(ctx context.Context) ([]AuthSource, error) {
|
|
var sources []AuthSource
|
|
if err := getDB(ctx).Where("is_active = ?", true).Order("id ASC").Find(&sources).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return sources, nil
|
|
}
|
|
|
|
// GetActiveAuthSourcesCached 获取所有启用的认证源（带缓存或直接查询）
|
|
func GetActiveAuthSourcesCached(ctx context.Context) ([]AuthSource, error) {
|
|
return ListActiveAuthSources(ctx)
|
|
}
|
|
|
|
// GetAuthSourceByNameCached 根据名称获取认证源（带缓存或直接查询）
|
|
func GetAuthSourceByNameCached(ctx context.Context, name string) (*AuthSource, error) {
|
|
return GetAuthSourceByName(ctx, name)
|
|
}
|
|
|
|
// FindExternalAccount 查询指定认证源的外部账号绑定
|
|
func FindExternalAccount(ctx context.Context, authSourceID uint64, externalID string) (*ExternalAccount, error) {
|
|
var account ExternalAccount
|
|
if err := getDB(ctx).Where("auth_source_id = ? AND external_id = ?", authSourceID, externalID).First(&account).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &account, nil
|
|
}
|
|
|
|
// BindExternalAccount 绑定外部账号
|
|
func BindExternalAccount(ctx context.Context, account *ExternalAccount) error {
|
|
return getDB(ctx).Create(account).Error
|
|
}
|
|
|
|
// ListExternalAccountsByUserID 获取用户绑定的所有外部账号
|
|
func ListExternalAccountsByUserID(ctx context.Context, userID uint64) ([]ExternalAccount, error) {
|
|
var accounts []ExternalAccount
|
|
if err := getDB(ctx).Where("user_id = ?", userID).Find(&accounts).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return accounts, nil
|
|
}
|
|
|
|
// UnbindExternalAccount 解绑外部账号
|
|
func UnbindExternalAccount(ctx context.Context, id, userID uint64) error {
|
|
return getDB(ctx).Where("id = ? AND user_id = ?", id, userID).Delete(&ExternalAccount{}).Error
|
|
}
|