package user

import (
	"errors"
	"go_dreamfactory/comm"
	"go_dreamfactory/lego/core"
	"go_dreamfactory/modules"
	"go_dreamfactory/pb"
	"go_dreamfactory/utils"
	"time"

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

// 签到
type ModelSign struct {
	modules.MCompModel
	module *User
}

func (this *ModelSign) Init(service core.IService, module core.IModule, comp core.IModuleComp, options core.IModuleOptions) (err error) {
	this.TableName = comm.TableSign
	err = this.MCompModel.Init(service, module, comp, options)
	this.module = module.(*User)
	return
}

func (this *ModelSign) GetUserSign(uid string) (result *pb.DBSign, err error) {
	result = &pb.DBSign{}
	if err = this.module.modelSign.Get(uid, result); err != nil {
		if mongo.ErrNoDocuments == err { // 创建一条新的数据
			_data := this.module.configure.GetSignConf(1, 1)
			if _data != nil {
				result = &pb.DBSign{
					Id:        primitive.NewObjectID().Hex(),
					Uid:       uid,
					SignTime:  0,
					SignCount: 0,
					Group:     _data.Group,
					Cid:       0,
					RTime:     0,
				}
				this.Add(uid, result)
			} else {
				err = errors.New("conf err")
				return nil, err
			}

		}
		err = nil
		return
	}
	err = nil
	return result, err
}

//修改用户签到数据
func (this *ModelSign) ChangeUserSign(uid string, value map[string]interface{}) (err error) {
	if len(value) == 0 {
		return nil
	}
	return this.module.modelSign.Change(uid, value)
}

// 累计登录天数
func (this *ModelSign) updateSignData(uid string, sign *pb.DBSign) (err error) {
	if !utils.IsToday(sign.SignTime) {
		sign.SignCount += 1
		count := sign.SignCount
		update := map[string]interface{}{
			"signCount": count,
			"signTime":  time.Now().UTC(),
		}
		this.ChangeUserSign(uid, update)
	}
	return
}

// 检测是否达到重置日期
func (this *ModelSign) checkResetSignData(session comm.IUserSession) {
	var (
		update  map[string]interface{}
		bReward bool
	)
	update = make(map[string]interface{}, 0)
	if sign, err := this.module.modelSign.GetUserSign(session.GetUserId()); err == nil {
		start, _ := utils.GetMonthStartEnd()
		if sign.RTime < start { // 重置
			sign.RTime = time.Now().Unix()
			sign.SignTime = sign.RTime
			sign.SignCount = 1
			if newGroup := this.module.configure.GetSignResetConf(sign.Cid + 1); newGroup != -1 { // 获取当前的组id
				sign.Cid += 1
				sign.Group = newGroup
				update["cid"] = sign.Cid
				update["group"] = sign.Group
			}
			update["rTime"] = sign.RTime
			update["signTime"] = sign.SignTime
			update["signCount"] = sign.SignCount
			this.Change(session.GetUserId(), update)
			bReward = true
		} else {
			if !utils.IsToday(sign.SignTime) {
				sign.SignCount += 1
				update["signCount"] = sign.SignCount
				sign.SignTime = time.Now().Unix()
				update["signTime"] = sign.SignTime
				sign.RTime = sign.SignTime
				update["rTime"] = sign.RTime
				this.Change(session.GetUserId(), update)
				bReward = true
			}
		}
		_data := this.module.configure.GetSignConf(sign.Cid, sign.Group)
		if bReward && _data != nil { // 发奖
			this.module.DispenseRes(session, _data.Loopgift, true)
		}
		session.SendMsg(string(this.module.GetType()), "sign", &pb.UserSignResp{
			Data:   sign,
			Reward: bReward,
		})
	}

	return
}