beyond/Assets/Scripts/InvectorDerivatives/bWeaponHolderManager.cs

using Invector;
using Invector.vCharacterController;
using Invector.vItemManager;
using System.Collections;
using System.Collections.Generic;
using UnityEngine;

namespace Beyond
{
    [vClassHeader("Weapon Holder Manager", "Create a new empty object inside a bone and add the vWeaponHolder script")]
    public class bWeaponHolderManager : vMonoBehaviour
    {
        public vWeaponHolder[] holders = new vWeaponHolder[0];
        public bool debugMode;
        internal bool inEquip;
        internal bool inUnequip;
        internal bItemManager itemManager;
        internal vThirdPersonController cc;
        public Dictionary<string, List<vWeaponHolder>> holderAreas = new Dictionary<string, List<vWeaponHolder>>();
        protected float equipTime;

        private void OnDrawGizmosSelected()
        {
            holders = GetComponentsInChildren<vWeaponHolder>(true);
        }

        protected virtual bool IsEquipping
        {
            get
            {
                if (cc) return cc.IsAnimatorTag("IsEquipping");
                return false;
            }
        }

        private void Start()
        {
            itemManager = GetComponent<bItemManager>();

            cc = GetComponent<vThirdPersonController>();
            if (itemManager)
            {
                itemManager.onFinishEquipItem.AddListener(EquipWeapon);
                itemManager.onFinishUnequipItem.AddListener(UnequipWeapon);
                itemManager.onSetLockedToEquip.AddListener(CheckAreaLockedToEquip);
                holders = GetComponentsInChildren<vWeaponHolder>(true);
                if (holders != null)
                {
                    foreach (vWeaponHolder holder in holders)
                    {
                        if (!holderAreas.ContainsKey(holder.equipPointName))
                        {
                            holderAreas.Add(holder.equipPointName, new List<vWeaponHolder>());
                            holderAreas[holder.equipPointName].Add(holder);
                        }
                        else
                        {
                            holderAreas[holder.equipPointName].Add(holder);
                        }

                        holder.SetActiveHolder(false);
                        holder.SetActiveWeapon(false);
                    }
                }
            }
        }

        public void CheckAreaLockedToEquip(bEquipArea equipArea)
        {
            if (equipArea.isLockedToEquip)
            {
                UnequipWeapon(equipArea, equipArea.currentEquippedItem);
            }
        }

        public void EquipWeapon(bEquipArea equipArea, bItem item)
        {
            if (item == null) return;
            var slotsInArea = equipArea.ValidSlots;

            if (slotsInArea != null && slotsInArea.Count > 0 && holderAreas.ContainsKey(equipArea.equipPointName))
            {
                // Check All Holders to Display
                for (int i = 0; i < slotsInArea.Count; i++)
                {
                    if (slotsInArea[i].item != null)
                    {
                        var holder = holderAreas[equipArea.equipPointName].Find(h => slotsInArea[i].item && slotsInArea[i].item.id == h.itemID
                        && ((equipArea.currentEquippedItem != null
                        && equipArea.currentEquippedItem != item
                        && equipArea.currentEquippedItem != slotsInArea[i].item
                        && equipArea.currentEquippedItem.id != item.id) || equipArea.currentEquippedItem == null || equipArea.isLockedToEquip));

                        if (holder)
                        {
                            holder.SetActiveHolder(true);
                            holder.SetActiveWeapon(true);
                            if (debugMode) Debug.Log("Hold: " + slotsInArea[i].item);
                        }
                    }
                }
                // Check Current Item to Equip with time
                if (!equipArea.isLockedToEquip && equipArea.currentEquippedItem != null && equipArea.currentEquippedItem == item)
                {
                    var holder = holderAreas[equipArea.equipPointName].Find(h => h.itemID == equipArea.currentEquippedItem.id);
                    if (holder)
                    {
                        // Unhide Holder and hide Equiped weapon
                        var immediate = (equipArea.currentEquippedItem != item || itemManager.playItemAnimation) &&
                            !string.IsNullOrEmpty(equipArea.currentEquippedItem.EnableAnim);

                        if (debugMode) Debug.Log("UnHold: " + item.name);
                        holder.SetActiveHolder(true);
                        holder.SetActiveWeapon(false);
                        //StartCoroutine(EquipRoutine(equipArea.currentEquipedItem.enableDelayTime, immediate,
                        //   () => { holder.SetActiveHolder(true); }, () => { holder.SetActiveWeapon(false); }, item.name));
                    }
                }
            }
        }

        public void UnequipWeapon(bEquipArea equipArea, bItem item)
        {
            if (holders.Length == 0 || item == null) return;
            if ((itemManager.inventory != null) && holderAreas.ContainsKey(equipArea.equipPointName))
            {
                var holder = holderAreas[equipArea.equipPointName].Find(h => item.id == h.itemID);
                if (holder)
                {
                    // Check if EquipArea contains unequipped item
                    var containsItem = equipArea.ValidSlots.Find(slot => slot.item == item) != null;
                    if (debugMode) Debug.Log(containsItem ? "Hold: " + item.name : "Hide :" + item.name + " Holder");

                    // Hide or unhide holder and weapon if contains item
                    if (containsItem)
                    {
                        //var immediate = itemManager.playItemAnimation && !string.IsNullOrEmpty(item.DisableAnim);
                        holder.SetActiveHolder(containsItem);
                        holder.SetActiveWeapon(containsItem);
                        //StartCoroutine(UnequipRoutine(item.disableDelayTime, immediate,
                        //       () => { holder.SetActiveHolder(containsItem); }, () => { holder.SetActiveWeapon(containsItem); }, item.name));
                    }
                    else
                    {
                        holder.SetActiveHolder(false);
                        holder.SetActiveWeapon(false);
                    }
                }
            }
        }

        internal vWeaponHolder GetHolder(GameObject equipment, int id)
        {
            var equipPoint = itemManager.equipPoints.Find(e => e.equipmentReference != null
                                                          && e.equipmentReference.item && e.equipmentReference.item.id == id
                                                          && e.equipmentReference.equipedObject == equipment);
            if (equipPoint != null && holderAreas.ContainsKey(equipPoint.equipPointName))
            {
                var holder = holderAreas[equipPoint.equipPointName].Find(h => id == h.itemID);
                return holder;
            }
            else
            {
                if (debugMode) Debug.LogWarning(this.ToString() + " fail to find a holder with equipPointName " + equipPoint.equipPointName);
                return null;
            }
        }

        internal IEnumerator UnequipRoutine(float equipDelay, bool immediate = false, UnityEngine.Events.UnityAction onStart = null, UnityEngine.Events.UnityAction onFinish = null, string itemName = "")
        {
            if (debugMode) Debug.Log("Start Unequip: " + itemName);
            if (!immediate && !inEquip) inUnequip = true;

            while (!IsEquipping && !immediate && !inEquip)
            {
                yield return null;
            }
            if (onStart != null)
                onStart.Invoke();

            if (!inEquip && !immediate) // ignore time if inEquip or immediate unequip
            {
                var equipTime = equipDelay;
                while (!immediate && !inEquip && equipTime > 0f)
                {
                    equipTime -= vTime.deltaTime;
                    yield return null;
                }
            }

            inUnequip = false;
            if (onFinish != null)
                onFinish.Invoke();
            if (debugMode) Debug.Log("Finish Unequip: " + itemName);
        }

        internal IEnumerator EquipRoutine(float equipDelay, bool immediate = false, UnityEngine.Events.UnityAction onStart = null, UnityEngine.Events.UnityAction onFinish = null, string itemName = "")
        {
            if (debugMode) Debug.Log("Start Equip: " + itemName);
            if (!immediate)
                inEquip = true;
            while (!IsEquipping && !immediate)
            {
                yield return null;
            }
            if (onStart != null) onStart.Invoke();
            if (!inUnequip && !immediate) // ignore time if inEquip or immediate unequip
            {
                var equipTime = equipDelay;
                while (!immediate && !inUnequip && equipTime > 0f)
                {
                    equipTime -= vTime.deltaTime;
                    yield return null;
                }
            }

            inEquip = false;
            if (onFinish != null) onFinish.Invoke();
            if (debugMode) Debug.Log("Finish Equip: " + itemName);
        }
    }
}