fix(Script/Spells): Validate Warlock spells (#9181)

This commit is contained in:
Kitzunu
2021-11-14 05:48:16 +01:00
committed by GitHub
parent 959ee62efe
commit 03c84db04d

View File

@@ -803,6 +803,11 @@ class spell_warl_demonic_circle_summon : public AuraScript
{
PrepareAuraScript(spell_warl_demonic_circle_summon);
bool Validate(SpellInfo const* /*spellInfo*/) override
{
return ValidateSpellInfo({ SPELL_WARLOCK_DEMONIC_CIRCLE_ALLOW_CAST, SPELL_WARLOCK_DEMONIC_CIRCLE_TELEPORT });
}
void HandleRemove(AuraEffect const* /*aurEff*/, AuraEffectHandleModes mode)
{
// If effect is removed by expire remove the summoned demonic circle too.
@@ -844,6 +849,11 @@ class spell_warl_demonic_circle_teleport : public AuraScript
{
PrepareAuraScript(spell_warl_demonic_circle_teleport);
bool Validate(SpellInfo const* /*spellInfo*/) override
{
return ValidateSpellInfo({ SPELL_WARLOCK_DEMONIC_CIRCLE_SUMMON });
}
void HandleTeleport(AuraEffect const* /*aurEff*/, AuraEffectHandleModes /*mode*/)
{
if (Player* player = GetTarget()->ToPlayer())
@@ -1010,6 +1020,16 @@ class spell_warl_health_funnel : public AuraScript
{
PrepareAuraScript(spell_warl_health_funnel);
bool Validate(SpellInfo const* /*spellInfo*/) override
{
return ValidateSpellInfo({
SPELL_WARLOCK_IMPROVED_HEALTH_FUNNEL_R2,
SPELL_WARLOCK_IMPROVED_HEALTH_FUNNEL_BUFF_R2,
SPELL_WARLOCK_IMPROVED_HEALTH_FUNNEL_R1,
SPELL_WARLOCK_IMPROVED_HEALTH_FUNNEL_BUFF_R1
});
}
void ApplyEffect(AuraEffect const* /*aurEff*/, AuraEffectHandleModes /*mode*/)
{
Unit* caster = GetCaster();