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

using System;
using System.Collections.Generic;
using System.Linq;
using Cielonos.MainGame.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()
        {
            if (icons == null || icons.Count == 0)
            {
                icons = GetComponentsInChildren<SupportEquipmentIcon>(true).ToList();
            }
            
            icons.ForEach(icon => icon.DisableAllParts());
        }

        public void Initialize(SupportEquipmentBase supportEquipment, int slotIndex)
        {
            Debug.Log($"Initializing SupportEquipmentIcon at slot {slotIndex}, total icons: {icons.Count}");
            icons[slotIndex].Initialize(supportEquipment);
        }
        
        private void Update()
        {
            icons.ForEach(icon => icon.UpdateUI());
        }

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