/**

 * vim: set ts=4 :

 * ===============================================================

 * SourceMod (C)2004-2007 AlliedModders LLC.  All rights reserved.

 * ===============================================================

 *

 * This file is not open source and may not be copied without explicit

 * written permission of AlliedModders LLC.  This file may not be redistributed 

 * in whole or significant part.

 * For information, see LICENSE.txt or http://www.sourcemod.net/license.php

 *

 * Version: $Id$

 */

#include "PlayerManager.h"

#include "ForwardSys.h"

#include "ShareSys.h"

#include "AdminCache.h"

#include "ConCmdManager.h"

#include "MenuStyle_Valve.h"

#include "MenuStyle_Radio.h"

#include "sm_stringutil.h"

PlayerManager g_Players;

bool g_OnMapStarted = false;

SH_DECL_HOOK5(IServerGameClients, ClientConnect, SH_NOATTRIB, 0, bool, edict_t *, const char *, const char *, char *, int);

SH_DECL_HOOK2_void(IServerGameClients, ClientPutInServer, SH_NOATTRIB, 0, edict_t *, const char *);

SH_DECL_HOOK1_void(IServerGameClients, ClientDisconnect, SH_NOATTRIB, 0, edict_t *);

SH_DECL_HOOK1_void(IServerGameClients, ClientCommand, SH_NOATTRIB, 0, edict_t *);

SH_DECL_HOOK1_void(IServerGameClients, ClientSettingsChanged, SH_NOATTRIB, 0, edict_t *);

SH_DECL_HOOK3_void(IServerGameDLL, ServerActivate, SH_NOATTRIB, 0, edict_t *, int, int);

PlayerManager::PlayerManager()

{

	m_AuthQueue = NULL;

	m_FirstPass = true;

	m_UserIdLookUp = new int[USHRT_MAX];

	memset(m_UserIdLookUp, 0, sizeof(int) * USHRT_MAX);

}

PlayerManager::~PlayerManager()

{

	delete [] m_AuthQueue;

	delete [] m_UserIdLookUp;

}

void PlayerManager::OnSourceModAllInitialized()

{

	SH_ADD_HOOK_MEMFUNC(IServerGameClients, ClientConnect, serverClients, this, &PlayerManager::OnClientConnect, false);

	SH_ADD_HOOK_MEMFUNC(IServerGameClients, ClientConnect, serverClients, this, &PlayerManager::OnClientConnect_Post, true);

	SH_ADD_HOOK_MEMFUNC(IServerGameClients, ClientPutInServer, serverClients, this, &PlayerManager::OnClientPutInServer, true);

	SH_ADD_HOOK_MEMFUNC(IServerGameClients, ClientDisconnect, serverClients, this, &PlayerManager::OnClientDisconnect, false);

	SH_ADD_HOOK_MEMFUNC(IServerGameClients, ClientDisconnect, serverClients, this, &PlayerManager::OnClientDisconnect_Post, true);

	SH_ADD_HOOK_MEMFUNC(IServerGameClients, ClientCommand, serverClients, this, &PlayerManager::OnClientCommand, false);

	SH_ADD_HOOK_MEMFUNC(IServerGameClients, ClientSettingsChanged, serverClients, this, &PlayerManager::OnClientSettingsChanged, true);

	SH_ADD_HOOK_MEMFUNC(IServerGameDLL, ServerActivate, gamedll, this, &PlayerManager::OnServerActivate, true);

	g_ShareSys.AddInterface(NULL, this);

	ParamType p1[] = {Param_Cell, Param_String, Param_Cell};

	ParamType p2[] = {Param_Cell};

	m_clconnect = g_Forwards.CreateForward("OnClientConnect", ET_Event, 3, p1);

	m_clputinserver = g_Forwards.CreateForward("OnClientPutInServer", ET_Ignore, 1, p2);

	m_cldisconnect = g_Forwards.CreateForward("OnClientDisconnect", ET_Ignore, 1, p2);

	m_cldisconnect_post = g_Forwards.CreateForward("OnClientDisconnect_Post", ET_Ignore, 1, p2);

	m_clcommand = g_Forwards.CreateForward("OnClientCommand", ET_Hook, 2, NULL, Param_Cell, Param_Cell);

	m_clinfochanged = g_Forwards.CreateForward("OnClientSettingsChanged", ET_Ignore, 1, p2);

	m_clauth = g_Forwards.CreateForward("OnClientAuthorized", ET_Ignore, 2, NULL, Param_Cell, Param_String);

	m_onActivate = g_Forwards.CreateForward("OnServerLoad", ET_Ignore, 0, NULL);

	m_onActivate2 = g_Forwards.CreateForward("OnMapStart", ET_Ignore, 0, NULL);

}

void PlayerManager::OnSourceModShutdown()

{

	SH_REMOVE_HOOK_MEMFUNC(IServerGameClients, ClientConnect, serverClients, this, &PlayerManager::OnClientConnect, false);

	SH_REMOVE_HOOK_MEMFUNC(IServerGameClients, ClientPutInServer, serverClients, this, &PlayerManager::OnClientPutInServer, true);

	SH_REMOVE_HOOK_MEMFUNC(IServerGameClients, ClientDisconnect, serverClients, this, &PlayerManager::OnClientDisconnect, false);

	SH_REMOVE_HOOK_MEMFUNC(IServerGameClients, ClientDisconnect, serverClients, this, &PlayerManager::OnClientDisconnect_Post, true);

	SH_REMOVE_HOOK_MEMFUNC(IServerGameClients, ClientCommand, serverClients, this, &PlayerManager::OnClientCommand, false);

	SH_REMOVE_HOOK_MEMFUNC(IServerGameClients, ClientSettingsChanged, serverClients, this, &PlayerManager::OnClientSettingsChanged, true);

	SH_REMOVE_HOOK_MEMFUNC(IServerGameDLL, ServerActivate, gamedll, this, &PlayerManager::OnServerActivate, true);

	/* Release forwards */

	g_Forwards.ReleaseForward(m_clconnect);

	g_Forwards.ReleaseForward(m_clputinserver);

	g_Forwards.ReleaseForward(m_cldisconnect);

	g_Forwards.ReleaseForward(m_cldisconnect_post);

	g_Forwards.ReleaseForward(m_clcommand);

	g_Forwards.ReleaseForward(m_clinfochanged);

	g_Forwards.ReleaseForward(m_clauth);

	g_Forwards.ReleaseForward(m_onActivate);

	g_Forwards.ReleaseForward(m_onActivate2);

	delete [] m_Players;

}

void PlayerManager::OnServerActivate(edict_t *pEdictList, int edictCount, int clientMax)

{

	if (m_FirstPass)

	{

		/* Initialize all players */

		m_maxClients = clientMax;

		m_PlayerCount = 0;

		m_Players = new CPlayer[m_maxClients + 1];

		m_AuthQueue = new unsigned int[m_maxClients + 1];

		m_FirstPass = false;

		memset(m_AuthQueue, 0, sizeof(unsigned int) * (m_maxClients + 1));

	}

	m_onActivate->Execute(NULL);

	m_onActivate2->Execute(NULL);

	g_OnMapStarted = true;

}

bool CheckSetAdmin(int index, CPlayer *pPlayer, AdminId id)

{

	const char *password = g_Admins.GetAdminPassword(id);

	if (password != NULL)

	{

		/* Whoa... the user needs a password! */

		const char *given = engine->GetClientConVarValue(index, "_password");

		if (!given || strcmp(given, password) != 0)

		{

			return false;

		}

	}

	pPlayer->SetAdminId(id, false);

	return true;

}

void PlayerManager::RunAuthChecks()

{

	CPlayer *pPlayer;

	const char *authstr;

	unsigned int removed = 0;

	for (unsigned int i=1; i<=m_AuthQueue[0]; i++)

	{

		pPlayer = GetPlayerByIndex(m_AuthQueue[i]);

		authstr = engine->GetPlayerNetworkIDString(pPlayer->m_pEdict);

		if (authstr && authstr[0] != '\0'

			&& (strcmp(authstr, "STEAM_ID_PENDING") != 0))

		{

			/* Set authorization */

			pPlayer->m_AuthID.assign(authstr);

			pPlayer->m_IsAuthorized = true;

			/* Mark as removed from queue */

			unsigned int client = m_AuthQueue[i];

			m_AuthQueue[i] = 0;

			removed++;

			/* Do admin lookups */

			AdminId id = g_Admins.FindAdminByIdentity("steam", authstr);

			if (id != INVALID_ADMIN_ID)

			{

				CheckSetAdmin(client, pPlayer, id);

			}

			/* Send to extensions */

			List<IClientListener *>::iterator iter;

			IClientListener *pListener;

			for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)

			{

				pListener = (*iter);

				pListener->OnClientAuthorized(client, authstr);

				if (!pPlayer->IsConnected())

				{

					break;

				}

			}

			/* Send to plugins if player is still connected */

			if (pPlayer->IsConnected() && m_clauth->GetFunctionCount())

			{

				/* :TODO: handle the case of a player disconnecting in the middle */

				m_clauth->PushCell(client);

				m_clauth->PushString(authstr);

				m_clauth->Execute(NULL);

			}

		}

	}

	/* Clean up the queue */

	if (removed)

	{

		/* We don't have to compact the list if it's empty */

		if (removed != m_AuthQueue[0])

		{

			unsigned int diff = 0;

			for (unsigned int i=1; i<=m_AuthQueue[0]; i++)

			{

				/* If this member is removed... */

				if (m_AuthQueue[i] == 0)

				{

					/* Increase the differential */

					diff++;

				} else {

					/* diff cannot increase faster than i+1 */

					assert(i > diff);

					assert(i - diff >= 1);

					/* move this index down */

					m_AuthQueue[i - diff] = m_AuthQueue[i];

				}

			}

			m_AuthQueue[0] -= removed;

		} else {

			m_AuthQueue[0] = 0;

			g_SourceMod.SetAuthChecking(false);

		}

	}

}

bool PlayerManager::OnClientConnect(edict_t *pEntity, const char *pszName, const char *pszAddress, char *reject, int maxrejectlen)

{

	int client = engine->IndexOfEdict(pEntity);

	List<IClientListener *>::iterator iter;

	IClientListener *pListener = NULL;

	for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)

	{

		pListener = (*iter);

		if (!pListener->InterceptClientConnect(client, reject, maxrejectlen))

		{

			return false;

		}

	}

	cell_t res = 1;

	m_Players[client].Initialize(pszName, pszAddress, pEntity);

	m_clconnect->PushCell(client);

	m_clconnect->PushStringEx(reject, maxrejectlen, SM_PARAM_STRING_UTF8, SM_PARAM_COPYBACK);

	m_clconnect->PushCell(maxrejectlen);

	m_clconnect->Execute(&res, NULL);

	if (res)

	{

		if (!m_Players[client].IsAuthorized())

		{

			m_AuthQueue[++m_AuthQueue[0]] = client;

			g_SourceMod.SetAuthChecking(true);

		}

	} else {

		RETURN_META_VALUE(MRES_SUPERCEDE, false);

	}

	m_UserIdLookUp[engine->GetPlayerUserId(pEntity)] = client;

	return true;

}

bool PlayerManager::OnClientConnect_Post(edict_t *pEntity, const char *pszName, const char *pszAddress, char *reject, int maxrejectlen)

{

	int client = engine->IndexOfEdict(pEntity);

	bool orig_value = META_RESULT_ORIG_RET(bool);

	CPlayer *pPlayer = GetPlayerByIndex(client);

	if (orig_value)

	{

		List<IClientListener *>::iterator iter;

		IClientListener *pListener = NULL;

		for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)

		{

			pListener = (*iter);

			pListener->OnClientConnected(client);

			if (!pPlayer->IsConnected())

			{

				break;

			}

		}

	}

	if (pPlayer->IsConnected())

	{

		/* Do an ip based lookup */

		char ip[24], *ptr;

		strncopy(ip, pszAddress, sizeof(ip));

		if ((ptr = strchr(ip, ':')) != NULL)

		{

			*ptr = '\0';

		}

		AdminId id = g_Admins.FindAdminByIdentity("ip", pPlayer->GetIPAddress());

		if (id != INVALID_ADMIN_ID)

		{

			CheckSetAdmin(client, pPlayer, id);

		}

	}

	return true;

}

void PlayerManager::OnClientPutInServer(edict_t *pEntity, const char *playername)

{

	cell_t res;

	int client = engine->IndexOfEdict(pEntity);

	CPlayer *pPlayer = GetPlayerByIndex(client);

	/* If they're not connected, they're a bot */

	if (!pPlayer->IsConnected())

	{

		/* Run manual connection routines */

		char error[255];

		const char *authid = engine->GetPlayerNetworkIDString(pEntity);

		pPlayer->Authorize(authid);

		if (!OnClientConnect(pEntity, playername, "127.0.0.1", error, sizeof(error)))

		{

			/* :TODO: kick the bot if it's rejected */

			return;

		}

		List<IClientListener *>::iterator iter;

		IClientListener *pListener = NULL;

		for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)

		{

			pListener = (*iter);

			pListener->OnClientConnected(client);

			/* See if bot was kicked */

			if (!pPlayer->IsConnected())

			{

				return;

			}

		}

		/* Now do authorization */

		for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)

		{

			pListener = (*iter);

			pListener->OnClientAuthorized(client, authid);

			/* See if bot was kicked */

			if (!pPlayer->IsConnected())

			{

				return;

			}

		}

		/* Finally, tell plugins */

		if (m_clauth->GetFunctionCount())

		{

			m_clauth->PushCell(client);

			m_clauth->PushString(authid);

			m_clauth->Execute(NULL);

		}

	}

	if (playerinfo)

	{

		pPlayer->m_Info = playerinfo->GetPlayerInfo(pEntity);

	}

	List<IClientListener *>::iterator iter;

	IClientListener *pListener = NULL;

	for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)

	{

		pListener = (*iter);

		pListener->OnClientPutInServer(client);

		/* See if player was kicked */

		if (!pPlayer->IsConnected())

		{

			return;

		}

	}

	m_Players[client].Connect();

	m_PlayerCount++;

	m_clputinserver->PushCell(client);

	m_clputinserver->Execute(&res, NULL);

}

void PlayerManager::OnSourceModLevelEnd()

{

	/* Disconnect all bots still in game */

	for (int i=1; i<=m_maxClients; i++)

	{

		if (m_Players[i].IsConnected() && m_Players[i].IsFakeClient())

		{

			OnClientDisconnect(m_Players[i].GetEdict());

		}

	}

}

void PlayerManager::OnClientDisconnect(edict_t *pEntity)

{

	cell_t res;

	int client = engine->IndexOfEdict(pEntity);

	if (m_Players[client].IsConnected())

	{

		m_cldisconnect->PushCell(client);

		m_cldisconnect->Execute(&res, NULL);

	}

	if (m_Players[client].IsInGame())

	{

		m_PlayerCount--;

	}

	List<IClientListener *>::iterator iter;

	IClientListener *pListener = NULL;

	for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)

	{

		pListener = (*iter);

		pListener->OnClientDisconnecting(client);

	}

	/**

	 * Remove client from auth queue if necessary

	 */

	if (!m_Players[client].IsAuthorized())

	{

		for (unsigned int i=1; i<=m_AuthQueue[0]; i++)

		{

			if (m_AuthQueue[i] == (unsigned)client)

			{

				/* Move everything ahead of us back by one */

				for (unsigned int j=i+1; j<=m_AuthQueue[0]; j++)

				{

					m_AuthQueue[j-1] = m_AuthQueue[j];

				}

				/* Remove us and break */

				m_AuthQueue[0]--;

				break;

			}

		}

	}

	m_Players[client].Disconnect();

	m_UserIdLookUp[engine->GetPlayerUserId(pEntity)] = 0;

}

void PlayerManager::OnClientDisconnect_Post(edict_t *pEntity)

{

	cell_t res;

	int client = engine->IndexOfEdict(pEntity);

	m_cldisconnect_post->PushCell(client);

	m_cldisconnect_post->Execute(&res, NULL);

	List<IClientListener *>::iterator iter;

	IClientListener *pListener = NULL;

	for (iter=m_hooks.begin(); iter!=m_hooks.end(); iter++)

	{

		pListener = (*iter);

		pListener->OnClientDisconnected(client);

	}

}

void PlayerManager::OnClientCommand(edict_t *pEntity)

{

	int client = engine->IndexOfEdict(pEntity);

	cell_t res = Pl_Continue;

	bool result = g_ValveMenuStyle.OnClientCommand(client);

	if (result)

	{

		res = Pl_Handled;

	} else {

		result = g_RadioMenuStyle.OnClientCommand(client);

		if (result)

		{

			res = Pl_Handled;

		}

	}

	int args = engine->Cmd_Argc() - 1;

	cell_t res2 = Pl_Continue;

	m_clcommand->PushCell(client);

	m_clcommand->PushCell(args);

	m_clcommand->Execute(&res2, NULL);

	if (res2 > res)

	{

		res = res2;

	}

	if (res >= Pl_Stop)

	{

		RETURN_META(MRES_SUPERCEDE);

	}

	res = g_ConCmds.DispatchClientCommand(client, (ResultType)res);

	if (res >= Pl_Handled)

	{

		RETURN_META(MRES_SUPERCEDE);

	}

}

void PlayerManager::OnClientSettingsChanged(edict_t *pEntity)

{

	cell_t res;

	int client = engine->IndexOfEdict(pEntity);

	m_clinfochanged->PushCell(engine->IndexOfEdict(pEntity));

	m_clinfochanged->Execute(&res, NULL);

	m_Players[client].SetName(engine->GetClientConVarValue(client, "name"));

}

int PlayerManager::GetMaxClients()

{

	return m_maxClients;

}

CPlayer *PlayerManager::GetPlayerByIndex(int client) const

{

	if (client > m_maxClients || client < 1)

	{

		return NULL;

	}

	return &m_Players[client];

}

int PlayerManager::GetNumPlayers()

{

	return m_PlayerCount;

}

int PlayerManager::GetClientOfUserId(int userid)

{

	if (userid < 0 || userid > USHRT_MAX)

	{

		return 0;

	}

	int client = m_UserIdLookUp[userid];

	/* Verify the userid.  The cache can get messed up with older

	 * Valve engines.  :TODO: If this gets fixed, do an old engine 

	 * check before invoking this backwards compat code.

	 */

	if (client)

	{

		CPlayer *player = GetPlayerByIndex(client);

		if (player && player->IsConnected())

		{

			int realUserId = engine->GetPlayerUserId(player->GetEdict());

			if (realUserId == userid)