package dreamwarorder

import (
	"go_dreamfactory/comm"
	"go_dreamfactory/lego/core"
	"go_dreamfactory/lego/sys/mgo"
	"go_dreamfactory/modules"
	"go_dreamfactory/pb"
	"go_dreamfactory/sys/configure"
	cfg "go_dreamfactory/sys/configure/structs"

	"go.mongodb.org/mongo-driver/bson/primitive"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/x/bsonx"
)

type ModelDreamWarorder struct {
	modules.MCompModel
	module *DreamWarorder
}

func (this *ModelDreamWarorder) Init(service core.IService, module core.IModule, comp core.IModuleComp, options core.IModuleOptions) (err error) {
	err = this.MCompModel.Init(service, module, comp, options)
	this.TableName = comm.TablekfDreamwarorder
	this.module = module.(*DreamWarorder)
	this.DB.CreateIndex(core.SqlTable(this.TableName), mongo.IndexModel{
		Keys: bsonx.Doc{{Key: "uid", Value: bsonx.Int32(1)}},
	})
	return
}

// 获取用户全部的埋点数据
func (this *ModelDreamWarorder) getUserDreamwarorder(uid string) (results *pb.DBDreamWarorder, err error) {
	results = &pb.DBDreamWarorder{}
	if err = this.Get(uid, results); err != nil && err != mgo.MongodbNil {
		this.module.Errorln(err)
		return
	}
	if err == mgo.MongodbNil {
		results = &pb.DBDreamWarorder{
			Id:            primitive.NewObjectID().Hex(),
			Uid:           uid,
			Daytasks:      make([]int32, 0),
			Weektasks:     make([]int32, 0),
			Completetasks: make([]int32, 0),
		}

		err = this.Add(uid, results)
	}
	return
}

//刷新日常任务
func (this *ModelDreamWarorder) refreshDayTask(info *pb.DBDreamWarorder, confs []*cfg.GamePassCheckTaskData, num int32) {
	var (
		tasks  []*cfg.GamePassCheckTaskData = make([]*cfg.GamePassCheckTaskData, 0)
		weight []int32                      = make([]int32, 0)
		indexs []int                        = make([]int, 0)
	)
	for _, v := range confs {
		if v.Page == 1 {
			tasks = append(tasks, v)
			weight = append(weight, v.Pro)
		}
	}

	indexs = comm.GetRandWs(weight, num)

	for _, i := range indexs {
		info.Daytasks = append(info.Daytasks, tasks[i].Id)
	}
	info.Daytime = configure.Now().Unix()
}

//刷新日常任务
func (this *ModelDreamWarorder) refreshWeekTask(info *pb.DBDreamWarorder, confs []*cfg.GamePassCheckTaskData, num int32) {
	var (
		tasks  []*cfg.GamePassCheckTaskData = make([]*cfg.GamePassCheckTaskData, 0)
		weight []int32                      = make([]int32, 0)
		indexs []int                        = make([]int, 0)
	)
	for _, v := range confs {
		if v.Page == 2 {
			tasks = append(tasks, v)
			weight = append(weight, v.Pro)
		}
	}

	indexs = comm.GetRandWs(weight, num)

	for _, i := range indexs {
		info.Weektasks = append(info.Daytasks, tasks[i].Id)
	}
	info.Weektime = configure.Now().Unix()
}