Core/Spells: Validate known heirlooms for SPELL_EFFECT_CREATE_HEIRLOOM_ITEM

Port From (https://github.com/TrinityCore/TrinityCore/commit/cd7c5eff49bc877fd39950c94b04a1cb2168e380)
2023-06-20 09:39:06 -04:00
parent 1fb0002097
commit 378b0ffccd
2 changed files with 15 additions and 0 deletions
@@ -229,6 +229,11 @@ namespace Game.Entities
                _owner.GetPlayer().AddHeirloom(itemId, (uint)flags);
        }

+        public bool HasHeirloom(uint itemId)
+        {
+            return _heirlooms.ContainsKey(itemId);
+        }
+        
        public void UpgradeHeirloom(uint itemId, uint castItem)
        {
            Player player = _owner.GetPlayer();
@@ -5575,6 +5575,16 @@ namespace Game.Spells
                        }
                        break;
                    }
+                    case SpellEffectName.CreateHeirloomItem:
+                    {
+                        if (!m_caster.IsPlayer())
+                            return SpellCastResult.BadTargets;
+
+                        if (!m_caster.ToPlayer().GetSession().GetCollectionMgr().HasHeirloom(m_misc.ItemId))
+                            return SpellCastResult.BadTargets;
+
+                        break;
+                    }
                    case SpellEffectName.GiveArtifactPower:
                    case SpellEffectName.GiveArtifactPowerNoBonus:
                    {