package hero

import (
	"crypto/rand"
	"go_dreamfactory/comm"
	"go_dreamfactory/lego/core"
	"go_dreamfactory/lego/sys/event"
	"go_dreamfactory/lego/sys/log"
	"go_dreamfactory/modules"
	"go_dreamfactory/pb"
	"go_dreamfactory/sys/db"
	"go_dreamfactory/utils"
	"math/big"

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

func NewModule() core.IModule {
	m := new(Hero)
	return m
}

type Hero struct {
	modules.ModuleBase
	api          *apiComp
	configure    *configureComp
	modelHero    *ModelHero
	modelRecord  *ModelRecord // 英雄抽卡保底，次数等数据
	modelTalent  *ModelTalent // 天赋系统
	moduleFetter comm.IHeroFetter
	service      core.IService
}

//模块名
func (this *Hero) GetType() core.M_Modules {
	return comm.ModuleHero
}

//模块初始化接口 注册用户创建角色事件
func (this *Hero) Init(service core.IService, module core.IModule, options core.IModuleOptions) (err error) {
	err = this.ModuleBase.Init(service, module, options)
	this.service = service
	return
}

//装备组件
func (this *Hero) OnInstallComp() {
	this.ModuleBase.OnInstallComp()
	this.api = this.RegisterComp(new(apiComp)).(*apiComp)
	this.modelHero = this.RegisterComp(new(ModelHero)).(*ModelHero)
	this.modelRecord = this.RegisterComp(new(ModelRecord)).(*ModelRecord)
	this.modelTalent = this.RegisterComp(new(ModelTalent)).(*ModelTalent)
	this.configure = this.RegisterComp(new(configureComp)).(*configureComp)
}
func (this *Hero) Start() (err error) {
	var module core.IModule
	if module, err = this.service.GetModule(comm.ModuleLibrary); err != nil {
		return
	}
	this.moduleFetter = module.(comm.IHeroFetter)
	err = this.ModuleBase.Start()
	event.RegisterGO(comm.EventUserOffline, this.EventUserOffline)
	return
}

//创建单个叠加英雄
func (this *Hero) CreateRepeatHero(session comm.IUserSession, heroCfgId string, num int32, bPush bool) (hero *pb.DBHero, code pb.ErrorCode) {
	var err error
	hero, err = this.modelHero.createHeroOverlying(session.GetUserId(), heroCfgId, num)
	if err == nil {
		go func(uid string, heroCfgId string) { // 携程处理 图鉴数据
			if db.IsCross() {
				this.moduleFetter.SendRpcAddHero(session, heroCfgId)
			} else {
				this.moduleFetter.AddHeroFetterData(uid, heroCfgId)
			}
			heroConf := this.modelHero.moduleHero.configure.GetHeroConfig(heroCfgId)
			if heroConf == nil {
				return
			}
			if heroConf.Handbook == -1 { // 不需要记录图鉴
				return
			}
			if result, err1 := this.ModuleUser.GetUserExpand(uid); err1 == nil {
				initUpdate := map[string]interface{}{}
				sz := result.GetTujian()
				if len(sz) == 0 {
					sz = make(map[string]int32, 0)
				}

				if _, ok := result.GetTujian()[heroCfgId]; !ok {
					sz[heroCfgId] = 0
					initUpdate["tujian"] = sz
				}

				if len(initUpdate) != 0 {

					this.ModuleUser.ChangeUserExpand(uid, initUpdate)
				}
			}
		}(session.GetUserId(), heroCfgId)
		// 统计任务
		this.ModuleRtask.SendToRtask(session, comm.Rtype1, utils.ToInt32(heroCfgId))
		// 查品质
		cfg := this.configure.GetHeroConfig(heroCfgId)
		if cfg != nil {

			this.ModuleRtask.SendToRtask(session, comm.Rtype30, 1, cfg.Color)
			this.ModuleRtask.SendToRtask(session, comm.Rtype31, 1, cfg.Color)
		}

		if bPush { //推送
			session.SendMsg("hero", "change", &pb.HeroChangePush{List: []*pb.DBHero{hero}})
		}
		return
	}
	code = pb.ErrorCode_HeroCreate
	return
}

//获取英雄
func (this *Hero) GetHeroByObjID(uid, heroId string) (*pb.DBHero, pb.ErrorCode) {
	hero := this.modelHero.getOneHero(uid, heroId)
	if hero == nil {
		return nil, pb.ErrorCode_HeroNoExist
	}
	return hero, pb.ErrorCode_Success
}

//佩戴装备
func (this *Hero) UpdateEquipment(session comm.IUserSession, hero *pb.DBHero, equip []*pb.DB_Equipment) (code pb.ErrorCode) {

	if hero == nil {
		code = pb.ErrorCode_HeroNoExist
		return
	}

	list := make([]*pb.DBHero, 0)
	if newHero, err := this.modelHero.setEquipment(hero); err != nil {
		code = pb.ErrorCode_HeroEquipUpdate
		return
	} else {
		if newHero != nil {
			list = append(list, newHero)
		}
	}
	list = append(list, hero)

	this.modelHero.setEquipProperty(hero, equip)
	session.SendMsg("hero", "change", &pb.HeroChangePush{List: list})
	//	随机任务统计
	if hero.SuiteId != 0 || hero.SuiteExtId != 0 {
		this.SendRdTask(session)
	}

	return
}

func (this *Hero) SendRdTask(session comm.IUserSession) {
	equipmap := make(map[int32]map[int32]int32, 0) // k 套装id k1 xx星 v 数量
	list := this.GetHeroList(session.GetUserId())
	for _, v := range list {
		if v.SuiteId != 0 {
			if _, ok := equipmap[v.SuiteId]; !ok {
				equipmap[v.SuiteId] = make(map[int32]int32, 0)
			}
			equipmap[v.SuiteId][v.Suite1Star]++
		}
		if v.SuiteExtId != 0 {
			if _, ok := equipmap[v.SuiteExtId]; !ok {
				equipmap[v.SuiteExtId] = make(map[int32]int32, 0)
			}
			equipmap[v.SuiteExtId][v.Suite2Star]++
		}
	}

	for k, v := range equipmap {
		for k1, v1 := range v {
			this.ModuleRtask.SendToRtask(session, comm.Rtype46, v1, k1, k)
		}
	}
}

//英雄列表
func (this *Hero) GetHeroList(uid string) []*pb.DBHero {
	return this.modelHero.getHeroList(uid)
}

//查询英雄数量
func (this *Hero) QueryHeroAmount(uId string, heroCfgId string) (amount uint32) {
	heroes := this.GetHeroList(uId)
	for _, v := range heroes {
		if v.HeroID == heroCfgId {
			amount++
		}
	}
	return amount
}

// 删除指定卡牌
func (this *Hero) DelCard(udi string, hero *pb.DBHero, amount int32) (code pb.ErrorCode) {
	err := this.modelHero.consumeHeroCard(udi, hero, amount)
	if err != nil {
		code = pb.ErrorCode_DBError
		return
	}
	return
}

// 清空数据
func (this *Hero) CleanData(uid string) {
	this.modelHero.cleanData(uid)
}

// 创建一些特殊的英雄
func (this *Hero) GetSpecifiedHero(session comm.IUserSession, heroConfId string, star, lv, amount int32) (code pb.ErrorCode) {
	if session.GetUserId() == "" || heroConfId == "" || star == 0 || lv == 0 || amount == 0 {
		return pb.ErrorCode_ReqParameterError
	}
	hero, err := this.modelHero.createOneHero(session.GetUserId(), heroConfId)
	if err != nil {
		return pb.ErrorCode_HeroCreate
	}
	hero.Lv = lv
	hero.Star = star
	hero.SameCount = amount
	_heroMap := map[string]interface{}{
		"lv":          hero.Lv,
		"star":        hero.Star,
		"isOverlying": false,
		"sameCount":   amount,
	}
	// 保存数据
	err = this.modelHero.ChangeList(session.GetUserId(), hero.Id, _heroMap)
	if err != nil {
		log.Errorf("GetSpecified failed:%v", err)
		return
	}
	// push change
	session.SendMsg("hero", "change", &pb.HeroChangePush{List: []*pb.DBHero{hero}})
	return
}

//Event-------------------------------------------------------------------------------------------------玩家离线
func (this *Hero) EventUserOffline(session comm.IUserSession) {
	err := this.modelHero.RemoveUserHeroInfo(session)
	this.Debugf("EventUserOffline:%s err:%v", session, err)
}

// 批量创建多个英雄
func (this *Hero) CreateRepeatHeros(session comm.IUserSession, heros map[string]int32, bPush bool) (code pb.ErrorCode) {
	for heroCfgId, num := range heros {
		if num == 0 { // 数量为0 不做处理
			continue
		}
		if _, code = this.CreateRepeatHero(session, heroCfgId, num, bPush); code != pb.ErrorCode_Success {
			this.Errorf("create hero %s failed", heroCfgId)
		}
	}
	return
}

func (this *Hero) AddHeroExp(session comm.IUserSession, heroObjID string, exp int32) (code pb.ErrorCode) {
	var (
		_hero       *pb.DBHero
		newhero     *pb.DBHero
		_changeHero []*pb.DBHero // 变化的英雄
	)
	if heroObjID == "" {
		return
	}
	_hero, code = this.GetHeroByObjID(session.GetUserId(), heroObjID)
	if code != pb.ErrorCode_Success {
		return
	}

	newhero, code = this.modelHero.AddCardExp(session, _hero, exp)
	if code != pb.ErrorCode_Success {
		return
	}
	_changeHero = append(_changeHero, _hero) // 升级后的英雄 hero id 不变
	if newhero != nil {
		_changeHero = append(_changeHero, newhero) // 原来的英雄 只是数量变化了
	}
	session.SendMsg(string(this.GetType()), "change", &pb.HeroChangePush{List: _changeHero})
	return
}

// 英雄练功
func (this *Hero) KungFuHero(session comm.IUserSession, heroObjID string, bKongfu bool) (code pb.ErrorCode) {
	var (
		_hero       *pb.DBHero
		newhero     *pb.DBHero
		_changeHero []*pb.DBHero // 变化的英雄
	)
	_hero, code = this.GetHeroByObjID(session.GetUserId(), heroObjID)
	if code != pb.ErrorCode_Success {
		return
	}
	if bKongfu && _hero.Status == pb.HeroType_HeroTypeKongFu {
		code = pb.ErrorCode_HeroAlreadyKongFuStatus // 已经是练功状态
		return
	}
	if !bKongfu {
		_heroMap := map[string]interface{}{
			"status": pb.HeroType_HeroTypeNil,
		}
		err1 := this.modelHero.ChangeList(session.GetUserId(), heroObjID, _heroMap) // 修改英雄信息
		if err1 != nil {
			this.Errorf("update hero skill failed:%v", err1)
			code = pb.ErrorCode_DBError
			return
		}
		_hero.Status = pb.HeroType_HeroTypeNil
		_changeHero = append(_changeHero, _hero)
		session.SendMsg(string(this.GetType()), "change", &pb.HeroChangePush{List: _changeHero})
		return
	}
	if _hero.SameCount > 1 {
		_hero.SameCount -= 1
		newHero := this.modelHero.CloneNewHero(_hero)
		_changeHero = append(_changeHero, newHero)
	}
	_heroMap := map[string]interface{}{
		"status":      pb.HeroType_HeroTypeKongFu,
		"isOverlying": false,
		"sameCount":   1,
	}
	_hero.Status = pb.HeroType_HeroTypeKongFu
	_hero.SameCount = 1
	err1 := this.modelHero.ChangeList(session.GetUserId(), heroObjID, _heroMap) // 修改英雄信息
	if err1 != nil {
		this.Errorf("update hero skill failed:%v", err1)
		code = pb.ErrorCode_DBError
		return
	}
	_changeHero = append(_changeHero, _hero) // 升级后的英雄 hero id 不变
	if newhero != nil {
		_changeHero = append(_changeHero, newhero) // 原来的英雄 只是数量变化了
	}
	_changeHero = append(_changeHero, _hero)
	session.SendMsg(string(this.GetType()), "change", &pb.HeroChangePush{List: _changeHero})
	return
}

// 创建怪物英雄
func (this *Hero) CreateMonster(heroCid string, star, lv int32) (hero *pb.DBHero) {
	hero = this.modelHero.InitTempHero(heroCid, star, lv)
	return
}

// 只通过唯一id 查询英雄信息
func (this *Hero) QueryCrossHeroinfo(oid string) (hero *pb.DBHero, err error) {
	if this.IsCross() {
		for _, tag := range db.GetServerTags() {
			conn, err1 := db.ServerDBConn(tag) // 遍历连接对象
			if err1 != nil {
				continue
			}

			sr := conn.Mgo.FindOne(comm.TableHero, bson.M{
				"_id": oid,
			})
			hero = &pb.DBHero{}
			if err = sr.Decode(hero); err == nil {
				return
			} else {
				this.modelHero.moduleHero.Errorf("find hero error: %v", err)
			}
		}
	} else { // 不是跨服就查本服 注意 这个接口是给跨服玩法调用 理论上这个分支是不会执行的
		if res := this.modelHero.DB.FindOne(comm.TableHero, bson.M{
			"_id": oid,
		}); res == nil {
			hero = &pb.DBHero{}
			if err = res.Decode(hero); err != nil {
				this.modelHero.moduleHero.Errorf("find hero error: %v", err)
				return
			}
		}
	}

	return
}

// 获取非叠加英雄信息
func (this *Hero) GetHeroListByUse(uid string) []*pb.DBHero {
	tmp := make([]*pb.DBHero, 0)
	for _, v := range this.modelHero.getHeroList(uid) {
		if !v.IsOverlying {
			tmp = append(tmp, v)
		}
	}
	return tmp
}

func (this *Hero) PushHeroProperty(session comm.IUserSession, heros []*pb.DBHero) (err error) {

	err = session.SendMsg(string(this.GetType()), "change", &pb.HeroChangePush{List: heros})
	return
}

// 充值了多少钱
func (this *Hero) RechargeMoney(uid string, money int32) {
	conf := this.configure.GetGlobalConf()
	if conf != nil {
		if len(conf.DrawCardRechargeReward) != 2 {
			return
		}
		if money != conf.DrawCardRechargeReward[0] {
			return
		}
	}
	if record, err := this.modelRecord.GetHeroRecord(uid); err != nil {
		update := map[string]interface{}{}

		if v, ok := record.Condition["recharge"]; !ok {
			record.Condition["recharge"] = 0
		} else {
			if conf.DrawCardRechargeReward[1] < v {
				record.Condition["recharge"] = 0
			}
		}
		// 同步数据
		update["condition"] = record.Condition
		if err := this.modelRecord.ChangeHeroRecord(uid, update); err != nil {
			this.Errorf("ChangeHeroRecord error: %v", err)
		}
	}
}

// 多少天没登录
func (this *Hero) NoLoginDay(uid string, day int32) {
	conf := this.configure.GetGlobalConf()
	if conf != nil {
		if len(conf.DrawCardRegressionReward) != 2 {
			return
		}
		if day != conf.DrawCardRegressionReward[0] {
			return
		}
	}
	if record, err := this.modelRecord.GetHeroRecord(uid); err != nil {
		update := map[string]interface{}{}

		if v, ok := record.Condition["login"]; !ok {
			record.Condition["login"] = 0
		} else {
			if conf.DrawCardRegressionReward[1] < v {
				record.Condition["login"] = 0
			}
		}
		// 同步数据
		update["condition"] = record.Condition
		if err := this.modelRecord.ChangeHeroRecord(uid, update); err != nil {
			this.Errorf("ChangeHeroRecord error: %v", err)
		}
	}
}

// 连续抽卡最多连续出A个相同阵营的英雄（普通卡池）
/*
heroCid 抽到的5星英雄ID
drawCount 当前抽卡次数
poll 当前卡池
返回值 newCid不满足抽卡要求替换其他5星英雄
*/
func (this *Hero) ContinuousRestriction(uid string, heroCid string, drawCount int32, pool string) (newCid string) {
	var (
		update map[string]interface{}
	)
	record, err := this.modelRecord.GetHeroRecord(uid)
	if err != nil {
		return heroCid
	}
	for heroid, index := range record.Star5Hero {
		if heroid == heroCid {
			conf := this.configure.GetGlobalConf() //.
			iMaxCOunt := conf.DrawCardContinuousRestrictionStar5
			if drawCount-index <= iMaxCOunt { // 连续n次还获得该英雄 直接替换其他英雄

				_data := this.configure.GetPollByType(pool)
				if _data == nil {

					return heroCid
				}
				sz := make([]int32, 0)
				for _, v := range _data[5] {
					sz = append(sz, v.Weight)
				}
				//randomIndex := this.modelHero.GetRandW(sz)
				for i := 0; i < len(_data[5]); i++ {
					if v, ok := _data[5]; ok {
						if int32(len(v)) > int32(i) {
							newCid = v[i].Id
							if newCid == heroid {
								continue
							}
							record.Star5Hero[heroid] = drawCount
							update["star5Hero"] = record.Star5Hero
							this.modelRecord.ChangeHeroRecord(uid, update) // 更新信息
							return
						}
					}
				}

			}
		}
	}
	return heroCid
}

// 检查充值和未登录天数之内抽卡是否抽中
func (this *Hero) CheckCondition(uid string) bool {
	var (
		curCount int32
		update   map[string]interface{}
	)
	update = make(map[string]interface{}, 0)
	defer this.modelRecord.ChangeHeroRecord(uid, update)
	record, _ := this.modelRecord.GetHeroRecord(uid)
	if v, ok := record.Condition["recharge"]; ok {
		conf := this.configure.GetGlobalConf() //.
		if len(conf.DrawCardRechargeReward) == 2 {
			curCount = conf.DrawCardRechargeReward[1]
		}
		if v >= curCount { // 触发保底 直接给5星
			delete(record.Condition, "recharge")
			update["condition"] = record.Condition
			return true
		} else { // 1/curCount概率
			n, _ := rand.Int(rand.Reader, big.NewInt(int64(curCount)))
			if n.Int64() < 1 {

				delete(record.Condition, "recharge")
				update["condition"] = record.Condition
				return true
			}
			record.Condition["recharge"] += 1
		}
	}
	if v, ok := record.Condition["login"]; ok {
		conf := this.configure.GetGlobalConf() //.
		if len(conf.DrawCardRegressionReward) == 2 {
			curCount = conf.DrawCardRegressionReward[1]
		}
		if v >= curCount { // 触发保底 直接给5星
			delete(record.Condition, "login")
			update["login"] = record.Condition
			return true
		} else { // 1/curCount概率
			n, _ := rand.Int(rand.Reader, big.NewInt(int64(curCount)))

			if n.Int64() < 1 {
				delete(record.Condition, "login")
				update["login"] = record.Condition
				return true
			}
			record.Condition["login"] += 1
		}
	}
	return false
}

// 检查大于lv等级英雄的数量
func (this *Hero) CheckLvNum(uid string, lv int32) int32 {

	tmp := make([]*pb.DBHero, 0)
	for _, v := range this.modelHero.getHeroList(uid) {
		if v.Lv >= lv {
			tmp = append(tmp, v)
		}
	}
	return int32(len(tmp))
}

func (this *Hero) GetTujianHeroNum(uid string) int32 {
	if result, err1 := this.ModuleUser.GetUserExpand(uid); err1 == nil {
		tujian := result.GetTujian()
		return int32(len(tujian))
	}
	return 0
}

////拥有觉醒至A级的B星英雄N个
func (this *Hero) CheckJuexingHeroNum(uid string, juexingLv int32, star int32) int32 {
	tmp := make([]*pb.DBHero, 0)
	for _, v := range this.modelHero.getHeroList(uid) {
		if v.JuexingLv >= juexingLv && v.Star >= star {
			tmp = append(tmp, v)
		}
	}
	return int32(len(tmp))
}

//拥有共鸣至N级的英雄
func (this *Hero) CheckResonaceHeroNum(uid string, resonaceLv int32) int32 {
	tmp := make([]*pb.DBHero, 0)
	for _, v := range this.modelHero.getHeroList(uid) {
		if v.ResonateNum >= resonaceLv {
			tmp = append(tmp, v)
		}
	}
	return int32(len(tmp))
}