AlicizaX/Client/Assets/Scripts/CustomeModule/InputGlyph/InputGlyphUXButton.cs

using UnityEngine;
using UnityEngine.InputSystem;
using UnityEngine.UI;

[RequireComponent(typeof(UXButton))]
public sealed class InputGlyphUXButton : InputGlyphBehaviourBase
{
    [SerializeField] private UXButton button;
    [SerializeField] private Image targetImage;

    private Sprite _cachedSprite;

#if UNITY_EDITOR
    private void OnValidate()
    {
        if (button == null)
        {
            button = GetComponent<UXButton>();
        }
    }
#endif

    protected override void OnEnable()
    {
        if (button == null)
        {
            button = GetComponent<UXButton>();
        }

        if (targetImage == null)
        {
            targetImage = GetComponent<Image>();
        }

        base.OnEnable();
    }

    protected override void RefreshGlyph()
    {
        InputActionReference actionReference = button != null ? button.HotKeyRefrence : null;
        if (actionReference == null || actionReference.action == null || targetImage == null)
        {
            if (targetImage != null && _cachedSprite != null)
            {
                _cachedSprite = null;
                targetImage.sprite = null;
            }

            return;
        }

        bool hasSprite = GlyphService.TryGetUISpriteForActionPath(actionReference, string.Empty, CurrentCategory, out Sprite sprite);
        if (!hasSprite)
        {
            sprite = null;
        }

        if (_cachedSprite != sprite)
        {
            _cachedSprite = sprite;
            targetImage.sprite = sprite;
        }
    }
}