#include <Windows.h>

#include <Spectre/Input/InputEvent.h>
#include <Spectre/Input/InputModule.h>
#include "Win32Input.h"
#include "Win32Keyboard.h"
#include "Win32MsgBuffer.h"

namespace sp {

static Win32MsgBuffer msg_buf;

static const Keyboard::Key deviceToKey[256] = {
	/* 0		1		2		3		4		5		6		7 */
	/* 8		9		A		B		C		D		E		F */
	// 00-0F
	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,
	Keyboard::Key::Backspace, Keyboard::Key::Tab,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Enter,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,
	// 10-1F
	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Pause,	Keyboard::Key::Capslock,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,
	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Escape,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,
	// 20-2F
	Keyboard::Key::Space,	Keyboard::Key::PageUp,	Keyboard::Key::PageDown,	Keyboard::Key::End,	Keyboard::Key::Home,	Keyboard::Key::Left,	Keyboard::Key::Up,	Keyboard::Key::Right,
	Keyboard::Key::Down,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Insert,	Keyboard::Key::Delete,	Keyboard::Key::Unknown,
	// 30-3F
	Keyboard::Key::Zero,	Keyboard::Key::One,	Keyboard::Key::Two,	Keyboard::Key::Three,	Keyboard::Key::Four,	Keyboard::Key::Five,	Keyboard::Key::Six,	Keyboard::Key::Seven,
	Keyboard::Key::Eight,	Keyboard::Key::Nine,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,
	// 40-4F
	Keyboard::Key::Unknown,	Keyboard::Key::A,		Keyboard::Key::B,		Keyboard::Key::C,		Keyboard::Key::D,		Keyboard::Key::E,		Keyboard::Key::F,		Keyboard::Key::G,
	Keyboard::Key::H,		Keyboard::Key::I,		Keyboard::Key::J,		Keyboard::Key::K,		Keyboard::Key::L,		Keyboard::Key::M,		Keyboard::Key::N,		Keyboard::Key::O,
	// 50-5F
	Keyboard::Key::P,		Keyboard::Key::Q,		Keyboard::Key::R,		Keyboard::Key::S,		Keyboard::Key::T,		Keyboard::Key::U,		Keyboard::Key::V,		Keyboard::Key::W,
	Keyboard::Key::X,		Keyboard::Key::Y,		Keyboard::Key::Z,		Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,
	// 60-6F
	Keyboard::Key::NUMPAD_0,	Keyboard::Key::NUMPAD_1,	Keyboard::Key::NUMPAD_2,	Keyboard::Key::NUMPAD_3,	Keyboard::Key::NUMPAD_4,	Keyboard::Key::NUMPAD_5,	Keyboard::Key::NUMPAD_6,	Keyboard::Key::NUMPAD_7,
	Keyboard::Key::NUMPAD_8,	Keyboard::Key::NUMPAD_9,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,
	// 70-7F
	Keyboard::Key::F1,	Keyboard::Key::F2,	Keyboard::Key::F3,	Keyboard::Key::F4,	Keyboard::Key::F5,	Keyboard::Key::F6,	Keyboard::Key::F7,	Keyboard::Key::F8,
	Keyboard::Key::F9,	Keyboard::Key::F10,	Keyboard::Key::F11,	Keyboard::Key::F12,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,
	// 80-8F
	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,
	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,
	// 90-9F
	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,
	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,
	// A0-AF
	Keyboard::Key::LShift,	Keyboard::Key::RShift,	Keyboard::Key::LCtrl,	Keyboard::Key::RCtrl,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,
	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,
	// B0-BF
	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,
	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Comma,	Keyboard::Key::Unknown,	Keyboard::Key::Period,	Keyboard::Key::Unknown,
	// C0-CF
	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,
	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,
	// E0-EF
	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,
	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,
	// F0-FF
	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,
	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,	Keyboard::Key::Unknown,
};

void Win32Keyboard::init()
{
	Win32Input::inputMsgBuffer.enabled = true;

	memset(m_btnState, 0, sizeof(m_btnState) / sizeof(m_btnState[0]));
}

bool Win32Keyboard::isKeyDown(Keyboard::Key key)
{
	return m_btnState[key];
}

void Win32Keyboard::update(InputModule *input)
{
	for(int i = 0; i < msg_buf.index; i++) {

		MSG msg = msg_buf.messages[i];

		if (msg.message == WM_KILLFOCUS) {

			for(int i = 0; i < Keyboard::Key::NUM_KEYS; i++) {

				if (m_btnState[i]) {
					InputEvent event(InputEvent::Key);
					event.key.code = (Keyboard::Key) i;
					event.key.pressed = msg.message == WM_KEYDOWN;

					m_btnState[i] = false;
					input->postInputEvent(event);
				}
			}

			continue;
		}

		if (msg.message != WM_KEYDOWN && msg.message != WM_KEYUP) {
			continue;
		}

		Keyboard::Key key = deviceToKey[msg.wParam % 0xFF];

		if (key != Keyboard::Key::Unknown) {
			InputEvent event(InputEvent::Key);
			event.key.code = key;
			event.key.pressed = msg.message == WM_KEYDOWN;

			m_btnState[key] = event.key.pressed;
			input->postInputEvent(event);
		}
	}

	msg_buf.index = 0;
}

bool Win32Keyboard::handleMessage(MSG message)
{
	return msg_buf.postMessage(message);
}

} // namespace sp