using System; using UnityEngine; namespace COM3D2.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); } } }