From 9e4c74ed1d73eac40ed29fbdb1e5103c121844a9 Mon Sep 17 00:00:00 2001
From: Syntriax <Syntriax@gmail.com>
Date: Sun, 13 Apr 2025 13:18:02 +0300
Subject: [PATCH] feat: IGameManager.FindRequiredBehaviour extension method
 added

---
 Engine.Core/Extensions/GameManagerExtensions.cs | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/Engine.Core/Extensions/GameManagerExtensions.cs b/Engine.Core/Extensions/GameManagerExtensions.cs
index 577e80d..db991e6 100644
--- a/Engine.Core/Extensions/GameManagerExtensions.cs
+++ b/Engine.Core/Extensions/GameManagerExtensions.cs
@@ -9,11 +9,8 @@ public static class GameManagerExtensions
         => gameManager.InstantiateHierarchyObject<HierarchyObject>(args);
 
     public static T GetRequiredHierarchyObject<T>(this IGameManager gameManager) where T : class
-    {
-        foreach (IHierarchyObject hierarchyObject in gameManager)
-            if (hierarchyObject is T @object)
-                return @object;
+        => gameManager.GetHierarchyObject<T>() ?? throw new HierarchyObjectNotFoundException($"{gameManager.GetType().FullName}({gameManager.Id}) does not contain any {nameof(IHierarchyObject)} object of type {typeof(T).FullName}");
 
-        throw new HierarchyObjectNotFoundException($"{gameManager.GetType().FullName}({gameManager.Id}) does not contain any {typeof(T).FullName}");
-    }
+    public static T FindRequiredBehaviour<T>(this IGameManager gameManager) where T : class
+        => gameManager.FindBehaviour<T>() ?? throw new BehaviourNotFoundException($"{gameManager.GetType().FullName}({gameManager.Id}) does not contain any {nameof(IHierarchyObject)} with {nameof(IBehaviour)} of type {typeof(T).FullName}");
 }