package repository

import (
	"carrotskin/internal/model"

	"gorm.io/gorm"
)

// clientRepository ClientRepository的实现
type clientRepository struct {
	db *gorm.DB
}

// NewClientRepository 创建ClientRepository实例
func NewClientRepository(db *gorm.DB) ClientRepository {
	return &clientRepository{db: db}
}

func (r *clientRepository) Create(client *model.Client) error {
	return r.db.Create(client).Error
}

func (r *clientRepository) FindByClientToken(clientToken string) (*model.Client, error) {
	var client model.Client
	err := r.db.Where("client_token = ?", clientToken).First(&client).Error
	if err != nil {
		return nil, err
	}
	return &client, nil
}

func (r *clientRepository) FindByUUID(uuid string) (*model.Client, error) {
	var client model.Client
	err := r.db.Where("uuid = ?", uuid).First(&client).Error
	if err != nil {
		return nil, err
	}
	return &client, nil
}

func (r *clientRepository) FindByUserID(userID int64) ([]*model.Client, error) {
	var clients []*model.Client
	err := r.db.Where("user_id = ?", userID).Find(&clients).Error
	return clients, err
}

func (r *clientRepository) Update(client *model.Client) error {
	return r.db.Save(client).Error
}

func (r *clientRepository) IncrementVersion(clientUUID string) error {
	return r.db.Model(&model.Client{}).
		Where("uuid = ?", clientUUID).
		Update("version", gorm.Expr("version + 1")).Error
}

func (r *clientRepository) DeleteByClientToken(clientToken string) error {
	return r.db.Where("client_token = ?", clientToken).Delete(&model.Client{}).Error
}

func (r *clientRepository) DeleteByUserID(userID int64) error {
	return r.db.Where("user_id = ?", userID).Delete(&model.Client{}).Error
}