package repository

import (
	"context"
	"git.hlsq.asia/mmorpg/service-common/db/mysql"
	"git.hlsq.asia/mmorpg/service-common/db/redis"
	"git.hlsq.asia/mmorpg/service-common/utils"
	"git.hlsq.asia/mmorpg/service-qgdzs/internal/dao/model"
	"git.hlsq.asia/mmorpg/service-qgdzs/internal/dao/query"
	"gorm.io/gorm"
	"time"
)

type QuestionDao struct {
	ctx   context.Context
	query *query.Query
	cache *redis.CacheClient
}

func NewQuestionDao(ctx context.Context, cache ...*redis.CacheClient) *QuestionDao {
	dao := &QuestionDao{
		ctx:   ctx,
		query: query.Use(mysql.GetDB(dbName)),
	}
	if len(cache) > 0 {
		dao.cache = cache[0]
	}
	return dao
}

func (d *QuestionDao) Create(question *model.Question) (*model.Question, error) {
	err := d.query.Question.WithContext(d.ctx).
		Create(question)
	return question, err
}

func (d *QuestionDao) FindByRandom(categorySn string) (*model.Question, error) {
	q := d.query.Question.WithContext(d.ctx)
	if categorySn != "" {
		q = q.Where(d.query.Question.CategorySn.Eq(categorySn))
	}

	count, err := q.Count()
	if err != nil {
		return nil, err
	}
	if count == 0 {
		return nil, gorm.ErrRecordNotFound
	}

	first, err := q.Offset(utils.RandInt(0, int(count-1))).First()
	if err != nil {
		return nil, err
	}
	return first, nil
}

func (d *QuestionDao) FindBySn(sn string) (*model.Question, error) {
	if d.cache != nil {
		var question model.Question
		if ok := d.cache.Get(d.ctx, keyCacheBySn(sn, question.TableName()), &question); ok {
			return &question, nil
		}
	}
	first, err := d.query.Question.WithContext(d.ctx).
		Where(d.query.Question.Sn.Eq(sn)).
		First()
	if err != nil {
		return nil, err
	}
	if d.cache != nil {
		d.cache.Set(d.ctx, keyCacheBySn(sn, first.TableName()), first, 5*time.Minute)
	}
	return first, nil
}

func (d *QuestionDao) FindByCategory(categorySn string) (*model.Question, error) {
	count, err := d.query.Question.WithContext(d.ctx).
		Where(d.query.Question.CategorySn.Eq(categorySn)).
		Count()
	if err != nil {
		return nil, err
	}
	if count == 0 {
		return nil, gorm.ErrRecordNotFound
	}
	first, err := d.query.Question.WithContext(d.ctx).
		Where(d.query.Question.CategorySn.Eq(categorySn)).
		Offset(utils.RandInt(0, int(count-1))).
		First()
	if err != nil {
		return nil, err
	}
	return first, nil
}