package model

import (
	"context"
	"errors"

	"gorm.io/gorm"
)

// AIProviderInsert 插入AI供应商
func AIProviderInsert(ctx context.Context, db *gorm.DB, provider *AIProvider) (int64, error) {
	result := db.WithContext(ctx).Create(provider)
	if result.Error != nil {
		return 0, result.Error
	}
	return provider.Id, nil
}

// AIProviderFindOne 根据ID查询AI供应商
func AIProviderFindOne(ctx context.Context, db *gorm.DB, id int64) (*AIProvider, error) {
	var provider AIProvider
	result := db.WithContext(ctx).First(&provider, id)
	if result.Error != nil {
		if errors.Is(result.Error, gorm.ErrRecordNotFound) {
			return nil, ErrNotFound
		}
		return nil, result.Error
	}
	return &provider, nil
}

// AIProviderFindList 查询AI供应商列表（分页）
func AIProviderFindList(ctx context.Context, db *gorm.DB, page, pageSize int64) ([]AIProvider, int64, error) {
	var providers []AIProvider
	var total int64

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

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

	offset := (page - 1) * pageSize
	if offset < 0 {
		offset = 0
	}
	err := query.Order("sort_order ASC, id ASC").Offset(int(offset)).Limit(int(pageSize)).Find(&providers).Error
	if err != nil {
		return nil, 0, err
	}

	return providers, total, nil
}

// AIProviderUpdate 更新AI供应商
func AIProviderUpdate(ctx context.Context, db *gorm.DB, provider *AIProvider) error {
	result := db.WithContext(ctx).Save(provider)
	return result.Error
}

// AIProviderDelete 删除AI供应商
func AIProviderDelete(ctx context.Context, db *gorm.DB, id int64) error {
	result := db.WithContext(ctx).Delete(&AIProvider{}, id)
	return result.Error
}

// AIProviderFindByName 根据名称查询AI供应商
func AIProviderFindByName(ctx context.Context, db *gorm.DB, name string) (*AIProvider, error) {
	var provider AIProvider
	result := db.WithContext(ctx).Where("name = ?", name).First(&provider)
	if result.Error != nil {
		if errors.Is(result.Error, gorm.ErrRecordNotFound) {
			return nil, ErrNotFound
		}
		return nil, result.Error
	}
	return &provider, nil
}

// AIProviderFindAllActive 查询所有启用的AI供应商
func AIProviderFindAllActive(ctx context.Context, db *gorm.DB) ([]AIProvider, error) {
	var providers []AIProvider
	err := db.WithContext(ctx).Where("is_active = ?", true).Order("sort_order ASC, id ASC").Find(&providers).Error
	if err != nil {
		return nil, err
	}
	return providers, nil
}