using Microsoft.Xna.Framework;
using Microsoft.Xna.Framework.Input;

using Syntriax.Engine.Core;

namespace Syntriax.Engine.Input;

public class KeyboardInputsBehaviour : BehaviourOverride, IKeyboardInputs
{
    private readonly Dictionary<Keys, Action<IKeyboardInputs, Keys>> OnPressed = new(256);
    private readonly Dictionary<Keys, Action<IKeyboardInputs, Keys>> OnReleased = new(256);

    private int cachePressedCurrentlyCount = 0;
    private readonly Keys[] cachePressedCurrently = new Keys[256];

    private int cachePressedPreviouslyCount = 0;
    private readonly Keys[] cachePressedPreviously = new Keys[256];

    public void RegisterOnPress(Keys key, Action<IKeyboardInputs, Keys> callback)
    {
        if (OnPressed.TryGetValue(key, out var action))
        {
            action += callback;
            return;
        }

        OnPressed.Add(key, callback);
    }

    public void UnregisterOnPress(Keys key, Action<IKeyboardInputs, Keys> callback)
    {
        if (OnPressed.TryGetValue(key, out var action))
            action -= callback;
    }

    public void RegisterOnRelease(Keys key, Action<IKeyboardInputs, Keys> callback)
    {
        if (OnReleased.TryGetValue(key, out var action))
        {
            action += callback;
            return;
        }

        OnReleased.Add(key, callback);
    }

    public void UnregisterOnRelease(Keys key, Action<IKeyboardInputs, Keys> callback)
    {
        if (OnReleased.TryGetValue(key, out var action))
            action -= callback;
    }

    protected override void OnUpdate(GameTime time)
    {
        KeyboardState keyboardState = Keyboard.GetState();
        keyboardState.GetPressedKeys(cachePressedCurrently);
        cachePressedCurrentlyCount = keyboardState.GetPressedKeyCount();

        for (int i = 0; i < cachePressedCurrentlyCount; i++)
        {
            Keys currentlyPressedKey = cachePressedCurrently[i];

            if (!OnPressed.TryGetValue(currentlyPressedKey, out var action))
                continue;

            if (WasPressed(currentlyPressedKey))
                continue;

            action.Invoke(this, currentlyPressedKey);
        }

        for (int i = 0; i < cachePressedPreviouslyCount; i++)
        {
            Keys previouslyPressedKey = cachePressedPreviously[i];

            if (!OnReleased.TryGetValue(previouslyPressedKey, out var action))
                continue;

            if (IsPressed(previouslyPressedKey))
                continue;

            action.Invoke(this, previouslyPressedKey);
        }

        Array.Copy(cachePressedCurrently, cachePressedPreviously, cachePressedCurrentlyCount);
        cachePressedPreviouslyCount = cachePressedCurrentlyCount;
    }

    public bool IsPressed(Keys key)
    {
        for (int i = 0; i < cachePressedCurrentlyCount; i++)
            if (cachePressedCurrently[i] == key)
                return true;
        return false;
    }

    public bool WasPressed(Keys key)
    {
        for (int i = 0; i < cachePressedPreviouslyCount; i++)
            if (cachePressedPreviously[i] == key)
                return true;
        return false;
    }
}