Core/Spells: Implemented spell effect 204 (SPELL_EFFECT_CHANGE_BATTLEPET_QUALITY)

Port From (https://github.com/TrinityCore/TrinityCore/commit/c4fee8c5cf85aeddc45aad18d7a944efa59022bd)
2022-01-05 20:38:57 -05:00
parent 13f73adf73
commit 5e0fd36ada
4 changed files with 128 additions and 0 deletions
@@ -591,6 +591,42 @@ namespace Game.BattlePets
            }
        }

+        public void ChangeBattlePetQuality(ObjectGuid guid, BattlePetBreedQuality quality)
+        {
+            if (!HasJournalLock())
+                return;
+
+            BattlePet pet = GetPet(guid);
+            if (pet == null)
+                return;
+
+            if (quality > BattlePetBreedQuality.Rare)
+                return;
+
+            var battlePetSpecies = CliDB.BattlePetSpeciesStorage.LookupByKey(pet.PacketInfo.Species);
+            if (battlePetSpecies != null)
+                if (battlePetSpecies.GetFlags().HasFlag(BattlePetSpeciesFlags.CantBattle))
+                    return;
+
+            byte qualityValue = (byte)quality;
+            if (pet.PacketInfo.Quality >= qualityValue)
+                return;
+
+            pet.PacketInfo.Quality = qualityValue;
+            pet.CalculateStats();
+            pet.PacketInfo.Health = pet.PacketInfo.MaxHealth;
+
+            if (pet.SaveInfo != BattlePetSaveInfo.New)
+                pet.SaveInfo = BattlePetSaveInfo.Changed;
+
+            List<BattlePet> updates = new();
+            updates.Add(pet);
+            SendUpdates(updates, false);
+
+            // UF::PlayerData::CurrentBattlePetBreedQuality isn't updated (Intended)
+            // _owner->GetPlayer()->SetCurrentBattlePetBreedQuality(qualityValue);
+        }
+
        public void HealBattlePetsPct(byte pct)
        {
            // TODO: After each Pet Battle, any injured companion will automatically