BUTR/Bannerlord.BLSE

using System;
using System.Diagnostics;
using System.Linq;
using System.Threading;

using TaleWorlds.InputSystem;
using TaleWorlds.Library;

namespace Bannerlord.LauncherEx.Helpers;

// Can't believe I need to add a custom keyboard handler for the launcher
internal class BUTRInputManager : IInputManager, IDisposable
{
    public readonly IInputManager InputManager;
    public readonly int[] ReleasedChars = new int[10];

    private KeyboardState _currentState;
    private KeyboardState _previousState;

    public BUTRInputManager(IInputManager inputManager) => InputManager = inputManager;

    public void Update()
    {
        _currentState.Dispose();

        _previousState = _currentState;
        _currentState = Keyboard.GetState();

        var previousPressedKeys = _previousState.GetPressedKeys();
        var currentPressedKeys = _currentState.GetPressedKeys();

        var i = 0;
        foreach (var str in previousPressedKeys.Except(currentPressedKeys).Select(x => _currentState.AsString(x)))
        {
            if (string.IsNullOrEmpty(str)) continue;

            ReleasedChars[i] = str[0];
            i++;
        }
        for (; i < ReleasedChars.Length; i++)
        {
            ReleasedChars[i] = default;
        }
    }

    public bool IsMouseActive() => InputManager.IsMouseActive();
    public float GetMousePositionX() => InputManager.GetMousePositionX();
    public float GetMousePositionY() => InputManager.GetMousePositionY();
    public float GetMouseScrollValue() => InputManager.GetMouseScrollValue();
    public float GetMouseMoveY() => InputManager.GetMouseMoveY();
    public float GetMouseSensitivity() => InputManager.GetMouseSensitivity();
    public float GetMouseDeltaZ() => InputManager.GetMouseDeltaZ();

    public void SetClipboardText(string text)
    {
        var thread = new Thread(() => WindowsClipboard.SetText(text));
        thread.SetApartmentState(ApartmentState.STA);
        thread.Start();
        thread.Join();
        //_inputManagerImplementation.SetClipboardText(text);
    }
    public string GetClipboardText()
    {
        var text = string.Empty;
        var thread = new Thread(() => text = WindowsClipboard.GetText());
        thread.SetApartmentState(ApartmentState.STA);
        thread.Start();
        thread.Join();
        return text;
        //return _inputManagerImplementation.GetClipboardText();
    }

    public Vec2 GetKeyState(InputKey key) =>
        IsAction(key, rawKey => new Vec2(_currentState.IsKeyDown(rawKey) ? 1f : 0f, _previousState.IsKeyDown(rawKey) ? 1f : 0f), _key => InputManager.GetKeyState(_key));
    public bool IsKeyPressed(InputKey key) =>
        IsAction(key, rawKey => _currentState.IsKeyDown(rawKey) && _previousState.IsKeyUp(rawKey), _key => InputManager.IsKeyPressed(_key));
    public bool IsKeyDown(InputKey key) =>
        IsAction(key, rawKey => _currentState.IsKeyDown(rawKey) || _previousState.IsKeyDown(rawKey), _key => InputManager.IsKeyDown(_key));
    public bool IsKeyReleased(InputKey key) =>
        IsAction(key, rawKey => _currentState.IsKeyUp(rawKey) && _previousState.IsKeyDown(rawKey), _key => InputManager.IsKeyReleased(_key));
    public bool IsKeyDownImmediate(InputKey key) =>
        IsAction(key, rawKey => _currentState.IsKeyDown(rawKey), _key => InputManager.IsKeyDownImmediate(_key));

    public Vec2 GetResolution() => InputManager.GetResolution();
    public Vec2 GetDesktopResolution() => InputManager.GetDesktopResolution();

    public void SetCursorPosition(int x, int y) => InputManager.SetCursorPosition(x, y);
    public void SetCursorFriction(float frictionValue) => InputManager.SetCursorFriction(frictionValue);

    public void PressKey(InputKey key) => InputManager.PressKey(key);
    public void ClearKeys() => InputManager.ClearKeys();
    public int GetVirtualKeyCode(InputKey key) => InputManager.GetVirtualKeyCode(key);
    public float GetMouseMoveX() => InputManager.GetMouseMoveX();
    public void UpdateKeyData(byte[] keyData) => InputManager.UpdateKeyData(keyData);

    public bool IsControllerConnected() => InputManager.IsControllerConnected();

#if v100 || v110
        public InputKey GetControllerClickKey() => InputManager.GetControllerClickKey();
#endif

#if v110 || v120
    public void SetRumbleEffect(float[] lowFrequencyLevels, float[] lowFrequencyDurations, int numLowFrequencyElements, float[] highFrequencyLevels, float[] highFrequencyDurations, int numHighFrequencyElements) =>
        InputManager.SetRumbleEffect(lowFrequencyLevels, lowFrequencyDurations, numLowFrequencyElements, highFrequencyLevels, highFrequencyDurations, numHighFrequencyElements);
    public void SetTriggerFeedback(byte leftTriggerPosition, byte leftTriggerStrength, byte rightTriggerPosition, byte rightTriggerStrength) =>
        InputManager.SetTriggerFeedback(leftTriggerPosition, leftTriggerStrength, rightTriggerPosition, rightTriggerStrength);
    public void SetTriggerWeaponEffect(byte leftStartPosition, byte leftEnd_position, byte leftStrength, byte rightStartPosition, byte rightEndPosition, byte rightStrength) =>
        InputManager.SetTriggerWeaponEffect(leftStartPosition, leftEnd_position, leftStrength, rightStartPosition, rightEndPosition, rightStrength);
    public void SetTriggerVibration(float[] leftTriggerAmplitudes, float[] leftTriggerFrequencies, float[] leftTriggerDurations, int numLeftTriggerElements, float[] rightTriggerAmplitudes, float[] rightTriggerFrequencies, float[] rightTriggerDurations, int numRightTriggerElements) =>
        InputManager.SetTriggerVibration(leftTriggerAmplitudes, leftTriggerFrequencies, leftTriggerDurations, numLeftTriggerElements, rightTriggerAmplitudes, rightTriggerFrequencies, rightTriggerDurations, numRightTriggerElements);
    public void SetLightbarColor(float red, float green, float blue) => InputManager.SetLightbarColor(red, green, blue);
#endif

#if v120
    public TaleWorlds.InputSystem.Input.ControllerTypes GetControllerType() => InputManager.GetControllerType();

    public float GetGyroX() => InputManager.GetGyroX();
    public float GetGyroY() => InputManager.GetGyroY();
    public float GetGyroZ() => InputManager.GetGyroZ();

    public InputKey[] GetClickKeys() => InputManager.GetClickKeys();

    public bool IsAnyTouchActive() => InputManager.IsAnyTouchActive();
#endif

    private static TReturn IsAction<TReturn>(InputKey key, Func<Keys, TReturn> action, Func<InputKey, TReturn> fallback)
    {
        var rawKey = key switch
        {
            InputKey.BackSpace => Keys.Back,
            InputKey.Enter => Keys.Enter,
            _ => Enum.TryParse<Keys>(key.ToString(), out var keyVal) ? keyVal : Keys.None
        };
        if (key is >= InputKey.ControllerLStickUp and <= InputKey.ControllerRTrigger or InputKey.ControllerLStick or InputKey.ControllerRStick)
        {
            return fallback(key);
        }
        if (key is InputKey.LeftMouseButton or InputKey.RightMouseButton or InputKey.MiddleMouseButton)
        {
            return fallback(key);
        }
        if (rawKey == Keys.None)
        {
            Trace.TraceError($"Wrong key {key}");
            return fallback(key);
        }

        return action(rawKey);
    }


    public void Dispose()
    {
        _currentState.Dispose();
        _previousState.Dispose();
    }
}