package repository

import (
	"gorm.io/gorm"

	"filestoragesystem/internal/model"
)

// UserRepo 用户数据访问
type UserRepo struct{ DB *gorm.DB }

func (r *UserRepo) Create(u *model.User) error { return r.DB.Create(u).Error }

func (r *UserRepo) FindByID(id uint) (*model.User, error) {
	var u model.User
	err := r.DB.Preload("Role.Permissions").First(&u, id).Error
	return &u, err
}

func (r *UserRepo) FindByUsernameOrEmail(account string) (*model.User, error) {
	var u model.User
	err := r.DB.Preload("Role.Permissions").
		Where("username = ? OR email = ?", account, account).First(&u).Error
	return &u, err
}

func (r *UserRepo) FindByUsername(username string) (*model.User, error) {
	var u model.User
	err := r.DB.Where("username = ?", username).First(&u).Error
	return &u, err
}

func (r *UserRepo) FindByEmail(email string) (*model.User, error) {
	var u model.User
	err := r.DB.Where("email = ?", email).First(&u).Error
	return &u, err
}

// List 分页查询用户，keyword匹配用户名/邮箱，roleID/status可选
func (r *UserRepo) List(keyword, roleID, status string, page, pageSize int) ([]model.User, int64, error) {
	q := r.DB.Model(&model.User{})
	if keyword != "" {
		like := "%" + keyword + "%"
		q = q.Where("username LIKE ? OR email LIKE ?", like, like)
	}
	if roleID != "" {
		q = q.Where("role_id = ?", roleID)
	}
	if status != "" {
		q = q.Where("status = ?", status)
	}
	var total int64
	if err := q.Count(&total).Error; err != nil {
		return nil, 0, err
	}
	var users []model.User
	err := q.Preload("Role").Order("id DESC").
		Offset((page - 1) * pageSize).Limit(pageSize).Find(&users).Error
	return users, total, err
}

func (r *UserRepo) Update(u *model.User) error { return r.DB.Save(u).Error }

func (r *UserRepo) UpdateFields(id uint, fields map[string]interface{}) error {
	return r.DB.Model(&model.User{}).Where("id = ?", id).Updates(fields).Error
}

func (r *UserRepo) Delete(id uint) error { return r.DB.Delete(&model.User{}, id).Error }

// UpdateStorageUsed 增减用户已用存储(delta可正可负)
func (r *UserRepo) UpdateStorageUsed(id uint, delta int64) error {
	return r.DB.Model(&model.User{}).Where("id = ?", id).
		UpdateColumn("storage_used", gorm.Expr("storage_used + ?", delta)).Error
}

// ---- API Key ----

// APIKeyRepo API密钥数据访问
type APIKeyRepo struct{ DB *gorm.DB }

func (r *APIKeyRepo) Create(k *model.APIKey) error { return r.DB.Create(k).Error }

func (r *APIKeyRepo) FindByID(id uint) (*model.APIKey, error) {
	var k model.APIKey
	err := r.DB.First(&k, id).Error
	return &k, err
}

func (r *APIKeyRepo) FindByAccessKey(accessKey string) (*model.APIKey, error) {
	var k model.APIKey
	err := r.DB.Where("access_key = ?", accessKey).First(&k).Error
	return &k, err
}

func (r *APIKeyRepo) ListByUser(userID uint) ([]model.APIKey, error) {
	var keys []model.APIKey
	err := r.DB.Where("user_id = ?", userID).Order("id DESC").Find(&keys).Error
	return keys, err
}

func (r *APIKeyRepo) Delete(id, userID uint) error {
	return r.DB.Where("id = ? AND user_id = ?", id, userID).Delete(&model.APIKey{}).Error
}

func (r *APIKeyRepo) Touch(id uint) error {
	return r.DB.Model(&model.APIKey{}).Where("id = ?", id).
		UpdateColumn("last_used_at", gorm.Expr("CURRENT_TIMESTAMP")).Error
}