COM3D2.MeidoPhotoStudio/src/MeidoPhotoStudio.Plugin/GUI/Controls/KeyRebindButton.cs

using System;
using UnityEngine;

namespace MeidoPhotoStudio.Plugin
{
    public class KeyRebindButton : BaseControl
    {
        private readonly Button button;
        private bool listening;
        private KeyCode keyCode;
        public KeyCode KeyCode
        {
            get => keyCode;
            set
            {
                keyCode = value;
                button.Label = keyCode.ToString();
            }
        }
        public KeyRebindButton(KeyCode code)
        {
            button = new Button(code.ToString());
            button.ControlEvent += (s, a) => StartListening();
        }

        public void Draw(GUIStyle buttonStyle, params GUILayoutOption[] layoutOptions)
        {
            GUI.enabled = !listening && !InputManager.Listening;
            button.Draw(buttonStyle, layoutOptions);
            GUI.enabled = true;
        }

        public override void Draw(params GUILayoutOption[] layoutOptions)
        {
            GUIStyle buttonStyle = new GUIStyle(GUI.skin.button);
            Draw(buttonStyle, layoutOptions);
        }

        private void StartListening()
        {
            listening = true;
            button.Label = string.Empty;
            InputManager.StartListening();
            InputManager.KeyChange += KeyChange;
        }

        private void KeyChange(object sender, EventArgs args)
        {
            listening = false;
            if (InputManager.CurrentKeyCode != KeyCode.Escape) KeyCode = InputManager.CurrentKeyCode;
            else KeyCode = KeyCode;
            InputManager.KeyChange -= KeyChange;
            OnControlEvent(EventArgs.Empty);
        }
    }
}