BITKit/Src/Unity/Scripts/UX/Core/UXElement.cs

using System.Collections;
using System.Collections.Generic;
using UnityEngine;
using UnityEngine.UIElements;
using UnityEngine.Events;
using Cysharp.Threading.Tasks;
using System.Threading;
namespace BITKit.UX
{
    public abstract class UXElement : Provider, IProvider<bool>, IActivable, IEnabled
    {
        protected bool isActive = true;
        public virtual bool Get() => isActive;
        public virtual void Set(bool active) { }
        public virtual void OnAwake(){}
        public virtual void OnStart() { }
        public override void Set<T>(T obj)
        {
            (this as IProvider<T>).Set(obj);
        }
        public virtual void SetActive(bool active)
        {
            GetVisualElement().SetActive(active);
        }

        public bool Enabled
        {
            get => enabled;
            set => enabled = value;
        }

        public virtual void SetPosition(Vector3 worldPosition)
        {
            GetVisualElement().SetPosition(worldPosition);
        }
        protected virtual void SetPosition(ref Vector2 position)
        {
            
        }
        public virtual VisualElement GetVisualElement() => null;

        public virtual bool IsEnabled()
        {
            return GetVisualElement().enabledSelf;
        }
        public virtual void SetEnabled(bool enabled)
        {
            GetVisualElement().SetEnabled(enabled);
        }
    }
    public abstract class UXElement<UX> : UXElement, IProvider<UX> where UX : VisualElement
    {
        public static implicit operator UX(UXElement<UX> self)
        {
            return self.visualElement;
        }
        [Header(Constant.Header.Components)]
        public UIDocument document;
        [Header(Constant.Header.Settings)]
        public string bindName;
        [SerializeField, SerializeReference, SubclassSelector]
        protected IReference bindNameProvider = new GetNameFromGameobject();
        [Header(Constant.Header.InternalVariables)]
        private UX m_visualElement;
        protected IStyle style => visualElement.style;
        public UX visualElement
        {
            get
            {
                if (m_visualElement is null)
                {
                    var bindName = this.bindName;
                    if (string.IsNullOrEmpty(bindName))
                    {
                        bindName = bindNameProvider.Get();
                    }
                    var split = bindName.Split(@".");
                    var visualElement = document.rootVisualElement;
                    if (string.IsNullOrEmpty(bindName))
                    {
                        
                    }
                    else
                    {
                        foreach (var name in split)
                        {
                            try
                            {
                                visualElement = visualElement.Q(name);
                            }
                            catch (System.Exception)
                            {
                                Debug.LogWarning(name);
                                throw;
                            }
                        }
                    }
                    m_visualElement = visualElement as UX;

                    if (m_visualElement is null)
                    {
                        Debug.LogWarning(bindName);
                    }
                }
                return m_visualElement;
            }
        }

        public override bool Get() => isActive;
        public override VisualElement GetVisualElement() => visualElement;
        protected CancellationToken cancellationToken;
        public override async void Set(bool active)
        {
            await UniTask.SwitchToMainThread(cancellationToken);
            visualElement.SetEnabled(active);
            if (active != isActive)
            {
                isActive = active;
                visualElement.SetEnabled(active);
            }
        }
        private void Awake()
        {
            cancellationToken = gameObject.GetCancellationTokenOnDestroy();
            OnAwake();
            OnStart();
        }
        private void OnValidate()
        {
            if (bindNameProvider is GetNameFromGameobject x)
            {
                x.gameobject = gameObject;
            }
            if (document is null)
            {
                document = GetComponentInParent<UIDocument>();
            }
        }
        UX IProvider<UX>.Get() => visualElement;
        public override T Get<T>()
        {
            if (visualElement is T t)
            {
                return t;
            }
            return base.Get<T>();
        }
        public void Set(UX t)
        {
            var root = visualElement.parent;
            root.Remove(visualElement);
            root.Add(t);
            m_visualElement = t;
        }
        protected virtual void OnVistualElementDisabled()
        {

        }
        public void AddToClassList(string className)
        {
            visualElement.AddToClassList(className);
        }
        public void RemoveFromClassList(string className)
        {
            visualElement.RemoveFromClassList(className);
        }
    }
}