Merge d9d2af0 into 09b4d1f

Author: r-log

src/game/Object/PetMgr.cpp

@@ -0,0 +1,105 @@
+/**
+ * MaNGOS is a full featured server for World of Warcraft, supporting
+ * the following clients: 1.12.x, 2.4.3, 3.3.5a, 4.3.4a and 5.4.8
+ *
+ * Copyright (C) 2005-2025 MaNGOS <https://www.getmangos.eu>
+ *
+ * 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, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ *
+ * World of Warcraft, and all World of Warcraft or Warcraft art, images,
+ * and lore are copyrighted by Blizzard Entertainment, Inc.
+ */
+
+#include "PetMgr.h"
+#include "Player.h"
+#include "Pet.h"
+#include "Transports.h"
+#include "WorldPacket.h"
+#include "Opcodes.h"
+#include "Log.h"
+
+void PetMgr::LoadStableSlotsFromField(uint32 raw)
+{
+    m_stableSlots = raw;
+    if (m_stableSlots > MAX_PET_STABLES)
+    {
+        sLog.outError("Player can have not more %u stable slots, but have in DB %u", MAX_PET_STABLES, uint32(m_stableSlots));
+        m_stableSlots = MAX_PET_STABLES;
+    }
+}
+
+void PetMgr::Remove(PetSaveMode mode)
+{
+    if (Pet* pet = m_owner->GetPet())
+    {
+        pet->Unsummon(mode, m_owner);
+    }
+}
+
+void PetMgr::RemoveActionBar()
+{
+    WorldPacket data(SMSG_PET_SPELLS, 8);
+    data << ObjectGuid();
+    m_owner->SendDirectMessage(&data);
+}
+
+void PetMgr::UnsummonTemporaryIfAny()
+{
+    Pet* pet = m_owner->GetPet();
+    if (!pet)
+    {
+        return;
+    }
+
+    if (!m_temporaryUnsummonedPetNumber && pet->isControlled() && !pet->isTemporarySummoned())
+    {
+        m_temporaryUnsummonedPetNumber = pet->GetCharmInfo()->GetPetNumber();
+    }
+
+    if (Transport* petTransport = pet->GetTransport())
+    {
+        petTransport->RemovePassenger(pet);
+        pet->SetTransport(nullptr);
+    }
+
+    pet->Unsummon(PET_SAVE_AS_CURRENT, m_owner);
+}
+
+void PetMgr::ResummonTemporaryUnsummonedIfAny()
+{
+    if (!m_temporaryUnsummonedPetNumber)
+    {
+        return;
+    }
+
+    // not resummon in not appropriate state
+    if (m_owner->IsPetNeedBeTemporaryUnsummoned())
+    {
+        return;
+    }
+
+    if (m_owner->GetPetGuid())
+    {
+        return;
+    }
+
+    Pet* NewPet = new Pet;
+    if (!NewPet->LoadPetFromDB(m_owner, 0, m_temporaryUnsummonedPetNumber, true))
+    {
+        delete NewPet;
+    }
+
+    m_temporaryUnsummonedPetNumber = 0;
+}

src/game/Object/PetMgr.h

@@ -0,0 +1,96 @@
+/**
+ * MaNGOS is a full featured server for World of Warcraft, supporting
+ * the following clients: 1.12.x, 2.4.3, 3.3.5a, 4.3.4a and 5.4.8
+ *
+ * Copyright (C) 2005-2025 MaNGOS <https://www.getmangos.eu>
+ *
+ * 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, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ *
+ * World of Warcraft, and all World of Warcraft or Warcraft art, images,
+ * and lore are copyrighted by Blizzard Entertainment, Inc.
+ */
+
+#ifndef MANGOS_H_PETMGR
+#define MANGOS_H_PETMGR
+
+#include "Common.h"
+#include "Pet.h"  // PetSaveMode enum + MAX_PET_STABLES constant
+
+class Player;
+
+/**
+ * PetMgr — owns a Player's pet-ownership metadata: the stable-slot
+ * count persisted with the character row and the temporary-unsummon
+ * tracking number used by transports / mounts to bring the same pet
+ * back after a short interruption.
+ *
+ * What this DOES NOT own:
+ *  - The Pet creature itself (a Creature subclass living in the world,
+ *    managed by Object/Pet.{cpp,h}). Its lifecycle, AI, spells, and
+ *    persistence to the `character_pet` table all stay in Pet.cpp.
+ *  - The pet stable opcode handlers (CMSG_STABLE_PET and friends in
+ *    WorldHandlers/NPCHandler.cpp). They keep reading/updating the slot
+ *    count through Player's public accessors, so the stable-purchase
+ *    economics (gold cost via StableSlotPricesStore, MAX_PET_STABLES
+ *    cap) are untouched.
+ */
+class PetMgr
+{
+    public:
+        explicit PetMgr(Player* owner)
+            : m_owner(owner), m_stableSlots(0), m_temporaryUnsummonedPetNumber(0)
+        {
+        }
+
+        /// Number of paid stable slots the character has unlocked.
+        /// Persisted to `characters`.stable_slots; clamped to
+        /// MAX_PET_STABLES on load.
+        uint32 GetStableSlots() const { return m_stableSlots; }
+        void SetStableSlots(uint32 slots) { m_stableSlots = slots; }
+
+        /// Called from Player::LoadFromDB with the raw column value.
+        /// Clamps to MAX_PET_STABLES and logs a server error on
+        /// out-of-range data so an operator can detect a tampered row.
+        void LoadStableSlotsFromField(uint32 raw);
+
+        /// Pet number that was active before a temporary unsummon (e.g.
+        /// transport zone-in). Zero means no pending resummon.
+        uint32 GetTemporaryUnsummonedPetNumber() const { return m_temporaryUnsummonedPetNumber; }
+        void SetTemporaryUnsummonedPetNumber(uint32 petnumber) { m_temporaryUnsummonedPetNumber = petnumber; }
+
+        /// If the owner currently has a controlled pet, asks it to
+        /// unsummon with the given mode (see PetSaveMode).
+        void Remove(PetSaveMode mode);
+
+        /// SMSG_PET_SPELLS with an empty guid — clears the pet action
+        /// bar UI on the client.
+        void RemoveActionBar();
+
+        /// Stash the current non-temporary pet's number so we can bring
+        /// it back later, then unsummon with PET_SAVE_AS_CURRENT.
+        void UnsummonTemporaryIfAny();
+
+        /// Counterpart to UnsummonTemporaryIfAny: if we stashed a pet
+        /// number and it's now appropriate to resummon, load it from DB.
+        /// Clears the stash either way.
+        void ResummonTemporaryUnsummonedIfAny();
+
+    private:
+        Player* m_owner;
+        uint32  m_stableSlots;
+        uint32  m_temporaryUnsummonedPetNumber;
+};
+
+#endif // MANGOS_H_PETMGR