diff --git a/LuaFunctions.cpp b/LuaFunctions.cpp index 841a86c..fa89d37 100644 --- a/LuaFunctions.cpp +++ b/LuaFunctions.cpp @@ -389,6 +389,9 @@ ElunaRegister UnitMethods[] = { "AddAura", &LuaUnit::AddAura }, { "RemoveAura", &LuaUnit::RemoveAura }, { "RemoveAllAuras", &LuaUnit::RemoveAllAuras }, +#if !defined(CLASSIC) + { "RemoveArenaAuras", &LuaUnit::RemoveArenaAuras }, +#endif { "ClearInCombat", &LuaUnit::ClearInCombat }, { "DeMorph", &LuaUnit::DeMorph }, { "SendUnitWhisper", &LuaUnit::SendUnitWhisper }, @@ -663,6 +666,9 @@ ElunaRegister PlayerMethods[] = { "ModifyMoney", &LuaPlayer::ModifyMoney }, { "LearnSpell", &LuaPlayer::LearnSpell }, { "LearnTalent", &LuaPlayer::LearnTalent }, +#if !defined(CLASSIC) + { "RemoveArenaSpellCooldowns", &LuaPlayer::RemoveArenaSpellCooldowns }, +#endif { "RemoveItem", &LuaPlayer::RemoveItem }, { "RemoveLifetimeKills", &LuaPlayer::RemoveLifetimeKills }, { "ResurrectPlayer", &LuaPlayer::ResurrectPlayer }, diff --git a/PlayerMethods.h b/PlayerMethods.h index e60006c..d61e934 100644 --- a/PlayerMethods.h +++ b/PlayerMethods.h @@ -3728,6 +3728,17 @@ namespace LuaPlayer return 0; } +#if !defined(CLASSIC) + /** + * Remove cooldowns on spells that have less than 10 minutes of cooldown from the [Player], similarly to when you enter an arena. + */ + int RemoveArenaSpellCooldowns(lua_State* /*L*/, Player* player) + { + player->RemoveArenaSpellCooldowns(); + return 0; + } +#endif + /** * Resurrects the [Player]. * diff --git a/UnitMethods.h b/UnitMethods.h index 567e85b..3ff3c2d 100644 --- a/UnitMethods.h +++ b/UnitMethods.h @@ -2657,6 +2657,17 @@ namespace LuaUnit return 0; } +#if !defined(CLASSIC) + /** + * Removes all positive visible [Aura]'s from the [Unit]. + */ + int RemoveArenaAuras(lua_State* /*L*/, Unit* unit) + { + unit->RemoveArenaAuras(); + return 0; + } +#endif + /** * Adds the given unit state for the [Unit]. *