/*****************************************************************************
 * PokerTH - The open source texas holdem engine                             *
 * Copyright (C) 2006-2016 Felix Hammer, Florian Thauer, Lothar May          *
 *                                                                           *
 * This program is free software: you can redistribute it and/or modify      *
 * it under the terms of the GNU Affero General Public License as            *
 * published by the Free Software Foundation, either version 3 of the        *
 * License, or (at your option) any later version.                           *
 *                                                                           *
 * This program 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 Affero General Public License for more details.                       *
 *                                                                           *
 * You should have received a copy of the GNU Affero General Public License  *
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.     *
 *                                                                           *
 *                                                                           *
 * Additional permission under GNU AGPL version 3 section 7                  *
 *                                                                           *
 * If you modify this program, or any covered work, by linking or            *
 * combining it with the OpenSSL project's OpenSSL library (or a             *
 * modified version of that library), containing parts covered by the        *
 * terms of the OpenSSL or SSLeay licenses, the authors of PokerTH           *
 * (Felix Hammer, Florian Thauer, Lothar May) grant you additional           *
 * permission to convey the resulting work.                                  *
 * Corresponding Source for a non-source form of such a combination          *
 * shall include the source code for the parts of OpenSSL used as well       *
 * as that of the covered work.                                              *
 *****************************************************************************/

#include <dbofficial/serverdbthread.h>
#include <dbofficial/asyncdbauth.h>
#include <dbofficial/asyncdblogin.h>
#include <dbofficial/asyncdbavatarblacklist.h>
#include <dbofficial/asyncdbcreategame.h>
#include <dbofficial/asyncdbendgame.h>
#include <dbofficial/asyncdbgameplace.h>
#include <dbofficial/asyncdbupdatescore.h>
#include <dbofficial/asyncdbreportavatar.h>
#include <dbofficial/asyncdbreportgame.h>
#include <dbofficial/asyncdbadminplayers.h>
#include <dbofficial/asyncdbblockplayer.h>
#include <dbofficial/compositeasyncdbquery.h>
#include <dbofficial/db_table_defs.h>
#include <ctime>
#include <sstream>
#include <mysql++.h>

#define QUERY_NICK_PREPARE				"nick_template"
#define QUERY_LOGIN_PREPARE				"login_template"
#define QUERY_AVATAR_BLACKLIST_PREPARE	"avatar_blacklist_template"
#define QUERY_CREATE_GAME_PREPARE		"create_game_template"
#define QUERY_END_GAME_PREPARE			"end_game_template"
#define QUERY_GAME_PLAYER_PREPARE		"game_player_template"
#define QUERY_UPDATE_SCORE_PREPARE		"update_score_template"
#define QUERY_REPORT_AVATAR_PREPARE		"report_avatar_template"
#define QUERY_REPORT_GAME_PREPARE		"report_game_template"
#define QUERY_ADMIN_PLAYER_PREPARE		"admin_player_template"
#define QUERY_BLOCK_PLAYER_PREPARE		"block_player_template"

using namespace std;

struct DBConnectionData {
	DBConnectionData() : conn(false) {}
	string host;
	string user;
	string pwd;
	string database;
	string encryptionKey;
	mysqlpp::Connection conn;
};

ServerDBThread::ServerDBThread(ServerDBCallback &cb, boost::shared_ptr<boost::asio::io_service> ioService)
	: m_ioService(ioService), m_semaphore(0), m_callback(cb), m_isConnected(false), m_permanentError(false), m_previouslyConnected(false)
{
	m_connData.reset(new DBConnectionData);
}

ServerDBThread::~ServerDBThread()
{
}

void
ServerDBThread::SignalTermination()
{
	Thread::SignalTermination();
	m_semaphore.post();
}

void
ServerDBThread::Init(const string &host, const string &user, const string &pwd,
					 const string &database, const string &encryptionKey)
{
	m_connData->host = host;
	m_connData->user = user;
	m_connData->pwd = pwd;
	m_connData->database = database;
	m_connData->encryptionKey = encryptionKey;
}

void
ServerDBThread::Start()
{
	Run();
}

void
ServerDBThread::Stop()
{
	SignalTermination();
	this->Join(THREAD_WAIT_INFINITE);
}

void
ServerDBThread::AsyncPlayerLogin(unsigned requestId, const string &playerName)
{
	if (IsConnected()) {
		list<string> params;
		params.push_back(m_connData->encryptionKey);
		params.push_back(playerName);
		boost::shared_ptr<AsyncDBQuery> asyncQuery(
			new AsyncDBAuth(
				requestId,
				QUERY_NICK_PREPARE,
				params));

		{
			boost::mutex::scoped_lock lock(m_asyncQueueMutex);
			m_asyncQueue.push(asyncQuery);
		}
		m_semaphore.post();
	} else {
		// If not connected to database, login fails.
		m_ioService->post(boost::bind(&ServerDBCallback::PlayerLoginFailed, &m_callback, requestId));
	}
}

void
ServerDBThread::AsyncCheckAvatarBlacklist(unsigned requestId, const std::string &avatarHash)
{
	if (IsConnected()) {
		list<string> params;
		params.push_back(avatarHash);
		boost::shared_ptr<AsyncDBQuery> asyncQuery(
			new AsyncDBAvatarBlacklist(
				requestId,
				QUERY_AVATAR_BLACKLIST_PREPARE,
				params));

		{
			boost::mutex::scoped_lock lock(m_asyncQueueMutex);
			m_asyncQueue.push(asyncQuery);
		}
		m_semaphore.post();
	} else {
		// If not connected to database, all avatars are blacklisted.
		m_ioService->post(boost::bind(&ServerDBCallback::AvatarIsBlacklisted, &m_callback, requestId));
	}
}

void
ServerDBThread::PlayerPostLogin(DB_id playerId, const std::string &avatarHash, const std::string &avatarType)
{
	list<string> params;
	params.push_back(mysqlpp::DateTime(time(NULL)));
	params.push_back(avatarHash);
	params.push_back(avatarType);
	ostringstream paramStream;
	paramStream << playerId;
	params.push_back(paramStream.str());
	boost::shared_ptr<AsyncDBQuery> asyncQuery(
		new AsyncDBLogin(
			playerId,
			QUERY_LOGIN_PREPARE,
			params));

	{
		boost::mutex::scoped_lock lock(m_asyncQueueMutex);
		m_asyncQueue.push(asyncQuery);
	}
	m_semaphore.post();
}

void
ServerDBThread::PlayerLogout(DB_id /*playerId*/)
{
}

void
ServerDBThread::AsyncCreateGame(unsigned requestId, const string &gameName)
{
	list<string> params;
	params.push_back(gameName);
	params.push_back(mysqlpp::DateTime(time(NULL)));
	boost::shared_ptr<AsyncDBQuery> asyncQuery(
		new AsyncDBCreateGame(
			requestId,
			QUERY_CREATE_GAME_PREPARE,
			params));

	{
		boost::mutex::scoped_lock lock(m_asyncQueueMutex);
		m_asyncQueue.push(asyncQuery);
	}
	m_semaphore.post();
}

void
ServerDBThread::SetGamePlayerPlace(unsigned requestId, DB_id playerId, unsigned place)
{
	// The game id param is added later (during init of the async op), because it may be unknown.
	list<string> params;
	ostringstream paramStream;
	paramStream << playerId;
	params.push_back(paramStream.str());
	paramStream.str("");
	paramStream << place;
	params.push_back(paramStream.str());
	boost::shared_ptr<AsyncDBQuery> asyncQuery(
		new AsyncDBGamePlace(
			requestId,
			QUERY_GAME_PLAYER_PREPARE,
			params));

	{
		boost::mutex::scoped_lock lock(m_asyncQueueMutex);
		m_asyncQueue.push(asyncQuery);
	}
	m_semaphore.post();
}

void
ServerDBThread::EndGame(unsigned requestId)
{
	// Set the end time of the game.
	{
		list<string> params;
		params.push_back(mysqlpp::DateTime(time(NULL)));
		boost::shared_ptr<AsyncDBQuery> asyncQuery(
			new AsyncDBEndGame(
				requestId,
				QUERY_END_GAME_PREPARE,
				params));

		{
			boost::mutex::scoped_lock lock(m_asyncQueueMutex);
			m_asyncQueue.push(asyncQuery);
		}
		m_semaphore.post();
	}
	// Update the player scores.
	{
		list<string> params;
		boost::shared_ptr<AsyncDBQuery> asyncQuery(
			new AsyncDBUpdateScore(
				requestId,
				QUERY_UPDATE_SCORE_PREPARE,
				params));

		{
			boost::mutex::scoped_lock lock(m_asyncQueueMutex);
			m_asyncQueue.push(asyncQuery);
		}
		m_semaphore.post();
	}
}

void
ServerDBThread::AsyncReportAvatar(unsigned requestId, unsigned replyId, DB_id reportedPlayerId, const std::string &avatarHash, const std::string &avatarType, DB_id *byPlayerId)
{
	list<string> params;
	ostringstream paramStream;
	paramStream << reportedPlayerId;
	params.push_back(paramStream.str());
	params.push_back(avatarHash);
	params.push_back(avatarType);
	if (byPlayerId) {
		paramStream.str("");
		paramStream << *byPlayerId;
		params.push_back(paramStream.str());
	} else {
		params.push_back("NULL");
	}
	params.push_back(mysqlpp::DateTime(time(NULL)));

	boost::shared_ptr<AsyncDBQuery> asyncQuery(
		new AsyncDBReportAvatar(
			requestId,
			replyId,
			QUERY_REPORT_AVATAR_PREPARE,
			params));

	{
		boost::mutex::scoped_lock lock(m_asyncQueueMutex);
		m_asyncQueue.push(asyncQuery);
	}
	m_semaphore.post();
}

void
ServerDBThread::AsyncReportGame(unsigned requestId, unsigned replyId, DB_id *creatorPlayerId, unsigned gameId, const std::string &gameName, DB_id *byPlayerId)
{
	list<string> params;
	ostringstream paramStream;
	if (creatorPlayerId) {
		paramStream << *creatorPlayerId;
		params.push_back(paramStream.str());
	} else {
		params.push_back("NULL");
	}
	params.push_back(gameName);
	if (byPlayerId) {
		paramStream.str("");
		paramStream << *byPlayerId;
		params.push_back(paramStream.str());
	} else {
		params.push_back("NULL");
	}
	params.push_back(mysqlpp::DateTime(time(NULL)));

	boost::shared_ptr<AsyncDBQuery> asyncQuery(
		new AsyncDBReportGame(
			requestId,
			replyId,
			gameId,
			QUERY_REPORT_GAME_PREPARE,
			params));

	{
		boost::mutex::scoped_lock lock(m_asyncQueueMutex);
		m_asyncQueue.push(asyncQuery);
	}

	m_semaphore.post();
}

void
ServerDBThread::AsyncQueryAdminPlayers(unsigned requestId)
{
	boost::shared_ptr<AsyncDBQuery> asyncQuery(
		new AsyncDBAdminPlayers(
			requestId,
			QUERY_ADMIN_PLAYER_PREPARE));
	{
		boost::mutex::scoped_lock lock(m_asyncQueueMutex);
		m_asyncQueue.push(asyncQuery);
	}

	m_semaphore.post();
}

void
ServerDBThread::AsyncBlockPlayer(unsigned requestId, unsigned replyId, DB_id playerId, int valid, int active)
{
	list<string> params;
	ostringstream paramStream;
	paramStream << valid;
	params.push_back(paramStream.str());
	paramStream.str("");
	paramStream << active;
	params.push_back(paramStream.str());
	paramStream.str("");
	paramStream << playerId;
	params.push_back(paramStream.str());
	boost::shared_ptr<AsyncDBQuery> asyncQuery(
		new AsyncDBBlockPlayer(
			requestId,
			replyId,
			QUERY_BLOCK_PLAYER_PREPARE,
			params));

	{
		boost::mutex::scoped_lock lock(m_asyncQueueMutex);
		m_asyncQueue.push(asyncQuery);
	}
	m_semaphore.post();
}

bool
ServerDBThread::IsConnected() const
{
	boost::mutex::scoped_lock lock(m_isConnectedMutex);
	return m_isConnected;
}

void
ServerDBThread::Main()
{
	while (!ShouldTerminate() && !HasPermanentError()) {
		if (HasDBConnection()) {
			SetConnected(true);
			m_semaphore.wait();
			HandleNextQuery();
		} else {
			SetConnected(false);
			EstablishDBConnection();
		}
	}
	m_connData->conn.disconnect();
}

bool
ServerDBThread::HasPermanentError() const
{
	return m_permanentError;
}

bool
ServerDBThread::HasDBConnection() const
{
	return m_connData->conn.connected();
}

void
ServerDBThread::EstablishDBConnection()
{
	m_connData->conn.set_option(new mysqlpp::SetCharsetNameOption("utf8"));
	if (!m_connData->conn.connect(
				m_connData->database.c_str(), m_connData->host.c_str(), m_connData->user.c_str(), m_connData->pwd.c_str())) {
		m_ioService->post(boost::bind(&ServerDBCallback::ConnectFailed, &m_callback, m_connData->conn.error()));
		if (!m_previouslyConnected)
			m_permanentError = true;
		else
			Msleep(250);
	} else {
		mysqlpp::Query prepareNick = m_connData->conn.query();
		prepareNick
				<< "PREPARE " QUERY_NICK_PREPARE " FROM " << mysqlpp::quote
				<< "SELECT " DB_TABLE_PLAYER_COL_ID ", AES_DECRYPT(" DB_TABLE_PLAYER_COL_PASSWORD ", ?), " DB_TABLE_PLAYER_COL_VALID ", TRIM(" DB_TABLE_PLAYER_COL_COUNTRY "), " DB_TABLE_PLAYER_COL_LASTLOGIN ", " DB_TABLE_PLAYER_COL_ACTIVE " FROM " DB_TABLE_PLAYER " WHERE BINARY " DB_TABLE_PLAYER_COL_USERNAME " = ?";
		mysqlpp::Query prepareAvatarBlacklist = m_connData->conn.query();
		prepareAvatarBlacklist
				<< "PREPARE " QUERY_AVATAR_BLACKLIST_PREPARE " FROM " << mysqlpp::quote
				<< "SELECT " DB_TABLE_AVATAR_BLACKLIST_COL_ID " FROM " DB_TABLE_AVATAR_BLACKLIST " WHERE BINARY " DB_TABLE_AVATAR_BLACKLIST_COL_AVATAR_HASH " = ?";
		mysqlpp::Query prepareLogin = m_connData->conn.query();
		prepareLogin
				<< "PREPARE " QUERY_LOGIN_PREPARE " FROM " << mysqlpp::quote
				<< "UPDATE " DB_TABLE_PLAYER " SET " DB_TABLE_PLAYER_COL_LASTLOGIN " = ?, " DB_TABLE_PLAYER_COL_AVATARHASH " = ?, " DB_TABLE_PLAYER_COL_AVATARTYPE " = ? WHERE " DB_TABLE_PLAYER_COL_ID " = ?";
		mysqlpp::Query prepareCreateGame = m_connData->conn.query();
		prepareCreateGame
				<< "PREPARE " QUERY_CREATE_GAME_PREPARE " FROM " << mysqlpp::quote
				<< "INSERT INTO " DB_TABLE_GAME " (" DB_TABLE_GAME_COL_NAME ", " DB_TABLE_GAME_COL_STARTTIME ") VALUES (?, ?)";
		mysqlpp::Query prepareEndGame = m_connData->conn.query();
		prepareEndGame
				<< "PREPARE " QUERY_END_GAME_PREPARE " FROM " << mysqlpp::quote
				<< "UPDATE " DB_TABLE_GAME " SET "DB_TABLE_GAME_COL_ENDTIME " = ? WHERE " DB_TABLE_GAME_COL_ID " = ?";
		mysqlpp::Query prepareRelation = m_connData->conn.query();
		prepareRelation
				<< "PREPARE " QUERY_GAME_PLAYER_PREPARE " FROM " << mysqlpp::quote
				<< "INSERT INTO " DB_TABLE_GAMEPLAYER " (" DB_TABLE_GAMEPLAYER_COL_GAMEID ", " DB_TABLE_GAMEPLAYER_COL_PLAYERID ", " DB_TABLE_GAMEPLAYER_COL_PLACE ") VALUES (?, ?, ?)";
		mysqlpp::Query prepareScore = m_connData->conn.query();
		prepareScore
				<< "PREPARE " QUERY_UPDATE_SCORE_PREPARE " FROM " << mysqlpp::quote
				<< "CALL updatePointsForGame(?)";
		mysqlpp::Query prepareReportAvatar = m_connData->conn.query();
		prepareReportAvatar
				<< "PREPARE " QUERY_REPORT_AVATAR_PREPARE " FROM " << mysqlpp::quote
				<< "INSERT INTO " DB_TABLE_REP_AVATAR " (" DB_TABLE_REP_AVATAR_COL_PLAYERID ", " DB_TABLE_REP_AVATAR_COL_AVATARHASH ", " DB_TABLE_REP_AVATAR_COL_AVATARTYPE ", " DB_TABLE_REP_AVATAR_COL_BY_PLAYERID ", " DB_TABLE_REP_AVATAR_COL_TIMESTAMP ") VALUES (?, ?, ?, ?, ?)";
		mysqlpp::Query prepareReportGame = m_connData->conn.query();
		prepareReportGame
				<< "PREPARE " QUERY_REPORT_GAME_PREPARE " FROM " << mysqlpp::quote
				<< "INSERT INTO " DB_TABLE_REP_GAME " (" DB_TABLE_REP_GAME_COL_CREATOR ", " DB_TABLE_REP_GAME_COL_GAMENAME ", " DB_TABLE_REP_GAME_COL_BY_PLAYERID ", " DB_TABLE_REP_GAME_COL_TIMESTAMP ", " DB_TABLE_REP_GAME_COL_GAMEID ") VALUES (?, ?, ?, ?, ?)";
		mysqlpp::Query prepareAdminPlayer = m_connData->conn.query();
		prepareAdminPlayer
				<< "PREPARE " QUERY_ADMIN_PLAYER_PREPARE " FROM " << mysqlpp::quote
				<< "SELECT " DB_TABLE_ADMIN_PLAYER_COL_PLAYERID " FROM " DB_TABLE_ADMIN_PLAYER;
		mysqlpp::Query prepareBlockPlayer = m_connData->conn.query();
		prepareBlockPlayer
				<< "PREPARE " QUERY_BLOCK_PLAYER_PREPARE " FROM " << mysqlpp::quote
				<< "UPDATE " DB_TABLE_PLAYER " SET " DB_TABLE_PLAYER_COL_VALID " = ?, " DB_TABLE_PLAYER_COL_ACTIVE " = ? WHERE " DB_TABLE_PLAYER_COL_ID " = ?";

		if (!prepareNick.exec() || !prepareAvatarBlacklist.exec() || !prepareLogin.exec() || !prepareCreateGame.exec()
				|| !prepareEndGame.exec() || !prepareRelation.exec() || !prepareScore.exec() || !prepareReportAvatar.exec()
				|| !prepareReportGame.exec() || !prepareAdminPlayer.exec() || !prepareBlockPlayer.exec()) {
			m_connData->conn.disconnect();
			m_ioService->post(boost::bind(&ServerDBCallback::ConnectFailed, &m_callback,
										  string(prepareNick.error()) + prepareAvatarBlacklist.error() + prepareLogin.error() + prepareCreateGame.error()
										  + prepareEndGame.error() + prepareRelation.error() + prepareScore.error() + prepareReportAvatar.error()
										  + prepareReportGame.error() + prepareAdminPlayer.error() + prepareBlockPlayer.error()));
			m_permanentError = true;
		} else {
			m_ioService->post(boost::bind(&ServerDBCallback::ConnectSuccess, &m_callback));
			m_previouslyConnected = true;
		}
	}
}

void
ServerDBThread::HandleNextQuery()
{
	boost::shared_ptr<AsyncDBQuery> nextQuery;
	{
		boost::mutex::scoped_lock lock(m_asyncQueueMutex);
		if (!m_asyncQueue.empty()) {
			nextQuery = m_asyncQueue.front();
			m_asyncQueue.pop();
		}
	}
	if (nextQuery) {
		do {
			nextQuery->Init(m_dbIdManager);
			mysqlpp::Query executeQuery = m_connData->conn.query();
			executeQuery << "EXECUTE " << nextQuery->GetPreparedName();

			list<string> paramList;
			nextQuery->GetParams(paramList);
			if (!paramList.empty()) {
				executeQuery << " using ";
				mysqlpp::Query paramQuery = m_connData->conn.query();
				paramQuery << "SET ";
				unsigned counter = 1;
				list<string>::iterator i = paramList.begin();
				list<string>::iterator end = paramList.end();
				while (i != end) {
					if (counter > 1) {
						paramQuery << ", ";
						executeQuery << ", ";
					}
					paramQuery << "@param" << counter << " = ";
					if (*i == "NULL") {
						paramQuery << "NULL";
					} else {
						paramQuery << "_utf8" << mysqlpp::quote << *i;
					}
					executeQuery << "@param" << counter;
					++counter;
					++i;
				}
				if (!paramQuery.exec()) {
					m_connData->conn.disconnect();
					m_ioService->post(boost::bind(&ServerDBCallback::QueryError, &m_callback, paramQuery.error()));
					break;
				}
			}
			if (nextQuery->RequiresResultSet()) {
				mysqlpp::StoreQueryResult res = executeQuery.store();
				if (res)
					nextQuery->HandleResult(executeQuery, m_dbIdManager, res, *m_ioService, m_callback);
				else
					nextQuery->HandleError(*m_ioService, m_callback);
			} else {
				if (executeQuery.exec())
					nextQuery->HandleNoResult(executeQuery, m_dbIdManager, *m_ioService, m_callback);
				else
					nextQuery->HandleError(*m_ioService, m_callback);
			}
		} while (nextQuery->Next()); // Consider composite queries.
	}
}

void
ServerDBThread::SetConnected(bool isConnected)
{
	boost::mutex::scoped_lock lock(m_isConnectedMutex);
	m_isConnected = isConnected;
}