BABA_YAGA/Assets/Scripts/Player/Shooter/vAmmoManager.cs

using UnityEngine;
using System.Collections.Generic;

namespace Invector.vItemManager
{
    [vClassHeader("Ammo Manager", iconName = "ammoIcon")]
    public class vAmmoManager : vMonoBehaviour
    {
        public vAmmoListData ammoListData;
        [HideInInspector]
        public vItemManager itemManager;

        //[HideInInspector]
        public List<vAmmo> ammos = new List<vAmmo>();
        public delegate void OnUpdateTotalAmmo();
        public OnUpdateTotalAmmo updateTotalAmmo = delegate { };

        void Start()
        {
            itemManager = GetComponent<vItemManager>();
            if (itemManager)
            {
                itemManager.onAddItem.AddListener(AddAmmo);
                itemManager.onCollectItemRef.AddListener(AddAmmo);
                itemManager.onDropItem.AddListener(DropAmmo);
                itemManager.onDestroyItem.AddListener(LeaveAmmo);
                itemManager.onChangeItemAmount.AddListener(ChangeItemAmount);
                itemManager.onLoadItems.AddListener(ReloadAllAmmoItems);
            }

            if (ammoListData)
            {
                ammos.Clear();
                for (int i = 0; i < ammoListData.ammos.Count; i++)
                {
                    var ammo = new vAmmo(ammoListData.ammos[i]);
                    ammo.onDestroyAmmoItem = new vAmmo.OnDestroyItem(OnDestroyAmmoItem);
                    ammos.Add(ammo);
                }
            }
        }

        public vAmmo GetAmmo(int id)
        {
            return ammos.Find(a => a.ammoID == id);
        }

        /// <summary>
        /// Use with AmmoStandalone Solution
        /// </summary>      
        /// <param name="id"></param>
        /// <param name="amount"></param>
        public void AddAmmo(int id, int amount)
        {
            var ammo = ammos.Find(a => a.ammoID == id);
            if (ammo != null)
            {
                ammo.AddAmmo(amount);
            }
            UpdateTotalAmmo();
        }

        /// <summary>
        /// Use with AmmoStandalone Solution
        /// </summary>
        /// <param name="ammoName"></param>
        /// <param name="id"></param>
        /// <param name="amount"></param>
        public void AddAmmo(string ammoName, int id, int amount)
        {
            var ammo = ammos.Find(a => a.ammoID == id);
            if (ammo == null)
            {
                ammo = new vAmmo(ammoName, id, amount);
                ammos.Add(ammo);
                ammo.onDestroyAmmoItem = new vAmmo.OnDestroyItem(OnDestroyAmmoItem);
            }
            else if (ammo != null)
            {
                ammo.AddAmmo(amount);
            }
            UpdateTotalAmmo();
        }

        /// <summary>
        /// Use with ItemManager
        /// </summary>
        /// <param name="item"></param>
        public void AddAmmo(vItem item)
        {
            if (item.type == vItemType.Ammo)
            {
                var ammo = ammos.Find(a => a.ammoID == item.id);
                if (ammo == null)
                {
                    ammo = new vAmmo(item.name, item.id, 0);
                    ammos.Add(ammo);
                    ammo.onDestroyAmmoItem = new vAmmo.OnDestroyItem(OnDestroyAmmoItem);
                }

                ammo.ammoItems.Add(item);
            }
            else
            {
                var _ammo = item.GetItemAttribute(vItemAttributes.AmmoCount);

                if (_ammo != null)
                {
                    var weaponReference = item.originalObject.GetComponent<vShooter.vShooterWeapon>();
                    if (weaponReference != null)
                    {
                        if (_ammo != null && _ammo.value > weaponReference.clipSize)
                        {
                            var extraAmmo = _ammo.value - weaponReference.clipSize;
                            _ammo.value -= extraAmmo;
                            ItemReference newAmmoItem = new ItemReference(weaponReference.ammoID);
                            newAmmoItem.amount = extraAmmo;
                            itemManager.CollectItem(newAmmoItem, ignoreItemAnimation: true);
                        }
                    }
                }
            }
            UpdateTotalAmmo();
        }

        /// <summary>
        /// Add Ammo Attribute
        /// </summary>
        /// <param name="item"></param>
        public void AddAmmo(ItemReference item)
        {
            var _ammo = item.attributes.Find(a => a.name == vItemAttributes.AmmoCount);
            var _item = itemManager.itemListData.items.Find(i => i.id == item.id);

            if (_ammo != null && _item != null)
            {
                var weaponReference = _item.originalObject.GetComponent<vShooter.vShooterWeapon>();
                if (weaponReference != null)
                {
                    ItemReference newAmmoItem = new ItemReference(weaponReference.ammoID);
                    newAmmoItem.amount = _ammo.value;
                    itemManager.CollectItem(newAmmoItem, ignoreItemAnimation: true);
                }
            }

            UpdateTotalAmmo();
        }

        protected void ChangeItemAmount(vItem item)
        {
            if (item.type == vItemType.Ammo)
            {
                var ammo = ammos.Find(a => a.ammoID == item.id);
                if (ammo == null)
                {
                    ammo = new vAmmo(item.name, item.id, item.amount);
                    ammos.Add(ammo);
                    ammo.onDestroyAmmoItem = new vAmmo.OnDestroyItem(OnDestroyAmmoItem);
                }
            }
            UpdateTotalAmmo();
        }

        public void LeaveAmmo(vItem item, int amount)
        {
            if (item.type == vItemType.Ammo)
            {
                var ammo = ammos.Find(a => a.ammoID == item.id);
                if (ammo != null)
                {
                    if ((item.amount - amount) <= 0 && ammo.ammoItems.Contains(item))
                        ammo.ammoItems.Remove(item);
                }
            }
            UpdateTotalAmmo();
        }

        public void DropAmmo(vItem item, int amount)
        {
            if (item.type == vItemType.Ammo)
            {
                var ammo = ammos.Find(a => a.ammoID == item.id);
                if (ammo != null)
                {
                    if ((item.amount - amount) <= 0 && ammo.ammoItems.Contains(item))
                        ammo.ammoItems.Remove(item);
                }
            }
            UpdateTotalAmmo();
        }

        public void UpdateTotalAmmo()
        {
            updateTotalAmmo.Invoke();
        }

        public void ReloadAllAmmoItems()
        {
            var ammosInManager = itemManager.items.FindAll(item => item.type == vItemType.Ammo);
            ammos.Clear();
            for (int i = 0; i < ammosInManager.Count; i++)
            {
                AddAmmo(ammosInManager[i]);
            }
        }

        void OnDestroyAmmoItem(vItem item)
        {
            if (itemManager) itemManager.DestroyItem(item, item.amount);
        }
    }

    [System.Serializable]
    public class vAmmo
    {
        public string ammoName;
        [Tooltip("Ammo ID - if is using ItemManager, make sure your AmmoManager and ItemListData use the same ID")]
        public int ammoID;
        [Tooltip("Don't need to setup if you're using a Inventory System")]
        [SerializeField]
        private int _count;

        public List<vItem> ammoItems;
        public delegate void OnDestroyItem(vItem item);
        public OnDestroyItem onDestroyAmmoItem = delegate { };

        public vAmmo()
        {
            ammoItems = new List<vItem>();
        }

        public vAmmo(string ammoName, int ammoID, int amount = 0)
        {
            this.ammoName = ammoName;
            this.ammoID = ammoID;
            this._count = amount;
            ammoItems = new List<vItem>();
        }

        public vAmmo(int ammoID, int amount = 0)
        {
            this.ammoID = ammoID;
            this._count = amount;
            ammoItems = new List<vItem>();
        }

        public vAmmo(vAmmo ammo)
        {
            this.ammoName = ammo.ammoName;
            this.ammoID = ammo.ammoID;
            this.ammoItems = ammo.ammoItems;
            this._count = ammo.count;
            ammoItems = new List<vItem>();
        }

        public virtual int count
        {
            get
            {
                var value = 0;
                if (ammoItems != null && ammoItems.Count > 0)
                {
                    for (int i = 0; i < ammoItems.Count; i++)
                        if (ammoItems[i])
                            value += ammoItems[i].amount;
                }
                return _count + value;
            }
        }

        public virtual void Use()
        {
            var ammoItem = ammoItems.Find(a => a.amount > 0);
            if (ammoItem)
            {
                ammoItem.amount--;
                if (ammoItem.amount == 0)
                {
                    ammoItems.Remove(ammoItem);
                    onDestroyAmmoItem(ammoItem);
                }
                return;
            }
            else if (_count > 0) _count--;
        }

        public virtual void Use(int amout)
        {
            for (int i = 0; i < amout; i++) Use();
        }

        public void AddAmmo(int amount)
        {
            _count += amount;
        }
    }
}