/**

 * vim: set ts=4 :

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

 * SourceMod SDKTools Extension

 * Copyright (C) 2004-2008 AlliedModders LLC.  All rights reserved.

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

 *

 * This program is free software; you can redistribute it and/or modify it under

 * the terms of the GNU General Public License, version 3.0, as published by the

 * Free Software Foundation.

 *

 * 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 General Public License for more

 * details.

 *

 * You should have received a copy of the GNU General Public License along with

 * this program.  If not, see <http://www.gnu.org/licenses/>.

 *

 * As a special exception, AlliedModders LLC gives you permission to link the

 * code of this program (as well as its derivative works) to "Half-Life 2," the

 * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software

 * by the Valve Corporation.  You must obey the GNU General Public License in

 * all respects for all other code used.  Additionally, AlliedModders LLC grants

 * this exception to all derivative works.  AlliedModders LLC defines further

 * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),

 * or <http://www.sourcemod.net/license.php>.

 *

 * Version: $Id$

 */

 #include "hooks.h"

 #include "extension.h"

 #include "basehandle.h"

 #include "vector.h"

 #include "utlvector.h"

 #include <shareddefs.h>

 #include "usercmd.h"

 #define FEATURECAP_PLAYERRUNCMD_11PARAMS	"SDKTools PlayerRunCmd 11Params"

 CHookManager g_Hooks;

 static bool PRCH_enabled = false;

 static bool PRCH_used = false;

 SH_DECL_MANUALHOOK2_void(PlayerRunCmdHook, 0, 0, 0, CUserCmd *, IMoveHelper *);

 CHookManager::CHookManager()

 {

 	m_usercmdsFwd = NULL;

 }

 void CHookManager::Initialize()

 {

 	int offset;

 	if (g_pGameConf->GetOffset("PlayerRunCmd", &offset))

 	{

 		SH_MANUALHOOK_RECONFIGURE(PlayerRunCmdHook, offset, 0, 0);

 		PRCH_enabled = true;

 	}

 	else

 	{

 		g_pSM->LogError(myself, "Failed to find PlayerRunCmd offset - OnPlayerRunCmd forward disabled.");

 		PRCH_enabled = false;

 	}

 	plsys->AddPluginsListener(this);

 	sharesys->AddCapabilityProvider(myself, this, FEATURECAP_PLAYERRUNCMD_11PARAMS);

 	m_usercmdsFwd = forwards->CreateForward("OnPlayerRunCmd", ET_Event, 11, NULL,

 		Param_Cell,			// client

 		Param_CellByRef,	// buttons

 		Param_CellByRef,	// impulse

 		Param_Array,		// Float:vel[3]

 		Param_Array,		// Float:angles[3]

 		Param_CellByRef,	// weapon

 		Param_CellByRef,	// subtype

 		Param_CellByRef,	// cmdnum

 		Param_CellByRef,	// tickcount

 		Param_CellByRef,	// seed

 		Param_Array);		// mouse[2]

 }

 void CHookManager::Shutdown()

 {

 	forwards->ReleaseForward(m_usercmdsFwd);

 	plsys->RemovePluginsListener(this);

 	sharesys->DropCapabilityProvider(myself, this, FEATURECAP_PLAYERRUNCMD_11PARAMS);

 }

 void CHookManager::OnClientPutInServer(int client)

 {

 	if (!PRCH_enabled)

 		return;

 	if (!PRCH_used)

 		return;

 	edict_t *pEdict = PEntityOfEntIndex(client);

 	if (!pEdict)

 	{

 		return;

 	}

 	IServerUnknown *pUnknown = pEdict->GetUnknown();

 	if (!pUnknown)

 	{

 		return;

 	}

 	CBaseEntity *pEntity = pUnknown->GetBaseEntity();

 	if (!pEntity)

 	{

 		return;

 	}

 	CVTableHook hook(pEntity);

 	for (size_t i = 0; i < m_runUserCmdHooks.length(); ++i)

 	{

 		if (hook == m_runUserCmdHooks[i])

 		{

 			return;

 		}

 	}

 	int hookid = SH_ADD_MANUALVPHOOK(PlayerRunCmdHook, pEntity, SH_MEMBER(this, &CHookManager::PlayerRunCmd), false);

 	hook.SetHookID(hookid);

 	m_runUserCmdHooks.append(new CVTableHook(hook));

 }

 void CHookManager::PlayerRunCmd(CUserCmd *ucmd, IMoveHelper *moveHelper)

 {

 	if (m_usercmdsFwd->GetFunctionCount() == 0)

 	{

 		RETURN_META(MRES_IGNORED);

 	}

 	CBaseEntity *pEntity = META_IFACEPTR(CBaseEntity);

 	if (!pEntity)

 	{

 		RETURN_META(MRES_IGNORED);

 	}

 	edict_t *pEdict = gameents->BaseEntityToEdict(pEntity);

 	if (!pEdict)

 	{

 		RETURN_META(MRES_IGNORED);

 	}

 	int client = IndexOfEdict(pEdict);

 	cell_t result = 0;

 	/* Impulse is a byte so we copy it back manually */

 	cell_t impulse = ucmd->impulse;

 	cell_t vel[3] = {sp_ftoc(ucmd->forwardmove), sp_ftoc(ucmd->sidemove), sp_ftoc(ucmd->upmove)};

 	cell_t angles[3] = {sp_ftoc(ucmd->viewangles.x), sp_ftoc(ucmd->viewangles.y), sp_ftoc(ucmd->viewangles.z)};

 	cell_t mouse[2] = {ucmd->mousedx, ucmd->mousedy};

 	m_usercmdsFwd->PushCell(client);

 	m_usercmdsFwd->PushCellByRef(&ucmd->buttons);

 	m_usercmdsFwd->PushCellByRef(&impulse);

 	m_usercmdsFwd->PushArray(vel, 3, SM_PARAM_COPYBACK);

 	m_usercmdsFwd->PushArray(angles, 3, SM_PARAM_COPYBACK);

 	m_usercmdsFwd->PushCellByRef(&ucmd->weaponselect);

 	m_usercmdsFwd->PushCellByRef(&ucmd->weaponsubtype);

 	m_usercmdsFwd->PushCellByRef(&ucmd->command_number);

 	m_usercmdsFwd->PushCellByRef(&ucmd->tick_count);

 	m_usercmdsFwd->PushCellByRef(&ucmd->random_seed);

 	m_usercmdsFwd->PushArray(mouse, 2, SM_PARAM_COPYBACK);

 	m_usercmdsFwd->Execute(&result);

 	ucmd->impulse = impulse;

 	ucmd->forwardmove = sp_ctof(vel[0]);

 	ucmd->sidemove = sp_ctof(vel[1]);

 	ucmd->upmove = sp_ctof(vel[2]);

 	ucmd->viewangles.x = sp_ctof(angles[0]);

 	ucmd->viewangles.y = sp_ctof(angles[1]);

 	ucmd->viewangles.z = sp_ctof(angles[2]);

 	ucmd->mousedx = mouse[0];

 	ucmd->mousedy = mouse[1];

 	if (result == Pl_Handled)

 	{

 		RETURN_META(MRES_SUPERCEDE);

 	}

 	RETURN_META(MRES_IGNORED);

 }

 void CHookManager::OnPluginLoaded(IPlugin *plugin)

 {

 	if (!PRCH_enabled)

 		return;

 	if (PRCH_used)

 		return;

 	if (!m_usercmdsFwd->GetFunctionCount())

 		return;

 	PRCH_used = true;

 	int MaxClients = playerhelpers->GetMaxClients();

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

 	{

 		if (playerhelpers->GetGamePlayer(i)->IsInGame())

 		{

 			OnClientPutInServer(i);

 		}

 	}

 }

 void CHookManager::OnPluginUnloaded(IPlugin *plugin)

 {

 	if (!PRCH_enabled)

 		return;

 	if (!PRCH_used)

 		return;

 	if (m_usercmdsFwd->GetFunctionCount())

 		return;

 	for (size_t i = 0; i < m_runUserCmdHooks.length(); ++i)

 	{

 		delete m_runUserCmdHooks[i];

 	}

 	m_runUserCmdHooks.clear();

 	PRCH_used = false;

 }

 FeatureStatus CHookManager::GetFeatureStatus(FeatureType type, const char *name)

 {

 	return FeatureStatus_Available;

 }