package battle

import (
	"go_dreamfactory/comm"
	"go_dreamfactory/lego/core"
	"go_dreamfactory/lego/sys/log"
	"go_dreamfactory/modules"
	"go_dreamfactory/pb"
	"go_dreamfactory/sys/db"
	"time"

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

///数据组件
type modelBattleComp struct {
	modules.MCompModel
	module *Battle
}

//组件初始化接口
func (this *modelBattleComp) Init(service core.IService, module core.IModule, comp core.IModuleComp, opt core.IModuleOptions) (err error) {
	this.TableName = comm.TableForum
	this.MCompModel.Init(service, module, comp, opt)
	this.module = module.(*Battle)
	this.TableName = "battle"
	this.DB.CreateIndex(core.SqlTable(this.TableName), mongo.IndexModel{
		Keys: bsonx.Doc{{Key: "heroid", Value: bsonx.Int32(1)}},
	})
	return
}

//查询战斗记录
func (this *modelBattleComp) queryrecord(oid string) (record *pb.DBBattleRecord, err error) {
	record = &pb.DBBattleRecord{}
	if err = this.Get(oid, record); err != nil {
		this.module.Errorln(err)
	}
	return
}

//创建pve 战斗记录
func (this *modelBattleComp) createpve(session comm.IUserSession, conn *db.DBConn, btype pb.BattleType, req *pb.BattlePVEReq) (record *pb.DBBattleRecord, code pb.ErrorCode) {
	record = &pb.DBBattleRecord{
		Id:         primitive.NewObjectID().Hex(),
		Btype:      btype,
		Ptype:      req.Ptype,
		State:      pb.BBattleState_in,
		RedCompId:  session.GetUserId(),
		Redflist:   make([]*pb.DBBattleFormt, 1),
		BlueCompId: "",
		Buleflist:  make([]*pb.DBBattleFormt, len(req.Mformat)),
	}
	record.Redflist[0] = &pb.DBBattleFormt{
		Leadpos: req.Leadpos,
		Team:    make([]*pb.BattleRole, len(req.Teamids)),
	}
	model := db.NewDBModel(comm.TableHero, time.Hour, conn)
	for i, v := range req.Teamids {
		if v != "" {
			hero := &pb.DBHero{}
			if err := model.GetListObj(session.GetUserId(), v, hero); err != nil {
				return
			}
			tid := 100 + i
			record.Redflist[0].Team[i] = &pb.BattleRole{
				Tid:          int32(tid),
				Oid:          hero.Id,
				HeroID:       hero.HeroID,
				Pos:          int32(i),
				Star:         hero.Star,
				Lv:           hero.Lv,
				CaptainSkill: hero.CaptainSkill,
				NormalSkill:  hero.NormalSkill,
				Property:     make(map[string]int32),
			}
			for k, v := range hero.Property {
				record.Redflist[0].Team[i].Property[k] += v
			}
			for k, v := range hero.AddProperty {
				record.Redflist[0].Team[i].Property[k] += v
			}
			for k, v := range hero.Energy {
				record.Redflist[0].Team[i].Property[k] += v
			}
			for k, v := range hero.EnergyProperty {
				record.Redflist[0].Team[i].Property[k] += v
			}
			for k, v := range hero.JuexProperty {
				record.Redflist[0].Team[i].Property[k] += v
			}
			if hero.SuiteId != 0 { //主套装
				if suit, err := this.module.configure.Getequipsuit(hero.SuiteId); err != nil {
					code = pb.ErrorCode_ConfigNoFound
					return
				} else {
					record.Redflist[0].Team[i].MainSuitSkill = suit.Skill
				}
			}
			if hero.SuiteExtId != 0 { //副套装
				if suit, err := this.module.configure.Getequipsuit(hero.SuiteExtId); err != nil {
					code = pb.ErrorCode_ConfigNoFound
					return
				} else {
					record.Redflist[0].Team[i].SubSuitSkill = suit.Skill
				}
			}
		} else {
			record.Redflist[0].Team[i] = nil
		}
	}

	for i, v := range req.Mformat {
		if mf, err := this.module.configure.GetMonsterFormat(v); err != nil {
			code = pb.ErrorCode_ConfigNoFound
			return
		} else {
			record.Buleflist[i] = &pb.DBBattleFormt{
				Leadpos: mf.CaptainId,
				Team:    make([]*pb.BattleRole, len(req.Teamids)),
			}
			for i1, v1 := range mf.MonsterList {
				if v1 == -1 {
					record.Buleflist[i].Team[i1] = nil
				} else {
					if monst, err := this.module.configure.GetMonster(v1); err != nil {
						code = pb.ErrorCode_ConfigNoFound
					} else {
						hero := &pb.DBHero{}
						if hero = this.module.ModuleHero.CreateMonster(monst.HeroId, monst.Star, mf.Lv); hero == nil {
							log.Error("on found battle req data", log.Field{Key: "HeroId", Value: monst.HeroId})
							code = pb.ErrorCode_ReqParameterError
							return
						} else {
							record.Buleflist[i].Team[i1] = &pb.BattleRole{
								Tid:          int32(200 + i*10 + i1),
								Oid:          hero.Id,
								HeroID:       hero.HeroID,
								Pos:          int32(i1),
								Star:         hero.Star,
								Lv:           hero.Lv,
								CaptainSkill: hero.CaptainSkill,
								NormalSkill:  hero.NormalSkill,
								Property:     hero.Property,
							}
							if monst.Equip4 != 0 {
								if suit, err := this.module.configure.Getequipsuit(monst.Equip4); err != nil {
									code = pb.ErrorCode_ConfigNoFound
									return
								} else {
									record.Buleflist[i].Team[i1].MainSuitSkill = suit.Skill
								}
							}
							if monst.Equip2 != 0 {
								if suit, err := this.module.configure.Getequipsuit(monst.Equip2); err != nil {
									code = pb.ErrorCode_ConfigNoFound
									return
								} else {
									record.Buleflist[i].Team[i1].SubSuitSkill = suit.Skill
								}
							}
							record.Buleflist[i].Team[i1].Property[comm.Hp] = int32(float32(record.Buleflist[i].Team[i1].Property[comm.Hp]) * mf.Hppro)
							record.Buleflist[i].Team[i1].Property[comm.Atk] = int32(float32(record.Buleflist[i].Team[i1].Property[comm.Atk]) * mf.Atkpro)
							record.Buleflist[i].Team[i1].Property[comm.Def] = int32(float32(record.Buleflist[i].Team[i1].Property[comm.Def]) * mf.Defpro)
						}
					}
				}
			}
		}
	}
	// if err := this.Add(record.Id, record); err != nil {
	// 	this.module.Errorln(err)
	// 	return
	// }
	return
}