#include "StdAfx.h"
#include "PythonRanking.h"
#include "AbstractPlayer.h"
#include <memory>
#include <string>

void CPythonRanking::RegisterRankingData(const char* name, int level, BYTE job, BYTE empire, const char* guild)
{
	vRankingContainer.emplace_back(std::make_shared<SRankingData>(name, level, job, empire, guild));
}

void CPythonRanking::ClearRankData()
{
	vRankingContainer.clear();
	vRankingContainer.reserve(RANKING_MAX_NUM);
}

size_t CPythonRanking::GetRankCount() const
{
	return vRankingContainer.size();
}

CPythonRanking::SRankingData* CPythonRanking::GetRankByLine(WORD dwArrayIndex) const
{
	if (dwArrayIndex >= GetRankCount())
		return nullptr;

	return vRankingContainer.at(dwArrayIndex).get();
}

WORD CPythonRanking::GetRankMyLine() const
{
	auto it = std::find_if(vRankingContainer.begin(), vRankingContainer.end(), [] { return !r->name.compare(IAbstractPlayer::GetSingleton().GetName()); });
	if (it != vRankingContainer.end())
		return std::distance(vRankingContainer.begin(), it) + 1;

	return 0;
}

CPythonRanking::CPythonRanking()
{
	vRankingContainer.reserve(RANKING_MAX_NUM);
}

CPythonRanking::~CPythonRanking()
{
	vRankingContainer.clear();
}

/*Module*/

PyObject * rankingGetRankCount(PyObject * poSelf, PyObject * poArgs)
{
	return Py_BuildValue("i", CPythonRanking::Instance().GetRankCount());
}

PyObject * rankingGetRankByLine(PyObject * poSelf, PyObject * poArgs)
{
	int iIndex;
	if (!PyTuple_GetInteger(poArgs, 0, &iIndex))
		return Py_BadArgument();

	const auto Rank = CPythonRanking::Instance().GetRankByLine(iIndex);
	if (!Rank)
		return Py_BuildException("Failed to find rank by index %d", iIndex);

	return Py_BuildValue("siiis", Rank->name.c_str(), Rank->level, Rank->job, Rank->empire, Rank->guild.c_str());
}

PyObject* rankingGetRankMyLine(PyObject* poSelf, PyObject* poArgs)
{
	return Py_BuildValue("i", CPythonRanking::Instance().GetRankMyLine());
}

PyObject* rankingClear(PyObject* poSelf, PyObject* poArgs)
{
	CPythonRanking::Instance().ClearRankData();
	return Py_BuildNone();
}

void initranking()
{
	static PyMethodDef s_methods[] =
	{
		{ "GetRankCount", rankingGetRankCount, METH_VARARGS },
		{ "GetRankByLine", rankingGetRankByLine, METH_VARARGS },
		{ "GetRankMyLine", rankingGetRankMyLine, METH_VARARGS },
		{ "RankClear", rankingClear, METH_VARARGS },
		{ NULL, NULL, NULL }
	};

	PyObject * poModule = Py_InitModule("ranking", s_methods);
	PyModule_AddIntConstant(poModule, "RANK_PAGE_MAX_NUM", CPythonRanking::RANK_PAGE_MAX_NUM);
	PyModule_AddIntConstant(poModule, "RANK_SHOW_COUNT", CPythonRanking::RANK_SHOW_COUNT);
}