Core/Spells: Implement using different difficulty data from all spell related db2s, not just SpellEffect and SpellPower

1 files changed, 7 insertions, 0 deletions

diff --git a/src/server/game/Spells/SpellScript.h b/src/server/game/Spells/SpellScript.h
index 5b57a7df5b6..40920ba6f20 100644
--- a/src/server/game/Spells/SpellScript.h
+++ b/src/server/game/Spells/SpellScript.h
@@ -47,6 +47,7 @@ class WorldObject;
 struct SpellDestination;
 struct SpellModifier;
 struct SpellValue;
+enum Difficulty : uint8;
 enum class ItemContext : uint8;
 
 #define SPELL_EFFECT_ANY (uint16)-1
@@ -483,6 +484,9 @@ class TC_GAME_API SpellScript : public _SpellScript
         void FinishCast(SpellCastResult result, uint32* param1 = nullptr, uint32* param2 = nullptr);
 
         void SetCustomCastResultMessage(SpellCustomErrors result);
+
+        // returns desired cast difficulty for triggered spells
+        Difficulty GetCastDifficulty() const;
 };
 
 // AuraScript interface - enum used for runtime checks of script function calls
@@ -936,6 +940,9 @@ class TC_GAME_API AuraScript : public _SpellScript
         Unit* GetTarget() const;
         // returns AuraApplication object of currently processed target
         AuraApplication const* GetTargetApplication() const;
+
+        // returns desired cast difficulty for triggered spells
+        Difficulty GetCastDifficulty() const;
 };
 
 //