CMaNGOS: Add SpellScript & AuraScript hooks

Partial implementation. Still need to fix up method files.

Author: Niam5

ElunaSpellWrapper.cpp

@@ -19,7 +19,7 @@ ElunaProcInfo::ElunaProcInfo(Unit* actor, Unit* actionTarget, uint32 typeMask,
 #endif
 {
 }
-
+#if defined ELUNA_TRINITY
 ElunaProcInfo::ElunaProcInfo(ProcEventInfo& procInfo, Map* map)
     : _actor(procInfo.GetActor()), _actionTarget(procInfo.GetActionTarget()), _typeMask(procInfo.GetTypeMask()), _spellTypeMask(procInfo.GetSpellTypeMask()), _spellPhaseMask(procInfo.GetSpellPhaseMask())
     , _hitMask(procInfo.GetHitMask()), _spell(const_cast<Spell*>(procInfo.GetProcSpell())), _spellInfo(procInfo.GetSpellInfo()), _schoolMask(procInfo.GetSchoolMask()), _damage(0)
@@ -58,13 +58,29 @@ ElunaProcInfo::ElunaProcInfo(ProcEventInfo& procInfo, Map* map)
         }
     }
 }
+#elif defined ELUNA_CMANGOS
+ElunaProcInfo::ElunaProcInfo(ProcExecutionData& procInfo, Map* map)
+    : _actor(procInfo.attacker), _actionTarget(procInfo.victim), _spell(const_cast<Spell*>(procInfo.spell))
+    , _spellInfo(procInfo.spellInfo), _damage(0)
+    , _attackType(BASE_ATTACK), _damageAbsorb(0), _resist(0), _block(0)
+    , _heal(0), _effectiveHeal(0), _healAbsorb(0), _map(map)
+#ifdef TRACKABLE_PTR_NAMESPACE
+    ,m_scriptRef(this, NoopAuraDeleter())
+#endif
+{
+}
+#endif
 
 SpellInfo const* ElunaProcInfo::GetSpellInfo() const
 {
     if (_spellInfo)
         return _spellInfo;
     if (_spell)
+#if defined ELUNA_TRINITY
         return _spell->GetSpellInfo();
+#elif defined ELUNA_CMANGOS
+        return _spell->GetSpellProto();
+#endif
     return nullptr;
 }
 
@@ -80,7 +96,7 @@ void ElunaProcInfo::SetHeal(uint32 heal)
     _heal = heal;
     _effectiveHeal = heal;
 }
-
+#if defined ELUNA_TRINITY
 void ElunaProcInfo::ApplyToProcEventInfo(ProcEventInfo& procInfo) const
 {
     if (DamageInfo* damageInfo = procInfo.GetDamageInfo())
@@ -122,11 +138,45 @@ void ElunaProcInfo::ApplyToProcEventInfo(ProcEventInfo& procInfo) const
         }
     }
 }
+#elif defined ELUNA_CMANGOS
+void ElunaProcInfo::ApplyToProcEventInfo(ProcExecutionData& procInfo) const
+{
+    if (HasDamage())
+    {
+        int32 damageDiff = static_cast<int32>(_damage) - static_cast<int32>(procInfo.damage);
+        if (damageDiff != 0)
+            procInfo.damage = damageDiff;
+
+        uint32 currentAbsorb = procInfo.absorb;
+
+        uint32 absorbToAdd = (_damageAbsorb > currentAbsorb) ? (_damageAbsorb - currentAbsorb) : 0;
+
+        if (absorbToAdd > 0)
+            procInfo.absorb = absorbToAdd;
+    }
+
+    if (HasHeal())
+    {
+        uint32 currentAbsorb = procInfo.absorb;
+        uint32 absorbToAdd = (_healAbsorb > currentAbsorb) ? (_healAbsorb - currentAbsorb) : 0;
+
+        if (absorbToAdd > 0)
+            procInfo.absorb = absorbToAdd;
+
+        procInfo.healthGain = _effectiveHeal;
+    }
+}
+#endif
 
+#ifdef ELUNA_TRINITY
 ElunaSpellInfo::ElunaSpellInfo(uint32 spellId) : _spellInfo(sSpellMgr->GetSpellInfo(spellId))
 {
-#ifdef ELUNA_TRINITY
     if (_spellInfo)
         m_scriptRef = Trinity::unique_trackable_ptr<ElunaSpellInfo>(this, NoopAuraDeleter());
+#elif defined ELUNA_CMANGOS
+ElunaSpellInfo::ElunaSpellInfo(uint32 spellId) : _spellInfo(sSpellMgr.GetSpellEntry(spellId))
+{
+    if (_spellInfo)
+        m_scriptRef = MaNGOS::unique_trackable_ptr<ElunaSpellInfo>(this, NoopAuraDeleter());
 #endif
 }

ElunaSpellWrapper.h

@@ -9,17 +9,21 @@
 class Unit;
 class Spell;
 class Map;
+#ifdef ELUNA_TRINITY
 class SpellInfo;
 class ProcEventInfo;
 class DamageInfo;
 class HealInfo;
-#ifdef ELUNA_TRINITY
-enum SpellSchoolMask : uint32;
-#else
-enum SpellSchoolMask;
-#endif
 enum DamageEffectType : uint8;
 enum WeaponAttackType : uint8;
+#else
+class SpellEntry;
+class ProcExecutionData;
+enum DamageEffectType;
+enum WeaponAttackType;
+typedef SpellEntry SpellInfo;
+#endif
+enum SpellSchoolMask : uint32;
 #ifdef ELUNA_TRINITY
 namespace Trinity
 {
@@ -29,6 +33,15 @@ namespace Trinity
     template<typename T>
     class unique_weak_ptr;
 }
+#elif defined ELUNA_CMANGOS
+namespace MaNGOS
+{
+    template<typename T>
+    class unique_trackable_ptr;
+
+    template<typename T>
+    class unique_weak_ptr;
+}
 #endif
 
 class ElunaProcInfo
@@ -60,14 +73,20 @@ class ElunaProcInfo
     struct NoopAuraDeleter { void operator()(ElunaProcInfo*) const { } };
 #ifdef ELUNA_TRINITY
     Trinity::unique_trackable_ptr<ElunaProcInfo> m_scriptRef;
+#elif defined ELUNA_CMANGOS
+    MaNGOS::unique_trackable_ptr<ElunaProcInfo> m_scriptRef;
 #endif
 
 public:
     ElunaProcInfo(Unit* actor, Unit* actionTarget, uint32 typeMask,
         uint32 spellTypeMask, uint32 spellPhaseMask, uint32 hitMask,
         Spell* spell, SpellInfo const* spellInfo, SpellSchoolMask schoolMask, Map* map);
 
+#if defined ELUNA_TRINITY
     explicit ElunaProcInfo(ProcEventInfo& procInfo, Map* map);
+#elif defined ELUNA_CMANGOS
+    explicit ElunaProcInfo(ProcExecutionData& procInfo, Map* map);
+#endif
     ~ElunaProcInfo()
     {
 #ifdef TRACKABLE_PTR_NAMESPACE
@@ -114,8 +133,11 @@ class ElunaProcInfo
     const Map* GetMap() const { return _map; }
 #ifdef ELUNA_TRINITY
     Trinity::unique_weak_ptr<ElunaProcInfo> GetWeakPtr() const { return m_scriptRef; }
-#endif
     void ApplyToProcEventInfo(ProcEventInfo& procInfo) const;
+#elif defined ELUNA_CMANGOS
+    MaNGOS::unique_weak_ptr<ElunaProcInfo> GetWeakPtr() const { return m_scriptRef; }
+    void ApplyToProcEventInfo(ProcExecutionData& procInfo) const;
+#endif
 };
 
 class ElunaSpellInfo
@@ -125,6 +147,8 @@ class ElunaSpellInfo
     struct NoopAuraDeleter { void operator()(ElunaSpellInfo*) const {} };
 #ifdef ELUNA_TRINITY
     Trinity::unique_trackable_ptr<ElunaSpellInfo> m_scriptRef;
+#elif defined ELUNA_CMANGOS
+    MaNGOS::unique_trackable_ptr<ElunaSpellInfo> m_scriptRef;
 #endif
 public:
     ElunaSpellInfo(uint32 spellId);
@@ -137,6 +161,8 @@ class ElunaSpellInfo
     SpellInfo const* GetSpellInfo() const { return _spellInfo; }
 #ifdef ELUNA_TRINITY
     Trinity::unique_weak_ptr<ElunaSpellInfo> GetWeakPtr() const { return m_scriptRef; }
+#elif defined ELUNA_CMANGOS
+    MaNGOS::unique_weak_ptr<ElunaSpellInfo> GetWeakPtr() const { return m_scriptRef; }
 #endif
 };
 

ElunaTemplate.cpp

@@ -20,13 +20,13 @@ ElunaConstrainedObjectRef<Aura> GetWeakPtrFor(Aura const* obj)
 #endif
     return { obj->GetWeakPtr(), map };
 }
-
+#if defined ELUNA_TRINITY
 ElunaConstrainedObjectRef<AuraEffect> GetWeakPtrFor(AuraEffect const* obj)
 {
     Map* map = obj->GetBase()->GetOwner()->GetMap();
     return { obj->GetWeakPtr(), map };
 }
-
+#endif
 ElunaConstrainedObjectRef<ElunaProcInfo> GetWeakPtrFor(ElunaProcInfo const* obj)
 {
     return { obj->GetWeakPtr(), obj->GetMap()};

ElunaTemplate.h

@@ -63,7 +63,9 @@ struct ElunaConstrainedObjectRef
 };
 
 ElunaConstrainedObjectRef<Aura> GetWeakPtrFor(Aura const* obj);
+#if defined ELUNA_TRINITY
 ElunaConstrainedObjectRef<AuraEffect> GetWeakPtrFor(AuraEffect const* obj);
+#endif
 ElunaConstrainedObjectRef<ElunaProcInfo> GetWeakPtrFor(ElunaProcInfo const* obj);
 ElunaConstrainedObjectRef<BattleGround> GetWeakPtrFor(BattleGround const* obj);
 ElunaConstrainedObjectRef<Group> GetWeakPtrFor(Group const* obj);

LuaEngine.h

@@ -639,6 +639,7 @@ class ELUNA_GAME_API Eluna
     void OnBGDestroy(BattleGround* bg, BattleGroundTypeId bgId, uint32 instanceId);
 
     /* Spell */
+    #ifdef ELUNA_TRINITY
     void OnSpellCast(Spell* pSpell, bool skipCheck);
     bool OnAuraApplication(Aura* aura, AuraEffect const* auraEff, Unit* target, uint8 mode, bool apply);
     void OnAuraDispel(Aura* aura, DispelInfo* dispelInfo);
@@ -662,6 +663,31 @@ class ELUNA_GAME_API Eluna
     void OnSpellHit(Spell* pSpell);
     void OnAfterSpellHit(Spell* pSpell);
     void OnEffectCalcAbsorb(Spell* pSpell, DamageInfo const& damageInfo, uint32& resistAmount, int32& absorbAmount);
+    #else
+    void OnSpellCast(Spell* pSpell, bool skipCheck);
+    bool OnAuraApplication(Aura* aura, Unit* target, uint8 mode, bool apply);
+    void OnAuraDispel(Aura* aura, Unit* dispeller, uint32 dispellingSpellId, uint32 originalStacks);
+    bool OnPeriodicTick(Aura* aura, Unit* target);
+    void OnPeriodicUpdate(Aura* aura);
+    void OnAuraCalcAmount(Aura* aura, int32& amount, bool& canBeRecalculated);
+    void OnCalcPeriodic(Aura* aura, bool& isPeriodic, int32& amplitude);
+    bool OnAuraCanProc(Aura* aura, ProcExecutionData& procInfo);
+    bool OnAuraProc(Aura* aura, ProcExecutionData& procInfo);
+    uint32 OnCheckCast(Spell* pSpell);
+    void OnBeforeCast(Spell* pSpell);
+    void OnAfterCast(Spell* pSpell);
+    void OnObjectAreaTargetSelect(Spell* pSpell, uint8 effIndex, std::list<WorldObject*>& targets);
+    void OnObjectTargetSelect(Spell* pSpell, uint8 effIndex, WorldObject*& target);
+    void OnDestinationTargetSelect(Spell* pSpell, uint8 effIndex, SpellCastTargets& target);
+    bool OnEffectLaunch(Spell* pSpell, uint8 effIndex, uint8 mode, bool preventDefault);
+    bool OnEffectLaunchTarget(Spell* pSpell, uint8 effIndex, uint8 mode, bool preventDefault);
+    bool OnEffectHit(Spell* pSpell, uint8 effIndex, uint8 mode, bool preventDefault);
+    bool OnEffectHitTarget(Spell* pSpell, uint8 effIndex, uint8 mode, bool preventDefault);
+    void OnBeforeSpellHit(Spell* pSpell, uint8 missInfo);
+    void OnSpellHit(Spell* pSpell);
+    void OnAfterSpellHit(Spell* pSpell);
+    void OnEffectCalcAbsorb(Spell* pSpell, SpellNonMeleeDamage const& damageInfo, uint32& resistAmount, int32& absorbAmount);
+    #endif
 };
 template<> Unit* Eluna::CHECKOBJ<Unit>(int narg, bool error);
 template<> Object* Eluna::CHECKOBJ<Object>(int narg, bool error);