diff --git a/src/server/game/Scripting/ScriptDefines/LootScript.cpp b/src/server/game/Scripting/ScriptDefines/LootScript.cpp index 6b49ff5a1..a6231b4a4 100644 --- a/src/server/game/Scripting/ScriptDefines/LootScript.cpp +++ b/src/server/game/Scripting/ScriptDefines/LootScript.cpp @@ -23,13 +23,16 @@ void ScriptMgr::OnLootMoney(Player* player, uint32 gold) { ASSERT(player); - ExecuteScript([&](LootScript* script) - { - script->OnLootMoney(player, gold); - }); + CALL_ENABLED_HOOKS(LootScript, LOOTHOOK_ON_LOOT_MONEY, script->OnLootMoney(player, gold)); } -LootScript::LootScript(const char* name) : ScriptObject(name) +LootScript::LootScript(const char* name, std::vector enabledHooks) + : ScriptObject(name, LOOTHOOK_END) { - ScriptRegistry::AddScript(this); + // If empty - enable all available hooks. + if (enabledHooks.empty()) + for (uint16 i = 0; i < LOOTHOOK_END; ++i) + enabledHooks.emplace_back(i); + + ScriptRegistry::AddScript(this, std::move(enabledHooks)); } diff --git a/src/server/game/Scripting/ScriptDefines/LootScript.h b/src/server/game/Scripting/ScriptDefines/LootScript.h index d71711383..445abdddc 100644 --- a/src/server/game/Scripting/ScriptDefines/LootScript.h +++ b/src/server/game/Scripting/ScriptDefines/LootScript.h @@ -19,11 +19,18 @@ #define SCRIPT_OBJECT_LOOT_SCRIPT_H_ #include "ScriptObject.h" +#include + +enum LootHook +{ + LOOTHOOK_ON_LOOT_MONEY, + LOOTHOOK_END +}; class LootScript : public ScriptObject { protected: - LootScript(const char* name); + LootScript(const char* name, std::vector enabledHooks = std::vector()); public: [[nodiscard]] bool IsDatabaseBound() const override { return false; }