BITFALL/Assets/Artists/Scripts/UX/UXEquipSelector.cs

using System;
using System.Collections;
using System.Collections.Concurrent;
using System.Collections.Generic;
using UnityEngine;
using BITKit;
using BITKit.Entities;
using BITKit.Entities.Player;
using BITKit.UX;
using RotaryHeart.Lib.SerializableDictionary;
using UnityEngine.Playables;
using UnityEngine.UIElements;

namespace BITFALL
{
    public class UXEquipSelector : MonoBehaviour
    {
        [SerializeReference,SubclassSelector]  private IPlayerService playerService;
        [SerializeField]
        private SerializableDictionaryBase<int, UXImage> dictionary = new();
        [SerializeField] private UXImage currentUXEquip;
        private readonly ConcurrentDictionary<int,IBasicItem> cache=new();

        private IPlayerEquipSelector _equipSelector;
        public void OnEquip(IBasicItem item)
        {
            if(item is null) return;
            var assets = item.GetAssetable();
            currentUXEquip.SetTexture(assets.RectangleIcon);
        }
        public void OnDeEquip(IBasicItem item)
        {
            currentUXEquip.SetTexture(null);
        }
        public void OnUpdateEquip(IDictionary<int, IBasicItem> maps)
        {
            cache.Clear();
            foreach (var (key, image) in dictionary)
            {
                if (maps.TryGetValue(key, out var item))
                {
                    var asset = item.GetAssetable();
                    image.SetTexture(asset.RectangleIcon);
                }
                else
                {
                    image.SetTexture(null);
                }
                cache.TryAdd(key, item);
            }
        }
        private void Start()
        {
            playerService.OnPlayerInitialized += OnStartLocalPlayer;
            OnDeEquip(null);
            foreach (var x in dictionary)
            {
                x.Value.visualElement.RegisterCallback<MouseDownEvent>(evt =>
                {
                    if (evt.button is not 1) return;
                    if (cache.TryGetValue(x.Key, out var item))
                    {
                        _equipSelector.TryDeEquip(item);
                    }
                });
            }
        }
        private void OnDestroy()
        {
            playerService.OnPlayerInitialized -= OnStartLocalPlayer;
        }
        private void OnStartLocalPlayer(IEntity entity)
        {
            _equipSelector = entity.Get<IPlayerEquipSelector>();
            _equipSelector.OnEquip += OnEquip;
            _equipSelector.OnDeEquip += OnDeEquip;
            _equipSelector.OnUpdateEquip += OnUpdateEquip;
        }
    }
}