bl/logic/controller/pet_bag.go

package controller

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

// SavePetBagOrder 保存当前主背包和备用背包顺序
func (h Controller) SavePetBagOrder(
	data *pet.SavePetBagOrderInboundInfo,
	player *player.Player) (result *fight.NullOutboundInfo,
	err errorcode.ErrorCode) {
	if !player.SavePetBagOrder(data.PetList, data.BackupPetList) {
		return nil, errorcode.ErrorCodes.ErrPokemonIDMismatch
	}
	player.Service.Info.Save(*player.Info)
	return nil, 0
}

// PetRetrieveFromWarehouse 领回仓库精灵
func (h Controller) PetRetrieveFromWarehouse(
	data *pet.PET_RETRIEVE, player *player.Player) (result *fight.NullOutboundInfo, err errorcode.ErrorCode) {
	if _, ok := player.FindPetBagSlot(data.CatchTime); ok {
		return nil, 0
	}

	petInfo := player.Service.Pet.PetInfoOneByCatchTime(data.CatchTime)
	if petInfo == nil {
		return nil, 0
	}

	player.AddPetToAvailableBag(petInfo.Data)

	return nil, 0
}

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

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

	switch data.Flag {
	case 0:
		slot, ok := player.FindPetBagSlot(data.CatchTime)
		if !ok {
			return result, errorcode.ErrorCodes.ErrPokemonNotExists
		}
		if !slot.IsValid() {
			return result, errorcode.ErrorCodes.ErrPokemonIDMismatch
		}
		if !player.Service.Pet.Update(slot.PetInfo()) {
			return result, errorcode.ErrorCodes.ErrSystemError
		}

		slot.Remove()
		if slot.IsMainBag() {
			player.Service.Info.Save(*player.Info)
		}

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

		petInfo := player.Service.Pet.PetInfoOneByCatchTime(data.CatchTime)
		if petInfo == nil {
			return result, errorcode.ErrorCodes.ErrPokemonNotExists
		}
		player.AddPetToAvailableBag(petInfo.Data)
		result.PetInfo = petInfo.Data
	}

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

	return result, 0
}

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

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

	switch data.Flag {
	case 0:
		index, currentPet, ok := player.FindPet(data.CatchTime)
		if !ok {
			break
		}
		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.RemoveBackupPet(data.CatchTime)
		player.Info.BackupPetList = append(player.Info.BackupPetList, *currentPet)

	case 1:
		if len(player.Info.PetList) >= 6 {
			break
		}

		if _, _, ok := player.FindPet(data.CatchTime); ok {
			player.RemoveBackupPet(data.CatchTime)
			break
		}

		if index, backupPet, ok := player.FindBackupPet(data.CatchTime); ok {
			if index < 0 || index >= len(player.Info.BackupPetList) {
				return result, errorcode.ErrorCodes.ErrPokemonIDMismatch
			}

			result.PetInfo = *backupPet
			player.Info.PetList = append(player.Info.PetList, *backupPet)
			player.Info.BackupPetList = append(player.Info.BackupPetList[:index], player.Info.BackupPetList[index+1:]...)
			break
		}

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

		player.Info.PetList = append(player.Info.PetList, petInfo.Data)
		result.PetInfo = petInfo.Data
	}

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

	return result, 0
}