Auto create Unity Component

Author: Trojaner

Rocket.UnityEngine/Properties/DependencyRegistrator.cs

@@ -1,14 +1,19 @@
 using Rocket.API.DependencyInjection;
 using Rocket.API.Scheduler;
 using Rocket.UnityEngine.Scheduling;
+using UnityEngine;
 
 namespace Rocket.UnityEngine.Properties
 {
     public class DependencyRegistrator : IDependencyRegistrator
     {
         public void Register(IDependencyContainer container, IDependencyResolver resolver)
         {
-            container.RegisterSingletonType<ITaskScheduler, UnityTaskScheduler>();
+            GameObject o = new GameObject();
+            Object.DontDestroyOnLoad(o);
+            var component = o.AddComponent<UnityTaskScheduler>();
+            component.Load(container);
+            container.RegisterInstance<ITaskScheduler>(component);
         }
     }
 }

Rocket.UnityEngine/Scheduling/UnityTaskScheduler.cs

@@ -12,11 +12,11 @@ namespace Rocket.UnityEngine.Scheduling
 {
     public class UnityTaskScheduler : MonoBehaviour, ITaskScheduler
     {
-        private readonly IDependencyContainer container;
-        private readonly List<ITask> tasks;
+        private IDependencyContainer container;
+        private List<ITask> tasks;
         public ReadOnlyCollection<ITask> Tasks => tasks.AsReadOnly();
 
-        public UnityTaskScheduler(IDependencyContainer container)
+        public void Load(IDependencyContainer container)
         {
             this.container = container;
             tasks = new List<ITask>();