first commit

Co-Authored-By: Codex <noreply@openai.com>

repository/user.go

@@ -0,0 +1,94 @@
+package repository
+
+import (
+	"errors"
+
+	"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{})
+
+	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
+}
+
+// DeleteUser 删除指定用户。
+func DeleteUser(id string) error {
+	db, err := DB()
+	if err != nil {
+		return err
+	}
+	return db.Delete(&model.User{}, "id = ?", id).Error
+}
+
+// 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
+}