Cielonos/Assets/Scripts/MainGame/UI/PlayerUI/SupportEquipments/SupportEquipmentsUIArea.cs

using System;
using System.Collections.Generic;
using System.Linq;
using Cielonos.MainGame.Characters.Inventory;
using SLSUtilities.UI;
using UnityEngine;

namespace Cielonos.MainGame.UI
{
    public class SupportEquipmentsUIArea : UIElementBase
    {
        public List<SupportEquipmentIcon> icons;
        
        public SupportEquipmentIcon this[SupportEquipmentBase supportEquipment] =>
            icons.FirstOrDefault(icon => icon.supportEquipment == supportEquipment);

        private void Awake()
        {
            icons = GetComponentsInChildren<SupportEquipmentIcon>(true).ToList();
        }

        public void Initialize(SupportEquipmentBase supportEquipment, int slotIndex)
        {
            icons[slotIndex].Initialize(supportEquipment);
        }
        
        private void Update()
        {
            icons.ForEach(icon => icon.UpdateUI());
        }

        public void Remove(int slotIndex)
        {
            icons[slotIndex].Initialize(null);
        }
    }
}