package user

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

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

func (this *apiComp) SettingteamCheck(session comm.IUserSession, req *pb.UserSettingteamReq) (code pb.ErrorCode) {
	if len(req.HeroObjIds) == 0 {
		code = pb.ErrorCode_ReqParameterError
	}
	return
}

func (this *apiComp) Settingteam(session comm.IUserSession, req *pb.UserSettingteamReq) (code pb.ErrorCode, data proto.Message) {
	if code = this.SettingteamCheck(session, req); code != pb.ErrorCode_Success {
		return
	}

	rsp := &pb.UserSettingteamResp{}
	if result, err := this.module.modelExpand.GetUserExpand(session.GetUserId()); err != nil {
		this.module.Errorf("uid:%v err: %v", session.GetUserId(), err)
		return
	} else {
		result.TeamHeroIds = req.HeroObjIds
		update := map[string]interface{}{
			"teamHeroIds": req.HeroObjIds,
		}
		if err := this.module.modelExpand.ChangeUserExpand(session.GetUserId(), update); err != nil {
			code = pb.ErrorCode_DBError
			return
		}
	}
	rsp.Uid = session.GetUserId()
	if err := session.SendMsg(string(this.module.GetType()), UserSubTypeSettingteam, rsp); err != nil {
		code = pb.ErrorCode_SystemError
	}
	return
}