Cielonos/Assets/Scripts/MainGame/UI/PlayerUI/MainWeapon/MainWeaponUIArea.cs

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

namespace Cielonos.MainGame.UI
{
    public class MainWeaponUIArea : UIElementBase
    {
        public MainWeaponBase mainWeapon;
        
        public GameObject functionIconPrefab;
        public RectTransform functionIconContainer;
        
        public MainWeaponFunctionIcon mainFunctionIcon;
        public DodgeFunctionIcon dodgeFunctionIcon;
        public List<MainWeaponFunctionIcon> functionIcons;
        public Dictionary<string, MainWeaponFunctionIcon> functionIconDict;
        
        private void Awake()
        {
            functionIcons = new List<MainWeaponFunctionIcon>();
            functionIconDict = new Dictionary<string, MainWeaponFunctionIcon>();
        }

        private void Start()
        {
            if (MainGameManager.Player != null && MainGameManager.Player.eventSm is Characters.PlayerEventSubmodule playerEventSm)
            {
                playerEventSm.onDodgeAttempt.Add("UI_DodgeFeedback", new PrioritizedAction<bool>(OnDodgeAttempt, 0));
            }
        }
        
        private void OnDestroy()
        {
            if (MainGameManager.Player != null && MainGameManager.Player.eventSm is Characters.PlayerEventSubmodule playerEventSm)
            {
                playerEventSm.onDodgeAttempt.Remove("UI_DodgeFeedback");
            }
        }

        private void OnDodgeAttempt(bool success)
        {
            if (dodgeFunctionIcon == null) return;
            if (success)
            {
                dodgeFunctionIcon.SetFrameOutline(0.3f, Color.white);
            }
            else
            {
                dodgeFunctionIcon.SetFrameOutline(0.3f, Color.red);
            }
        }

        public void Initialize(MainWeaponBase mainWeapon)
        {
            this.mainWeapon = mainWeapon;
            string weaponName = mainWeapon.contentData.itemClass.Name;
            
            ClearIcons();

            foreach (KeyValuePair<string, RuntimeFunctionUnit> unit in mainWeapon.functionSm.functionUnits)
            {
                if (unit.Value.data.isMain)
                {
                    mainFunctionIcon.Initialize(unit.Value);
                    functionIconDict[unit.Key] = mainFunctionIcon;
                }
                else
                {
                    if (!unit.Value.data.shownInUI) continue;
                    
                    MainWeaponFunctionIcon icon = Instantiate(functionIconPrefab, functionIconContainer).GetComponent<MainWeaponFunctionIcon>();
                    icon.Initialize(unit.Value);
                    functionIcons.Add(icon);
                    functionIconDict[unit.Key] = icon;
                }
                
            }
        }

        private void Update()
        {
            dodgeFunctionIcon.UpdateUI();
            functionIcons.ForEach(icon => icon.UpdateUI());
        }

        private void ClearIcons()
        {
            foreach (MainWeaponFunctionIcon icon in functionIcons)
            {
                Destroy(icon.gameObject);
            }
            
            functionIcons.Clear();
            functionIconDict.Clear();
        }
    }
}