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

namespace XRLib
{
    [DefaultExecutionOrder(int.MinValue)]
    public partial class Initializer : MonoBehaviour,ISingle
    {
        Dictionary<MonoBehaviour, bool> activeTable;
        public List<MonoBehaviour> targets = new();
        public List<UIBase> baseUIs = new();
        HashSet<UIBase> allUIs = new();
        HashSet<CanvasBase> canvasBases = new();
        Dictionary<CanvasBase, List<PanelBase>> canvasToPanels = 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()
        {
            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;
                }
                canvasToPanels.Add(cb, new());
                return;
            }

            if (ui is PanelBase pb)
            {
                var parentCanvas = pb.GetComponentInParent<CanvasBase>();
                if (canvasBases.Add(parentCanvas))
                {
                    canvasToPanels.Add(parentCanvas, new List<PanelBase>());
                }

                canvasToPanels[parentCanvas].Add(pb);

                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);
            }
        }
    }
}