using EFT.InputSystem;

namespace UIFixes
{
    public enum ToggleHoldState
    {
        Idle = 13,
        ClickOrHold = 14,
        Holding = 15
    }

    public class ToggleHoldIdleState(KeyCombination keyCombination) : KeyCombination.KeyCombinationState(keyCombination)
    {
        public override ECommand GetCommand(float deltaTime)
        {
            if (!CanProcess())
            {
                return ECommand.None;
            }

            HandleKeys(false);
            KeyCombination.method_0((KeyCombination.EKeyState)ToggleHoldState.ClickOrHold);
            return GetCommandInternal();
        }

        protected bool CanProcess()
        {
            return GetKeysStatus(out EKeyPress ekeyPress) && (ekeyPress == EKeyPress.Down);
        }
    }

    public class ToggleHoldClickOrHoldState(KeyCombination keyCombination) : KeyCombination.KeyCombinationState(keyCombination)
    {
        public override void Enter()
        {
            timer = KeyCombination.DoubleClickTimeout;
        }

        public override ECommand GetCommand(float deltaTime)
        {
            if (GetKeysStatus(out EKeyPress ekeyPress))
            {
                if (ekeyPress == EKeyPress.Hold)
                {
                    HandleKeys(false);
                    if (LongEnough(deltaTime))
                    {
                        KeyCombination.method_0((KeyCombination.EKeyState)ToggleHoldState.Holding);
                    }

                    return ECommand.None;
                }
            }

            UnhandleKeys(null);
            KeyCombination.method_0((KeyCombination.EKeyState)ToggleHoldState.Idle);
            return ECommand.None;
        }

        private bool LongEnough(float deltaTime)
        {
            timer -= deltaTime;
            return timer <= 0f;
        }

        private float timer;
    }

    public class ToggleHoldHoldState(KeyCombination keyCombination, ECommand disableCommand) : KeyCombination.KeyCombinationState(keyCombination)
    {
        private readonly ECommand disableCommand = disableCommand;

        public override ECommand GetCommand(float deltaTime)
        {
            if (GetKeysStatus(out EKeyPress ekeyPress) && ekeyPress == EKeyPress.Hold)
            {
                HandleKeys(false);
                return ECommand.None;
            }

            UnhandleKeys(null);
            KeyCombination.method_0((KeyCombination.EKeyState)ToggleHoldState.Idle);
            return disableCommand;
        }
    }
}