Generic Getters

Author: IntoTheDev

Pool.cs

@@ -7,7 +7,7 @@ namespace ToolBox.Pools
 	[System.Serializable]
 	public class Pool
 	{
-		[SerializeField, AssetsOnly, TabGroup("Data")] private Poolable prefab = null;
+		[SerializeField, AssetsOnly, TabGroup("Data"), AssetSelector] private Poolable prefab = null;
 		[SerializeField, TabGroup("Data")] private int startCount = 0;
 		[SerializeField, TabGroup("Data")] private bool isResizable = false;
 		[SerializeField, SceneObjectsOnly, TabGroup("Data")] private Transform holder = null;
@@ -98,6 +98,18 @@ public Poolable GetEntity(Vector3 position, Quaternion rotation, Transform paren
 			return entity;
 		}
 
+		public T GetEntity<T>() where T : Component =>
+			GetEntity().Component as T;
+
+		public T GetEntity<T>(Transform parent, bool spawnInWorldSpace) where T : Component =>
+			GetEntity(parent, spawnInWorldSpace).Component as T;
+
+		public T GetEntity<T>(Vector3 position, Quaternion rotation) where T : Component =>
+			GetEntity(position, rotation).Component as T;
+
+		public T GetEntity<T>(Vector3 position, Quaternion rotation, Transform parent, bool spawnInWorldSpace) where T : Component =>
+			GetEntity(position, rotation, parent, spawnInWorldSpace).Component as T;
+
 		public void ReturnEntity(Poolable entity)
 		{
 			if (entity.Pool != this)

Poolable.cs

@@ -1,38 +1,37 @@
+using Sirenix.OdinInspector;
+using Sirenix.Serialization;
+using ToolBox.Observer;
 using UnityEngine;
-using UnityEngine.Events;
 
 namespace ToolBox.Pools
 {
 	[DisallowMultipleComponent]
-	public class Poolable : MonoBehaviour
+	public class Poolable : SerializedMonoBehaviour, IReactor
 	{
-		public Pool Pool { get; private set; } = null;
+		[SerializeField] private Component component = null;
+		[OdinSerialize] private IReactor[] onBackToPool = null;
+		[OdinSerialize] private IReactor[] onBackFromPool = null;
 
-		[SerializeField] private UnityEvent OnBackToPool = null;
-		[SerializeField] private UnityEvent OnBackFromPool = null;
+		public Pool Pool { get; private set; } = null;
+		public Component Component => component;
 
 		private bool isPooled = false;
 		private bool isEnabled = true;
-		private bool wasSpawnedBefore = false;
 
 		public void ReturnToPool()
 		{
 			if (!isEnabled)
 				return;
 
-			OnBackToPool?.Invoke();
+			onBackToPool.Dispatch();
 
 			Pool.ReturnEntity(this);
 			isEnabled = false;
 		}
 
 		public void ReturnFromPool()
 		{
-			if (wasSpawnedBefore)
-				OnBackFromPool?.Invoke();
-			else
-				wasSpawnedBefore = true;
-
+			onBackFromPool.Dispatch();
 			isEnabled = true;
 		}
 
@@ -44,5 +43,8 @@ public void SetPool(Pool pool)
 				isPooled = true;
 			}
 		}
+
+		public void HandleReaction() =>
+			ReturnToPool();
 	}
 }