package controller

import (
	"blazing/common/socket/errorcode"
	"blazing/logic/service/pet"
	"blazing/logic/service/player"
	"blazing/modules/player/model"
)

// GetPetInfo 获取精灵信息
func (h Controller) GetPetInfo(
	data *pet.InInfo,
	player *player.Player) (result *model.PetInfo,
	err errorcode.ErrorCode) {
	_, petInfo, found := player.FindPet(data.CatchTime)
	if found {
		result = petInfo
		return result, 0
	}

	ret := player.Service.Pet.PetInfoOneByCatchTime(data.CatchTime)
	if ret == nil {
		return nil, errorcode.ErrorCodes.ErrPokemonNotExists
	}

	result = &ret.Data
	return result, 0
}

// GetUserBagPetInfo 获取主背包和并列备用精灵列表
func (h Controller) GetUserBagPetInfo(
	data *pet.GetUserBagPetInfoInboundEmpty,
	player *player.Player) (result *pet.GetUserBagPetInfoOutboundInfo,
	err errorcode.ErrorCode) {
	return buildUserBagPetInfo(player), 0
}

// GetPetList 获取当前主背包列表
func (h Controller) GetPetList(
	data *pet.GetPetListInboundEmpty,
	player *player.Player) (result *pet.GetPetListOutboundInfo,
	err errorcode.ErrorCode) {
	return buildPetListOutboundInfo(player.Info.PetList), 0
}

// GetPetReleaseList 获取仓库可放生列表
func (h Controller) GetPetReleaseList(
	data *pet.GetPetListFreeInboundEmpty,
	player *player.Player) (result *pet.GetPetListOutboundInfo,
	err errorcode.ErrorCode) {
	syncBackupPetList(player)
	return buildPetListOutboundInfo(buildWarehousePetList(player)), 0
}

<<<<<<< ours
// PetReleaseToWarehouse 将精灵从仓库包中放生
// data: 包含精灵ID和捕获时间的输入信息
// player: 当前玩家对象
// 返回: 无数据和错误码
func (h Controller) PetReleaseToWarehouse(
	data *pet.PET_ROWEI, player *player.Player) (result *fight.NullOutboundInfo, err errorcode.ErrorCode) {
	_, _, inBag := player.FindPet(data.CatchTime)
	_, _, inBackup := findBackupPet(player, data.CatchTime)
	freeForbidden := xmlres.PetMAP[int(data.ID)].FreeForbidden
	// 如果背包没找到,再放入背包
	if inBag || inBackup || freeForbidden == 1 {
		return nil, errorcode.ErrorCodes.ErrCannotReleaseNonWarehouse
	}

	if !player.Service.Pet.UpdateFree(data.CatchTime, 1) {
		return nil, errorcode.ErrorCodes.ErrSystemError
	}
	return nil, err

}

// PetRetrieveFromWarehouse 领回包
func (h Controller) PetRetrieveFromWarehouse(
	data *pet.PET_RETRIEVE, player *player.Player) (result *fight.NullOutboundInfo, err errorcode.ErrorCode) { //这个时候player应该是空的

	//如果背包没找到,再放入背包
	if _, _, ok := player.FindPet(data.CatchTime); !ok {
		if petInfo := player.Service.Pet.PetInfoOneByCatchTime(data.CatchTime); petInfo != nil {
			syncBackupPetList(player)
			if len(player.Info.PetList) < 6 {
				player.Info.PetList = append(player.Info.PetList, petInfo.Data)
			} else if len(player.Info.BackupPetList) < 6 {
				player.Info.BackupPetList = append(player.Info.BackupPetList, petInfo.Data)
			}
			player.Service.Info.Save(*player.Info)
		}
	}

	return nil, 0

}

// TogglePetBagWarehouse 精灵背包仓库切换
func (h Controller) TogglePetBagWarehouse(
	data *pet.PetReleaseInboundInfo,
	player *player.Player) (result *pet.PetReleaseOutboundInfo, err errorcode.ErrorCode) {
	result = &pet.PetReleaseOutboundInfo{}
	result.Flag = uint32(data.Flag)

	if player.IsArenaSwitchLocked() {
		return result, errorcode.ErrorCodes.ErrChampionCannotSwitch
	}

	syncBackupPetList(player)

	switch data.Flag {
	case 0:
		if index, currentPet, ok := player.FindPet(data.CatchTime); ok {
			if index < 0 || index >= len(player.Info.PetList) {
				return result, errorcode.ErrorCodes.ErrPokemonIDMismatch
			}
			if !player.Service.Pet.Update(*currentPet) {
				return result, errorcode.ErrorCodes.ErrSystemError
			}
			player.Info.PetList = append(player.Info.PetList[:index], player.Info.PetList[index+1:]...)
			player.Service.Info.Save(*player.Info)
			break
		}

		index, currentPet, ok := findBackupPet(player, data.CatchTime)
		if !ok {
			return result, errorcode.ErrorCodes.ErrPokemonNotExists
		}
		if index < 0 || index >= len(player.Info.BackupPetList) {
			return result, errorcode.ErrorCodes.ErrPokemonIDMismatch
		}
		if !player.Service.Pet.Update(*currentPet) {
			return result, errorcode.ErrorCodes.ErrSystemError
		}
		player.Info.BackupPetList = append(player.Info.BackupPetList[:index], player.Info.BackupPetList[index+1:]...)

	case 1:
		if len(player.Info.PetList) >= 6 && len(player.Info.BackupPetList) >= 6 {
			return result, errorcode.ErrorCodes.ErrPokemonIDMismatch
		}
		if _, _, ok := player.FindPet(data.CatchTime); ok {
			return result, 0
		}
		if _, _, ok := findBackupPet(player, data.CatchTime); ok {
			return result, 0
		}

		petInfo := player.Service.Pet.PetInfoOneByCatchTime(data.CatchTime)
		if petInfo == nil {
			return result, errorcode.ErrorCodes.ErrPokemonNotExists
		}
		if len(player.Info.PetList) < 6 {
			player.Info.PetList = append(player.Info.PetList, petInfo.Data)
		} else {
			player.Info.BackupPetList = append(player.Info.BackupPetList, petInfo.Data)
		}
		result.PetInfo = petInfo.Data

	}

	if len(player.Info.PetList) > 0 {
		result.FirstPetTime = player.Info.PetList[0].CatchTime
	}

	return nil, 0
}

func (h Controller) TogglePetBagWarehouseLegacy(
	data *pet.PetReleaseLegacyInboundInfo,
	player *player.Player) (
	result *pet.PetReleaseOutboundInfo,
	err errorcode.ErrorCode) { //这个时候player应该是空的
	//放入背包=数据库置1+添加到背包+pet release发包    仓库=数据库置0+移除背包   设置首发等于取到首发精灵后重新排序
	//这里只修改，因为添加和移除背包在宠物获取时已经做了

	result = &pet.PetReleaseOutboundInfo{}
	result.Flag = uint32(data.Flag)
	//擂台住不能换精灵
	if player.IsArenaSwitchLocked() {
		return result, errorcode.ErrorCodes.ErrChampionCannotSwitch
	}
	switch data.Flag {
	case 0:

		index, pet, ok := player.FindPet(data.CatchTime)
		if ok {
			// ========== 新增：index合法性校验 ==========
			if index < 0 || index >= len(player.Info.PetList) {
				return result, errorcode.ErrorCodes.ErrPokemonIDMismatch
			}

			if !player.Service.Pet.Update(*pet) {
				return result, errorcode.ErrorCodes.ErrSystemError
			}

			player.Info.BackupPetList = removePetByCatchTime(player.Info.BackupPetList, data.CatchTime)
			player.Info.BackupPetList = append(player.Info.BackupPetList, *pet)
			player.Info.PetList = append(player.Info.PetList[:index], player.Info.PetList[index+1:]...)
		}

		// break // 只移除第一个匹配值，若需移除所有，可省略 break 继续循环
	case 1:
		if len(player.Info.PetList) < 6 {
			//todo 背包
			_, _, ok := player.FindPet(data.CatchTime)

			//如果背包没找到,再放入背包
			if !ok {
				r := player.Service.Pet.PetInfoOneByCatchTime(data.CatchTime)
				if r == nil {
					return result, errorcode.ErrorCodes.ErrPokemonNotExists
				}
				player.Info.PetList = append(player.Info.PetList, r.Data)
				player.Info.BackupPetList = removePetByCatchTime(player.Info.BackupPetList, data.CatchTime)
				result.PetInfo = r.Data
			} else {
				player.Info.BackupPetList = removePetByCatchTime(player.Info.BackupPetList, data.CatchTime)
			}

		}

	}

	if len(player.Info.PetList) > 0 {
		result.FirstPetTime = player.Info.PetList[0].CatchTime //设置首发
	}

	return result, 0
}

=======
>>>>>>> theirs
// PlayerShowPet 精灵展示
func (h Controller) PlayerShowPet(
	data *pet.PetShowInboundInfo, player *player.Player) (result *pet.PetShowOutboundInfo, err errorcode.ErrorCode) {
	result = &pet.PetShowOutboundInfo{
		UserID:    data.Head.UserID,
		CatchTime: data.CatchTime,
		Flag:      data.Flag,
	}

	_, currentPet, ok := player.FindPet(data.CatchTime)
	if data.Flag == 0 {
		player.SetPetDisplay(0, nil)
		player.GetSpace().RefreshUserInfo(player)
		defer player.GetSpace().Broadcast(player, data.Head.CMD, result)
		return
	}

	if !ok {
		return nil, errorcode.ErrorCodes.ErrPokemonNotExists
	}

	player.SetPetDisplay(data.Flag, currentPet)
	player.GetSpace().RefreshUserInfo(player)
	result = buildPetShowOutboundInfo(data.Head.UserID, data.Flag, currentPet)
	defer player.GetSpace().Broadcast(player, data.Head.CMD, result)
	return
<<<<<<< ours

}

// PetOneCure 单体治疗
func (h Controller) PetOneCure(
	data *pet.PetOneCureInboundInfo, player *player.Player) (result *pet.PetOneCureOutboundInfo, err errorcode.ErrorCode) { //这个时候player应该是空的
	if player.IsArenaHealLocked() {
		return result, errorcode.ErrorCodes.ErrChampionCannotHeal
	}

	_, currentPet, ok := player.FindPet(data.CatchTime)
	if ok {
		defer currentPet.Cure()

	}

	return &pet.PetOneCureOutboundInfo{
		CatchTime: data.CatchTime,
	}, 0

}

// PetFirst 精灵首发
func (h Controller) PetFirst(
	data *pet.PetDefaultInboundInfo, player *player.Player) (result *pet.PetDefaultOutboundInfo, err errorcode.ErrorCode) { //这个时候player应该是空的
	//擂台住不能换精灵
	if player.IsArenaSwitchLocked() {
		return result, errorcode.ErrorCodes.ErrChampionCannotSwitch
	}

	result = &pet.PetDefaultOutboundInfo{}

	index, _, ok := player.FindPet(data.CatchTime)
	if ok && index != 0 {

		player.Info.PetList[index], player.Info.PetList[0] = player.Info.PetList[0], player.Info.PetList[index]

		result.IsDefault = 1
	}

	return result, 0

}

// SetPetExp 设置宠物经验
func (h Controller) SetPetExp(data *pet.PetSetExpInboundInfo, player *player.Player) (result *pet.PetSetExpOutboundInfo, err errorcode.ErrorCode) {
	_, currentPet, found := player.FindPet(data.CatchTime)
	if found && currentPet.Level < 100 {
		player.AddPetExp(currentPet, data.Exp)
		return &pet.PetSetExpOutboundInfo{Exp: player.Info.ExpPool}, 0
	}

	return &pet.PetSetExpOutboundInfo{Exp: player.Info.ExpPool}, errorcode.ErrorCodes.ErrSystemError
=======
>>>>>>> theirs
}