ChunilENG/Assets/Plugins/XRLib/Core/Initializer.cs

using System.Collections.Generic;
using UnityEngine;
using UnityEngine.SceneManagement;

namespace WI
{
    [DefaultExecutionOrder(int.MaxValue)]
    public partial class Initializer : MonoBehaviour,ISingle
    {
        public List<MonoBehaviour> targets = new();
        public List<UIBase> baseUIs = new();
        HashSet<UIBase> allUIs = new();
        HashSet<CanvasBase> canvasBases = new();
        void GetAllInitializer()
        {
            var sceneRootObjs = SceneManager.GetActiveScene().GetRootGameObjects();

            targets.Clear();
            foreach (var obj in sceneRootObjs)
            {
                var inits = obj.transform.FindAll<MonoBehaviour>();
                targets.AddRange(inits);
            }
        }

        void Awake()
        {
            //Debug.Log("Initializing");
            GetAllInitializer();
            ForceActive();
        }

        void ForceActive()
        {
            var activeTable = new Dictionary<MonoBehaviour, bool>();
            foreach (var target in targets)
            {
                activeTable.Add(target, target.gameObject.activeSelf);
                target.gameObject.SetActive(true);
            }

            foreach (var a in activeTable)
            {
                //if (a.Key.gameObject == null)
                //{
                //    Debug.LogError($"Wrong Object!?");
                //    continue;
                //}

                if (a.Key is UIBase ui)
                    SetUI(ui);

                if (a.Value == false)
                    a.Key.gameObject.SetActive(false);
            }
        }

        void SetUI(UIBase ui)
        {
            if (!allUIs.Add(ui))
            {
                Debug.LogError($"Duplicate Initialize. {ui}");
                return;
            }

            if (ui is CanvasBase cb)
            {
                if (!canvasBases.Add(cb))
                {
                    Debug.LogError($"Canvas Initialize Failed! {cb}");
                }
                return;
            }

            if (ui is PanelBase pb)
            {
                var parentCanvas = pb.GetComponentInParent<CanvasBase>();
                canvasBases.Add(parentCanvas);

                var panelType = pb.GetType();
                var canvasPanelField = parentCanvas
                    .GetType()
                    .GetField(panelType.Name.ToLower(), System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Instance | System.Reflection.BindingFlags.Public);
                canvasPanelField?.SetValue(parentCanvas, pb);
            }
        }
    }
}