Core/Spells: Handle GAMEOBJECT_TYPE_UI_LINK spellFocus (#25132)

1 files changed, 20 insertions, 0 deletions

diff --git a/src/server/game/Entities/GameObject/GameObjectData.h b/src/server/game/Entities/GameObject/GameObjectData.h
index 638661e0420..7477b9b5137 100644
--- a/src/server/game/Entities/GameObject/GameObjectData.h
+++ b/src/server/game/Entities/GameObject/GameObjectData.h
@@ -926,6 +926,26 @@ struct GameObjectTemplate
         }
     }
 
+    uint32 GetSpellFocusType() const
+    {
+        switch (type)
+        {
+            case GAMEOBJECT_TYPE_SPELL_FOCUS:   return spellFocus.spellFocusType;
+            case GAMEOBJECT_TYPE_UI_LINK:       return UILink.spellFocusType;
+            default: return 0;
+        }
+    }
+
+    uint32 GetSpellFocusRadius() const
+    {
+        switch (type)
+        {
+            case GAMEOBJECT_TYPE_SPELL_FOCUS:   return spellFocus.radius;
+            case GAMEOBJECT_TYPE_UI_LINK:       return UILink.radius;
+            default: return 0;
+        }
+    }
+
     void InitializeQueryData();
     WorldPacket BuildQueryData(LocaleConstant loc) const;
 };