blob: dd9490d447c57ed0a049fdab9861f5e78be888cb (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
/*
* This file is part of the TrinityCore Project. See AUTHORS file for Copyright information
*
* This program is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License as published by the
* Free Software Foundation; either version 2 of the License, or (at your
* option) any later version.
*
* This program is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
* more details.
*
* You should have received a copy of the GNU General Public License along
* with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "WorldSession.h"
#include "BattlePetMgr.h"
#include "BattlePetPackets.h"
#include "ObjectAccessor.h"
#include "Player.h"
#include "TemporarySummon.h"
void WorldSession::HandleBattlePetRequestJournal(WorldPackets::BattlePet::BattlePetRequestJournal& /*battlePetRequestJournal*/)
{
GetBattlePetMgr()->SendJournal();
}
void WorldSession::HandleBattlePetRequestJournalLock(WorldPackets::BattlePet::BattlePetRequestJournalLock& /*battlePetRequestJournalLock*/)
{
GetBattlePetMgr()->SendJournalLockStatus();
if (GetBattlePetMgr()->HasJournalLock())
GetBattlePetMgr()->SendJournal();
}
void WorldSession::HandleBattlePetSetBattleSlot(WorldPackets::BattlePet::BattlePetSetBattleSlot& battlePetSetBattleSlot)
{
if (BattlePets::BattlePet* pet = GetBattlePetMgr()->GetPet(battlePetSetBattleSlot.PetGuid))
if (WorldPackets::BattlePet::BattlePetSlot* slot = GetBattlePetMgr()->GetSlot(BattlePets::BattlePetSlot(battlePetSetBattleSlot.Slot)))
slot->Pet = pet->PacketInfo;
}
void WorldSession::HandleQueryBattlePetName(WorldPackets::BattlePet::QueryBattlePetName& queryBattlePetName)
{
WorldPackets::BattlePet::QueryBattlePetNameResponse response;
response.BattlePetID = queryBattlePetName.BattlePetID;
Creature* summonedBattlePet = ObjectAccessor::GetCreatureOrPetOrVehicle(*_player, queryBattlePetName.UnitGUID);
if (!summonedBattlePet || !summonedBattlePet->IsSummon())
{
SendPacket(response.Write());
return;
}
response.CreatureID = summonedBattlePet->GetEntry();
response.Timestamp = summonedBattlePet->GetBattlePetCompanionNameTimestamp();
Unit* petOwner = summonedBattlePet->ToTempSummon()->GetSummonerUnit();
if (!petOwner->IsPlayer())
{
SendPacket(response.Write());
return;
}
BattlePets::BattlePet const* battlePet = petOwner->ToPlayer()->GetSession()->GetBattlePetMgr()->GetPet(queryBattlePetName.BattlePetID);
if (!battlePet)
{
SendPacket(response.Write());
return;
}
response.Name = battlePet->PacketInfo.Name;
if (battlePet->DeclinedName)
{
response.HasDeclined = true;
response.DeclinedNames = *battlePet->DeclinedName;
}
response.Allow = !response.Name.empty();
SendPacket(response.Write());
}
void WorldSession::HandleBattlePetSetFlags(WorldPackets::BattlePet::BattlePetSetFlags& battlePetSetFlags)
{
if (!GetBattlePetMgr()->HasJournalLock())
return;
if (BattlePets::BattlePet* pet = GetBattlePetMgr()->GetPet(battlePetSetFlags.PetGuid))
{
if (battlePetSetFlags.ControlType == BattlePets::FLAGS_CONTROL_TYPE_APPLY)
pet->PacketInfo.Flags |= battlePetSetFlags.Flags;
else // FLAGS_CONTROL_TYPE_REMOVE
pet->PacketInfo.Flags &= ~battlePetSetFlags.Flags;
if (pet->SaveInfo != BattlePets::BATTLE_PET_NEW)
pet->SaveInfo = BattlePets::BATTLE_PET_CHANGED;
}
}
void WorldSession::HandleBattlePetClearFanfare(WorldPackets::BattlePet::BattlePetClearFanfare& battlePetClearFanfare)
{
GetBattlePetMgr()->ClearFanfare(battlePetClearFanfare.PetGuid);
}
void WorldSession::HandleBattlePetSummon(WorldPackets::BattlePet::BattlePetSummon& battlePetSummon)
{
if (*_player->m_activePlayerData->SummonedBattlePetGUID != battlePetSummon.PetGuid)
GetBattlePetMgr()->SummonPet(battlePetSummon.PetGuid);
else
GetBattlePetMgr()->DismissPet();
}
void WorldSession::HandleBattlePetUpdateNotify(WorldPackets::BattlePet::BattlePetUpdateNotify& battlePetUpdateNotify)
{
GetBattlePetMgr()->UpdateBattlePetData(battlePetUpdateNotify.PetGuid);
}
|
