package model

import (
	"context"
	"errors"
	"gorm.io/gorm"
)

var (
	ErrNotFound = errors.New("record not found")
)

// Insert 插入用户
func Insert(ctx context.Context, db *gorm.DB, user *User) (int64, error) {
	result := db.WithContext(ctx).Create(user)
	if result.Error != nil {
		return 0, result.Error
	}
	return user.Id, nil
}

// FindOne 根据ID查询用户
func FindOne(ctx context.Context, db *gorm.DB, id int64) (*User, error) {
	var user User
	result := db.WithContext(ctx).First(&user, id)
	if result.Error != nil {
		if errors.Is(result.Error, gorm.ErrRecordNotFound) {
			return nil, ErrNotFound
		}
		return nil, result.Error
	}
	return &user, nil
}

// FindOneByEmail 根据邮箱查询用户
func FindOneByEmail(ctx context.Context, db *gorm.DB, email string) (*User, error) {
	var user User
	result := db.WithContext(ctx).Where("email = ?", email).First(&user)
	if result.Error != nil {
		if errors.Is(result.Error, gorm.ErrRecordNotFound) {
			return nil, ErrNotFound
		}
		return nil, result.Error
	}
	return &user, nil
}

// Update 更新用户
func Update(ctx context.Context, db *gorm.DB, user *User) error {
	result := db.WithContext(ctx).Save(user)
	return result.Error
}

// Delete 删除用户
func Delete(ctx context.Context, db *gorm.DB, id int64) error {
	result := db.WithContext(ctx).Delete(&User{}, id)
	return result.Error
}

// FindList 查询用户列表
func FindList(ctx context.Context, db *gorm.DB, page, pageSize int64, keyword string, status int64) ([]User, int64, error) {
	var users []User
	var total int64

	query := db.WithContext(ctx).Model(&User{})

	// 关键字搜索
	if keyword != "" {
		query = query.Where("username LIKE ? OR email LIKE ? OR phone LIKE ?",
			"%"+keyword+"%", "%"+keyword+"%", "%"+keyword+"%")
	}

	// 状态筛选
	if status > 0 {
		query = query.Where("status = ?", status)
	}

	// 统计总数
	if err := query.Count(&total).Error; err != nil {
		return nil, 0, err
	}

	// 分页查询
	offset := (page - 1) * pageSize
	if offset < 0 {
		offset = 0
	}
	err := query.Offset(int(offset)).Limit(int(pageSize)).Find(&users).Error
	if err != nil {
		return nil, 0, err
	}

	return users, total, nil
}