Unity_Utils/JohnsonUtils/Utilities/Icons/BindingsIconsUtil.cs

using System.Collections.Generic;
using System.Linq;
using UnityEngine;
using UnityEngine.InputSystem;
using static UnityEngine.InputSystem.InputSystem;

namespace JohnsonUtils.Utilities.Icons
{
    public static class BindingsIconsUtil
    {
        private static GamepadIcons xbox;
        private static GamepadIcons ps4;
        private static MouseKeyboardIcons mouseKeyboard;

        private static void Init()
        {
            xbox = Resources.Load<GamepadIcons>("InputSystem/GamepadIcons_Xbox");
            ps4 = Resources.Load<GamepadIcons>("InputSystem/GamepadIcons_PS");
            mouseKeyboard = Resources.Load<MouseKeyboardIcons>("InputSystem/MouseKeyboardIcons");
        }

        public static Sprite GetSprite(string deviceLayoutName, string mainControlPath)
        {
            if (xbox == null) Init();

            if (string.IsNullOrEmpty(deviceLayoutName) || string.IsNullOrEmpty(mainControlPath))
                return null;

            Sprite icon = default;
            if (IsFirstLayoutBasedOnSecond(deviceLayoutName, "DualShockGamepad"))
            {
                icon = ps4.GetSprite(mainControlPath);
            }
            else if (IsFirstLayoutBasedOnSecond(deviceLayoutName, "Gamepad"))
            {
                icon = xbox.GetSprite(mainControlPath);
            }
            else if (IsFirstLayoutBasedOnSecond(deviceLayoutName, "Keyboard") || IsFirstLayoutBasedOnSecond(deviceLayoutName, "Mouse"))
            {
                icon = mouseKeyboard.GetSprite(mainControlPath);
            }

            return icon;
        }

        public static Sprite GetSprite(InputAction inputAction, string deviceName)
        {
            if (inputAction == null || string.IsNullOrEmpty(deviceName))
                return null;

            var mainBindings = GetRelevantMainBindings(inputAction, deviceName, true);
            if (!mainBindings.Any())
                return null;

            var mainBinding = mainBindings.First();

            inputAction.GetBindingDisplayString(mainBinding, out var deviceLayoutName, out var mainControlPath);

            return GetSprite(deviceLayoutName, mainControlPath);
        }

        // So we don't create an instance each time the next method is called
        private static readonly List<int> RelevantMainBindings = new List<int>();

        public static List<int> GetRelevantMainBindings(InputAction inputAction, string deviceName, bool checkUnrebindableBindings = false)
        {
            RelevantMainBindings.Clear();
            if (inputAction.bindings.Count >= 4)
            {
                if (inputAction.bindings[0].isComposite)
                {
                    if (deviceName == "Gamepad")
                    {
                        RelevantMainBindings.Add(inputAction.bindings.Count - 2);
                    }
                    else
                    {
                        for (int i = 1; i < inputAction.bindings.Count - 2; i += 2)
                        {
                            RelevantMainBindings.Add(i);
                        }
                    }
                }
                else
                {
                    RelevantMainBindings.Add(deviceName == "Gamepad" ? 2 : 0);
                }
            }
            else if (inputAction.bindings.Count >= 2 && checkUnrebindableBindings)
            {
                RelevantMainBindings.Add(deviceName == "Gamepad" ? 1 : 0);
            }

            return RelevantMainBindings.ToList();
        }
    }
}