using System;
using UnityEngine;
using UnityEngine.XR;
using UnityEngine.XR.Interaction.Toolkit;

public static class InputHelpers
{
    /// <summary>
    /// A list of buttons that can be bound to.
    /// </summary>
    public enum Button
    {
        None = 0,
        MenuButton,
        Trigger,
        Grip,
        TriggerPressed,
        GripPressed,
        PrimaryButton,
        PrimaryTouch,
        SecondaryButton,
        SecondaryTouch,
        Primary2DAxisTouch,
        Primary2DAxisClick,
        Secondary2DAxisTouch,
        Secondary2DAxisClick,
        PrimaryAxis2DUp,
        PrimaryAxis2DDown,
        PrimaryAxis2DLeft,
        PrimaryAxis2DRight,
        SecondaryAxis2DUp,
        SecondaryAxis2DDown,
        SecondaryAxis2DLeft,
        SecondaryAxis2DRight,
    };

    enum ButtonReadType
    {
        None = 0,
        Binary,
        Axis1D,
        Axis2DUp,
        Axis2DDown,
        Axis2DLeft,
        Axis2DRight,
    }

    struct ButtonInfo
    {
        public ButtonInfo(string name, ButtonReadType type)
        {
            this.name = name;
            this.type = type;
        }

        public string name;
        public ButtonReadType type;
    }

    static readonly ButtonInfo[] s_ButtonData =
    {
        new ButtonInfo("", ButtonReadType.None),
        new ButtonInfo("MenuButton", ButtonReadType.Binary),
        new ButtonInfo("Trigger", ButtonReadType.Axis1D),
        new ButtonInfo("Grip", ButtonReadType.Axis1D),
        new ButtonInfo("TriggerPressed", ButtonReadType.Binary),
        new ButtonInfo("GripPressed", ButtonReadType.Binary),
        new ButtonInfo("PrimaryButton", ButtonReadType.Binary),
        new ButtonInfo("PrimaryTouch", ButtonReadType.Binary),
        new ButtonInfo("SecondaryButton", ButtonReadType.Binary),
        new ButtonInfo("SecondaryTouch", ButtonReadType.Binary),
        new ButtonInfo("Primary2DAxisTouch", ButtonReadType.Binary),
        new ButtonInfo("Primary2DAxisClick", ButtonReadType.Binary),
        new ButtonInfo("Secondary2DAxisTouch", ButtonReadType.Binary),
        new ButtonInfo("Secondary2DAxisClick", ButtonReadType.Binary),
        new ButtonInfo("Primary2DAxis", ButtonReadType.Axis2DUp),
        new ButtonInfo("Primary2DAxis", ButtonReadType.Axis2DDown),
        new ButtonInfo("Primary2DAxis", ButtonReadType.Axis2DLeft),
        new ButtonInfo("Primary2DAxis", ButtonReadType.Axis2DRight),
        new ButtonInfo("Secondary2DAxis", ButtonReadType.Axis2DUp),
        new ButtonInfo("Secondary2DAxis", ButtonReadType.Axis2DDown),
        new ButtonInfo("Secondary2DAxis", ButtonReadType.Axis2DLeft),
        new ButtonInfo("Secondary2DAxis", ButtonReadType.Axis2DRight),
    };

    const float k_DefaultPressThreshold = 0.1f;

    public static bool IsPressed(this InputDevice device, Button button, out bool isPressed, float pressThreshold = -1.0f)
    {
        if ((int)button >= s_ButtonData.Length)
        {
            throw new ArgumentException("[InputHelpers.IsPressed] The value of <button> is out of the supported range.");
        }

        if (!device.isValid)
        {
            isPressed = false;
            return false;
        }

        var info = s_ButtonData[(int)button];
        switch (info.type)
        {
            case ButtonReadType.Binary:
                {
                    if(device.TryGetFeatureValue(new InputFeatureUsage<bool>(info.name), out var value))
                    {
                        isPressed = value;
                        return true;
                    }
                }
                break;
            case ButtonReadType.Axis1D:
                {
                    if (device.TryGetFeatureValue(new InputFeatureUsage<float>(info.name), out var value))
                    {
                        var threshold = (pressThreshold >= 0f) ? pressThreshold : k_DefaultPressThreshold;
                        isPressed = value >= threshold;
                        return true;
                    }
                }
                break;
            case ButtonReadType.Axis2DUp:
                {
                    if (device.TryGetFeatureValue(new InputFeatureUsage<Vector2>(info.name), out var value))
                    {
                        var threshold = (pressThreshold >= 0f) ? pressThreshold : k_DefaultPressThreshold;
                        isPressed = value.y >= threshold;
                        return true;
                    }
                }
                break;
            case ButtonReadType.Axis2DDown:
                {
                    if (device.TryGetFeatureValue(new InputFeatureUsage<Vector2>(info.name), out var value))
                    {
                        var threshold = (pressThreshold >= 0f) ? pressThreshold : k_DefaultPressThreshold;
                        isPressed = value.y <= -threshold;
                        return true;
                    }
                }
                break;
            case ButtonReadType.Axis2DLeft:
                {
                    if (device.TryGetFeatureValue(new InputFeatureUsage<Vector2>(info.name), out var value))
                    {
                        var threshold = (pressThreshold >= 0f) ? pressThreshold : k_DefaultPressThreshold;
                        isPressed = value.x <= -threshold;
                        return true;
                    }
                }
                break;
            case ButtonReadType.Axis2DRight:
                {
                    if (device.TryGetFeatureValue(new InputFeatureUsage<Vector2>(info.name), out var value))
                    {
                        var threshold = (pressThreshold >= 0f) ? pressThreshold : k_DefaultPressThreshold;
                        isPressed = value.x >= threshold;
                        return true;
                    }
                }
                break;
        }

        isPressed = false;
        return false;
    }

    public static bool IsPressed(this InputDeviceWrapper device, Button button, out bool isPressed, float pressThreshold = -1.0f)
    {
        if ((int)button >= s_ButtonData.Length)
        {
            throw new ArgumentException("[InputHelpers.IsPressed] The value of <button> is out of the supported range.");
        }

        if (!device.isValid)
        {
            isPressed = false;
            return false;
        }

        var info = s_ButtonData[(int)button];
        switch (info.type)
        {
            case ButtonReadType.Binary:
                {
                    if (device.TryGetFeatureValue(new InputFeatureUsage<bool>(info.name), out var value))
                    {
                        isPressed = value;
                        return true;
                    }
                }
                break;
            case ButtonReadType.Axis1D:
                {
                    if (device.TryGetFeatureValue(new InputFeatureUsage<float>(info.name), out var value))
                    {
                        var threshold = (pressThreshold >= 0f) ? pressThreshold : k_DefaultPressThreshold;
                        isPressed = value >= threshold;
                        return true;
                    }
                }
                break;
            case ButtonReadType.Axis2DUp:
                {
                    if (device.TryGetFeatureValue(new InputFeatureUsage<Vector2>(info.name), out var value))
                    {
                        var threshold = (pressThreshold >= 0f) ? pressThreshold : k_DefaultPressThreshold;
                        isPressed = value.y >= threshold;
                        return true;
                    }
                }
                break;
            case ButtonReadType.Axis2DDown:
                {
                    if (device.TryGetFeatureValue(new InputFeatureUsage<Vector2>(info.name), out var value))
                    {
                        var threshold = (pressThreshold >= 0f) ? pressThreshold : k_DefaultPressThreshold;
                        isPressed = value.y <= -threshold;
                        return true;
                    }
                }
                break;
            case ButtonReadType.Axis2DLeft:
                {
                    if (device.TryGetFeatureValue(new InputFeatureUsage<Vector2>(info.name), out var value))
                    {
                        var threshold = (pressThreshold >= 0f) ? pressThreshold : k_DefaultPressThreshold;
                        isPressed = value.x <= -threshold;
                        return true;
                    }
                }
                break;
            case ButtonReadType.Axis2DRight:
                {
                    if (device.TryGetFeatureValue(new InputFeatureUsage<Vector2>(info.name), out var value))
                    {
                        var threshold = (pressThreshold >= 0f) ? pressThreshold : k_DefaultPressThreshold;
                        isPressed = value.x >= threshold;
                        return true;
                    }
                }
                break;
        }

        isPressed = false;
        return false;
    }
}