package dao

import (
	"cid/model/entity"
	"context"

	"github.com/gogf/gf/v2/frame/g"
)

var CIDRequest = &cidRequestDao{}

type cidRequestDao struct{}

// Create 创建CID请求记录
func (d *cidRequestDao) Create(ctx context.Context, request *entity.CidRequest) (id int64, err error) {
	result, err := g.DB().Model("cid_requests").Insert(request)
	if err != nil {
		return 0, err
	}
	id, err = result.LastInsertId()
	return
}

// GetHistory 获取CID请求历史
func (d *cidRequestDao) GetHistory(ctx context.Context, userId int64, page, size int) (list []*entity.CidRequest, total int64, err error) {
	model := g.DB().Model("cid_requests")

	// 根据用户ID筛选
	model = model.Where("user_id = ?", userId)

	// 获取总数
	count, err := model.Count()
	if err != nil {
		return
	}
	total = int64(count)

	// 分页查询
	offset := (page - 1) * size
	err = model.Order("created_at DESC").
		Offset(offset).
		Limit(size).
		Scan(&list)

	return
}

// GetStatistics 获取统计信息
func (d *cidRequestDao) GetStatistics(ctx context.Context, userId int64) (stats map[string]interface{}, err error) {
	stats = make(map[string]interface{})

	// 总请求数
	totalRequests, err := g.DB().Model("cid_requests").
		Where("user_id = ?", userId).
		Count()
	if err != nil {
		return nil, err
	}
	stats["total_requests"] = totalRequests

	// 成功请求数
	successfulRequests, err := g.DB().Model("cid_requests").
		Where("user_id = ? AND status = ?", userId, "completed").
		Count()
	if err != nil {
		return nil, err
	}
	stats["successful_requests"] = successfulRequests

	// 平均处理时间
	avgProcessTime, err := g.DB().Model("cid_requests").
		Fields("AVG(process_time)").
		Where("user_id = ?", userId).
		Value()
	if err != nil {
		return nil, err
	}
	stats["average_process_time"] = avgProcessTime

	return
}