Update

Author: IntoTheDev

Pool.cs

@@ -1,24 +1,26 @@
 using Sirenix.OdinInspector;
 using System.Collections.Generic;
+using ToolBox.Modules;
 using UnityEngine;
 
 namespace ToolBox.Pools
 {
-	[System.Serializable]
 	public class Pool
 	{
 		[SerializeField, AssetsOnly, ValueDropdown(nameof(GetPoolables))] private Poolable prefab = null;
 		[SerializeField] private int startCount = 0;
 		[SerializeField, SceneObjectsOnly] private Transform holder = null;
-		[SerializeField, ReadOnly] private int currentCount = 0;
+		[SerializeField] private ModulesContainer<GameObject> objectInitializator = null; 
 
+		private int currentCount = 0;
 		private Queue<Poolable> entities = null;
 
-		public Pool(Poolable prefab, int startCount, Transform holder)
+		public Pool(Poolable prefab, int startCount, Transform holder, ModulesContainer<GameObject> objectInitializator)
 		{
 			this.prefab = prefab;
 			this.startCount = startCount;
 			this.holder = holder;
+			this.objectInitializator = objectInitializator;
 		}
 
 		private IEnumerable<Poolable> GetPoolables() =>
@@ -29,19 +31,22 @@ public void Fill()
 			entities = new Queue<Poolable>(startCount);
 			currentCount = startCount;
 
-			Poolable original = CreateObject(prefab);
+			Poolable original = Object.Instantiate(prefab, holder);
+			objectInitializator.Process(original.gameObject);
 
-			for (int i = 0; i < startCount - 1; i++)
-				CreateObject(original);
+			AddToPool(original);
 
-			Poolable CreateObject(Poolable prototype)
+			for (int i = 0; i < startCount - 1; i++)
 			{
-				Poolable entity = Object.Instantiate(prototype, holder);
-				entity.SetPool(this);
-				entities.Enqueue(entity);
-				entity.gameObject.SetActive(false);
+				Poolable entity = Object.Instantiate(original, holder);
+				AddToPool(entity);
+			}
 
-				return entity;
+			void AddToPool(Poolable newEntity)
+			{
+				newEntity.SetPool(this);
+				entities.Enqueue(newEntity);
+				newEntity.gameObject.SetActive(false);
 			}
 		}
 
@@ -117,6 +122,7 @@ private Poolable TakeEntity()
 			{
 				entity = Object.Instantiate(prefab, holder);
 				entity.SetPool(this);
+
 				return entity;
 			}
 

Poolable.cs

@@ -9,8 +9,8 @@ namespace ToolBox.Pools
 	public class Poolable : SerializedMonoBehaviour, IModule
 	{
 		[SerializeField] private Component component = null;
-		[OdinSerialize] private ModulesContainer onBackToPool = default;
-		[OdinSerialize] private ModulesContainer onBackFromPool = default;
+		[SerializeField] private ModulesContainer onBackToPool = default;
+		[SerializeField] private ModulesContainer onBackFromPool = default;
 
 		public Pool Pool { get; private set; } = null;
 		public Component Component => component;