#include "sm_srvcmds.h"

 #include "sm_version.h"

 #include "sm_stringutil.h"

 RootConsoleMenu g_RootMenu;

 RootConsoleMenu::RootConsoleMenu()

 {

 	m_pCommands = sm_trie_create();

 }

 RootConsoleMenu::~RootConsoleMenu()

 {

 	sm_trie_destroy(m_pCommands);

 	List<ConsoleEntry *>::iterator iter;

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

 	{

 		delete (*iter);

 	}

 	m_Menu.clear();

 }

 void RootConsoleMenu::OnSourceModStartup(bool late)

 {

 	ConCommandBaseMgr::OneTimeInit(this);

 	AddRootConsoleCommand("version", "Display version information", this);

 	AddRootConsoleCommand("credits", "Display credits listing", this);

 }

 void RootConsoleMenu::OnSourceModShutdown()

 {

 	RemoveRootConsoleCommand("credits", this);

 	RemoveRootConsoleCommand("version", this);

 }

 bool RootConsoleMenu::RegisterConCommandBase(ConCommandBase *pCommand)

 {

 	META_REGCVAR(pCommand);

 	return true;

 }

 void RootConsoleMenu::ConsolePrint(const char *fmt, ...)

 {

 	char buffer[512];

 	va_list ap;

 	va_start(ap, fmt);

 	size_t len = vsnprintf(buffer, sizeof(buffer), fmt, ap);

 	va_end(ap);

 	if (len >= sizeof(buffer) - 1)

 	{

 		buffer[510] = '\n';

 		buffer[511] = '\0';

 	} else {

 		buffer[len++] = '\n';

 		buffer[len] = '\0';

 	}

 	META_CONPRINT(buffer);

 }

 bool RootConsoleMenu::AddRootConsoleCommand(const char *cmd, const char *text, IRootConsoleCommand *pHandler)

 {

 	if (sm_trie_retrieve(m_pCommands, cmd, NULL))

 	{

 		return false;

 	}

 	sm_trie_insert(m_pCommands, cmd, pHandler);

 	/* Sort this into the menu */

 	List<ConsoleEntry *>::iterator iter = m_Menu.begin();

 	ConsoleEntry *pEntry;

 	bool inserted = false;

 	while (iter != m_Menu.end())

 	{

 		pEntry = (*iter);

 		if (strcmp(cmd, pEntry->command.c_str()) < 0)

 		{

 			ConsoleEntry *pNew = new ConsoleEntry;

 			pNew->command.assign(cmd);

 			pNew->description.assign(text);

 			m_Menu.insert(iter, pNew);

 			inserted = true;

 			break;

 		}

 		iter++;

 	}

 	if (!inserted)

 	{

 		ConsoleEntry *pNew = new ConsoleEntry;

 		pNew->command.assign(cmd);

 		pNew->description.assign(text);

 		m_Menu.push_back(pNew);

 	}

 	return true;

 }

 bool RootConsoleMenu::RemoveRootConsoleCommand(const char *cmd, IRootConsoleCommand *pHandler)

 {

 	/* Sanity tests */

 	IRootConsoleCommand *object;

 	if (sm_trie_retrieve(m_pCommands, cmd, (void **)&object))

 	{

 		if (object != pHandler)

 		{

 			return false;

 		}

 	} else {

 		return false;

 	}

 	sm_trie_delete(m_pCommands, cmd);

 	List<ConsoleEntry *>::iterator iter;

 	ConsoleEntry *pEntry;

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

 	{

 		pEntry = (*iter);

 		if (pEntry->command.compare(cmd) == 0)

 		{

 			delete pEntry;

 			m_Menu.erase(iter);

 			break;

 		}

 	}

 	return true;

 }

 void RootConsoleMenu::DrawGenericOption(const char *cmd, const char *text)

 {

 	char buffer[255];

 	size_t len, cmdlen = strlen(cmd);

 	len = snprintf(buffer, sizeof(buffer), "    %s", cmd);

 	if (cmdlen < 16)

 	{

 		size_t num = 16 - cmdlen;

 		for (size_t i = 0; i < num; i++)

 		{

 			buffer[len++] = ' ';

 		}

 		len += snprintf(&buffer[len], sizeof(buffer) - len, " - %s", text);

 		ConsolePrint("%s", buffer);

 	}

 }

 void RootConsoleMenu::GotRootCmd()

 {

 	unsigned int argnum = GetArgumentCount();

 	if (argnum >= 2)

 	{

 		const char *cmd = GetArgument(1);

 		IRootConsoleCommand *pHandler;

 		if (sm_trie_retrieve(m_pCommands, cmd, (void **)&pHandler))

 		{

 			pHandler->OnRootConsoleCommand(cmd, argnum);

 			return;

 		}

 	}

 	ConsolePrint("SourceMod Menu:");

 	ConsolePrint("Usage: sm <command> [arguments]");

 	List<ConsoleEntry *>::iterator iter;

 	ConsoleEntry *pEntry;

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

 	{

 		pEntry = (*iter);

 		DrawGenericOption(pEntry->command.c_str(), pEntry->description.c_str());

 	}

 }

 const char *RootConsoleMenu::GetArgument(unsigned int argno)

 {

 	return engine->Cmd_Argv(argno);

 }

 const char *RootConsoleMenu::GetArguments()

 {

 	return engine->Cmd_Args();

 }

 unsigned int RootConsoleMenu::GetArgumentCount()

 {

 	return engine->Cmd_Argc();

 }

 void RootConsoleMenu::OnRootConsoleCommand(const char *cmd, unsigned int argcount)

 {

 	if (strcmp(cmd, "credits") == 0)

 	{

 		ConsolePrint(" SourceMod was developed by AlliedModders, LLC.");

 		ConsolePrint(" Development would not have been possible without the following people:");

 		ConsolePrint("  David \"BAILOPAN\" Anderson, lead developer");

 		ConsolePrint("  Borja \"faluco\" Ferrer, Core developer");

 		ConsolePrint("  Scott \"Damaged Soul\" Ehlert, SourceMM developer");

 		ConsolePrint("  Pavol \"PM OnoTo\" Marko, SourceHook developer");

 		ConsolePrint(" Special thanks to Viper of GameConnect");

 		ConsolePrint(" Special thanks to Mani of Mani-Admin-Plugin");

 		ConsolePrint(" http://www.sourcemod.net/");

 	} else if (strcmp(cmd, "version") == 0) {

 		ConsolePrint(" SourceMod Version Information:");

 		ConsolePrint("    SourceMod Version: %s", SOURCEMOD_VERSION);

 		ConsolePrint("    JIT Version: %s, %s", g_pVM->GetVMName(), g_pVM->GetVersionString());

 		ConsolePrint("    JIT Settings: %s", g_pVM->GetCPUOptimizations());

 		ConsolePrint("    http://www.sourcemod.net/");

 	}

 }

 CON_COMMAND(sm, "SourceMod Menu")

 {

 	g_RootMenu.GotRootCmd();

 }