ds@579: /**
dvander@4076:  * vim: set ts=4 sw=4 tw=99 noet :
ds@1330:  * =============================================================================
dvander@1231:  * SourceMod
ds@1947:  * Copyright (C) 2004-2008 AlliedModders LLC.  All rights reserved.
ds@1330:  * =============================================================================
ds@579:  *
ds@1330:  * This program is free software; you can redistribute it and/or modify it under
ds@1330:  * the terms of the GNU General Public License, version 3.0, as published by the
ds@1330:  * Free Software Foundation.
dvander@1231:  * 
ds@1330:  * This program is distributed in the hope that it will be useful, but WITHOUT
ds@1330:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
ds@1330:  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
ds@1330:  * details.
dvander@1231:  *
ds@1330:  * You should have received a copy of the GNU General Public License along with
ds@1330:  * this program.  If not, see <http://www.gnu.org/licenses/>.
dvander@1231:  *
ds@1330:  * As a special exception, AlliedModders LLC gives you permission to link the
ds@1330:  * code of this program (as well as its derivative works) to "Half-Life 2," the
ds@1330:  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
ds@1330:  * by the Valve Corporation.  You must obey the GNU General Public License in
ds@1330:  * all respects for all other code used.  Additionally, AlliedModders LLC grants
ds@1330:  * this exception to all derivative works.  AlliedModders LLC defines further
ds@1330:  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
ds@1330:  * or <http://www.sourcemod.net/license.php>.
ds@579:  *
ds@579:  * Version: $Id$
ds@579:  */
ds@579: 
ds@579: #include "ConCmdManager.h"
ds@579: #include "sm_srvcmds.h"
ds@579: #include "AdminCache.h"
ds@579: #include "sm_stringutil.h"
dvander@591: #include "PlayerManager.h"
dvander@895: #include "HalfLife2.h"
dvander@895: #include "ChatTriggers.h"
dvander@3004: #include "logic_bridge.h"
ds@579: 
dvander@4113: using namespace ke;
dvander@4113: 
ds@579: ConCmdManager g_ConCmds;
ds@579: 
skamonkey@3899: #if SOURCE_ENGINE == SE_DOTA
skamonkey@3942: 	SH_DECL_HOOK2_void(ConCommand, Dispatch, SH_NOATTRIB, false, const CCommandContext &, const CCommand &);
skamonkey@3899: #elif SOURCE_ENGINE >= SE_ORANGEBOX
faluco@1541: 	SH_DECL_HOOK1_void(ConCommand, Dispatch, SH_NOATTRIB, false, const CCommand &);
faluco@1541: #else
faluco@1541: 	SH_DECL_HOOK0_void(ConCommand, Dispatch, SH_NOATTRIB, false);
faluco@1541: #endif
faluco@1541: 
ds@579: SH_DECL_HOOK1_void(IServerGameClients, SetCommandClient, SH_NOATTRIB, false, int);
ds@579: 
dvander@4112: typedef ke::LinkedList<CmdHook *> PluginHookList;
dvander@4112: void RegisterInPlugin(CmdHook *hook);
ds@579: 
dvander@4113: ConCmdManager::ConCmdManager()
ds@579: {
dvander@769: 	m_CmdClient = 0;
ds@579: }
ds@579: 
ds@579: ConCmdManager::~ConCmdManager()
ds@579: {
ds@579: }
ds@579: 
ds@579: void ConCmdManager::OnSourceModAllInitialized()
ds@579: {
dvander@3872: 	scripts->AddPluginsListener(this);
ds@579: 	g_RootMenu.AddRootConsoleCommand("cmds", "List console commands", this);
kyle@3819: 	SH_ADD_HOOK(IServerGameClients, SetCommandClient, serverClients, SH_MEMBER(this, &ConCmdManager::SetCommandClient), false);
ds@579: }
ds@579: 
ds@579: void ConCmdManager::OnSourceModShutdown()
ds@579: {
dvander@3872: 	scripts->RemovePluginsListener(this);
ds@579: 	/* All commands should already be removed by the time we're done */
kyle@3819: 	SH_REMOVE_HOOK(IServerGameClients, SetCommandClient, serverClients, SH_MEMBER(this, &ConCmdManager::SetCommandClient), false);
ds@579: 	g_RootMenu.RemoveRootConsoleCommand("cmds", this);
ds@579: }
ds@579: 
dvander@1584: void ConCmdManager::OnUnlinkConCommandBase(ConCommandBase *pBase, const char *name, bool is_read_safe)
dvander@1584: {
dvander@1584: 	/* Whoa, first get its information struct */
dvander@1584: 	ConCmdInfo *pInfo;
dvander@1584: 
dvander@4076: 	if (!m_Cmds.retrieve(name, &pInfo))
dvander@1584: 		return;
dvander@1584: 
dvander@4112: 	CmdHookList::iterator iter = pInfo->hooks.begin();
dvander@4112: 	while (iter != pInfo->hooks.end())
dvander@4112: 	{
dvander@4112: 		CmdHook *hook = *iter;
dvander@4112: 
dvander@4112: 		IPluginContext *pContext = hook->pf->GetParentContext();
dvander@4112: 		IPlugin *pPlugin = scripts->FindPluginByContext(pContext->GetContext());
dvander@4112: 
dvander@4112: 		// The list is guaranteed to exist.
dvander@4112: 		PluginHookList *list;
dvander@4112: 		pPlugin->GetProperty("CommandList", (void **)&list, false);
dvander@4112: 		for (PluginHookList::iterator hiter = list->begin(); hiter != list->end(); hiter++)
dvander@4112: 		{
dvander@4112: 			if (*hiter == hook)
dvander@4112: 			{
dvander@4112: 				list->erase(hiter);
dvander@4112: 				break;
dvander@4112: 			}
dvander@4112: 		}
dvander@4112: 
dvander@4113: 		if (hook->admin)
dvander@4113: 			hook->admin->group->hooks.remove(hook);
dvander@4113: 
dvander@4112: 		iter = pInfo->hooks.erase(iter);
dvander@4112: 		delete hook;
dvander@4112: 	}
dvander@1584: 
dvander@1584: 	RemoveConCmd(pInfo, name, is_read_safe, false);
dvander@1584: }
dvander@1584: 
dvander@4112: void ConCmdManager::OnPluginDestroyed(IPlugin *plugin)
dvander@1584: {
dvander@4112: 	PluginHookList *pList;
dvander@4112: 	if (!plugin->GetProperty("CommandList", (void **)&pList, true))
dvander@4112: 		return;
dvander@1584: 
dvander@4112: 	PluginHookList::iterator iter = pList->begin();
dvander@4112: 	while (iter != pList->end())
dvander@1584: 	{
dvander@4112: 		CmdHook *hook = *iter;
dvander@1584: 
dvander@4112: 		hook->info->hooks.remove(hook);
dvander@4113: 		if (hook->admin)
dvander@4113: 			hook->admin->group->hooks.remove(hook);
dvander@4112: 
dvander@4112: 		if (hook->info->hooks.empty())
dvander@4112: 			RemoveConCmd(hook->info, hook->info->pCmd->GetName(), true, true);
dvander@1584: 
dvander@4112: 		iter = pList->erase(iter);
dvander@4112: 		delete hook;
dvander@1584: 	}
dvander@4112: 
dvander@4112: 	delete pList;
dvander@1584: }
dvander@1584: 
skamonkey@3899: #if SOURCE_ENGINE == SE_DOTA
skamonkey@3942: void CommandCallback(const CCommandContext &context, const CCommand &command)
skamonkey@3899: {
skamonkey@3899: #elif SOURCE_ENGINE >= SE_ORANGEBOX
faluco@1541: void CommandCallback(const CCommand &command)
faluco@1541: {
faluco@1541: #else
ds@579: void CommandCallback()
ds@579: {
faluco@1541: 	CCommand command;
faluco@1541: #endif
dvander@1544: 
dvander@1544: 	g_HL2.PushCommandStack(&command);
dvander@1544: 
faluco@1541: 	g_ConCmds.InternalDispatch(command);
dvander@1544: 
dvander@1544: 	g_HL2.PopCommandStack();
ds@579: }
ds@579: 
ds@579: void ConCmdManager::SetCommandClient(int client)
ds@579: {
ds@579: 	m_CmdClient = client + 1;
ds@579: }
ds@579: 
dvander@3302: ConCmdInfo *ConCmdManager::FindInTrie(const char *name)
dvander@3302: {
dvander@3302: 	ConCmdInfo *pInfo;
dvander@4076: 	if (!m_Cmds.retrieve(name, &pInfo))
dvander@3302: 		return NULL;
dvander@3302: 	return pInfo;
dvander@3302: }
dvander@3302: 
dvander@3302: ConCmdList::iterator ConCmdManager::FindInList(const char *cmd)
dvander@3302: {
dvander@3302: 	List<ConCmdInfo *>::iterator iter = m_CmdList.begin();
dvander@3302: 
dvander@3302: 	while (iter != m_CmdList.end())
dvander@3302: 	{
dvander@3302: 		if (strcasecmp((*iter)->pCmd->GetName(), cmd) == 0)
dvander@3302: 			break;
dvander@3302: 		iter++;
dvander@3302: 	}
dvander@3302: 
dvander@3302: 	return iter;
dvander@3302: }
dvander@3302: 
dvander@1335: ResultType ConCmdManager::DispatchClientCommand(int client, const char *cmd, int args, ResultType type)
ds@579: {
ds@579: 	ConCmdInfo *pInfo;
dvander@2546: 
dvander@3302: 	if ((pInfo = FindInTrie(cmd)) == NULL)
ds@579: 	{
dvander@3302: 		ConCmdList::iterator item = FindInList(cmd);
dvander@3302: 		if (item == m_CmdList.end())
dvander@2546: 			return type;
dvander@3302: 		pInfo = *item;
dvander@2546: 	}
dvander@2546: 
dvander@2546: 	cell_t result = type;
dvander@4112: 	for (CmdHookList::iterator iter = pInfo->hooks.begin(); iter != pInfo->hooks.end(); iter++)
dvander@2546: 	{
dvander@4112: 		CmdHook *hook = *iter;
dvander@4112: 
dvander@4112: 		if (hook->type == CmdHook::Server || !hook->pf->IsRunnable())
dvander@2546: 			continue;
dvander@4112: 
dvander@4112: 		if (hook->admin && !CheckAccess(client, cmd, hook->admin))
dvander@2546: 		{
dvander@2546: 			if (result < Pl_Handled)
dvander@2546: 				result = Pl_Handled;
dvander@2546: 			continue;
dvander@2546: 		}
dvander@4112: 
dvander@4112: 		hook->pf->PushCell(client);
dvander@4112: 		hook->pf->PushCell(args);
dvander@4112: 
dvander@4112: 		cell_t tempres = result;
dvander@4112: 		if (hook->pf->Execute(&tempres) == SP_ERROR_NONE)
dvander@2546: 		{
dvander@2546: 			if (tempres > result)
dvander@2546: 				result = tempres;
dvander@2546: 			if (result == Pl_Stop)
dvander@2546: 				break;
ds@579: 		}
ds@579: 	}
ds@579: 
dvander@4112: 	return (ResultType)result;
ds@579: }
ds@579: 
faluco@1541: void ConCmdManager::InternalDispatch(const CCommand &command)
ds@579: {
dvander@1369: 	int client = m_CmdClient;
dvander@1369: 
dvander@1369: 	if (client)
dvander@1369: 	{
dvander@1369: 		CPlayer *pPlayer = g_Players.GetPlayerByIndex(client);
dvander@1369: 		if (!pPlayer || !pPlayer->IsConnected())
dvander@1369: 			return;
dvander@1369: 	}
dvander@1369: 
ds@579: 	/**
ds@579: 	 * Note: Console commands will EITHER go through IServerGameDLL::ClientCommand,
ds@579: 	 * OR this dispatch.  They will NEVER go through both.
ds@579: 	 * -- 
ds@579: 	 * Whether or not it goes through the callback is determined by FCVAR_GAMEDLL
ds@579: 	 */
dvander@1544: 	const char *cmd = g_HL2.CurrentCommandName();
ds@579: 
dvander@3302: 	ConCmdInfo *pInfo = FindInTrie(cmd);
dvander@3302: 	if (pInfo == NULL)
ds@579: 	{
dvander@2245:         /* Unfortunately, we now have to do a slow lookup because Valve made client commands 
dvander@2245:          * case-insensitive.  We can't even use our sortedness.
dvander@2245:          */
dvander@2245:         if (client == 0 && !engine->IsDedicatedServer())
dvander@2245:             return;
dvander@2245: 
dvander@3302: 		ConCmdList::iterator item = FindInList(cmd);
dvander@3302: 		if (item == m_CmdList.end())
dvander@3302: 			return;
dvander@3302: 
dvander@3302: 		pInfo = *item;
ds@579: 	}
ds@579: 
dvander@1925: 	/* This is a hack to prevent say triggers from firing on messages that were 
dvander@1925: 	 * blocked because of flooding.  We won't remove this, but the hack will get 
dvander@1925: 	 * "nicer" when we expose explicit say hooks.
dvander@1925: 	 */
dvander@1927: 	if (g_ChatTriggers.WasFloodedMessage())
dvander@1925: 		return;
dvander@1925: 
ds@579: 	cell_t result = Pl_Continue;
faluco@1541: 	int args = command.ArgC() - 1;
ds@579: 
dvander@4112: 	// On a listen server, sometimes the server host's client index can be set
dvander@4112: 	// as 0. So index 1 is passed to the command callback to correct this
dvander@4112: 	// potential problem.
dvander@4112: 	int realClient = (!engine->IsDedicatedServer() && client == 0)
dvander@4112: 	                 ? g_Players.ListenClient()
dvander@4112: 	                 : client;
dvander@4112: 	int dedicatedClient = engine->IsDedicatedServer() ? 0 : g_Players.ListenClient();
ds@579: 
dvander@4112: 	for (CmdHookList::iterator iter = pInfo->hooks.begin(); iter != pInfo->hooks.end(); iter++)
ds@579: 	{
dvander@4112: 		CmdHook *hook = *iter;
dvander@4112: 
dvander@4112: 		if (!hook->pf->IsRunnable())
dvander@4112: 			continue;
dvander@4112: 
dvander@4112: 		if (hook->type == CmdHook::Server)
ds@579: 		{
dvander@4112: 			// Don't execute unless we're in the server console.
dvander@4112: 			if (realClient != dedicatedClient)
ds@579: 				continue;
dvander@4112: 		} else {
dvander@4112: 			// Check admin rights if needed. realClient isn't needed since we
dvander@4112: 			// should bypass admin checks if client == 0 anyway.
dvander@4112: 			if (client && hook->admin && !CheckAccess(client, cmd, hook->admin))
ds@579: 			{
ds@579: 				if (result < Pl_Handled)
ds@579: 					result = Pl_Handled;
ds@579: 				continue;
ds@579: 			}
ds@856: 
dvander@4112: 			// Client hooks get a client argument.
dvander@4112: 			hook->pf->PushCell(realClient);
dvander@4112: 		}
ds@856: 
dvander@4112: 		hook->pf->PushCell(args);
dvander@1996: 
dvander@4112: 		cell_t tempres = result;
dvander@4112: 		if (hook->pf->Execute(&tempres) == SP_ERROR_NONE)
dvander@4112: 		{
dvander@4112: 			if (tempres > result)
dvander@4112: 				result = tempres;
dvander@4112: 			if (result == Pl_Stop)
dvander@4112: 				break;
ds@579: 		}
ds@579: 	}
ds@579: 
ds@579: 	if (result >= Pl_Handled)
ds@579: 	{
ds@579: 		if (!pInfo->sourceMod)
ds@579: 			RETURN_META(MRES_SUPERCEDE);
ds@579: 		return;
ds@579: 	}
ds@579: }
ds@579: 
skamonkey@3389: bool ConCmdManager::CheckClientCommandAccess(int client, const char *cmd, FlagBits cmdflags)
ds@579: {
dvander@1349: 	if (cmdflags == 0 || client == 0)
ds@579: 	{
ds@579: 		return true;
ds@579: 	}
ds@579: 
ds@856: 	/* If running listen server, then client 1 is the server host and should have 'root' access */
ds@856: 	if (client == 1 && !engine->IsDedicatedServer())
ds@856: 	{
ds@856: 		return true;
ds@856: 	}
ds@856: 
ds@579: 	CPlayer *player = g_Players.GetPlayerByIndex(client);
dvander@1119: 	if (!player 
dvander@1119: 		|| player->GetEdict() == NULL
dvander@1119: 		|| player->IsFakeClient())
ds@579: 	{
ds@579: 		return false;
ds@579: 	}
ds@579: 
skamonkey@3389: 	return CheckAdminCommandAccess(player->GetAdminId(), cmd, cmdflags);
skamonkey@3389: }
skamonkey@3389: 
skamonkey@3389: bool ConCmdManager::CheckAdminCommandAccess(AdminId adm, const char *cmd, FlagBits cmdflags)
skamonkey@3389: {
ds@579: 	if (adm != INVALID_ADMIN_ID)
ds@579: 	{
ds@579: 		FlagBits bits = g_Admins.GetAdminFlags(adm, Access_Effective);
ds@579: 
ds@579: 		/* root knows all, WHOA */
ds@579: 		if ((bits & ADMFLAG_ROOT) == ADMFLAG_ROOT)
ds@579: 		{
ds@579: 			return true;
ds@579: 		}
ds@579: 
dvander@1300: 		/* Check for overrides
dvander@1300: 		 * :TODO: is it worth optimizing this?
dvander@1300: 		 */
ds@579: 		unsigned int groups = g_Admins.GetAdminGroupCount(adm);
ds@579: 		GroupId gid;
ds@579: 		OverrideRule rule;
ds@579: 		bool override = false;
ds@579: 		for (unsigned int i=0; i<groups; i++)
ds@579: 		{
ds@579: 			gid = g_Admins.GetAdminGroup(adm, i, NULL);
ds@579: 			/* First get group-level override */
ds@579: 			override = g_Admins.GetGroupCommandOverride(gid, cmd, Override_CommandGroup, &rule);
ds@579: 			/* Now get the specific command override */
ds@579: 			if (g_Admins.GetGroupCommandOverride(gid, cmd, Override_Command, &rule))
ds@579: 			{
ds@579: 				override = true;
ds@579: 			}
ds@579: 			if (override)
ds@579: 			{
ds@579: 				if (rule == Command_Allow)
ds@579: 				{
ds@579: 					return true;
dvander@1925: 				}
dvander@1925: 				else if (rule == Command_Deny)
dvander@1925: 				{
dvander@1300: 					return false;
ds@579: 				}
ds@579: 			}
ds@579: 		}
dvander@1300: 
dvander@1300: 		/* See if our other flags match */
dvander@1300: 		if ((bits & cmdflags) == cmdflags)
dvander@1300: 		{
dvander@1300: 			return true;
dvander@1300: 		}
ds@579: 	}
ds@579: 
dvander@1119: 	return false;
dvander@1119: }
dvander@1119: 
dvander@1119: bool ConCmdManager::CheckAccess(int client, const char *cmd, AdminCmdInfo *pAdmin)
dvander@1119: {
skamonkey@3389: 	if (CheckClientCommandAccess(client, cmd, pAdmin->eflags))
ds@579: 	{
dvander@1119: 		return true;
ds@579: 	}
dvander@1119: 
kyle@4217: 	CPlayer *pPlayer = g_Players.GetPlayerByIndex(client);
kyle@4217: 	if (!pPlayer)
kyle@4217: 	{
kyle@4217: 		return false;
kyle@4217: 	}
kyle@4217: 
ds@579: 	/* If we got here, the command failed... */
ds@579: 	char buffer[128];
dvander@3004: 	if (!logicore.CoreTranslate(buffer, sizeof(buffer), "%T", 2, NULL, "No Access", &client))
ds@579: 	{
dvander@1925: 		UTIL_Format(buffer, sizeof(buffer), "You do not have access to this command");
ds@579: 	}
ds@579: 
dvander@895: 	unsigned int replyto = g_ChatTriggers.GetReplyTo();
dvander@895: 	if (replyto == SM_REPLY_CONSOLE)
dvander@895: 	{
dvander@895: 		char fullbuffer[192];
dvander@1925: 		UTIL_Format(fullbuffer, sizeof(fullbuffer), "[SM] %s.\n", buffer);
kyle@4217: 		pPlayer->PrintToConsole(fullbuffer);
dvander@1925: 	}
dvander@1925: 	else if (replyto == SM_REPLY_CHAT)
dvander@1925: 	{
dvander@895: 		char fullbuffer[192];
dvander@1925: 		UTIL_Format(fullbuffer, sizeof(fullbuffer), "[SM] %s.", buffer);
dvander@895: 		g_HL2.TextMsg(client, HUD_PRINTTALK, fullbuffer);
dvander@895: 	}
ds@579: 	
ds@579: 	return false;
ds@579: }
ds@579: 
ds@579: bool ConCmdManager::AddAdminCommand(IPluginFunction *pFunction, 
ds@579: 									 const char *name, 
ds@579: 									 const char *group,
ds@579: 									 int adminflags,
ds@579: 									 const char *description, 
ds@579: 									 int flags)
ds@579: {
ds@579: 	ConCmdInfo *pInfo = AddOrFindCommand(name, description, flags);
ds@579: 
ds@579: 	if (!pInfo)
ds@579: 		return false;
ds@579: 
dvander@4113: 	GroupMap::Insert i = m_CmdGrps.findForAdd(group);
dvander@4113: 	if (!i.found())
ds@579: 	{
dvander@4113: 		if (!m_CmdGrps.add(i, group))
dvander@4113: 			return false;
dvander@4113: 		i->value = NoAddRef(new CommandGroup());
ds@579: 	}
dvander@4113: 	Ref<CommandGroup> cmdgroup = i->value;
ds@579: 
dvander@4113: 	CmdHook *pHook = new CmdHook(CmdHook::Client, pInfo, pFunction, description);
dvander@4113: 	pHook->admin = new AdminCmdInfo(cmdgroup, adminflags);
ds@579: 
ds@579: 	/* First get the command group override, if any */
ds@579: 	bool override = g_Admins.GetCommandOverride(group, 
ds@579: 		Override_CommandGroup,
dvander@4112: 		&(pHook->admin->eflags));
ds@579: 
ds@579: 	/* Next get the command override, if any */
ds@579: 	if (g_Admins.GetCommandOverride(name,
ds@579: 		Override_Command,
dvander@4112: 		&(pHook->admin->eflags)))
ds@579: 	{
ds@579: 		override = true;
ds@579: 	}
ds@579: 
ds@579: 	/* Assign normal flags if there were no overrides */
ds@579: 	if (!override)
dvander@4112: 		pHook->admin->eflags = pHook->admin->flags;
dvander@4112: 	pInfo->eflags = pHook->admin->eflags;
ds@579: 
dvander@4113: 	cmdgroup->hooks.append(pHook);
dvander@4112: 	pInfo->hooks.append(pHook);
dvander@4112: 	RegisterInPlugin(pHook);
ds@579: 	return true;
ds@579: }
ds@579: 
ds@579: bool ConCmdManager::AddServerCommand(IPluginFunction *pFunction, 
ds@579: 									  const char *name, 
ds@579: 									  const char *description, 
ds@579: 									  int flags)
ds@579: 
ds@579: {
ds@579: 	ConCmdInfo *pInfo = AddOrFindCommand(name, description, flags);
ds@579: 
ds@579: 	if (!pInfo)
ds@579: 		return false;
ds@579: 
dvander@4112: 	CmdHook *pHook = new CmdHook(CmdHook::Server, pInfo, pFunction, description);
ds@579: 
dvander@4112: 	pInfo->hooks.append(pHook);
dvander@4112: 	RegisterInPlugin(pHook);
ds@579: 	return true;
ds@579: }
ds@579: 
dvander@4112: void RegisterInPlugin(CmdHook *hook)
ds@579: {
dvander@4112: 	PluginHookList *pList;
ds@579: 
dvander@4112: 	IPlugin *pPlugin = scripts->FindPluginByContext(hook->pf->GetParentContext());
dvander@4112: 	if (!pPlugin->GetProperty("CommandList", (void **)&pList))
dvander@4112: 	{
dvander@4112: 		pList = new PluginHookList();
dvander@4112: 		pPlugin->SetProperty("CommandList", pList);
dvander@4112: 	}
dvander@4112: 
dvander@4112: 	const char *orig = hook->info->pCmd->GetName();
ds@579: 
ds@579: 	/* Insert this into the help list, SORTED alphabetically. */
dvander@4112: 	PluginHookList::iterator iter = pList->begin();
ds@579: 	while (iter != pList->end())
ds@579: 	{
dvander@4112: 		const char *cmd = (*iter)->info->pCmd->GetName();
ds@579: 		if (strcmp(orig, cmd) < 0)
ds@579: 		{
dvander@4112: 			pList->insertBefore(iter, hook);
dvander@4112: 			return;
ds@579: 		}
ds@579: 		iter++;
ds@579: 	}
ds@579: 
dvander@4112: 	pList->append(hook);
ds@579: }
ds@579: 
ds@579: void ConCmdManager::AddToCmdList(ConCmdInfo *info)
ds@579: {
ds@579: 	List<ConCmdInfo *>::iterator iter = m_CmdList.begin();
ds@579: 	ConCmdInfo *pInfo;
ds@579: 	bool inserted = false;
ds@579: 	const char *orig = NULL;
ds@579: 
ds@579: 	orig = info->pCmd->GetName();
ds@579: 
ds@579: 	/* Insert this into the help list, SORTED alphabetically. */
ds@579: 	while (iter != m_CmdList.end())
ds@579: 	{
ds@579: 		const char *cmd = NULL;
ds@579: 		pInfo = (*iter);
ds@579: 		cmd = pInfo->pCmd->GetName();
ds@579: 		if (strcmp(orig, cmd) < 0)
ds@579: 		{
ds@579: 			m_CmdList.insert(iter, info);
ds@579: 			inserted = true;
ds@579: 			break;
ds@579: 		}
ds@579: 		iter++;
ds@579: 	}
ds@579: 
ds@579: 	if (!inserted)
ds@579: 	{
ds@579: 		m_CmdList.push_back(info);
ds@579: 	}
ds@579: }
ds@579: 
dvander@1435: void ConCmdManager::UpdateAdminCmdFlags(const char *cmd, OverrideType type, FlagBits bits, bool remove)
ds@579: {
ds@579: 	if (type == Override_Command)
ds@579: 	{
dvander@4112: 		ConCmdInfo *pInfo;
dvander@4076: 		if (!m_Cmds.retrieve(cmd, &pInfo))
ds@579: 			return;
ds@579: 
dvander@4112: 		for (CmdHookList::iterator iter = pInfo->hooks.begin(); iter != pInfo->hooks.end(); iter++)
ds@579: 		{
dvander@4112: 			if (!iter->admin)
dvander@4112: 				continue;
dvander@4112: 
dvander@4112: 			if (!remove)
dvander@4112: 				iter->admin->eflags = bits;
dvander@4112: 			else
dvander@4112: 				iter->admin->eflags = iter->admin->flags;
dvander@4112: 			pInfo->eflags = iter->admin->eflags;
ds@579: 		}
dvander@1584: 	}
dvander@1584: 	else if (type == Override_CommandGroup)
dvander@1584: 	{
dvander@4113: 		GroupMap::Result r = m_CmdGrps.find(cmd);
dvander@4113: 		if (!r.found())
ds@579: 			return;
ds@579: 
dvander@4113: 		Ref<CommandGroup> group(r->value);
dvander@4113: 
dvander@4113: 		for (PluginHookList::iterator iter = group->hooks.begin(); iter != group->hooks.end(); iter++)
ds@579: 		{
dvander@4113: 			CmdHook *hook = *iter;
dvander@4113: 			if (remove)
dvander@4113: 				hook->admin->eflags = bits;
dvander@4113: 			else
dvander@4113: 				hook->admin->eflags = hook->admin->flags;
dvander@4113: 			hook->info->eflags = hook->admin->eflags;
ds@579: 		}
ds@579: 	}
ds@579: }
ds@579: 
dvander@1584: void ConCmdManager::RemoveConCmd(ConCmdInfo *info, const char *name, bool is_read_safe, bool untrack)
ds@579: {
dvander@1584: 	/* Remove from the trie */
dvander@4076: 	m_Cmds.remove(name);
dvander@1584: 
ds@579: 	/* Remove console-specific information
ds@579: 	 * This should always be true as of right now
ds@579: 	 */
ds@579: 	if (info->pCmd)
ds@579: 	{
ds@579: 		if (info->sourceMod)
ds@579: 		{
ds@579: 			/* Unlink from SourceMM */
faluco@1541: 			g_SMAPI->UnregisterConCommandBase(g_PLAPI, info->pCmd);
ds@579: 			/* Delete the command's memory */
ds@579: 			char *new_help = const_cast<char *>(info->pCmd->GetHelpText());
ds@579: 			char *new_name = const_cast<char *>(info->pCmd->GetName());
ds@579: 			delete [] new_help;
ds@579: 			delete [] new_name;
ds@579: 			delete info->pCmd;
dvander@1584: 		}
dvander@1584: 		else
dvander@1584: 		{
dvander@1584: 			if (is_read_safe)
dvander@1584: 			{
dvander@1584: 				/* Remove the external hook */
kyle@3819: 				SH_REMOVE_HOOK(ConCommand, Dispatch, info->pCmd, SH_STATIC(CommandCallback), false);
dvander@1584: 			}
dvander@1584: 			if (untrack)
dvander@1584: 			{
dvander@1584: 				UntrackConCommandBase(info->pCmd, this);
dvander@1584: 			}
ds@579: 		}
ds@579: 	}
ds@579: 	
ds@579: 	/* Remove from list */
ds@579: 	m_CmdList.remove(info);
ds@579: 
ds@579: 	delete info;
ds@579: }
ds@579: 
dvander@893: bool ConCmdManager::LookForSourceModCommand(const char *cmd)
dvander@893: {
dvander@893: 	ConCmdInfo *pInfo;
dvander@4076: 	if (!m_Cmds.retrieve(cmd, &pInfo))
dvander@893: 		return false;
dvander@893: 
dvander@4112: 	return pInfo->sourceMod && !pInfo->hooks.empty();
dvander@893: }
dvander@893: 
dvander@1349: bool ConCmdManager::LookForCommandAdminFlags(const char *cmd, FlagBits *pFlags)
dvander@1349: {
dvander@1349: 	ConCmdInfo *pInfo;
dvander@4076: 	if (!m_Cmds.retrieve(cmd, &pInfo))
dvander@1349: 		return false;
dvander@1349: 
dvander@4112: 	*pFlags = pInfo->eflags;
asherkin@3687: 	return true;
dvander@1349: }
dvander@1349: 
ds@579: ConCmdInfo *ConCmdManager::AddOrFindCommand(const char *name, const char *description, int flags)
ds@579: {
ds@579: 	ConCmdInfo *pInfo;
dvander@4076: 	if (!m_Cmds.retrieve(name, &pInfo))
ds@579: 	{
dvander@3302: 		ConCmdList::iterator item = FindInList(name);
dvander@3302: 		if (item != m_CmdList.end())
dvander@3302: 			return *item;
dvander@3302: 
ds@579: 		pInfo = new ConCmdInfo();
ds@579: 		/* Find the commandopan */
ds@2400: 		ConCommand *pCmd = FindCommand(name);
ds@579: 
ds@579: 		if (!pCmd)
ds@579: 		{
ds@579: 			/* Note that we have to duplicate because the source might not be 
ds@579: 			 * a static string, and these expect static memory.
ds@579: 			 */
ds@579: 			if (!description)
ds@579: 			{
ds@579: 				description = "";
ds@579: 			}
ds@579: 			char *new_name = sm_strdup(name);
ds@579: 			char *new_help = sm_strdup(description);
ds@579: 			pCmd = new ConCommand(new_name, CommandCallback, new_help, flags);
ds@579: 			pInfo->sourceMod = true;
dvander@1584: 		}
dvander@1584: 		else
dvander@1584: 		{
dvander@1584: 			TrackConCommandBase(pCmd, this);
kyle@3819: 			SH_ADD_HOOK(ConCommand, Dispatch, pCmd, SH_STATIC(CommandCallback), false);
ds@579: 		}
ds@579: 
ds@579: 		pInfo->pCmd = pCmd;
ds@579: 
dvander@4076: 		m_Cmds.insert(name, pInfo);
ds@579: 		AddToCmdList(pInfo);
ds@579: 	}
ds@579: 
ds@579: 	return pInfo;
ds@579: }
ds@579: 
faluco@1541: void ConCmdManager::OnRootConsoleCommand(const char *cmdname, const CCommand &command)
ds@579: {
faluco@1541: 	if (command.ArgC() >= 3)
ds@579: 	{
faluco@1541: 		const char *text = command.Arg(2);
dvander@723: 
dvander@3872: 		IPlugin *pPlugin = scripts->FindPluginByConsoleArg(text);
ds@579: 
ds@579: 		if (!pPlugin)
ds@579: 		{
dvander@1902: 			g_RootMenu.ConsolePrint("[SM] Plugin \"%s\" was not found.", text);
ds@579: 			return;
ds@579: 		}
ds@579: 
ds@582: 		const sm_plugininfo_t *plinfo = pPlugin->GetPublicInfo();
ds@582: 		const char *plname = IS_STR_FILLED(plinfo->name) ? plinfo->name : pPlugin->GetFilename();
ds@582: 
dvander@4112: 		PluginHookList *pList;
ds@579: 		if (!pPlugin->GetProperty("CommandList", (void **)&pList))
ds@579: 		{
ds@582: 			g_RootMenu.ConsolePrint("[SM] No commands found for: %s", plname);
ds@579: 			return;
ds@579: 		}
dvander@4112: 		if (pList->empty())
ds@579: 		{
ds@582: 			g_RootMenu.ConsolePrint("[SM] No commands found for: %s", plname);
ds@579: 			return;
ds@579: 		}
ds@579: 
ds@579: 		const char *type = NULL;
ds@579: 		const char *name;
ds@579: 		const char *help;
dvander@4112: 		g_RootMenu.ConsolePrint("[SM] Listing commands for: %s", plname);
ds@582: 		g_RootMenu.ConsolePrint("  %-17.16s %-8.7s %s", "[Name]", "[Type]", "[Help]");
dvander@4112: 		for (PluginHookList::iterator iter = pList->begin(); iter != pList->end(); iter++)
ds@579: 		{
dvander@4112: 			CmdHook *hook = *iter;
dvander@4112: 			if (hook->type == CmdHook::Server)
ds@579: 				type = "server";
dvander@4112: 			else 
dvander@4112: 				type = hook->info->eflags == 0 ? "console" : "admin";
dvander@4112: 
dvander@4112: 			name = hook->info->pCmd->GetName();
dvander@4112: 			if (hook->helptext.length())
dvander@4112: 				help = hook->helptext.chars();
dvander@1584: 			else
dvander@4112: 				help = hook->info->pCmd->GetHelpText();
ds@582: 			g_RootMenu.ConsolePrint("  %-17.16s %-12.11s %s", name, type, help);		
ds@579: 		}
ds@579: 
ds@579: 		return;
ds@579: 	}
ds@579: 
ds@579: 	g_RootMenu.ConsolePrint("[SM] Usage: sm cmds <plugin #>");
ds@579: }