package rtask

import (
	"go_dreamfactory/comm"
	"go_dreamfactory/pb"
	"go_dreamfactory/utils"

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

func (this *apiComp) BattleFinishCheck(session comm.IUserSession, req *pb.RtaskBattleFinishReq) (code pb.ErrorCode) {
	if req.RtaskId == 0 || req.RtaskSubId == 0 {
		code = pb.ErrorCode_ReqParameterError
	}
	return
}

func (this *apiComp) BattleFinish(session comm.IUserSession, req *pb.RtaskBattleFinishReq) (code pb.ErrorCode, data proto.Message) {
	if code = this.BattleFinishCheck(session, req); code != pb.ErrorCode_Success {
		return
	}
	// 获取当前任务配置
	conf := this.moduleRtask.configure.getRtaskById(req.RtaskId)
	if conf == nil {
		code = pb.ErrorCode_ConfigNoFound
		return
	}

	// 获取支线任务配置
	sideConf := this.moduleRtask.configure.getRtaskSidById(req.RtaskSubId)
	if sideConf == nil {
		code = pb.ErrorCode_ConfigNoFound
		return
	}

	// 获取玩家的任务
	rtask := &pb.DBRtask{}
	if err := this.moduleRtask.modelRtask.Get(session.GetUserId(), rtask); err != nil {
		return
	}

	var (
		frtaskArr *pb.FrtaskIds //完成的任务
		ok        bool
	)

	// 查找任务组
	if frtaskArr, ok = rtask.FrtaskIds[conf.Group]; !ok {
		frtaskArr = &pb.FrtaskIds{}
	}

	if _, ok := utils.Findx(frtaskArr.RtaskIds, req.RtaskId); !ok {
		// 更新完成的任务
		frtaskArr.RtaskIds = append(frtaskArr.RtaskIds, req.RtaskId)
		if rtask.FrtaskIds == nil {
			rtask.FrtaskIds = make(map[int32]*pb.FrtaskIds)
		}
		rtask.FrtaskIds[conf.Group] = frtaskArr

		update := map[string]interface{}{
			"frtaskIds": rtask.FrtaskIds,
		}
		if err := this.moduleRtask.modelRtask.Change(session.GetUserId(), update); err != nil {
			code = pb.ErrorCode_SystemError
			return
		}

		// 发奖
		for _, v := range sideConf.Reward {
			if v.ChooseId == 0 {
				this.moduleRtask.DispenseRes(session, v.Reward, true)
			}
		}
	}

	if err := session.SendMsg(string(this.moduleRtask.GetType()), RtaskSubTypeBattleFinish,
		&pb.RtaskBattleFinishResp{
			RtaskId:    req.RtaskId,
			RtaskSubId: req.RtaskSubId,
		}); err != nil {
		code = pb.ErrorCode_SystemError
	}

	return
}