AppleHillsProduction/Assets/Scripts/Bootstrap/CustomBootSettings.cs

using System.Threading.Tasks;
using UnityEngine;

namespace Bootstrap
{
    /// <summary>
    /// Settings for the custom boot process
    /// </summary>
    public class CustomBootSettings : ScriptableObject
    {
        /// <summary>
        /// A list of prefabs which should be loaded during boot
        /// </summary>
        public GameObject[] BootPrefabs;
        
        /// <summary>
        /// Internal references to instances of the prefabs from <see cref="BootPrefabs"/>
        /// </summary>
        private GameObject[] Instances;

        /// <summary>
        /// Runtime container object which acts as the parent for any BootPrefab instances
        /// </summary>
        private GameObject RuntimeContainer;
        
        /// <summary>
        /// Initialise the boot settings object asynchronously, loading each prefab in <see cref="BootPrefabs"/>
        /// </summary>
        public async Task Initialise()
        {
            RuntimeContainer = new GameObject($"{name}_Container");
            DontDestroyOnLoad(RuntimeContainer);
            Instances = new GameObject[BootPrefabs.Length];
            for (var i = 0; i < BootPrefabs.Length; i++)
            {
                if (!BootPrefabs[i]) continue;
                
                var instance = GameObject.InstantiateAsync(BootPrefabs[i], RuntimeContainer.transform);
                while (!instance.isDone)
                    await Task.Yield();

                Instances[i] = instance.Result[0];
            }
        }
        
        /// <summary>
        /// Initialise the boot settings object synchronously, loading each prefab in <see cref="BootPrefabs"/>
        /// </summary>
        public void InitialiseSync()
        {
            RuntimeContainer = new GameObject($"{name}_Container");
            if (Application.isPlaying)
            {
                DontDestroyOnLoad(RuntimeContainer);
            }

            Instances = new GameObject[BootPrefabs.Length];
            for (var i = 0; i < BootPrefabs.Length; i++)
            {
                if (!BootPrefabs[i]) continue;
                
                var instance = GameObject.Instantiate(BootPrefabs[i], RuntimeContainer.transform);
                Instances[i] = instance;
            }
        }

        /// <summary>
        /// Destroy all loaded instances referenced by <see cref="Instances"/>
        /// </summary>
        public void Cleanup()
        {
            foreach (var t in Instances)
            {
                if (t)
                {
                    if (Application.isPlaying)
                    {
                        GameObject.Destroy(t);
                    }
                    else
                    {
                        GameObject.DestroyImmediate(t);
                    }
                    
                }
            }

            Instances = null;

            if (Application.isPlaying)
            {
                GameObject.Destroy(RuntimeContainer);
            }
            else
            {
                GameObject.DestroyImmediate(RuntimeContainer);
            }
        }
    }
}