package dao

import (
	"errors"
	"opencatd-open/team/model"
	"time"

	"gorm.io/gorm"
)

var _ ApiKeyRepository = (*ApiKeyDAO)(nil)

type ApiKeyRepository interface {
	Create(apiKey *model.ApiKey) error
	GetByID(id int64) (*model.ApiKey, error)
	GetByName(name string) (*model.ApiKey, error)
	GetByApiKey(apiKeyValue string) (*model.ApiKey, error)
	Update(apiKey *model.ApiKey) error
	Delete(id int64) error
	List(offset, limit int, status *int) ([]model.ApiKey, error)
	Enable(id int64) error
	Disable(id int64) error
	BatchEnable(ids []int64) error
	BatchDisable(ids []int64) error
	BatchDelete(ids []int64) error
	Count() (int64, error)
}

type ApiKeyDAO struct {
	db *gorm.DB
}

func NewApiKeyDAO(db *gorm.DB) *ApiKeyDAO {
	return &ApiKeyDAO{db: db}
}

// CreateApiKey 创建ApiKey
func (dao *ApiKeyDAO) Create(apiKey *model.ApiKey) error {
	if apiKey == nil {
		return errors.New("apiKey is nil")
	}
	return dao.db.Create(apiKey).Error
}

// GetApiKeyByID 根据ID获取ApiKey
func (dao *ApiKeyDAO) GetByID(id int64) (*model.ApiKey, error) {
	var apiKey model.ApiKey
	err := dao.db.First(&apiKey, id).Error
	if err != nil {
		return nil, err
	}
	return &apiKey, nil
}

// GetApiKeyByName 根据名称获取ApiKey
func (dao *ApiKeyDAO) GetByName(name string) (*model.ApiKey, error) {
	var apiKey model.ApiKey
	err := dao.db.Where("name = ?", name).First(&apiKey).Error
	if err != nil {
		return nil, err
	}
	return &apiKey, nil
}

// GetApiKeyByApiKey 根据ApiKey值获取ApiKey
func (dao *ApiKeyDAO) GetByApiKey(apiKeyValue string) (*model.ApiKey, error) {
	var apiKey model.ApiKey
	err := dao.db.Where("api_key = ?", apiKeyValue).First(&apiKey).Error
	if err != nil {
		return nil, err
	}
	return &apiKey, nil
}

// UpdateApiKey 更新ApiKey信息
func (dao *ApiKeyDAO) Update(apiKey *model.ApiKey) error {
	if apiKey == nil {
		return errors.New("apiKey is nil")
	}
	apiKey.UpdatedAt = time.Now()
	return dao.db.Save(apiKey).Error
}

// DeleteApiKey 删除ApiKey
func (dao *ApiKeyDAO) Delete(id int64) error {
	return dao.db.Delete(&model.ApiKey{}, id).Error
}

// ListApiKeys 获取ApiKey列表
func (dao *ApiKeyDAO) List(offset, limit int, status *int) ([]model.ApiKey, error) {
	var apiKeys []model.ApiKey
	db := dao.db.Offset(offset).Limit(limit)
	if status != nil {
		db = db.Where("status = ?", *status)
	}
	err := db.Find(&apiKeys).Error
	if err != nil {
		return nil, err
	}
	return apiKeys, nil
}

// EnableApiKey 启用ApiKey
func (dao *ApiKeyDAO) Enable(id int64) error {
	return dao.db.Model(&model.ApiKey{}).Where("id = ?", id).Update("status", 0).Error
}

// DisableApiKey 禁用ApiKey
func (dao *ApiKeyDAO) Disable(id int64) error {
	return dao.db.Model(&model.ApiKey{}).Where("id = ?", id).Update("status", 1).Error
}

// BatchEnableApiKeys 批量启用ApiKey
func (dao *ApiKeyDAO) BatchEnable(ids []int64) error {
	if len(ids) == 0 {
		return errors.New("ids is empty")
	}
	return dao.db.Model(&model.ApiKey{}).Where("id IN ?", ids).Update("status", 0).Error
}

// BatchDisableApiKeys 批量禁用ApiKey
func (dao *ApiKeyDAO) BatchDisable(ids []int64) error {
	if len(ids) == 0 {
		return errors.New("ids is empty")
	}
	return dao.db.Model(&model.ApiKey{}).Where("id IN ?", ids).Update("status", 1).Error
}

// BatchDeleteApiKey 批量删除ApiKey
func (dao *ApiKeyDAO) BatchDelete(ids []int64) error {
	if len(ids) == 0 {
		return errors.New("ids is empty")
	}
	return dao.db.Delete(&model.ApiKey{}, ids).Error
}

// CountApiKeys 获取ApiKey总数
func (dao *ApiKeyDAO) Count() (int64, error) {
	var count int64
	err := dao.db.Model(&model.ApiKey{}).Count(&count).Error
	if err != nil {
		return 0, err
	}
	return count, nil
}