HouYunFei
b2cae2471d
- 新增AdminUser类型定义及用户管理API接口 - 实现用户列表查询、保存和删除功能 - 添加Linux.do第三方登录支持 - 集成Linux.do OAuth认证流程 - 在系统设置中添加Linux.do登录配置选项 - 实现算力点余额调整记录功能 - 优化搜索组件关键词状态管理 - 更新数据库迁移添加credit_logs表 - 完善用户状态和信用等级验证逻辑
118 lines
2.8 KiB
Go
118 lines
2.8 KiB
Go
package repository
|
|
|
|
import (
|
|
"errors"
|
|
"strings"
|
|
|
|
"github.com/basketikun/infinite-canvas/model"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// ListUsers 分页查询用户。
|
|
func ListUsers(q model.Query) ([]model.User, int64, error) {
|
|
db, err := DB()
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
q.Normalize()
|
|
tx := db.Model(&model.User{})
|
|
if keyword := strings.TrimSpace(q.Keyword); keyword != "" {
|
|
like := "%" + keyword + "%"
|
|
tx = tx.Where("username LIKE ? OR display_name LIKE ? OR email LIKE ? OR linux_do_id LIKE ?", like, like, like, like)
|
|
}
|
|
|
|
var total int64
|
|
if err := tx.Count(&total).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
var users []model.User
|
|
err = tx.Order("created_at desc").Offset(q.Offset()).Limit(q.PageSize).Find(&users).Error
|
|
return users, total, err
|
|
}
|
|
|
|
// CountUsers 返回用户总数。
|
|
func CountUsers() (int64, error) {
|
|
db, err := DB()
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
var total int64
|
|
return total, db.Model(&model.User{}).Count(&total).Error
|
|
}
|
|
|
|
// HasAdmin 判断系统中是否存在管理员。
|
|
func HasAdmin() (bool, error) {
|
|
db, err := DB()
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
var total int64
|
|
err = db.Model(&model.User{}).Where("role = ?", model.UserRoleAdmin).Count(&total).Error
|
|
return total > 0, err
|
|
}
|
|
|
|
// GetUserByID 根据 ID 查询用户。
|
|
func GetUserByID(id string) (model.User, bool, error) {
|
|
db, err := DB()
|
|
if err != nil {
|
|
return model.User{}, false, err
|
|
}
|
|
return findUser(db, "id = ?", id)
|
|
}
|
|
|
|
// GetUserByUsername 根据用户名查询用户。
|
|
func GetUserByUsername(username string) (model.User, bool, error) {
|
|
db, err := DB()
|
|
if err != nil {
|
|
return model.User{}, false, err
|
|
}
|
|
return findUser(db, "username = ?", username)
|
|
}
|
|
|
|
// SaveUser 保存用户信息。
|
|
func SaveUser(user model.User) (model.User, error) {
|
|
db, err := DB()
|
|
if err != nil {
|
|
return user, err
|
|
}
|
|
return user, db.Save(&user).Error
|
|
}
|
|
|
|
// SaveCreditLog 保存算力点变更流水。
|
|
func SaveCreditLog(log model.CreditLog) (model.CreditLog, error) {
|
|
db, err := DB()
|
|
if err != nil {
|
|
return log, err
|
|
}
|
|
return log, db.Save(&log).Error
|
|
}
|
|
|
|
// DeleteUser 删除指定用户。
|
|
func DeleteUser(id string) error {
|
|
db, err := DB()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return db.Delete(&model.User{}, "id = ?", id).Error
|
|
}
|
|
|
|
// GetUserByLinuxDoID 根据 Linux.do ID 查询用户。
|
|
func GetUserByLinuxDoID(id string) (model.User, bool, error) {
|
|
db, err := DB()
|
|
if err != nil {
|
|
return model.User{}, false, err
|
|
}
|
|
return findUser(db, "linux_do_id = ?", id)
|
|
}
|
|
|
|
// findUser 查询单个用户，并将未命中转换为 ok=false。
|
|
func findUser(db *gorm.DB, query string, args ...any) (model.User, bool, error) {
|
|
user := model.User{}
|
|
err := db.Where(query, args...).First(&user).Error
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return model.User{}, false, nil
|
|
}
|
|
return user, err == nil, err
|
|
}
|