// Copyright (C) 2023 Nicholas Maltbie
//
// Permission is hereby granted, free of charge, to any person obtaining a copy of this software and
// associated documentation files (the "Software"), to deal in the Software without restriction,
// including without limitation the rights to use, copy, modify, merge, publish, distribute,
// sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in all copies or
// substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING
// BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
// CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
// ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
// SOFTWARE.

using nickmaltbie.ScreenManager;
using nickmaltbie.ScreenManager.Actions;
using UnityEngine;
using UnityEngine.InputSystem;
using UnityEngine.UI;

namespace nickmaltbie.OpenKCC.UI.Actions
{
    /// <summary>
    /// Rebind an individual button input action
    /// </summary>
    public class SelectiveRebindInput : MonoBehaviour, IBindingControl
    {
        /// <summary>
        /// Prefix for input mapping for saving to player preferences
        /// </summary>
        public const string inputMappingPlayerPrefPrefix = "Input Mapping";

        /// <summary>
        /// Path used to cancel binding mid operation
        /// </summary>
        public string cancelPath = "<Keyboard>/escape";

        /// <summary>
        /// Input action being modified
        /// </summary>
        public InputActionReference inputAction = null;
        /// <summary>
        /// Binding display name for showing the control button description
        /// </summary>
        public UnityEngine.UI.Text bindingDisplayNameText = null;
        /// <summary>
        /// Button to start rebinding for the given input action
        /// </summary>
        public Button startRebinding = null;

        /// <summary>
        /// Text to display when waiting for the player to press a new input action
        /// </summary>
        public GameObject waitingForInputObject = null;
        /// <summary>
        /// Menu controller related to this selected object
        /// </summary>
        public MenuController menuController;

        /// <summary>
        /// Rebinding operation action waiting for player command to change button bindings
        /// </summary>
        public InputActionRebindingExtensions.RebindingOperation rebindingOperation { get; private set; }

        /// <summary>
        /// Get a readable display name for a binding index
        /// </summary>
        /// <returns>Human readable information of button for this binding index</returns>
        public string GetKeyReadableName() =>
            InputControlPath.ToHumanReadableString(
                inputAction.action.bindings[0].effectivePath,
                InputControlPath.HumanReadableStringOptions.OmitDevice);

        /// <summary>
        /// Get the input mapping player preference key from a given index
        /// </summary>
        public string InputMappingKey => $"{inputMappingPlayerPrefPrefix} {inputAction.action.name}";

        public void Awake()
        {
            // Load the default mapping saved to the file
            menuController = gameObject.GetComponentInParent<MenuController>();
            string inputMapping = PlayerPrefs.GetString(InputMappingKey, string.Empty);
            if (!string.IsNullOrEmpty(inputMapping))
            {
                inputAction.action.ApplyBindingOverride(0, inputMapping);
            }
        }

        public void Start()
        {
            startRebinding.onClick.AddListener(() => StartRebinding());
            bindingDisplayNameText.text = GetKeyReadableName();
        }

        /// <summary>
        /// Start the rebinding process for a given component of this composite axis.
        /// </summary>
        public void StartRebinding()
        {
            startRebinding.gameObject.SetActive(false);
            waitingForInputObject.SetActive(true);
            menuController.allowInputChanges = false;

            inputAction.action.Disable();
            inputAction.action.actionMap.Disable();
            rebindingOperation = inputAction.action.PerformInteractiveRebinding(0)
                .WithControlsExcluding("<Pointer>/position") // Don't bind to mouse position
                .WithControlsExcluding("<Pointer>/delta")    // To avoid accidental input from mouse motion
                .WithCancelingThrough(cancelPath)
                .OnMatchWaitForAnother(0.1f)
                .WithTimeout(5.0f)
                .OnComplete(operation => RebindComplete())
                .OnCancel(operation => RebindCancel())
                .Start();
        }

        /// <summary>
        /// Cancel the rebinding process for a given component of this composite axis.
        /// </summary>
        public void RebindCancel()
        {
            bindingDisplayNameText.text = GetKeyReadableName();
            rebindingOperation.Dispose();

            startRebinding.gameObject.SetActive(true);
            waitingForInputObject.SetActive(false);
            menuController.allowInputChanges = true;
            inputAction.action.Enable();
            inputAction.action.actionMap.Enable();
        }

        /// <summary>
        /// Finish the rebinding process for a given component of this composite axis.
        /// </summary>
        public void RebindComplete()
        {
            string overridePath = inputAction.action.bindings[0].overridePath;
#if UNITY_2020_3_OR_NEWER
            foreach (PlayerInput input in GameObject.FindObjectsByType<PlayerInput>(FindObjectsSortMode.None))
#else
            foreach (PlayerInput input in GameObject.FindObjectsOfType<PlayerInput>())
#endif
            {
                input.actions.FindAction(inputAction.name)?.ApplyBindingOverride(0, overridePath);
            }

            bindingDisplayNameText.text = GetKeyReadableName();
            rebindingOperation.Dispose();

            PlayerPrefs.SetString(InputMappingKey, inputAction.action.bindings[0].overridePath);

            startRebinding.gameObject.SetActive(true);
            waitingForInputObject.SetActive(false);
            menuController.allowInputChanges = true;
            inputAction.action.Enable();
            inputAction.action.actionMap.Enable();
        }

        public void ResetBinding()
        {
            PlayerPrefs.DeleteKey(InputMappingKey);
            inputAction.action.RemoveAllBindingOverrides();
        }

        public void UpdateDisplay()
        {
            bindingDisplayNameText.text = GetKeyReadableName();
        }
    }
}