package svc

import (
	"errors"
	"net/rpc"
	"time"

	"github.com/globalsign/mgo"
	"github.com/globalsign/mgo/bson"
	"mongo.games.com/goserver/core/logger"

	"mongo.games.com/game/dbproxy/mongo"
	"mongo.games.com/game/model"
)

var (
	LotteryLogDBErr = errors.New("log_lottery db open failed.")
)

func LotteryLogsCollection(plt string) *mongo.Collection {
	s := mongo.MgoSessionMgrSington.GetPltMgoSession(plt, model.LotteryLogDBName)
	if s != nil {
		c, first := s.DB().C(model.LotteryLogCollName)
		if first {
			c.EnsureIndex(mgo.Index{Key: []string{"cid"}, Background: true, Sparse: true})
			c.EnsureIndex(mgo.Index{Key: []string{"ctime"}, Background: true, Sparse: true})
			c.EnsureIndex(mgo.Index{Key: []string{"-ctime"}, Background: true, Sparse: true})
			c.EnsureIndex(mgo.Index{Key: []string{"num"}, Background: true, Sparse: true})
			c.EnsureIndex(mgo.Index{Key: []string{"snid"}, Background: true, Sparse: true})
			c.EnsureIndex(mgo.Index{Key: []string{"name"}, Background: true, Sparse: true})
			c.EnsureIndex(mgo.Index{Key: []string{"playernum"}, Background: true, Sparse: true})
			c.EnsureIndex(mgo.Index{Key: []string{"code"}, Background: true, Sparse: true})
			c.EnsureIndex(mgo.Index{Key: []string{"isrobot"}, Background: true, Sparse: true})
			c.EnsureIndex(mgo.Index{Key: []string{"ismust"}, Background: true, Sparse: true})
			c.EnsureIndex(mgo.Index{Key: []string{"ts"}, Background: true, Sparse: true})
			c.EnsureIndex(mgo.Index{Key: []string{"-ts"}, Background: true, Sparse: true})
		}
		return c
	}
	return nil
}

func InsertLotteryLogs(plt string, logs ...*model.LotteryLog) (err error) {
	clog := LotteryLogsCollection(plt)
	if clog == nil {
		return
	}
	switch len(logs) {
	case 0:
		return errors.New("no data")
	case 1:
		err = clog.Insert(logs[0])
	default:
		docs := make([]interface{}, 0, len(logs))
		for _, log := range logs {
			docs = append(docs, log)
		}
		err = clog.Insert(docs...)
	}
	if err != nil {
		logger.Logger.Warn("InsertLotteryLogs error:", err)
		return
	}
	return
}

func GetLotteryLogs(plt string, count int) (ret []*model.LotteryLog, err error) {
	clog := LotteryLogsCollection(plt)
	if clog == nil {
		return nil, LotteryLogDBErr
	}
	err = clog.Find(bson.M{"ts": bson.M{"$lt": time.Now().Unix()}}).Sort("-ts").Limit(count).All(&ret)
	if err != nil && !errors.Is(err, mgo.ErrNotFound) {
		return nil, err
	}
	return
}

type LotteryLogSvc struct{}

func (svc *LotteryLogSvc) GetLotteryLogs(req *model.GetLotteryLogReq, ret *model.GetLotteryLogResp) (err error) {
	ret.LotteryLog, err = GetLotteryLogs(req.Platform, req.Num)
	return
}

func init() {
	rpc.Register(new(LotteryLogSvc))
}