/**

 * vim: set ts=4 :

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

 * SourceMod

 * Copyright (C) 2004-2007 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 "ConVarManager.h"

#include "HalfLife2.h"

#include "PluginSys.h"

#include "ForwardSys.h"

#include "HandleSys.h"

#include "sm_srvcmds.h"

#include "sm_stringutil.h"

#include <sh_vector.h>

#include "PlayerManager.h"

ConVarManager g_ConVarManager;

SH_DECL_HOOK5_void(IServerGameDLL, OnQueryCvarValueFinished, SH_NOATTRIB, 0, QueryCvarCookie_t, edict_t *, EQueryCvarValueStatus, const char *, const char *);

SH_DECL_HOOK5_void(IServerPluginCallbacks, OnQueryCvarValueFinished, SH_NOATTRIB, 0, QueryCvarCookie_t, edict_t *, EQueryCvarValueStatus, const char *, const char *);

const ParamType CONVARCHANGE_PARAMS[] = {Param_Cell, Param_String, Param_String};

typedef List<const ConVar *> ConVarList;

ConVarManager::ConVarManager() : m_ConVarType(0), m_VSPIface(NULL), m_CanQueryConVars(false)

{

#if PLAPI_VERSION < 12

	m_IgnoreHandle = false;

#endif

	/* Create a convar lookup trie */

	m_ConVarCache = sm_trie_create();

}

ConVarManager::~ConVarManager()

{

	List<ConVarInfo *>::iterator iter;

	/* Destroy our convar lookup trie */

	sm_trie_destroy(m_ConVarCache);

	/* Destroy all the ConVarInfo structures */

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

	{

		delete (*iter);

	}

	m_ConVars.clear();

}

void ConVarManager::OnSourceModAllInitialized()

{

	/* Only valid with ServerGameDLL006 or greater */

	if (g_SMAPI->GetGameDLLVersion() >= 6)

	{

		SH_ADD_HOOK_MEMFUNC(IServerGameDLL, OnQueryCvarValueFinished, gamedll, this, &ConVarManager::OnQueryCvarValueFinished, false);

		m_CanQueryConVars = true;

	}

	HandleAccess sec;

	/* Set up access rights for the 'ConVar' handle type */

	sec.access[HandleAccess_Read] = 0;

	sec.access[HandleAccess_Delete] = HANDLE_RESTRICT_IDENTITY | HANDLE_RESTRICT_OWNER;

	sec.access[HandleAccess_Clone] = HANDLE_RESTRICT_IDENTITY | HANDLE_RESTRICT_OWNER;

	/* Create the 'ConVar' handle type */

	m_ConVarType = g_HandleSys.CreateType("ConVar", this, 0, NULL, &sec, g_pCoreIdent, NULL);

	/* Add the 'convars' option to the 'sm' console command */

	g_RootMenu.AddRootConsoleCommand("cvars", "View convars created by a plugin", this);

}

void ConVarManager::OnSourceModShutdown()

{

	if (m_CanQueryConVars)

	{

		SH_REMOVE_HOOK_MEMFUNC(IServerGameDLL, OnQueryCvarValueFinished, gamedll, this, &ConVarManager::OnQueryCvarValueFinished, false);

		SH_REMOVE_HOOK_MEMFUNC(IServerPluginCallbacks, OnQueryCvarValueFinished, m_VSPIface, this, &ConVarManager::OnQueryCvarValueFinished, false);

	}

	IChangeableForward *fwd;

	List<ConVarInfo *>::iterator i;

	/* Iterate list of ConVarInfo structures */

	for (i = m_ConVars.begin(); i != m_ConVars.end(); i++)

	{

		fwd = (*i)->pChangeForward;

		/* Free any convar-change forwards that still exist */

		if (fwd)

		{

			g_Forwards.ReleaseForward(fwd);

		}

	}

	/* Remove the 'convars' option from the 'sm' console command */

	g_RootMenu.RemoveRootConsoleCommand("cvars", this);

	/* Remove the 'ConVar' handle type */

	g_HandleSys.RemoveType(m_ConVarType, g_pCoreIdent);

}

void ConVarManager::OnSourceModVSPReceived(IServerPluginCallbacks *iface)

{

	/* This will be called after OnSourceModAllInitialized(), so...

	 *

	 * If we haven't been able to hook the IServerGameDLL version at this point,

	 * then use hook IServerPluginCallbacks version from the engine.

	 */

	/* The Ship currently only supports ServerPluginCallbacks001, but we need 002 */

	if (g_IsOriginalEngine)

	{

		return;

	}

	if (!m_CanQueryConVars)

	{

		m_VSPIface = iface;

		SH_ADD_HOOK_MEMFUNC(IServerPluginCallbacks, OnQueryCvarValueFinished, iface, this, &ConVarManager::OnQueryCvarValueFinished, false);

		m_CanQueryConVars = true;

	}

}

#if PLAPI_VERSION >= 12

void ConVarManager::OnUnlinkConCommandBase(PluginId id, ConCommandBase *pCommand)

{

	/* Only check convars that have not been created by SourceMod's core */

	if (id != g_PLID && !pCommand->IsCommand())

	{

		ConVarInfo *pInfo;

		const char *cvarName = pCommand->GetName();

		HandleSecurity sec(NULL, g_pCoreIdent);

		bool handleExists = sm_trie_retrieve(m_ConVarCache, cvarName, reinterpret_cast<void **>(&pInfo));

		if (handleExists)

		{

			g_HandleSys.FreeHandle(pInfo->handle, &sec);

			sm_trie_delete(m_ConVarCache, cvarName);

			m_ConVars.remove(pInfo);

		}

	}

}

#else

/* I truly detest this code */

void ConVarManager::OnMetamodPluginUnloaded(PluginId id)

{

	ConVarInfo *pInfo;

	const char *cvarName;

	HandleSecurity sec(NULL, g_pCoreIdent);

	List<ConVarInfo *>::iterator i;

	for (i = m_ConVars.begin(); i != m_ConVars.end(); i++)

	{

		pInfo = (*i);

		cvarName = pInfo->name;

		if (cvarName && !icvar->FindVar(cvarName))

		{

			m_IgnoreHandle = true;

			g_HandleSys.FreeHandle(pInfo->handle, &sec);

			sm_trie_delete(m_ConVarCache, cvarName);

			m_ConVars.erase(i);

			delete [] cvarName;

		}

	}

}

#endif

void ConVarManager::OnPluginUnloaded(IPlugin *plugin)

{

	ConVarList *pConVarList;

	/* If plugin has a convar list, free its memory */

	if (plugin->GetProperty("ConVarList", (void **)&pConVarList, true))

	{

		delete pConVarList;

	}

}

void ConVarManager::OnHandleDestroy(HandleType_t type, void *object)

{

#if PLAPI_VERSION < 12

	/* Lovely workaround for our workaround! */

	if (m_IgnoreHandle)

	{

		m_IgnoreHandle = false;

		return;

	}

#endif

	ConVarInfo *info;

	ConVar *pConVar = static_cast<ConVar *>(object);

	/* Find convar in lookup trie */

	sm_trie_retrieve(m_ConVarCache, pConVar->GetName(), reinterpret_cast<void **>(&info));

	/* If convar was created by SourceMod plugin... */

	if (info->sourceMod)

	{

		/* Delete string allocations */

		delete [] pConVar->GetName(); 

		delete [] pConVar->GetDefault();

		delete [] pConVar->GetHelpText();

		/* Then unlink it from SourceMM */

		g_SMAPI->UnregisterConCmdBase(g_PLAPI, pConVar);

	}

}

void ConVarManager::OnRootConsoleCommand(const char *command, unsigned int argcount)

{

	if (argcount >= 3)

	{

		/* Get plugin index that was passed */

		int id = atoi(g_RootMenu.GetArgument(2));

		/* Get plugin object */

		CPlugin *plugin = g_PluginSys.GetPluginByOrder(id);

		if (!plugin)

		{

			g_RootMenu.ConsolePrint("[SM] Plugin index %d not found.", id);

			return;

		}

		/* Get plugin name */

		const sm_plugininfo_t *plinfo = plugin->GetPublicInfo();

		const char *plname = IS_STR_FILLED(plinfo->name) ? plinfo->name : plugin->GetFilename();

		ConVarList *pConVarList;

		ConVarList::iterator iter;

		/* If no convar list... */

		if (!plugin->GetProperty("ConVarList", (void **)&pConVarList))

		{

			g_RootMenu.ConsolePrint("[SM] No convars found for: %s", plname);

			return;

		}

		g_RootMenu.ConsolePrint("[SM] Listing %d convars for: %s", pConVarList->size(), plname);

		g_RootMenu.ConsolePrint("  %-32.31s %s", "[Name]", "[Value]");

		/* Iterate convar list and display each one */

		for (iter = pConVarList->begin(); iter != pConVarList->end(); iter++)

		{

			const ConVar *pConVar = (*iter);

			g_RootMenu.ConsolePrint("  %-32.31s %s", pConVar->GetName(), pConVar->GetString()); 

		}

		return;

	}

	/* Display usage of subcommand */

	g_RootMenu.ConsolePrint("[SM] Usage: sm convars <plugin #>");

}

Handle_t ConVarManager::CreateConVar(IPluginContext *pContext, const char *name, const char *defaultVal, const char *description, int flags, bool hasMin, float min, bool hasMax, float max)

{

	ConVar *pConVar = NULL;

	ConVarInfo *pInfo = NULL;

	Handle_t hndl = 0;

	/* Find out if the convar exists already */

	pConVar = icvar->FindVar(name);

	/* If the convar already exists... */

	if (pConVar)

	{

		/* Add convar to plugin's list */

		AddConVarToPluginList(pContext, pConVar);

		/* First find out if we already have a handle to it */

		if (sm_trie_retrieve(m_ConVarCache, name, (void **)&pInfo))

		{

			return pInfo->handle;

		} else {

			/* If we don't, then create a new handle from the convar */

			hndl = g_HandleSys.CreateHandle(m_ConVarType, pConVar, NULL, g_pCoreIdent, NULL);

			/* Create and initialize ConVarInfo structure */

			pInfo = new ConVarInfo();

			pInfo->handle = hndl;

			pInfo->sourceMod = false;

			pInfo->pChangeForward = NULL;

			pInfo->origCallback = pConVar->GetCallback();

		#if PLAPI_VERSION < 12

			pInfo->name = sm_strdup(pConVar->GetName());

		#endif

			/* Insert struct into caches */

			m_ConVars.push_back(pInfo);

			sm_trie_insert(m_ConVarCache, name, pInfo);

			return hndl;

		}

	}

	/* To prevent creating a convar that has the same name as a console command... ugh */

	ConCommandBase *pBase = icvar->GetCommands();

	while (pBase)

	{

		if (pBase->IsCommand() && strcmp(pBase->GetName(), name) == 0)

		{

			return BAD_HANDLE;

		}

		pBase = const_cast<ConCommandBase *>(pBase->GetNext());

	}

	/* Since an existing convar (or concmd with the same name) was not found , now we can finally create it */

	pConVar = new ConVar(sm_strdup(name), sm_strdup(defaultVal), flags, sm_strdup(description), hasMin, min, hasMax, max);

	/* Add convar to plugin's list */

	AddConVarToPluginList(pContext, pConVar);

	/* Create a handle from the new convar */

	hndl = g_HandleSys.CreateHandle(m_ConVarType, pConVar, NULL, g_pCoreIdent, NULL);

	/* Create and initialize ConVarInfo structure */

	pInfo = new ConVarInfo();

	pInfo->handle = hndl;

	pInfo->sourceMod = true;

	pInfo->pChangeForward = NULL;

	pInfo->origCallback = NULL;

#if PLAPI_VERSION < 12

	pInfo->name = NULL;

#endif

	/* Insert struct into caches */

	m_ConVars.push_back(pInfo);

	sm_trie_insert(m_ConVarCache, name, pInfo);

	return hndl;

}

Handle_t ConVarManager::FindConVar(const char *name)

{

	ConVar *pConVar = NULL;

	ConVarInfo *pInfo;

	Handle_t hndl;

	/* Search for convar */

	pConVar = icvar->FindVar(name);

	/* If it doesn't exist, then return an invalid handle */

	if (!pConVar)

	{

		return BAD_HANDLE;

	}

	/* At this point, the convar exists. So, find out if we already have a handle */

	if (sm_trie_retrieve(m_ConVarCache, name, (void **)&pInfo))

	{

		return pInfo->handle;

	}

	/* If we don't have a handle, then create a new one */

	hndl = g_HandleSys.CreateHandle(m_ConVarType, pConVar, NULL, g_pCoreIdent, NULL);

	/* Create and initialize ConVarInfo structure */

	pInfo = new ConVarInfo();

	pInfo->handle = hndl;

	pInfo->sourceMod = false;

	pInfo->pChangeForward = NULL;

	pInfo->origCallback = pConVar->GetCallback();

#if PLAPI_VERSION < 12

	pInfo->name = sm_strdup(pConVar->GetName());

#endif

	/* Insert struct into our caches */

	m_ConVars.push_back(pInfo);

	sm_trie_insert(m_ConVarCache, name, pInfo);

	return hndl;

}

void ConVarManager::HookConVarChange(ConVar *pConVar, IPluginFunction *pFunction)

{

	ConVarInfo *pInfo;

	IChangeableForward *pForward;

	/* Find the convar in the lookup trie */

	if (sm_trie_retrieve(m_ConVarCache, pConVar->GetName(), (void **)&pInfo))

	{

		/* Get the forward */

		pForward = pInfo->pChangeForward;

		/* If forward does not exist, create it */

		if (!pForward)

		{

			pForward = g_Forwards.CreateForwardEx(NULL, ET_Ignore, 3, CONVARCHANGE_PARAMS);

			pInfo->pChangeForward = pForward;

			/* Install our own callback */

			pConVar->InstallChangeCallback(OnConVarChanged);

		}

		/* Add function to forward's list */

		pForward->AddFunction(pFunction);

	}

}

void ConVarManager::UnhookConVarChange(ConVar *pConVar, IPluginFunction *pFunction)

{

	ConVarInfo *pInfo;

	IChangeableForward *pForward;

	IPluginContext *pContext = pFunction->GetParentContext();

	/* Find the convar in the lookup trie */

	if (sm_trie_retrieve(m_ConVarCache, pConVar->GetName(), (void **)&pInfo))

	{

		/* Get the forward */

		pForward = pInfo->pChangeForward;

		/* If the forward doesn't exist, we can't unhook anything */

		if (!pForward)

		{

			pContext->ThrowNativeError("Convar \"%s\" has no active hook", pConVar->GetName());

			return;

		}

		/* Remove the function from the forward's list */

		if (!pForward->RemoveFunction(pFunction))

		{

			pContext->ThrowNativeError("Invalid hook callback specified for convar \"%s\"", pConVar->GetName());

			return;

		}

		/* If the forward now has 0 functions in it... */

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

		{

			/* Free this forward */

			g_Forwards.ReleaseForward(pForward);

			pInfo->pChangeForward = NULL;

			/* Put back the original convar callback */

			pConVar->InstallChangeCallback(pInfo->origCallback);

		}

	}

}

QueryCvarCookie_t ConVarManager::QueryClientConVar(edict_t *pPlayer, const char *name, IPluginFunction *pCallback, Handle_t hndl)

{

	QueryCvarCookie_t cookie;

	/* Call StartQueryCvarValue() in either the IVEngineServer or IServerPluginHelpers depending on situation */

	if (!m_VSPIface)

	{

		cookie = engine->StartQueryCvarValue(pPlayer, name);	

	} else {

		cookie = serverpluginhelpers->StartQueryCvarValue(pPlayer, name);

	}

	ConVarQuery query = {cookie, pCallback, hndl};

	m_ConVarQueries.push_back(query);

	return cookie;

}

void ConVarManager::AddConVarToPluginList(IPluginContext *pContext, const ConVar *pConVar)

{

	ConVarList *pConVarList;

	ConVarList::iterator iter;

	bool inserted = false;

	const char *orig = pConVar->GetName();

	IPlugin *plugin = g_PluginSys.FindPluginByContext(pContext->GetContext());

	/* Check plugin for an existing convar list */

	if (!plugin->GetProperty("ConVarList", (void **)&pConVarList))

	{

		pConVarList = new ConVarList();

		plugin->SetProperty("ConVarList", pConVarList);

	} else if (pConVarList->find(pConVar) != pConVarList->end()) {

		/* If convar is already in list, then don't add it */

		return;

	}

	/* Insert convar into list which is sorted alphabetically */

	for (iter = pConVarList->begin(); iter != pConVarList->end(); iter++)

	{

		if (strcmp(orig, (*iter)->GetName()) < 0)

		{

			pConVarList->insert(iter, pConVar);

			inserted = true;

			break;

		}

	}

	if (!inserted)

	{

		pConVarList->push_back(pConVar);

	}

}

void ConVarManager::OnConVarChanged(ConVar *pConVar, const char *oldValue)

{

	/* If the values are the same, exit early in order to not trigger callbacks */

	if (strcmp(pConVar->GetString(), oldValue) == 0)

	{

		return;

	}

	Trie *pCache = g_ConVarManager.GetConVarCache();

	ConVarInfo *pInfo;

	/* Find the convar in the lookup trie */

	sm_trie_retrieve(pCache, pConVar->GetName(), (void **)&pInfo);

	FnChangeCallback origCallback = pInfo->origCallback;

	IChangeableForward *pForward = pInfo->pChangeForward;

	/* If there was a change callback installed previously, call it */

	if (origCallback)

	{

		origCallback(pConVar, oldValue);

	}

	/* Now call forwards in plugins that have hooked this */

	pForward->PushCell(pInfo->handle);

	pForward->PushString(oldValue);

	pForward->PushString(pConVar->GetString());

	pForward->Execute(NULL);

}

void ConVarManager::OnQueryCvarValueFinished(QueryCvarCookie_t cookie, edict_t *pPlayer, EQueryCvarValueStatus result, const char *cvarName, const char *cvarValue)

{

	IPluginFunction *pCallback = NULL;

	cell_t value = 0;

	List<ConVarQuery>::iterator iter;

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

	{

		ConVarQuery &query = (*iter);

		if (query.cookie == cookie)

		{

			pCallback = query.pCallback;

			value = query.value;

			if (!pCallback)

			{

				g_Players.HandleLangQuery(engine->GetPlayerUserId(pPlayer), (result == eQueryCvarValueStatus_ValueIntact) ? cvarValue : "", cookie);

				m_ConVarQueries.erase(iter);

				return;

			}

			break;

		}

	}

	if (pCallback)

	{

		cell_t ret;

		pCallback->PushCell(cookie);