#include "input/Input.h"
#include "Quirks.h"
#include "platform.h"

#define INPUT WalrusRPG::Input
using WalrusRPG::Input::Key;
using WalrusRPG::Input::KeyState;

struct InputMap
{
    Key key;
    keycode_t key_code;
};

struct KeyBuffer
{
    bool current;
    bool previous;
};
struct KeyBuffer key_states[Key::K_SIZE] = {{false, false}};

// TODO: make these software-mappable
#ifdef SFML
using sf::Keyboard;
InputMap key_map[] = {
    {Key::K_A, Keyboard::W},
    {Key::K_B, Keyboard::X},
    {Key::K_L, Keyboard::Q},
    {Key::K_R, Keyboard::S},

    {Key::K_UP, Keyboard::Up},
    {Key::K_DOWN, Keyboard::Down},
    {Key::K_LEFT, Keyboard::Left},
    {Key::K_RIGHT, Keyboard::Right},

    {Key::K_START, Keyboard::Return},
    {Key::K_SELECT, Keyboard::BackSpace},
};
#endif
#ifdef NSPIRE
static InputMap key_map[] = {
    {Key::K_A, KEY_NSPIRE_CTRL},     {Key::K_B, KEY_NSPIRE_SHIFT},
    {Key::K_L, KEY_NSPIRE_TAB},      {Key::K_R, KEY_NSPIRE_MENU},

    {Key::K_UP, KEY_NSPIRE_8},       {Key::K_DOWN, KEY_NSPIRE_5},
    {Key::K_LEFT, KEY_NSPIRE_4},     {Key::K_RIGHT, KEY_NSPIRE_6},

    {Key::K_START, KEY_NSPIRE_DOC}, {Key::K_SELECT, KEY_NSPIRE_ESC},
};
#endif

KeyState INPUT::key_get_state(Key key)
{
    // "Just" pressed/released before held/inactive to let these events through
    if (key_pressed(key))
        return KS_JUST_PRESSED;
    if (key_down(key))
        return KS_PRESSED;
    if (key_released(key))
        return KS_JUST_RELEASED;
    if (key_up(key))
        return KS_RELEASED;

    // Default case to mute compiler warnings, shouldn't even reach here in practice
    return KS_RELEASED;
}

void INPUT::key_poll()
{
    for (unsigned i = 0; i < K_SIZE; i++)
    {
        key_states[i].previous = key_states[i].current;
        key_states[i].current = WalrusRPG::Quirks::get_key(key_map[i].key_code);
    }
}

bool INPUT::key_pressed(Key key)
{
    return !key_states[key].previous && key_states[key].current;
}
bool INPUT::key_released(Key key)
{
    return key_states[key].previous && !key_states[key].current;
}
bool INPUT::key_down(Key key)
{
    return key_states[key].current;
}
bool INPUT::key_up(Key key)
{
    return !key_states[key].current;
}