siege-song/Data/Assets/Malbers Animations/Common/Scripts/Utilities/Triggers/TriggerProxy.cs

using UnityEngine;
using MalbersAnimations.Events;
using MalbersAnimations.Scriptables;
using System.Collections.Generic;
using System;

#if UNITY_EDITOR
using UnityEditor;
#endif

namespace MalbersAnimations.Utilities
{
    /// <summary>
    /// This is used when the collider is in a different gameObject and you need to check the Collider Events
    /// Create this component at runtime and subscribe to the UnityEvents
    /// </summary>
    [AddComponentMenu("Malbers/Utilities/Colliders/Trigger Proxy")]
    public class TriggerProxy : MonoBehaviour, IMLayer
    {
        //[Tooltip("Proxy ID, can be used to Identify which is the Proxy Trigger used")]
        //[SerializeField] private IntReference m_ID = new IntReference(0);
        [Tooltip("Hit Layer for the Trigger Proxy")]
        [SerializeField] private LayerReference hitLayer = new LayerReference(-1);
        [SerializeField] private QueryTriggerInteraction triggerInteraction = QueryTriggerInteraction.Ignore;
        [Tooltip("Search only Tags")]
        public Tag[] Tags;

        public ColliderEvent OnTrigger_Enter = new ColliderEvent();
        public ColliderEvent OnTrigger_Exit = new ColliderEvent();
        public ColliderEvent OnTrigger_Stay = new ColliderEvent();

        public GameObjectEvent OnGameObjectEnter = new GameObjectEvent();
        public GameObjectEvent OnGameObjectExit = new GameObjectEvent();
        public GameObjectEvent OnGameObjectStay = new GameObjectEvent();

        [SerializeField] private bool m_debug = false;

        public BoolReference useOnTriggerStay = new BoolReference();


        ///// <summary>All the Gameobjects using the Proxy</summary>
        //internal List<Type> AllowedTypes = new List<Type>();
        //public Type ForcedType;



        internal List<Collider> m_colliders = new List<Collider>();
        /// <summary>All the Gameobjects using the Proxy</summary>
        internal List<GameObject> EnteringGameObjects = new List<GameObject>();
        internal List<TriggerTarget> TriggerTargets= new List<TriggerTarget>();

        public Action<GameObject, Collider> EnterTriggerInteraction = delegate { };
        public Action<GameObject, Collider> ExitTriggerInteraction = delegate { };

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

        //public int ID { get => m_ID.Value; set => m_ID.Value = value; }
        public LayerMask Layer { get => hitLayer.Value; set => hitLayer.Value = value; }
        public QueryTriggerInteraction TriggerInteraction { get => triggerInteraction; set => triggerInteraction = value; }

        /// <summary> Collider Component used for the Trigger Proxy </summary>
        [RequiredField] public Collider trigger;
        public Transform Owner { get; set; }

        public bool TrueConditions(Collider other)
        {
            if (!Active) return false;

            if (Tags != null && Tags.Length > 0)
            {
                if (!other.gameObject.HasMalbersTagInParent(Tags)) return false;
            }

            if (trigger == null) return false; // you are 
            if (triggerInteraction == QueryTriggerInteraction.Ignore && other.isTrigger) return false; // Check Trigger Interactions 
            if (!MTools.Layer_in_LayerMask(other.gameObject.layer, Layer)) return false;

            if (transform.IsChildOf(other.transform)) return false;         // Do not Interact with yourself
            if (Owner != null && other.transform.IsChildOf(Owner)) return false;             // Do not Interact with yourself

            return true;
        }


        public GameObject FindRealParent(Transform other)
        {
            if (other.parent == null)
                return other.gameObject;
            else
            {
                if (MTools.Layer_in_LayerMask(other.gameObject.layer, Layer))               //If this object is in the same Layer
                {
                    if (MTools.Layer_in_LayerMask(other.parent.gameObject.layer, Layer))    //Check the Parent
                    {
                        return FindRealParent(other.parent);                                //If the Parent is also on the Layer; Keep searching Upwards
                    }
                    else
                    {
                        return other.gameObject;                                            //If the Parent is not on the same Layer ... return the Child
                    }
                }
                else
                {
                    return other.gameObject;                                                 //If the Child is not on the same Layer ... return the Child
                }
            }
        }

        public void OnTriggerEnter(Collider other)
        {
            if (TrueConditions(other))
            {
                var realRoot = other.transform.root.gameObject;        //Get the animal on the entering collider

                if (!MTools.Layer_in_LayerMask(realRoot.layer, Layer))  //If the Root is not on the same Layer
                    realRoot = FindRealParent(other.transform);         //Means the Root is not on the real root since its not on the search layer 

                OnTrigger_Enter.Invoke(other); //Invoke when a Collider enters the Trigger
                if (m_debug) Debug.Log($"<b>{name}</b> [Entering Collider] -> [{other.name}]", this);

                ////Check Recently destroyed Colliders (Strange bug)
                //CheckMissingColliders();

                if (m_colliders.Find(coll => coll == other) == null)                               //if the entering collider is not already on the list add it
                {
                    m_colliders.Add(other);
                    AddTarget(other);
                }

                if (EnteringGameObjects.Contains(realRoot))
                {
                    return;
                }
                else
                {
                    EnteringGameObjects.Add(realRoot);
                    EnterTriggerInteraction(realRoot, other);
                    OnGameObjectEnter.Invoke(realRoot);

                    if (m_debug) Debug.Log($"<b>{name}</b> [Entering GameObject] -> [{realRoot.name}]", this);
                }
            }
        }

        /// <summary>Check Recently destroyed Colliders (Strange bug)</summary>
        private void CheckMissingColliders()
        {
            for (var i = m_colliders.Count - 1; i > -1; i--)
            {
                if (m_colliders[i] == null) m_colliders.RemoveAt(i);
            }

            if (m_colliders.Count == 0)
            {
                EnteringGameObjects = new List<GameObject>();
                TriggerTargets = new List<TriggerTarget>();
            }
        } 
        /// <summary>Add a Trigger Target to every new Collider found</summary>
        private void AddTarget(Collider other)
        {
            if (!TriggerTargets.Exists(x => x.m_collider == other))
            {
                var hasTarget = other.GetComponent<TriggerTarget>();
                if (hasTarget == null) hasTarget = other.gameObject.AddComponent<TriggerTarget>();

                TriggerTargets.Add(hasTarget);
                hasTarget.AddProxy(this,other);
            }
        }

        public void OnTriggerExit(Collider other) => TriggerExit(other, true);

        /// <summary>OnTrigger exit Logic</summary>
        public void TriggerExit(Collider other, bool remove)
        {
            if (TrueConditions(other))
            {
                OnTrigger_Exit.Invoke(other);

                m_colliders.Remove(other);
                RemoveTarget(other, remove);

                if (m_debug) Debug.Log($"<b>{name}</b> [Exit Collider] -> [{other.name}]", this);

                var realRoot = other.transform.root.gameObject;         //Get the gameObject on the entering collider

                if (!MTools.Layer_in_LayerMask(realRoot.layer, Layer))  //If the Root is not on the same Layer
                    realRoot = FindRealParent(other.transform);         //Means the Root is not on the real root since its not on the search layer

                if (EnteringGameObjects.Contains(realRoot))             //Means that the Entering GameObject still exist
                {
                    if (!m_colliders.Exists(c => c != null && c.transform.root.gameObject == realRoot)) //Means that all that root colliders are out
                    {
                        EnteringGameObjects.Remove(realRoot);

                        OnGameObjectExit.Invoke(realRoot);

                        ExitTriggerInteraction(realRoot, other);

                        if (m_debug) Debug.Log($"<b>{name}</b> [Leaving Gameobject] -> [{realRoot.name}]", this);
                    }
                }
                //CheckMissingColliders();
            }
        }

        internal void RemoveTarget(Collider other, bool remove)
        {
            var HasTarget = TriggerTargets.Find(x => x.m_collider == other);

            if (HasTarget)
            {
                if (remove) HasTarget.RemoveProxy(this);
                TriggerTargets.Remove(HasTarget);
            }
        }
  

        public void ResetTrigger()
        {
            m_colliders = new List<Collider>();
            EnteringGameObjects = new List<GameObject>();
            TriggerTargets = new List<TriggerTarget>(); 
        }

        private void OnDisable()
        {
            if (m_colliders.Count > 0)
            {
                foreach (var c in m_colliders)
                {
                    if (c)
                    {
                        OnTrigger_Exit.Invoke(c); //the colliders may be destroyed
                        RemoveTarget(c, true);
                    }
                }
            }

            if (EnteringGameObjects.Count > 0)
            {
                foreach (var c in EnteringGameObjects)
                {

                    if (c) OnGameObjectExit.Invoke(c);  //the gameobjects  may be destroyed
                }
            }

            if (m_debug) Debug.Log($"<b>{name}</b> [Exit All Colliders and Triggers] ");

            ResetTrigger();
        }

        private void OnEnable() => ResetTrigger();

        private void Awake()
        {
            if (trigger == null) trigger = GetComponent<Collider>();

            if (trigger) trigger.isTrigger = true;
            else
                Debug.LogWarning("This Script requires a Collider, please add any type of collider", this);

            if (Owner == null) Owner = transform;

            ResetTrigger();
        }


        private void Update()
        {
            CheckOntriggerStay();
        }
        void CheckOntriggerStay()
        {
            if (useOnTriggerStay.Value)
            {
                foreach (var gos in EnteringGameObjects)
                {
                    OnGameObjectStay.Invoke(gos);
                }

                foreach (var col in m_colliders)
                {
                    OnTrigger_Stay.Invoke(col);
                }
            }
        }

        public void SetLayer(LayerMask mask, QueryTriggerInteraction triggerInteraction, Transform Owner, Tag[] tags = null)
        {
            TriggerInteraction = triggerInteraction;
            Tags = tags;
            Layer = mask;
            this.Owner = Owner;
        }


        public static TriggerProxy CheckTriggerProxy(Collider trigger, LayerMask Layer, QueryTriggerInteraction TriggerInteraction, Transform Owner)
        {
            TriggerProxy Proxy = null;
            if (trigger != null)
            {
                Proxy = trigger.GetComponent<TriggerProxy>();

                if (Proxy == null)
                {
                    Proxy = trigger.gameObject.AddComponent<TriggerProxy>();
                    Proxy.SetLayer(Layer, TriggerInteraction, Owner);
                    Proxy.hideFlags = HideFlags.HideInInspector;
                }
                else
                {
                    Proxy.Layer = Proxy.Layer | Layer; //combine both layers
                }
                if (TriggerInteraction != QueryTriggerInteraction.Ignore) Proxy.TriggerInteraction = TriggerInteraction;

                trigger.gameObject.SetLayer(2, false); //Force the Trigger Area to be on the Ignore Raycast Layer
                trigger.isTrigger = true;

                Proxy.Active = true;
            }

            return Proxy;
        }

       

        [HideInInspector] public int Editor_Tabs1;
    }


    #region Inspector


#if UNITY_EDITOR
    [CanEditMultipleObjects, CustomEditor(typeof(TriggerProxy))]
    public class TriggerProxyEditor : Editor
    {
        SerializedProperty debug, OnTrigger_Enter, OnTrigger_Exit, useOnTriggerStay, OnTrigger_Stay, Editor_Tabs1,
            triggerInteraction, hitLayer, OnGameObjectEnter, OnGameObjectExit, OnGameObjectStay, Tags;

        TriggerProxy m;

        protected string[] Tabs1 = new string[] { "General", "Events" };

        private void OnEnable()
        {
            m = (TriggerProxy)target;
            triggerInteraction = serializedObject.FindProperty("triggerInteraction");
            useOnTriggerStay = serializedObject.FindProperty("useOnTriggerStay");
            hitLayer = serializedObject.FindProperty("hitLayer");
            debug = serializedObject.FindProperty("m_debug");
            OnTrigger_Enter = serializedObject.FindProperty("OnTrigger_Enter");
            OnTrigger_Exit = serializedObject.FindProperty("OnTrigger_Exit");
            OnGameObjectEnter = serializedObject.FindProperty("OnGameObjectEnter");
            OnGameObjectExit = serializedObject.FindProperty("OnGameObjectExit");
            Tags = serializedObject.FindProperty("Tags");
            OnGameObjectStay = serializedObject.FindProperty("OnGameObjectStay");
            OnTrigger_Stay = serializedObject.FindProperty("OnTrigger_Stay");
            Editor_Tabs1 = serializedObject.FindProperty("Editor_Tabs1");
        }


        public override void OnInspectorGUI()
        {
            serializedObject.Update();
            MalbersEditor.DrawDescription("Use this component to do quick OnTrigger Enter/Exit logics");

            Editor_Tabs1.intValue = GUILayout.Toolbar(Editor_Tabs1.intValue, Tabs1);
            if (Editor_Tabs1.intValue == 0) DrawGeneral();
            else DrawEvents();
            if (Application.isPlaying && debug.boolValue)
            {
                EditorGUI.BeginDisabledGroup(true);
                EditorGUILayout.BeginVertical(EditorStyles.helpBox);    

                EditorGUILayout.LabelField("Debug", EditorStyles.boldLabel);

                //   EditorGUILayout.ObjectField("Own Collider", m.trigger, typeof(Collider), false);

                EditorGUILayout.LabelField("GameObjects (" + m.EnteringGameObjects.Count + ")", EditorStyles.boldLabel);
                foreach (var item in m.EnteringGameObjects)
                {
                    if (item != null) EditorGUILayout.ObjectField(item.name, item, typeof(GameObject), false);
                }

                EditorGUILayout.LabelField("Colliders (" + m.m_colliders.Count + ")", EditorStyles.boldLabel);

                foreach (var item in m.m_colliders)
                {
                    if (item != null)   EditorGUILayout.ObjectField(item.name, item, typeof(Collider), false);
                }

                EditorGUILayout.LabelField("Targets (" + m.TriggerTargets.Count + ")", EditorStyles.boldLabel);

                foreach (var item in m.TriggerTargets)
                {
                    if (item != null) EditorGUILayout.ObjectField(item.name, item, typeof(Collider), false);
                }

                EditorGUILayout.EndVertical();
                Repaint();
                EditorGUI.EndDisabledGroup();
            }

            serializedObject.ApplyModifiedProperties();
        }

        private void DrawGeneral()
        {
            EditorGUILayout.BeginVertical(EditorStyles.helpBox);
            EditorGUILayout.BeginHorizontal();
            EditorGUILayout.PropertyField(hitLayer, new GUIContent("Layer"));
            MalbersEditor.DrawDebugIcon(debug);
            EditorGUILayout.EndHorizontal();
            EditorGUILayout.PropertyField(triggerInteraction);
            EditorGUILayout.PropertyField(useOnTriggerStay);
            EditorGUI.indentLevel++;
            EditorGUILayout.PropertyField(Tags, true);
            EditorGUI.indentLevel--;
            EditorGUILayout.EndVertical();
        }

        private void DrawEvents()
        {
            EditorGUILayout.BeginVertical(EditorStyles.helpBox);

            EditorGUILayout.PropertyField(OnTrigger_Enter, new GUIContent("On Trigger Enter"));
            EditorGUILayout.PropertyField(OnTrigger_Exit, new GUIContent("On Trigger Exit"));
            if (m.useOnTriggerStay.Value)
                EditorGUILayout.PropertyField(OnTrigger_Stay, new GUIContent("On Trigger Stay"));


            EditorGUILayout.PropertyField(OnGameObjectEnter, new GUIContent("On GameObject Enter "));
            EditorGUILayout.PropertyField(OnGameObjectExit, new GUIContent("On GameObject Exit"));
            if (m.useOnTriggerStay.Value)
                EditorGUILayout.PropertyField(OnGameObjectStay, new GUIContent("On GameObject Stay"));
            EditorGUILayout.EndVertical();
        }
    }
#endif
    #endregion
}