/*
 * =====================================================================================
 *
 *  OpenMiner
 *
 *  Copyright (C) 2018-2020 Unarelith, Quentin Bazin <openminer@unarelith.net>
 *  Copyright (C) 2019-2020 the OpenMiner contributors (see CONTRIBUTORS.md)
 *
 *  This file is part of OpenMiner.
 *
 *  OpenMiner is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU Lesser General Public
 *  License as published by the Free Software Foundation; either
 *  version 2.1 of the License, or (at your option) any later version.
 *
 *  OpenMiner is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 *  Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public License
 *  along with OpenMiner; if not, write to the Free Software Foundation, Inc.,
 *  51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
 *
 * =====================================================================================
 */
#include <fstream>

#include <gk/core/Debug.hpp>
#include <gk/core/Filesystem.hpp>

#include <sol/sol.hpp>

#include "GameKey.hpp"
#include "KeyboardHandler.hpp"

KeyboardHandler::KeyboardHandler() {
	addKey(GameKey::Left,            "Left",              SDLK_a);
	addKey(GameKey::Right,           "Right",             SDLK_d);
	addKey(GameKey::Forward,         "Forward",           SDLK_w);
	addKey(GameKey::Back,            "Back",              SDLK_s);

	addKey(GameKey::Jump,            "Jump",              SDLK_SPACE);
	addKey(GameKey::Fly,             "Fly",               SDLK_x);
	addKey(GameKey::Sneak,           "Sneak",             SDLK_LSHIFT);
	addKey(GameKey::Sprint,          "Sprint",            SDLK_LCTRL);

	addKey(GameKey::Chat,            "Chat",              SDLK_t);
	addKey(GameKey::Command,         "Command",           SDLK_KP_DIVIDE);

	addKey(GameKey::Shift,           "Shift",             SDLK_LSHIFT);

	addKey(GameKey::BlockInfoToggle, "Block info toggle", SDLK_p);
}

void KeyboardHandler::loadKeysFromFile(const std::string &filename) {
	if (gk::Filesystem::fileExists(filename)) {
		sol::state lua;

		try {
			lua.safe_script_file(filename);

			sol::optional<sol::table> keys = lua["keys"];
			if (keys != sol::nullopt) {
				for (auto &it : keys.value()) {
					const std::string &keyID = it.first.as<std::string>();
					const std::string &keyName = it.second.as<sol::table>()["name"].get<std::string>();
					const std::string &keyValue = it.second.as<sol::table>()["value"].get<std::string>();

					SDL_Keycode keycode = SDL_GetKeyFromName(keyValue.c_str());

					auto keyit = m_keysID.find(keyID);
					if (keyit != m_keysID.end()) {
						gk::GameKey key = keyit->second;
						m_keys[key].setKeycode(keycode);
						if (m_keys[key].keycode() == SDLK_UNKNOWN) {
							gkWarning() << "Key name '" + keyValue + "' not recognized";
						}
					}
					else {
						addKey(gk::GameKey(m_keys.size()), keyName, keycode, keyID);
					}
				}
			}

			gkInfo() << "Key mapping loaded successfully";
		}
		catch (sol::error &e) {
			gkError() << e.what();
		}
	}
}

void KeyboardHandler::saveKeysToFile(const std::string &filename) {
	std::ofstream file{filename, std::ofstream::out | std::ofstream::trunc};
	file << "keys = {" << std::endl;
	for (auto &it : m_keys) {
		file << "\t[\"" << it.second.stringID() << "\"] = {" << std::endl;
		file << "\t\tname = \"" + it.second.name() + "\"," << std::endl;
		file << "\t\tvalue = \"" << getKeyName(it.first) << "\"" << std::endl;
		file << "\t}," << std::endl;
	}
	file << "}" << std::endl;
}

bool KeyboardHandler::isKeyPressed(gk::GameKey key) {
	const u8 *keyboardState = SDL_GetKeyboardState(nullptr);
	SDL_Keycode keyScancode = m_keys[key].keycode();

	m_keysPressed[key] = keyboardState[SDL_GetScancodeFromKey(keyScancode)];

	return m_keysPressed[key];
}

void KeyboardHandler::addKey(gk::GameKey id, const std::string &name, SDL_Keycode defaultKey, const std::string &stringID, Key *key) {
	auto keyit = m_keysID.find(stringID);
	if (keyit == m_keysID.end()) {
		auto it = m_keys.emplace(id, Key((u16)id, (stringID.empty() ? "_" + name : stringID), name));
		it.first->second.setKeycode(defaultKey);

		m_keysID.emplace(it.first->second.stringID(), id);

		if (key) {
			key->setKeycode(it.first->second.keycode());
			it.first->second.setParent(key);
		}
	}
	else {
		Key &keyitem = m_keys.at(keyit->second);
		keyitem.setName(name);

		if (key) {
			key->setKeycode(keyitem.keycode());
			keyitem.setParent(key);
		}
	}

	InputHandler::addKey(id);
}