130 lines
2.5 KiB
Go
130 lines
2.5 KiB
Go
package repo
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"trustcontact/internal/models"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type UserRepo struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
type UserListParams struct {
|
|
Query string
|
|
Sort string
|
|
Dir string
|
|
Page int
|
|
PageSize int
|
|
}
|
|
|
|
func NewUserRepo(db *gorm.DB) *UserRepo {
|
|
return &UserRepo{db: db}
|
|
}
|
|
|
|
func (r *UserRepo) FindByID(id uint) (*models.User, error) {
|
|
var user models.User
|
|
if err := r.db.First(&user, id).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
return &user, nil
|
|
}
|
|
|
|
func (r *UserRepo) FindByEmail(email string) (*models.User, error) {
|
|
var user models.User
|
|
if err := r.db.Where("email = ?", email).First(&user).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
return &user, nil
|
|
}
|
|
|
|
func (r *UserRepo) Create(user *models.User) error {
|
|
return r.db.Create(user).Error
|
|
}
|
|
|
|
func (r *UserRepo) SetEmailVerified(userID uint, verified bool) error {
|
|
return r.db.Model(&models.User{}).
|
|
Where("id = ?", userID).
|
|
Update("email_verified", verified).Error
|
|
}
|
|
|
|
func (r *UserRepo) UpdatePasswordHash(userID uint, passwordHash string) error {
|
|
return r.db.Model(&models.User{}).
|
|
Where("id = ?", userID).
|
|
Update("password_hash", passwordHash).Error
|
|
}
|
|
|
|
func (r *UserRepo) List(params UserListParams) ([]models.User, int64, error) {
|
|
query := r.db.Model(&models.User{})
|
|
|
|
search := strings.TrimSpace(params.Query)
|
|
if search != "" {
|
|
like := "%" + strings.ToLower(search) + "%"
|
|
query = query.Where("LOWER(name) LIKE ? OR LOWER(email) LIKE ?", like, like)
|
|
}
|
|
|
|
var total int64
|
|
if err := query.Count(&total).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
orderBy := sanitizeSort(params.Sort)
|
|
orderDir := sanitizeDir(params.Dir)
|
|
orderClause := fmt.Sprintf("%s %s", orderBy, orderDir)
|
|
|
|
page := params.Page
|
|
if page < 1 {
|
|
page = 1
|
|
}
|
|
pageSize := params.PageSize
|
|
if pageSize <= 0 {
|
|
pageSize = 10
|
|
}
|
|
if pageSize > 100 {
|
|
pageSize = 100
|
|
}
|
|
|
|
offset := (page - 1) * pageSize
|
|
|
|
var users []models.User
|
|
if err := query.Order(orderClause).Limit(pageSize).Offset(offset).Find(&users).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
return users, total, nil
|
|
}
|
|
|
|
func sanitizeSort(sort string) string {
|
|
switch strings.ToLower(strings.TrimSpace(sort)) {
|
|
case "id":
|
|
return "id"
|
|
case "name":
|
|
return "name"
|
|
case "email":
|
|
return "email"
|
|
default:
|
|
return "id"
|
|
}
|
|
}
|
|
|
|
func sanitizeDir(dir string) string {
|
|
switch strings.ToLower(strings.TrimSpace(dir)) {
|
|
case "desc":
|
|
return "desc"
|
|
default:
|
|
return "asc"
|
|
}
|
|
}
|