package smithy

import (
	"go_dreamfactory/comm"
	"go_dreamfactory/pb"
	"go_dreamfactory/sys/configure"
	cfg "go_dreamfactory/sys/configure/structs"

	"google.golang.org/protobuf/proto"
)

//参数校验
func (this *apiComp) ForgeEquipCheck(session comm.IUserSession, req *pb.SmithyForgeEquipReq) (code pb.ErrorCode) {
	if req.ReelId == 0 || req.Lava < 0 || req.Quality < 0 {
		code = pb.ErrorCode_ReqParameterError
		return
	}
	return
}

// 打造装备
func (this *apiComp) ForgeEquip(session comm.IUserSession, req *pb.SmithyForgeEquipReq) (code pb.ErrorCode, data proto.Message) {
	var (
		stove  *pb.DBStove
		err    error
		update map[string]interface{}

		costRes            []*cfg.Gameatn
		customLv           int32 // 定制装备的等级
		rsp                *pb.SmithyForgeEquipResp
		needTemperatureCos int32 // 需要消耗的温度
		maxT               int32
	)
	rsp = &pb.SmithyForgeEquipResp{}
	update = make(map[string]interface{})
	code = this.ForgeEquipCheck(session, req)
	if code != pb.ErrorCode_Success {
		return // 参数校验失败直接返回
	}
	stove, err = this.module.modelStove.getSmithyStoveList(session.GetUserId())
	if err != nil {
		code = pb.ErrorCode_DBError
		return
	}
	reelcfg := this.module.configure.GetSmithyReelConfigData(req.ReelId)
	if reelcfg == nil {
		code = pb.ErrorCode_ReqParameterError // 没有找到该类型的图纸信息
		return
	}
	// 校验图纸是否激活
	if _, ok := stove.Data[req.ReelId]; !ok { // 是不是首次打造
		if !this.module.configure.CheckSmithyFirstReelConfigData(reelcfg.Type, req.ReelId) { // 没有激活图纸
			code = pb.ErrorCode_SmithyNoReel
			return
		}
		stove.Data[req.ReelId] = &pb.Mastery{
			Lv:    1,
			Value: 0,
		}
	}

	// 是否是精益打造
	if req.Quality > 0 {
		costRes = append(costRes, reelcfg.RefineCos)
	}
	// 校验是不是装备定制打造
	if req.SuiteId != 0 {
		// 检查消耗
		if code = this.module.CheckRes(session, []*cfg.Gameatn{reelcfg.CustomizedCos1}); code != pb.ErrorCode_Success {
			return
		}
		costRes = append(costRes, reelcfg.CustomizedCos1)
		if req.Position == -1 {
			if code = this.module.CheckRes(session, []*cfg.Gameatn{reelcfg.CustomizedCos2}); code != pb.ErrorCode_Success {
				return
			}
			costRes = append(costRes, reelcfg.CustomizedCos2)
		}
		// 随机权重 获取等级
		index := this.module.modelStove.GetRandEquipLv(reelcfg.CustomizedLvDistribution)
		if int32(len(reelcfg.CustomizedLv)) > index {
			customLv = reelcfg.CustomizedLv[index]
		}
	}
	needTemperatureCos = reelcfg.TemperatureCos
	// 熟练度减少的温度
	t := this.module.modelStove.CheckTemperature(req.ReelId, stove.Data[req.ReelId].Lv)
	needTemperatureCos = reelcfg.TemperatureCos * (1000 - t) / 1000 // 千分比

	if req.Lava > 0 { // 是否是熔岩打造
		exemption := this.module.configure.GetGlobalConf().ExemptionTemperatureCos // 预计消耗温度
		exemption_TemperatureCosNum := this.module.configure.GetGlobalConf().ExemptionTemperatureCosNum
		exemption_TemperatureCosNum = req.Lava * exemption_TemperatureCosNum
		if needTemperatureCos > exemption_TemperatureCosNum {
			code = pb.ErrorCode_SmithyLackLava // 缺少熔岩
			return
		}
		// 检查消耗
		lavaCost := &cfg.Gameatn{
			A: exemption.A,
			T: exemption.T,
			N: exemption.N * req.Lava,
		}
		if code = this.module.CheckRes(session, []*cfg.Gameatn{lavaCost}); code != pb.ErrorCode_Success {
			return
		}
		costRes = append(costRes, lavaCost)
	} else {
		// 检查炉温 是否够
		if stove.Temperature < needTemperatureCos {
			code = pb.ErrorCode_SmithyNoTemperature // 炉温不够 直接返回
			return
		}
		stove.Temperature -= needTemperatureCos // 消耗温度
		update["temperature"] = stove.Temperature
	}

	costRes = append(costRes, reelcfg.BasicCos...)
	subAtn := this.module.modelStove.CheckForgeConsume(req.ReelId, stove.Data[req.ReelId].Lv)
	//costRes = append(costRes, subAtn...)
	// 异常处理 防止不消耗该道具 但是技能做了该道具的减免
	for _, v := range costRes {
		for _, v1 := range subAtn {
			if v.A == v1.A && v.T == v1.T {
				v.N += v1.N
			}
		}
		if v.N < 0 {
			v.N = 0
		}
	}
	if code = this.module.CheckRes(session, costRes); code != pb.ErrorCode_Success {
		return
	}

	if code = this.module.ConsumeRes(session, costRes, true); code != pb.ErrorCode_Success {
		return
	}

	// 装备资源分发
	if customLv > 0 { //

		sz := make([]int32, 4) // 最高 4个品质
		// 获得极品权重
		sz[3] = this.module.modelStove.StoveToolsQualityProbability(stove)
		if equip, code1 := this.module.ModuleEquipment.GetForgeEquip(session, req.SuiteId, req.Position, customLv, sz); code1 == pb.ErrorCode_Success {
			rsp.Equip = append(rsp.Equip, equip)
		} else {
			code = code1
			return
		}
	} else {
		res := this.module.configure.GetDropReward(reelcfg.BasicDrop)
		//res := this.module.configure.GetSmithyDropReward(reelcfg.BasicDrop)
		if err, atno := this.module.DispenseAtno(session, res, true); err == pb.ErrorCode_Success {
			for _, v := range atno {
				if eq, err1 := this.module.ModuleEquipment.QueryEquipment(session.GetUserId(), v.O); err1 == pb.ErrorCode_Success {
					rsp.Equip = append(rsp.Equip, eq)
				}
			}
		}
	}

	if stoveLvConf := this.module.configure.GetSmithyStoveConf(stove.Lv); stoveLvConf != nil {
		maxT = stoveLvConf.MaxTemperature
	}
	if stove.RecoveTime == 0 && stove.Temperature <= maxT {
		stove.RecoveTime = configure.Now().Unix()
		update["recoveTime"] = stove.RecoveTime
	}
	stove.Data[req.ReelId].Value += 1
	stove.Forge[req.ReelId] += 1
	update["data"] = stove.Data

	update["forge"] = stove.Forge // 打造次数
	this.module.modelStove.updateSmithyStove(session.GetUserId(), update)
	// 检查是否提升了熟练度等级
	nextProficiency := this.module.configure.GetSmithyProficileData(req.ReelId, stove.Data[req.ReelId].Lv+1)
	if nextProficiency != nil && nextProficiency.Proficiency <= stove.Data[req.ReelId].Value { // 提升熟练度
		stove.Data[req.ReelId].Lv += 1
		stove.Data[req.ReelId].Value = 0
		// 校验是否解锁了新的图纸
		if nextProficiency.Type == comm.SmithyReelType3 {
			stove.Data[nextProficiency.Value1] = &pb.Mastery{
				Lv:    1,
				Value: 0,
			}
		}
		// 是否解锁新套装
		if nextProficiency.Type == comm.SmithyReelType4 {
			if rst, err := this.module.ModuleUser.GetUserExpand(session.GetUserId()); err == nil {
				rst.SuiteId = append(rst.SuiteId, nextProficiency.Value1)
				update := make(map[string]interface{}, 0)
				update["suiteId"] = rst.SuiteId
				this.module.ModuleUser.ChangeUserExpand(session.GetUserId(), update)
			}
		}
	}
	rsp.Data = stove
	session.SendMsg(string(this.module.GetType()), "forgeequip", rsp)
	// 更新图鉴信息
	for _, v := range rsp.Equip {
		if equipConf := this.module.configure.GetEquipmentConfigureById(v.CId); equipConf != nil { // 获取装备等级
			this.module.modelAtlas.CheckActivateAtlas(session.GetUserId(), v.CId, equipConf.Star, int32(len(v.AdverbEntry)+1), stove.Forge[req.ReelId])
		}
	}

	return
}