package dao

import (
	"context"

	"cidservice/model/entity"
	"github.com/gogf/gf/v2/frame/g"
)

var (
	AdSource = adSourceDao{}
)

type adSourceDao struct{}

// GetByName 根据名称获取广告源
func (d *adSourceDao) GetByName(ctx context.Context, name string) (adSource *entity.AdSource, err error) {
	err = g.DB().Model("ad_sources").
		Where("name = ?", name).
		Scan(&adSource)
	return
}

// GetAvailableSources 获取可用的广告源
func (d *adSourceDao) GetAvailableSources(ctx context.Context) (list []*entity.AdSource, err error) {
	err = g.DB().Model("ad_sources").
		Where("status = ?", "active").
		Order("priority DESC, created_at ASC").
		Scan(&list)
	return
}

// GetSourcesByProvider 根据提供商获取广告源
func (d *adSourceDao) GetSourcesByProvider(ctx context.Context, provider string) (list []*entity.AdSource, err error) {
	err = g.DB().Model("ad_sources").
		Where("provider = ? AND status = ?", provider, "active").
		Order("priority DESC").
		Scan(&list)
	return
}

// Create 创建广告源
func (d *adSourceDao) Create(ctx context.Context, adSource *entity.AdSource) (id int64, err error) {
	result, err := g.DB().Model("ad_sources").Insert(adSource)
	if err != nil {
		return 0, err
	}
	id, err = result.LastInsertId()
	return
}

// Update 更新广告源
func (d *adSourceDao) Update(ctx context.Context, adSource *entity.AdSource) (affected int64, err error) {
	result, err := g.DB().Model("ad_sources").
		Where("id = ?", adSource.Id).
		Update(adSource)
	if err != nil {
		return 0, err
	}
	return result.RowsAffected()
}

// Delete 删除广告源
func (d *adSourceDao) Delete(ctx context.Context, id int64) (affected int64, err error) {
	result, err := g.DB().Model("ad_sources").
		Where("id = ?", id).
		Delete()
	if err != nil {
		return 0, err
	}
	return result.RowsAffected()
}

// GetByID 根据ID获取广告源
func (d *adSourceDao) GetByID(ctx context.Context, id int64) (adSource *entity.AdSource, err error) {
	err = g.DB().Model("ad_sources").
		Where("id = ?", id).
		Scan(&adSource)
	return
}

// UpdateFields 更新广告源部分字段
func (d *adSourceDao) UpdateFields(ctx context.Context, id int64, data *entity.AdSource) (affected int64, err error) {
	result, err := g.DB().Model("ad_sources").
		Where("id = ?", id).
		Update(data)
	if err != nil {
		return 0, err
	}
	return result.RowsAffected()
}