ds@504: /**
dvander@547:  * ===============================================================
dvander@547:  * SourceMod (C)2004-2007 AlliedModders LLC.  All rights reserved.
dvander@547:  * ===============================================================
dvander@547:  *
dvander@547:  * This file is not open source and may not be copied without explicit
dvander@547:  * written permission of AlliedModders LLC.  This file may not be redistributed 
dvander@547:  * in whole or significant part.
dvander@547:  * For information, see LICENSE.txt or http://www.sourcemod.net/license.php
dvander@547:  *
dvander@547:  * Version: $Id$
dvander@547:  */
ds@504: 
dvander@498: #include "CConCmdManager.h"
dvander@498: #include "sm_srvcmds.h"
dvander@513: #include "AdminCache.h"
ds@504: #include "sm_stringutil.h"
dvander@514: #include "CPlayerManager.h"
dvander@514: #include "CTranslator.h"
dvander@498: 
dvander@498: CConCmdManager g_ConCmds;
dvander@498: 
dvander@498: SH_DECL_HOOK0_void(ConCommand, Dispatch, SH_NOATTRIB, false);
dvander@498: SH_DECL_HOOK1_void(IServerGameClients, SetCommandClient, SH_NOATTRIB, false, int);
dvander@498: 
dvander@498: struct PlCmdInfo
dvander@498: {
dvander@498: 	ConCmdInfo *pInfo;
dvander@513: 	CmdHook *pHook;
dvander@498: 	CmdType type;
dvander@498: };
dvander@513: typedef List<PlCmdInfo> CmdList;
dvander@513: void AddToPlCmdList(CmdList *pList, const PlCmdInfo &info);
dvander@498: 
dvander@513: CConCmdManager::CConCmdManager() : m_Strings(1024)
dvander@498: {
dvander@498: 	m_pCmds = sm_trie_create();
dvander@513: 	m_pCmdGrps = sm_trie_create();
dvander@498: }
dvander@498: 
dvander@498: CConCmdManager::~CConCmdManager()
dvander@498: {
dvander@498: 	sm_trie_destroy(m_pCmds);
dvander@513: 	sm_trie_destroy(m_pCmdGrps);
dvander@498: }
dvander@498: 
dvander@498: void CConCmdManager::OnSourceModAllInitialized()
dvander@498: {
dvander@498: 	g_PluginSys.AddPluginsListener(this);
dvander@498: 	g_RootMenu.AddRootConsoleCommand("cmds", "List console commands", this);
dvander@498: 	SH_ADD_HOOK_MEMFUNC(IServerGameClients, SetCommandClient, serverClients, this, &CConCmdManager::SetCommandClient, false);
dvander@498: }
dvander@498: 
dvander@498: void CConCmdManager::OnSourceModShutdown()
dvander@498: {
dvander@500: 	/* All commands should already be removed by the time we're done */
dvander@498: 	SH_REMOVE_HOOK_MEMFUNC(IServerGameClients, SetCommandClient, serverClients, this, &CConCmdManager::SetCommandClient, false);
dvander@498: 	g_RootMenu.RemoveRootConsoleCommand("cmds", this);
dvander@498: 	g_PluginSys.RemovePluginsListener(this);
dvander@498: }
dvander@498: 
dvander@513: void CConCmdManager::RemoveConCmds(List<CmdHook *> &cmdlist, IPluginContext *pContext)
dvander@498: {
dvander@513: 	List<CmdHook *>::iterator iter = cmdlist.begin();
dvander@513: 	CmdHook *pHook;
dvander@513: 
dvander@513: 	while (iter != cmdlist.end())
dvander@513: 	{
dvander@513: 		pHook = (*iter);
dvander@513: 		if (pHook->pf->GetParentContext() == pContext)
dvander@513: 		{
dvander@513: 			delete pHook->pAdmin;
dvander@513: 			delete pHook;
dvander@513: 			iter = cmdlist.erase(iter);
dvander@513: 		} else {
dvander@513: 			iter++;
dvander@513: 		}
dvander@513: 	}
dvander@498: }
dvander@498: 
dvander@498: void CConCmdManager::OnPluginDestroyed(IPlugin *plugin)
dvander@498: {
dvander@498: 	CmdList *pList;
dvander@500: 	List<ConCmdInfo *> removed;
dvander@498: 	if (plugin->GetProperty("CommandList", (void **)&pList, true))
dvander@498: 	{
dvander@513: 		IPluginContext *pContext = plugin->GetBaseContext();
dvander@498: 		CmdList::iterator iter;
dvander@513: 		/* First pass!
dvander@513: 	 	 * Only bother if there's an actual command list on this plugin... 
dvander@513: 	 	 */
dvander@498: 		for (iter=pList->begin();
dvander@513: 			iter!=pList->end();
dvander@513: 			iter++)
dvander@498: 		{
dvander@498: 			PlCmdInfo &cmd = (*iter);
dvander@513: 			ConCmdInfo *pInfo = cmd.pInfo;
dvander@513: 
dvander@513: 			/* Has this chain already been fully cleaned/removed? */
dvander@513: 			if (removed.find(pInfo) != removed.end())
dvander@498: 			{
dvander@513: 				continue;
dvander@498: 			}
dvander@513: 
dvander@513: 			/* Remove any hooks from us on this command */
dvander@513: 			RemoveConCmds(pInfo->conhooks, pContext);
dvander@513: 			RemoveConCmds(pInfo->srvhooks, pContext);
dvander@513: 
dvander@513: 			/* See if there are still hooks */
dvander@513: 			if (pInfo->srvhooks.size())
dvander@513: 			{
dvander@513: 				continue;
dvander@513: 			}
dvander@513: 			if (pInfo->conhooks.size())
dvander@513: 			{
dvander@513: 				continue;
dvander@513: 			}
dvander@513: 
dvander@513: 			/* Remove the command, it should be safe now */
dvander@513: 			RemoveConCmd(pInfo);
dvander@513: 			removed.push_back(pInfo);
dvander@498: 		}
dvander@498: 		delete pList;
dvander@498: 	}
dvander@498: }
dvander@498: 
dvander@498: void CommandCallback()
dvander@498: {
dvander@498: 	g_ConCmds.InternalDispatch();
dvander@498: }
dvander@498: 
dvander@498: void CConCmdManager::SetCommandClient(int client)
dvander@498: {
dvander@498: 	m_CmdClient = client + 1;
dvander@498: }
dvander@498: 
dvander@500: ResultType CConCmdManager::DispatchClientCommand(int client, ResultType type)
dvander@500: {
dvander@500: 	const char *cmd = engine->Cmd_Argv(0);
dvander@500: 	int args = engine->Cmd_Argc() - 1;
dvander@500: 
dvander@500: 	ConCmdInfo *pInfo;
dvander@500: 	if (sm_trie_retrieve(m_pCmds, cmd, (void **)&pInfo))
dvander@500: 	{
dvander@500: 		cell_t result = type;
dvander@513: 		cell_t tempres = result;
dvander@513: 		List<CmdHook *>::iterator iter;
dvander@513: 		CmdHook *pHook;
dvander@513: 		for (iter=pInfo->conhooks.begin();
dvander@513: 			 iter!=pInfo->conhooks.end();
dvander@513: 			 iter++)
dvander@500: 		{
dvander@513: 			pHook = (*iter);
dvander@525: 			if (!pHook->pf->IsRunnable())
dvander@525: 			{
dvander@525: 				continue;
dvander@525: 			}
dvander@514: 			if (pHook->pAdmin && !CheckAccess(client, cmd, pHook->pAdmin))
dvander@513: 			{
dvander@514: 				if (result < Pl_Handled)
dvander@514: 				{
dvander@514: 					result = Pl_Handled;
dvander@514: 				}
dvander@514: 				continue;
dvander@513: 			}
dvander@513: 			pHook->pf->PushCell(client);
dvander@513: 			pHook->pf->PushCell(args);
dvander@513: 			if (pHook->pf->Execute(&tempres) == SP_ERROR_NONE)
dvander@513: 			{
dvander@513: 				if (tempres > result)
dvander@513: 				{
dvander@513: 					result = tempres;
dvander@513: 				}
dvander@513: 				if (result == Pl_Stop)
dvander@513: 				{
dvander@513: 					break;
dvander@513: 				}
dvander@513: 			}
dvander@500: 		}
dvander@500: 		type = (ResultType)result;
dvander@500: 	}
dvander@500: 
dvander@500: 	return type;
dvander@500: }
dvander@500: 
dvander@498: void CConCmdManager::InternalDispatch()
dvander@498: {
dvander@500: 	/**
dvander@500: 	 * Note: Console commands will EITHER go through IServerGameDLL::ClientCommand,
dvander@500: 	 * OR this dispatch.  They will NEVER go through both.
dvander@500: 	 * -- 
dvander@500: 	 * Whether or not it goes through the callback is determined by FCVAR_GAMEDLL
dvander@500: 	 */
dvander@498: 
dvander@498: 	const char *cmd = engine->Cmd_Argv(0);
dvander@498: 	ConCmdInfo *pInfo;
dvander@498: 	if (!sm_trie_retrieve(m_pCmds, cmd, (void **)&pInfo))
dvander@498: 	{
dvander@498: 		return;
dvander@498: 	}
dvander@498: 
dvander@498: 	cell_t result = Pl_Continue;
dvander@500: 	int args = engine->Cmd_Argc() - 1;
dvander@500: 
dvander@513: 	List<CmdHook *>::iterator iter;
dvander@513: 	CmdHook *pHook;
dvander@513: 
dvander@513: 	/* Execute server-only commands if viable */
dvander@513: 	if (m_CmdClient == 0 && pInfo->srvhooks.size())
dvander@498: 	{
dvander@513: 		cell_t tempres = result;
dvander@513: 		for (iter=pInfo->srvhooks.begin();
dvander@513: 			 iter!=pInfo->srvhooks.end();
dvander@513: 			 iter++)
dvander@500: 		{
dvander@513: 			pHook = (*iter);
dvander@525: 			if (!pHook->pf->IsRunnable())
dvander@525: 			{
dvander@525: 				continue;
dvander@525: 			}
dvander@513: 			pHook->pf->PushCell(args);
dvander@513: 			if (pHook->pf->Execute(&tempres) == SP_ERROR_NONE)
dvander@513: 			{
dvander@513: 				if (tempres > result)
dvander@513: 				{
dvander@513: 					result = tempres;
dvander@513: 				}
dvander@513: 				if (result == Pl_Stop)
dvander@513: 				{
dvander@513: 					break;
dvander@513: 				}
dvander@513: 			}
dvander@500: 		}
dvander@500: 	
dvander@500: 		/* Check if there's an early stop */
dvander@500: 		if (result >= Pl_Stop)
dvander@500: 		{
dvander@500: 			if (!pInfo->sourceMod)
dvander@500: 			{
dvander@500: 				RETURN_META(MRES_SUPERCEDE);
dvander@500: 			}
dvander@500: 			return;
dvander@500: 		}
dvander@500: 	}
dvander@500: 
dvander@513: 	/* Execute console commands */
dvander@513: 	if (pInfo->conhooks.size())
dvander@500: 	{
dvander@513: 		cell_t tempres = result;
dvander@513: 		for (iter=pInfo->conhooks.begin();
dvander@513: 			iter!=pInfo->conhooks.end();
dvander@513: 			iter++)
dvander@513: 		{
dvander@513: 			pHook = (*iter);
dvander@525: 			if (!pHook->pf->IsRunnable())
dvander@525: 			{
dvander@525: 				continue;
dvander@525: 			}
dvander@513: 			if (m_CmdClient 
dvander@513: 				&& pHook->pAdmin
dvander@514: 				&& !CheckAccess(m_CmdClient, cmd, pHook->pAdmin))
dvander@513: 			{
dvander@514: 				if (result < Pl_Handled)
dvander@514: 				{
dvander@514: 					result = Pl_Handled;
dvander@514: 				}
dvander@514: 				continue;
dvander@513: 			}
dvander@513: 			pHook->pf->PushCell(m_CmdClient);
dvander@513: 			pHook->pf->PushCell(args);
dvander@546: 			if (pHook->pf->Execute(&tempres) == SP_ERROR_NONE)
dvander@513: 			{
dvander@513: 				if (tempres > result)
dvander@513: 				{
dvander@513: 					result = tempres;
dvander@513: 				}
dvander@513: 				if (result == Pl_Stop)
dvander@513: 				{
dvander@513: 					break;
dvander@513: 				}
dvander@513: 			}
dvander@513: 		}
dvander@498: 	}
dvander@498: 
dvander@498: 	if (result >= Pl_Handled)
dvander@498: 	{
dvander@498: 		if (!pInfo->sourceMod)
dvander@498: 		{
dvander@498: 			RETURN_META(MRES_SUPERCEDE);
dvander@498: 		}
dvander@498: 		return;
dvander@498: 	}
dvander@498: }
dvander@498: 
dvander@514: bool CConCmdManager::CheckAccess(int client, const char *cmd, AdminCmdInfo *pAdmin)
dvander@513: {
dvander@514: 	FlagBits cmdflags = pAdmin->eflags;
dvander@514: 	if (cmdflags == 0)
dvander@514: 	{
dvander@514: 		return true;
dvander@514: 	}
dvander@514: 
dvander@514: 	CPlayer *player = g_Players.GetPlayerByIndex(client);
dvander@514: 	if (!player)
dvander@514: 	{
dvander@514: 		return false;
dvander@514: 	}
dvander@514: 
dvander@514: 	AdminId adm = player->GetAdminId();
dvander@514: 	if (adm != INVALID_ADMIN_ID)
dvander@514: 	{
dvander@514: 		FlagBits bits = g_Admins.GetAdminFlags(adm, Access_Effective);
dvander@514: 
dvander@514: 		/* root knows all, WHOA */
dvander@514: 		if ((bits & ADMFLAG_ROOT) == ADMFLAG_ROOT)
dvander@514: 		{
dvander@514: 			return true;
dvander@514: 		}
dvander@514: 
dvander@514: 		/* See if our other flags match */
dvander@520: 		if ((bits & cmdflags) == cmdflags)
dvander@514: 		{
dvander@514: 			return true;
dvander@514: 		}
dvander@514: 
dvander@514: 		/* Check for overrides */
dvander@514: 		unsigned int groups = g_Admins.GetAdminGroupCount(adm);
dvander@514: 		GroupId gid;
dvander@514: 		OverrideRule rule;
dvander@514: 		bool override = false;
dvander@514: 		for (unsigned int i=0; i<groups; i++)
dvander@514: 		{
dvander@514: 			gid = g_Admins.GetAdminGroup(adm, i, NULL);
dvander@514: 			/* First get group-level override */
dvander@514: 			override = g_Admins.GetGroupCommandOverride(gid, cmd, Override_CommandGroup, &rule);
dvander@514: 			/* Now get the specific command override */
dvander@514: 			if (g_Admins.GetGroupCommandOverride(gid, cmd, Override_Command, &rule))
dvander@514: 			{
dvander@514: 				override = true;
dvander@514: 			}
dvander@514: 			if (override)
dvander@514: 			{
dvander@514: 				if (rule == Command_Allow)
dvander@514: 				{
dvander@514: 					return true;
dvander@514: 				} else if (rule == Command_Deny) {
dvander@514: 					break;
dvander@514: 				}
dvander@514: 			}
dvander@514: 		}
dvander@514: 	}
dvander@514: 
dvander@520: 	if (player->IsFakeClient()
dvander@520: 		|| player->GetEdict() == NULL)
dvander@520: 	{
dvander@520: 		return false;
dvander@520: 	}
dvander@520: 	
dvander@514: 	/* If we got here, the command failed... */
dvander@520: 	char buffer[128];
dvander@520: 	if (g_Translator.CoreTrans(client, buffer, sizeof(buffer), "No Access", NULL, NULL)
dvander@520: 		!= Trans_Okay)
dvander@520: 	{
dvander@520: 		snprintf(buffer, sizeof(buffer), "You do not have access to this command");
dvander@520: 	}
dvander@520: 
dvander@520: 	char fullbuffer[192];
dvander@520: 	snprintf(fullbuffer, sizeof(fullbuffer), "[SM] %s.\n", buffer);
dvander@520: 
dvander@520: 	engine->ClientPrintf(player->GetEdict(), fullbuffer);
dvander@520: 	
dvander@514: 	return false;
dvander@513: }
dvander@513: 
dvander@500: void CConCmdManager::AddConsoleCommand(IPluginFunction *pFunction, 
dvander@500: 									   const char *name, 
dvander@500: 									   const char *description, 
dvander@500: 									   int flags)
dvander@500: {
dvander@500: 	ConCmdInfo *pInfo = AddOrFindCommand(name, description, flags);
dvander@513: 	CmdHook *pHook = new CmdHook();
dvander@500: 
dvander@513: 	pHook->pf = pFunction;
dvander@513: 	if (description && description[0])
dvander@500: 	{
dvander@513: 		pHook->helptext.assign(description);
dvander@500: 	}
dvander@513: 	pInfo->conhooks.push_back(pHook);
dvander@500: 
dvander@500: 	/* Add to the plugin */
dvander@500: 	CmdList *pList;
dvander@500: 	IPlugin *pPlugin = g_PluginSys.GetPluginByCtx(pFunction->GetParentContext()->GetContext());
dvander@500: 	if (!pPlugin->GetProperty("CommandList", (void **)&pList))
dvander@500: 	{
dvander@500: 		pList = new CmdList();
dvander@500: 		pPlugin->SetProperty("CommandList", pList);
dvander@500: 	}
dvander@500: 	PlCmdInfo info;
dvander@500: 	info.pInfo = pInfo;
dvander@500: 	info.type = Cmd_Console;
dvander@513: 	info.pHook = pHook;
dvander@513: 	AddToPlCmdList(pList, info);
dvander@513: }
dvander@513: 
dvander@513: bool CConCmdManager::AddAdminCommand(IPluginFunction *pFunction, 
dvander@513: 									 const char *name, 
dvander@513: 									 const char *group,
dvander@513: 									 int adminflags,
dvander@513: 									 const char *description, 
dvander@513: 									 int flags)
dvander@513: {
dvander@513: 	ConCmdInfo *pInfo = AddOrFindCommand(name, description, flags);
dvander@513: 
dvander@513: 	CmdHook *pHook = new CmdHook();
dvander@513: 	AdminCmdInfo *pAdmin = new AdminCmdInfo();
dvander@513: 
dvander@513: 	pHook->pf = pFunction;
dvander@513: 	if (description && description[0])
dvander@513: 	{
dvander@513: 		pHook->helptext.assign(description);
dvander@513: 	}
dvander@513: 	pHook->pAdmin = pAdmin;
dvander@513: 
dvander@513: 	void *object;
dvander@513: 	int grpid;
dvander@513: 	if (!sm_trie_retrieve(m_pCmdGrps, group, (void **)&object))
dvander@513: 	{
dvander@513: 		grpid = m_Strings.AddString(group);
dvander@513: 		sm_trie_insert(m_pCmdGrps, group, (void *)grpid);
dvander@513: 	} else {
dvander@513: 		grpid = (int)object;
dvander@513: 	}
dvander@513: 
dvander@513: 	pAdmin->cmdGrpId = grpid;
dvander@520: 	pAdmin->flags = adminflags;
dvander@513: 
dvander@513: 	/* First get the command group override, if any */
dvander@513: 	bool override = g_Admins.GetCommandOverride(group, 
dvander@513: 		Override_CommandGroup,
dvander@513: 		&(pAdmin->eflags));
dvander@513: 
dvander@513: 	/* Next get the command override, if any */
dvander@513: 	if (g_Admins.GetCommandOverride(name,
dvander@513: 		Override_Command,
dvander@513: 		&(pAdmin->eflags)))
dvander@513: 	{
dvander@513: 		override = true;
dvander@513: 	}
dvander@513: 
dvander@513: 	/* Assign normal flags if there were no overrides */
dvander@513: 	if (!override)
dvander@513: 	{
dvander@513: 		pAdmin->eflags = pAdmin->flags;
dvander@513: 	}
dvander@513: 
dvander@513: 	/* Finally, add the hook */
dvander@513: 	pInfo->conhooks.push_back(pHook);
dvander@513: 
dvander@513: 	/* Now add to the plugin */
dvander@513: 	CmdList *pList;
dvander@513: 	IPlugin *pPlugin = g_PluginSys.GetPluginByCtx(pFunction->GetParentContext()->GetContext());
dvander@513: 	if (!pPlugin->GetProperty("CommandList", (void **)&pList))
dvander@513: 	{
dvander@513: 		pList = new CmdList();
dvander@513: 		pPlugin->SetProperty("CommandList", pList);
dvander@513: 	}
dvander@513: 	PlCmdInfo info;
dvander@513: 	info.pInfo = pInfo;
dvander@513: 	info.type = Cmd_Admin;
dvander@513: 	info.pHook = pHook;
dvander@513: 	AddToPlCmdList(pList, info);
dvander@513: 
dvander@513: 	return true;
dvander@500: }
dvander@500: 
dvander@498: void CConCmdManager::AddServerCommand(IPluginFunction *pFunction, 
dvander@498: 									  const char *name, 
dvander@498: 									  const char *description, 
dvander@498: 									  int flags)
dvander@498: 
dvander@498: {
dvander@498: 	ConCmdInfo *pInfo = AddOrFindCommand(name, description, flags);
dvander@513: 	CmdHook *pHook = new CmdHook();
dvander@498: 
dvander@513: 	pHook->pf = pFunction;
dvander@513: 	if (description && description[0])
dvander@498: 	{
dvander@513: 		pHook->helptext.assign(description);
dvander@498: 	}
dvander@498: 
dvander@513: 	pInfo->srvhooks.push_back(pHook);
dvander@498: 
dvander@498: 	/* Add to the plugin */
dvander@498: 	CmdList *pList;
dvander@498: 	IPlugin *pPlugin = g_PluginSys.GetPluginByCtx(pFunction->GetParentContext()->GetContext());
dvander@498: 	if (!pPlugin->GetProperty("CommandList", (void **)&pList))
dvander@498: 	{
dvander@498: 		pList = new CmdList();
dvander@498: 		pPlugin->SetProperty("CommandList", pList);
dvander@498: 	}
dvander@498: 	PlCmdInfo info;
dvander@498: 	info.pInfo = pInfo;
dvander@500: 	info.type = Cmd_Server;
dvander@513: 	info.pHook = pHook;
dvander@513: 	AddToPlCmdList(pList, info);
dvander@513: }
dvander@513: 
dvander@513: void AddToPlCmdList(CmdList *pList, const PlCmdInfo &info)
dvander@513: {
dvander@513: 	CmdList::iterator iter = pList->begin();
dvander@513: 	bool inserted = false;
dvander@513: 	const char *orig = NULL;
dvander@513: 
dvander@513: 	orig = info.pInfo->pCmd->GetName();
dvander@513: 
dvander@513: 	/* Insert this into the help list, SORTED alphabetically. */
dvander@513: 	while (iter != pList->end())
dvander@513: 	{
dvander@513: 		PlCmdInfo &obj = (*iter);
dvander@513: 		const char *cmd = obj.pInfo->pCmd->GetName();
dvander@513: 		if (strcmp(orig, cmd) < 0)
dvander@513: 		{
dvander@513: 			pList->insert(iter, info);
dvander@513: 			inserted = true;
dvander@513: 			break;
dvander@513: 		}
dvander@513: 		iter++;
dvander@513: 	}
dvander@513: 
dvander@513: 	if (!inserted)
dvander@513: 	{
dvander@513: 		pList->push_back(info);
dvander@513: 	}
dvander@498: }
dvander@498: 
dvander@498: void CConCmdManager::AddToCmdList(ConCmdInfo *info)
dvander@498: {
dvander@498: 	List<ConCmdInfo *>::iterator iter = m_CmdList.begin();
dvander@498: 	ConCmdInfo *pInfo;
dvander@498: 	bool inserted = false;
dvander@503: 	const char *orig = NULL;
dvander@498: 
dvander@503: 	orig = info->pCmd->GetName();
dvander@498: 
dvander@498: 	/* Insert this into the help list, SORTED alphabetically. */
dvander@498: 	while (iter != m_CmdList.end())
dvander@498: 	{
dvander@498: 		const char *cmd = NULL;
dvander@498: 		pInfo = (*iter);
dvander@503: 		cmd = pInfo->pCmd->GetName();
dvander@498: 		if (strcmp(orig, cmd) < 0)
dvander@498: 		{
dvander@498: 			m_CmdList.insert(iter, info);
dvander@498: 			inserted = true;
dvander@498: 			break;
dvander@498: 		}
dvander@498: 		iter++;
dvander@498: 	}
dvander@498: 
dvander@498: 	if (!inserted)
dvander@498: 	{
dvander@498: 		m_CmdList.push_back(info);
dvander@498: 	}
dvander@498: }
dvander@498: 
dvander@514: void CConCmdManager::UpdateAdminCmdFlags(const char *cmd, OverrideType type, FlagBits bits)
dvander@514: {
dvander@514: 	ConCmdInfo *pInfo;
dvander@514: 
dvander@514: 	if (type == Override_Command)
dvander@514: 	{
dvander@514: 		if (!sm_trie_retrieve(m_pCmds, cmd, (void **)&pInfo))
dvander@514: 		{
dvander@514: 			return;
dvander@514: 		}
dvander@514: 
dvander@514: 		List<CmdHook *>::iterator iter;
dvander@514: 		CmdHook *pHook;
dvander@514: 
dvander@514: 		for (iter=pInfo->conhooks.begin(); iter!=pInfo->conhooks.end(); iter++)
dvander@514: 		{
dvander@514: 			pHook = (*iter);
dvander@514: 			if (pHook->pAdmin)
dvander@514: 			{
dvander@514: 				if (bits)
dvander@514: 				{
dvander@514: 					pHook->pAdmin->eflags = bits;
dvander@514: 				} else {
dvander@514: 					pHook->pAdmin->eflags = pHook->pAdmin->flags;
dvander@514: 				}
dvander@514: 			}
dvander@514: 		}
dvander@514: 	} else if (type == Override_CommandGroup) {
dvander@514: 		void *object;
dvander@514: 		if (!sm_trie_retrieve(m_pCmdGrps, cmd, &object))
dvander@514: 		{
dvander@514: 			return;
dvander@514: 		}
dvander@514: 		int grpid = (int)object;
dvander@514: 
dvander@514: 		/* This is bad :( loop through all commands */
dvander@514: 		List<ConCmdInfo *>::iterator iter;
dvander@514: 		CmdHook *pHook;
dvander@514: 		for (iter=m_CmdList.begin(); iter!=m_CmdList.end(); iter++)
dvander@514: 		{
dvander@514: 			pInfo = (*iter);
dvander@514: 			for (List<CmdHook *>::iterator citer=pInfo->conhooks.begin();
dvander@514: 				 citer!=pInfo->conhooks.end();
dvander@514: 				 citer++)
dvander@514: 			{
dvander@514: 				pHook = (*citer);
dvander@514: 				if (pHook->pAdmin && pHook->pAdmin->cmdGrpId == grpid)
dvander@514: 				{
dvander@514: 					if (bits)
dvander@514: 					{
dvander@514: 						pHook->pAdmin->eflags = bits;
dvander@514: 					} else {
dvander@514: 						pHook->pAdmin->eflags = pHook->pAdmin->flags;
dvander@514: 					}
dvander@514: 				}
dvander@514: 			}
dvander@514: 		}
dvander@514: 	}
dvander@514: }
dvander@514: 
dvander@498: void CConCmdManager::RemoveConCmd(ConCmdInfo *info)
dvander@498: {
dvander@503: 	/* Remove console-specific information
dvander@503: 	 * This should always be true as of right now
dvander@503: 	 */
dvander@498: 	if (info->pCmd)
dvander@498: 	{
dvander@498: 		/* Remove from the trie */
dvander@498: 		sm_trie_delete(m_pCmds, info->pCmd->GetName());
dvander@498: 
dvander@498: 		if (info->sourceMod)
dvander@498: 		{
dvander@498: 			/* Unlink from SourceMM */
dvander@498: 			g_SMAPI->UnregisterConCmdBase(g_PLAPI, info->pCmd);
dvander@498: 			/* Delete the command's memory */
dvander@498: 			char *new_help = const_cast<char *>(info->pCmd->GetHelpText());
dvander@498: 			char *new_name = const_cast<char *>(info->pCmd->GetName());
dvander@498: 			delete [] new_help;
dvander@498: 			delete [] new_name;
dvander@498: 			delete info->pCmd;
dvander@498: 		} else {
dvander@498: 			/* Remove the external hook */
dvander@498: 			SH_REMOVE_HOOK_STATICFUNC(ConCommand, Dispatch, info->pCmd, CommandCallback, false);
dvander@498: 		}
dvander@498: 	}
dvander@498: 	
dvander@498: 	/* Remove from list */
dvander@498: 	m_CmdList.remove(info);
dvander@513: 
dvander@513: 	delete info;
dvander@498: }
dvander@498: 
dvander@498: ConCmdInfo *CConCmdManager::AddOrFindCommand(const char *name, const char *description, int flags)
dvander@498: {
dvander@498: 	ConCmdInfo *pInfo;
dvander@498: 	if (!sm_trie_retrieve(m_pCmds, name, (void **)&pInfo))
dvander@498: 	{
dvander@498: 		pInfo = new ConCmdInfo();
dvander@498: 		/* Find the commandopan */
dvander@498: 		ConCommandBase *pBase = icvar->GetCommands();
dvander@498: 		ConCommand *pCmd = NULL;
dvander@498: 		while (pBase)
dvander@498: 		{
dvander@498: 			if (pBase->IsCommand()
dvander@498: 				&& (strcmp(pBase->GetName(), name) == 0))
dvander@498: 			{
dvander@498: 				pCmd = (ConCommand *)pBase;
dvander@498: 				break;
dvander@498: 			}
dvander@498: 			pBase = const_cast<ConCommandBase *>(pBase->GetNext());
dvander@498: 		}
dvander@498: 
dvander@498: 		if (!pCmd)
dvander@498: 		{
dvander@498: 			/* Note that we have to duplicate because the source might not be 
dvander@498: 			 * a static string, and these expect static memory.
dvander@498: 			 */
dvander@498: 			if (!description)
dvander@498: 			{
dvander@498: 				description = "";
dvander@498: 			}
dvander@498: 			char *new_name = sm_strdup(name);
dvander@498: 			char *new_help = sm_strdup(description);
dvander@498: 			pCmd = new ConCommand(new_name, CommandCallback, new_help, flags);
dvander@498: 			pInfo->sourceMod = true;
dvander@498: 		} else {
dvander@498: 			SH_ADD_HOOK_STATICFUNC(ConCommand, Dispatch, pCmd, CommandCallback, false);
dvander@498: 		}
dvander@498: 
dvander@498: 		pInfo->pCmd = pCmd;
dvander@498: 
dvander@498: 		sm_trie_insert(m_pCmds, name, pInfo);
dvander@498: 		AddToCmdList(pInfo);
dvander@498: 	}
dvander@498: 
dvander@498: 	return pInfo;
dvander@498: }
dvander@498: 
dvander@498: void CConCmdManager::OnRootConsoleCommand(const char *command, unsigned int argcount)
dvander@498: {
dvander@499: 	if (argcount >= 3)
dvander@499: 	{
dvander@499: 		const char *text = engine->Cmd_Argv(2);
dvander@499: 		int id = atoi(text);
dvander@499: 		CPlugin *pPlugin = g_PluginSys.GetPluginByOrder(id);
dvander@499: 
dvander@499: 		if (!pPlugin)
dvander@499: 		{
dvander@499: 			g_RootMenu.ConsolePrint("[SM] Plugin index %d not found.", id);
dvander@499: 			return;
dvander@499: 		}
dvander@499: 
dvander@499: 		CmdList *pList;
dvander@499: 		if (!pPlugin->GetProperty("CommandList", (void **)&pList))
dvander@499: 		{
dvander@499: 			g_RootMenu.ConsolePrint("[SM] No commands found for %s", pPlugin->GetFilename());
dvander@499: 			return;
dvander@499: 		}
dvander@499: 		if (!pList->size())
dvander@499: 		{
dvander@499: 			g_RootMenu.ConsolePrint("[SM] No commands found for %s", pPlugin->GetFilename());
dvander@499: 			return;
dvander@499: 		}
dvander@499: 
dvander@499: 		CmdList::iterator iter;
dvander@499: 		const char *type = NULL;
dvander@499: 		const char *name;
dvander@499: 		const char *help;
dvander@499: 		g_RootMenu.ConsolePrint(" %-17.16s %-8.7s %s", "[Name]", "[Type]", "[Help]");
dvander@499: 		for (iter=pList->begin();
dvander@499: 			 iter!=pList->end();
dvander@499: 			 iter++)
dvander@499: 		{
dvander@499: 			PlCmdInfo &cmd = (*iter);
dvander@499: 			if (cmd.type == Cmd_Server)
dvander@499: 			{
dvander@499: 				type = "server";
dvander@499: 			} else if (cmd.type == Cmd_Console) {
dvander@499: 				type = "console";
dvander@513: 			} else if (cmd.type == Cmd_Admin) {
dvander@513: 				type = "admin";
dvander@499: 			}
dvander@499: 			name = cmd.pInfo->pCmd->GetName();
dvander@513: 			if (cmd.pHook->helptext.size())
dvander@513: 			{
dvander@513: 				help = cmd.pHook->helptext.c_str();
dvander@513: 			} else {
dvander@513: 				help = cmd.pInfo->pCmd->GetHelpText();
dvander@513: 			}
dvander@500: 			g_RootMenu.ConsolePrint(" %-17.16s %-12.11s %s", name, type, help);		
dvander@499: 		}
dvander@499: 
dvander@499: 		return;
dvander@499: 	}
dvander@499: 
dvander@499: 	g_RootMenu.ConsolePrint("[SM] Usage: sm cmds <plugin #>");
dvander@498: }