dvander@1488: /**
dvander@1488:  * vim: set ts=4 :
dvander@1488:  * =============================================================================
dvander@1488:  * SourceMod Sample Extension
dvander@1488:  * Copyright (C) 2004-2007 AlliedModders LLC.  All rights reserved.
dvander@1488:  * =============================================================================
dvander@1488:  *
dvander@1488:  * This program is free software; you can redistribute it and/or modify it under
dvander@1488:  * the terms of the GNU General Public License, version 3.0, as published by the
dvander@1488:  * Free Software Foundation.
dvander@1488:  * 
dvander@1488:  * This program is distributed in the hope that it will be useful, but WITHOUT
dvander@1488:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
dvander@1488:  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
dvander@1488:  * details.
dvander@1488:  *
dvander@1488:  * You should have received a copy of the GNU General Public License along with
dvander@1488:  * this program.  If not, see <http://www.gnu.org/licenses/>.
dvander@1488:  *
dvander@1488:  * As a special exception, AlliedModders LLC gives you permission to link the
dvander@1488:  * code of this program (as well as its derivative works) to "Half-Life 2," the
dvander@1488:  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
dvander@1488:  * by the Valve Corporation.  You must obey the GNU General Public License in
dvander@1488:  * all respects for all other code used.  Additionally, AlliedModders LLC grants
dvander@1488:  * this exception to all derivative works.  AlliedModders LLC defines further
dvander@1488:  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
dvander@1488:  * or <http://www.sourcemod.net/license.php>.
dvander@1488:  *
dvander@1491:  * Version: $Id$
dvander@1488:  */
dvander@1488: 
dvander@1488: #include <stdlib.h>
dvander@1489: #include <stdarg.h>
dvander@1488: #include "TopMenu.h"
dvander@1488: 
dvander@1488: struct obj_by_name_t
dvander@1488: {
dvander@1488: 	unsigned int obj_index;
dvander@1488: 	char name[64];
dvander@1488: };
dvander@1488: 
dvander@1488: int _SortObjectNamesDescending(const void *ptr1, const void *ptr2);
dvander@1488: unsigned int strncopy(char *dest, const char *src, size_t count);
dvander@1488: size_t UTIL_Format(char *buffer, size_t maxlength, const char *fmt, ...);
dvander@1488: 
dvander@1488: TopMenu::TopMenu(ITopMenuObjectCallbacks *callbacks)
dvander@1488: {
dvander@1488: 	m_clients = NULL;
dvander@1488: 	m_SerialNo = 1;
dvander@1488: 	m_pTitle = callbacks;
dvander@1488: 	m_max_clients = 0;
dvander@1488: 
dvander@1488: 	if (playerhelpers->IsServerActivated())
dvander@1488: 	{
dvander@1488: 		CreatePlayers(playerhelpers->GetMaxClients());
dvander@1488: 	}
dvander@1488: }
dvander@1488: 
dvander@1488: TopMenu::~TopMenu()
dvander@1488: {
dvander@1488: 	/* Delete all categories */
dvander@1497: 	while (m_Categories.size())
dvander@1488: 	{
dvander@1497: 		RemoveFromMenu(m_Categories[0]->obj->object_id);
dvander@1488: 	}
dvander@1488: 
dvander@1495: 	/* Remove all objects */
dvander@1497: 	for (size_t i = 0; i < m_Objects.size(); i++)
dvander@1488: 	{
dvander@1495: 		assert(m_Objects[i]->is_free == true);
dvander@1488: 		delete m_Objects[i];
dvander@1488: 	}
dvander@1488: 
dvander@1495: 	m_pTitle->OnTopMenuObjectRemoved(this, 0);
dvander@1495: 
dvander@1488: 	/* Delete all cached config entries */
dvander@1497: 	for (size_t i = 0; i < m_Config.cats.size(); i++)
dvander@1488: 	{
dvander@1488: 		delete m_Config.cats[i];
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	/* Sweep players */
dvander@1497: 	for (size_t i = 0; i <= (size_t)m_max_clients; i++)
dvander@1488: 	{
dvander@1488: 		TearDownClient(&m_clients[i]);
dvander@1488: 	}
dvander@1488: 	delete [] m_clients;
dvander@1488: }
dvander@1488: 
dvander@1488: void TopMenu::OnClientConnected(int client)
dvander@1488: {
dvander@1488: 	if (m_clients == NULL)
dvander@1488: 	{
dvander@1488: 		return;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	topmenu_player_t *player = &m_clients[client];
dvander@1488: 	TearDownClient(player);
dvander@1488: }
dvander@1488: 
dvander@1488: void TopMenu::OnClientDisconnected(int client)
dvander@1488: {
dvander@1488: 	if (m_clients == NULL)
dvander@1488: 	{
dvander@1488: 		return;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	topmenu_player_t *player = &m_clients[client];
dvander@1488: 	TearDownClient(player);
dvander@1488: }
dvander@1488: 
dvander@1488: void TopMenu::OnServerActivated(int max_clients)
dvander@1488: {
dvander@1488: 	if (m_clients == NULL)
dvander@1488: 	{
dvander@1488: 		CreatePlayers(max_clients);
dvander@1488: 	}
dvander@1488: }
dvander@1488: 
dvander@1488: unsigned int TopMenu::AddToMenu(const char *name,
dvander@1488: 								TopMenuObjectType type,
dvander@1488: 								ITopMenuObjectCallbacks *callbacks,
dvander@1488: 								IdentityToken_t *owner,
dvander@1488: 								const char *cmdname,
dvander@1488: 								FlagBits flags,
dvander@1488: 								unsigned int parent)
dvander@1488: {
dvander@1488: 	/* Sanity checks */
dvander@1488: 	if (type == TopMenuObject_Category && parent != 0)
dvander@1488: 	{
dvander@1488: 		return 0;
dvander@1488: 	}
dvander@1488: 	else if (type == TopMenuObject_Item && parent == 0)
dvander@1488: 	{
dvander@1488: 		return 0;
dvander@1488: 	}
dvander@1488: 	else if (m_ObjLookup.retrieve(name) != NULL)
dvander@1488: 	{
dvander@1488: 		return 0;
dvander@1488: 	}
dvander@1495: 	else if (type != TopMenuObject_Item && type != TopMenuObject_Category)
dvander@1495: 	{
dvander@1495: 		return 0;
dvander@1495: 	}
dvander@1488: 
dvander@1488: 	/* If we're adding an item, make sure the parent is valid, 
dvander@1488: 	 * and that the parent is a category.
dvander@1488: 	 */
dvander@1488: 	topmenu_object_t *parent_obj = NULL;
dvander@1488: 	topmenu_category_t *parent_cat = NULL;
dvander@1488: 	if (type == TopMenuObject_Item)
dvander@1488: 	{
dvander@1488: 		/* Check parent index.  Note it will be >= 1 here. */
dvander@1488: 		if (parent > m_Objects.size() || m_Objects[parent - 1]->is_free)
dvander@1488: 		{
dvander@1488: 			return 0;
dvander@1488: 		}
dvander@1488: 		parent_obj = m_Objects[parent - 1];
dvander@1488: 
dvander@1488: 		/* Find an equivalent pointer in the category array. */
dvander@1488: 		for (size_t i = 0; i < m_Categories.size(); i++)
dvander@1488: 		{
dvander@1488: 			if (m_Categories[i]->obj == parent_obj)
dvander@1488: 			{
dvander@1488: 				parent_cat = m_Categories[i];
dvander@1488: 				break;
dvander@1488: 			}
dvander@1488: 		}
dvander@1488: 
dvander@1488: 		/* If none was found, leave. */
dvander@1488: 		if (parent_cat == NULL)
dvander@1488: 		{
dvander@1488: 			return 0;
dvander@1488: 		}
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	/* Re-use an old object pointer if we can. */
dvander@1488: 	topmenu_object_t *obj = NULL;
dvander@1488: 	for (size_t i = 0; i < m_Objects.size(); i++)
dvander@1488: 	{
dvander@1488: 		if (m_Objects[i]->is_free == true)
dvander@1488: 		{
dvander@1488: 			obj = m_Objects[i];
dvander@1488: 			break;
dvander@1488: 		}
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	/* Otherwise, allocate a new one. */
dvander@1488: 	if (obj == NULL)
dvander@1488: 	{
dvander@1488: 		obj = new topmenu_object_t;
dvander@1488: 		obj->object_id = ((unsigned int)m_Objects.size()) + 1;
dvander@1488: 		m_Objects.push_back(obj);
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	/* Initialize the object's properties. */
dvander@1488: 	obj->callbacks = callbacks;
dvander@1488: 	obj->flags = flags;
dvander@1488: 	obj->owner = owner;
dvander@1488: 	obj->type = type;
dvander@1488: 	obj->is_free = false;
dvander@1488: 	obj->parent = parent_obj;
dvander@1488: 	strncopy(obj->name, name, sizeof(obj->name));
dvander@1488: 	strncopy(obj->cmdname, cmdname ? cmdname : "", sizeof(obj->cmdname));
dvander@1488: 
dvander@1488: 	if (obj->type == TopMenuObject_Category)
dvander@1488: 	{
dvander@1488: 		/* Create a new category entry */
dvander@1488: 		topmenu_category_t *cat = new topmenu_category_t;
dvander@1488: 		cat->obj = obj;
dvander@1488: 		cat->reorder = false;
dvander@1488: 		cat->serial = 1;
dvander@1488: 
dvander@1488: 		/* Add it, then update our serial change number. */
dvander@1488: 		m_Categories.push_back(cat);
dvander@1488: 		m_SerialNo++;
dvander@1488: 
dvander@1488: 		/* Updating sorting info */
dvander@1488: 		m_bCatsNeedResort = true;
dvander@1488: 	}
dvander@1488: 	else if (obj->type == TopMenuObject_Item)
dvander@1488: 	{
dvander@1488: 		/* Update the category, mark it as needing changes */
dvander@1488: 		parent_cat->obj_list.push_back(obj);
dvander@1488: 		parent_cat->reorder = true;
dvander@1488: 		parent_cat->serial++;
dvander@1496: 
dvander@1496: 		/* If the category just went from 0 to 1 items, mark it as 
dvander@1496: 		 * changed, so clients get the category drawn.
dvander@1496: 		 */
dvander@1496: 		if (parent_cat->obj_list.size() == 1)
dvander@1496: 		{
dvander@1496: 			m_SerialNo++;
dvander@1496: 		}
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	m_ObjLookup.insert(name, obj);
dvander@1488: 
dvander@1488: 	return obj->object_id;
dvander@1488: }
dvander@1488: 
dvander@1488: void TopMenu::RemoveFromMenu(unsigned int object_id)
dvander@1488: {
dvander@1488: 	if (object_id == 0 
dvander@1488: 		|| object_id > m_Objects.size() 
dvander@1488: 		|| m_Objects[object_id - 1]->is_free)
dvander@1488: 	{
dvander@1488: 		return;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	topmenu_object_t *obj = m_Objects[object_id - 1];
dvander@1488: 
dvander@1488: 	m_ObjLookup.remove(obj->name);
dvander@1488: 
dvander@1488: 	if (obj->type == TopMenuObject_Category)
dvander@1488: 	{
dvander@1495: 		/* Find it in the category list. */
dvander@1488: 		for (size_t i = 0; i < m_Categories.size(); i++)
dvander@1488: 		{
dvander@1488: 			if (m_Categories[i]->obj == obj)
dvander@1488: 			{
dvander@1495: 				/* Mark all children as removed + free.  Note we could
dvander@1495: 				 * call into RemoveMenuItem() for this, but it'd be very 
dvander@1495: 				 * inefficient!
dvander@1495: 				 */
dvander@1495: 				topmenu_category_t *cat = m_Categories[i];
dvander@1495: 				for (size_t j = 0; j < m_Categories[i]->obj_list.size(); j++)
dvander@1495: 				{
dvander@1495: 					cat->obj_list[j]->callbacks->OnTopMenuObjectRemoved(this, cat->obj_list[j]->object_id);
dvander@1495: 					cat->obj_list[j]->is_free = true;
dvander@1495: 				}
dvander@1495: 				
dvander@1495: 				/* Remove the category from the list, then delete it. */
dvander@1488: 				m_Categories.erase(m_Categories.iterAt(i));
dvander@1495: 				delete cat;
dvander@1488: 				break;
dvander@1488: 			}
dvander@1488: 		}
dvander@1495: 
dvander@1495: 		/* Update the root as changed. */
dvander@1488: 		m_SerialNo++;
dvander@1488: 		m_bCatsNeedResort = true;
dvander@1488: 	}
dvander@1488: 	else if (obj->type == TopMenuObject_Item)
dvander@1488: 	{
dvander@1488: 		/* Find the category this item is in. */
dvander@1488: 		topmenu_category_t *parent_cat = NULL;
dvander@1488: 		for (size_t i = 0; i < m_Categories.size(); i++)
dvander@1488: 		{
dvander@1488: 			if (m_Categories[i]->obj == obj->parent)
dvander@1488: 			{
dvander@1488: 				parent_cat = m_Categories[i];
dvander@1488: 				break;
dvander@1488: 			}
dvander@1488: 		}
dvander@1488: 
dvander@1488: 		/* Erase it from the category's lists. */
dvander@1488: 		if (parent_cat)
dvander@1488: 		{
dvander@1488: 			for (size_t i = 0; i < parent_cat->obj_list.size(); i++)
dvander@1488: 			{
dvander@1488: 				if (parent_cat->obj_list[i] == obj)
dvander@1488: 				{
dvander@1488: 					parent_cat->obj_list.erase(parent_cat->obj_list.iterAt(i));
dvander@1496: 
dvander@1496: 					/* If this category now has no items, mark root as changed 
dvander@1496: 					 * so clients won't get the category drawn anymore.
dvander@1496: 					 */
dvander@1496: 					if (parent_cat->obj_list.size() == 0)
dvander@1496: 					{
dvander@1496: 						m_SerialNo++;
dvander@1496: 					}
dvander@1488: 					break;
dvander@1488: 				}
dvander@1488: 			}
dvander@1495: 
dvander@1495: 			/* Update the category as changed. */
dvander@1488: 			parent_cat->reorder = true;
dvander@1488: 			parent_cat->serial++;
dvander@1488: 		}
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	/* Finally, mark the object as free. */
dvander@1488: 	obj->is_free = true;
dvander@1495: 
dvander@1495: 	/* The callbacks pointer is still valid, so fire away! */
dvander@1495: 	obj->callbacks->OnTopMenuObjectRemoved(this, object_id);
dvander@1488: }
dvander@1488: 
dvander@1488: bool TopMenu::DisplayMenu(int client, unsigned int hold_time, TopMenuPosition position)
dvander@1488: {
dvander@1488: 	if (m_clients == NULL)
dvander@1488: 	{
dvander@1488: 		return false;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	IGamePlayer *pPlayer = playerhelpers->GetGamePlayer(client);
dvander@1488: 	if (!pPlayer->IsInGame())
dvander@1488: 	{
dvander@1488: 		return false;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	UpdateClientRoot(client, pPlayer);
dvander@1488: 
dvander@1488: 	topmenu_player_t *pClient = &m_clients[client];
dvander@1488: 	if (pClient->root == NULL)
dvander@1488: 	{
dvander@1488: 		return false;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	bool return_value = false;
dvander@1488: 
dvander@1488: 	if (position == TopMenuPosition_LastCategory && 
dvander@1488: 		pClient->last_category < m_Categories.size())
dvander@1488: 	{
dvander@1488: 		return_value = DisplayCategory(client, pClient->last_category, hold_time, true);
dvander@1488: 	}
dvander@1488: 	else if (position == TopMenuPosition_LastRoot)
dvander@1488: 	{
dvander@1488: 		pClient->root->DisplayAtItem(client, hold_time, pClient->last_root_pos);
dvander@1488: 	}
dvander@1488: 	else if (position == TopMenuPosition_Start)
dvander@1488: 	{
dvander@1488: 		pClient->last_position = 0;
dvander@1488: 		pClient->last_category = 0;
dvander@1488: 		return_value = pClient->root->Display(client, hold_time);
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	return return_value;
dvander@1488: }
dvander@1488: 
dvander@1488: bool TopMenu::DisplayCategory(int client, unsigned int category, unsigned int hold_time, bool last_position)
dvander@1488: {
dvander@1488: 	UpdateClientCategory(client, category);
dvander@1488: 
dvander@1488: 	topmenu_player_t *pClient = &m_clients[client];
dvander@1488: 	if (category >= pClient->cat_count
dvander@1488: 		|| pClient->cats[category].menu == NULL)
dvander@1488: 	{
dvander@1488: 		return false;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	bool return_value = false;
dvander@1488: 
dvander@1488: 	topmenu_player_category_t *player_cat = &(pClient->cats[category]);
dvander@1488: 
dvander@1488: 	pClient->last_category = category;
dvander@1488: 	if (last_position)
dvander@1488: 	{
dvander@1488: 		return_value = player_cat->menu->DisplayAtItem(client, hold_time, pClient->last_position);
dvander@1488: 	}
dvander@1488: 	else
dvander@1488: 	{
dvander@1488: 		return_value = player_cat->menu->Display(client, hold_time);
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	return return_value;
dvander@1488: }
dvander@1488: 
dvander@1488: void TopMenu::OnMenuSelect2(IBaseMenu *menu, int client, unsigned int item, unsigned int item_on_page)
dvander@1488: {
dvander@1488: 	const char *item_name = menu->GetItemInfo(item, NULL);
dvander@1488: 	if (!item_name)
dvander@1488: 	{
dvander@1488: 		return;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	topmenu_object_t *obj;
dvander@1488: 	topmenu_player_t *pClient = &m_clients[client];
dvander@1488: 	topmenu_object_t **pObject = m_ObjLookup.retrieve(item_name);
dvander@1488: 	if (pObject == NULL)
dvander@1488: 	{
dvander@1488: 		return;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	obj = *pObject;
dvander@1488: 
dvander@1488: 	/* We now have the object... what do we do with it? */
dvander@1488: 	if (obj->type == TopMenuObject_Category)
dvander@1488: 	{
dvander@1488: 		/* If it's a category, the user wants to view it.. */
dvander@1488: 		for (size_t i = 0; i < m_Categories.size(); i++)
dvander@1488: 		{
dvander@1488: 			if (m_Categories[i]->obj == obj)
dvander@1488: 			{
dvander@1488: 				pClient->last_root_pos = item_on_page;
dvander@1488: 				DisplayCategory(client, (unsigned int)i, MENU_TIME_FOREVER, false);
dvander@1488: 				break;
dvander@1488: 			}
dvander@1488: 		}
dvander@1488: 	}
dvander@1488: 	else
dvander@1488: 	{
dvander@1488: 		pClient->last_position = item_on_page;
dvander@1488: 		
dvander@1488: 		/* Re-check access in case this user had their credentials revoked */
dvander@1488: 		if (obj->cmdname[0] != '\0' && !adminsys->CheckAccess(client, obj->cmdname, obj->flags, false))
dvander@1488: 		{
dvander@1488: 			DisplayMenu(client, 0, TopMenuPosition_LastCategory);
dvander@1488: 			return;
dvander@1488: 		}
dvander@1488: 		
dvander@1488: 		/* Pass the information on to the callback */
dvander@1495: 		obj->callbacks->OnTopMenuSelectOption(this, client, obj->object_id);
dvander@1488: 	}
dvander@1488: }
dvander@1488: 
dvander@1488: void TopMenu::OnMenuDrawItem(IBaseMenu *menu, int client, unsigned int item, unsigned int &style)
dvander@1488: {
dvander@1488: 	const char *item_name = menu->GetItemInfo(item, NULL);
dvander@1488: 	if (!item_name)
dvander@1488: 	{
dvander@1488: 		return;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	topmenu_object_t *obj;
dvander@1488: 	topmenu_object_t **pObject = m_ObjLookup.retrieve(item_name);
dvander@1488: 	if (pObject == NULL)
dvander@1488: 	{
dvander@1488: 		return;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	obj = *pObject;
dvander@1488: 
dvander@1488: 	if (obj->cmdname[0] == '\0')
dvander@1488: 	{
dvander@1488: 		return;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	if (!adminsys->CheckAccess(client, obj->cmdname, obj->flags, false))
dvander@1488: 	{
dvander@1488: 		style = ITEMDRAW_IGNORE;
dvander@1488: 	}
dvander@1488: }
dvander@1488: 
dvander@1488: unsigned int TopMenu::OnMenuDisplayItem(IBaseMenu *menu,
dvander@1488: 								int client,
dvander@1488: 								IMenuPanel *panel,
dvander@1488: 								unsigned int item,
dvander@1488: 								const ItemDrawInfo &dr)
dvander@1488: {
dvander@1488: 	const char *item_name = menu->GetItemInfo(item, NULL);
dvander@1488: 	if (!item_name)
dvander@1488: 	{
dvander@1488: 		return 0;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	topmenu_object_t *obj;
dvander@1488: 	topmenu_object_t **pObject = m_ObjLookup.retrieve(item_name);
dvander@1488: 	if (pObject == NULL)
dvander@1488: 	{
dvander@1488: 		return 0;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	obj = *pObject;
dvander@1488: 
dvander@1488: 	/* Ask the object to render the text for this client */
dvander@1488: 	char renderbuf[64];
dvander@1495: 	obj->callbacks->OnTopMenuDrawOption(this, client, obj->object_id, renderbuf, sizeof(renderbuf));
dvander@1488: 
dvander@1488: 	/* Build the new draw info */
dvander@1488: 	ItemDrawInfo new_dr = dr;
dvander@1488: 	new_dr.display = renderbuf;
dvander@1488: 
dvander@1488: 	/* Man I love the menu API.  Ask the panel to draw the item and give the position 
dvander@1488: 	 * back to Core's renderer.  This way we don't have to worry about keeping the 
dvander@1488: 	 * render buffer static!
dvander@1488: 	 */
dvander@1488: 	return panel->DrawItem(new_dr);
dvander@1488: }
dvander@1488: 
dvander@1488: void TopMenu::OnMenuCancel(IBaseMenu *menu, int client, MenuCancelReason reason)
dvander@1488: {
dvander@1488: 	if (reason == MenuCancel_ExitBack)
dvander@1488: 	{
dvander@1488: 		/* If the client chose exit back, they were on a category menu, and we can
dvander@1488: 		 * now display the root menu from the last known position.
dvander@1488: 		 */
dvander@1488: 		DisplayMenu(client, 0, TopMenuPosition_LastRoot);
dvander@1488: 	}
dvander@1488: }
dvander@1488: 
dvander@1488: void TopMenu::UpdateClientRoot(int client, IGamePlayer *pGamePlayer)
dvander@1488: {
dvander@1488: 	topmenu_player_t *pClient = &m_clients[client];
dvander@1488: 	IGamePlayer *pPlayer = pGamePlayer ? pGamePlayer : playerhelpers->GetGamePlayer(client);
dvander@1488: 
dvander@1488: 	/* Determine if an update is necessary */
dvander@1488: 	bool is_update_needed = false;
dvander@1488: 	if (pClient->menu_serial != m_SerialNo)
dvander@1488: 	{
dvander@1488: 		is_update_needed = true;
dvander@1488: 	}
dvander@1488: 	else if (pPlayer->GetUserId() != pClient->user_id)
dvander@1488: 	{
dvander@1488: 		is_update_needed = true;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	/* If no update is needed at the root level, just leave now */
dvander@1488: 	if (!is_update_needed)
dvander@1488: 	{
dvander@1488: 		return;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	/* First we need to flush the cache... */
dvander@1488: 	TearDownClient(pClient);
dvander@1488: 
dvander@1488: 	/* Now, rebuild the category list, but don't create menus */
dvander@1488: 	if (m_Categories.size() == 0)
dvander@1488: 	{
dvander@1488: 		pClient->cat_count = 0;
dvander@1488: 		pClient->cats = NULL;
dvander@1488: 	}
dvander@1488: 	else
dvander@1488: 	{
dvander@1488: 		pClient->cat_count = (unsigned int)m_Categories.size();
dvander@1488: 		pClient->cats = new topmenu_player_category_t[pClient->cat_count];
dvander@1488: 		memset(pClient->cats, 0, sizeof(topmenu_player_category_t) * pClient->cat_count);
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	/* Re-sort the root categories if needed */
dvander@1488: 	SortCategoriesIfNeeded();
dvander@1488: 
dvander@1488: 	/* Build the root menu */
dvander@1488: 	IBaseMenu *root_menu = menus->GetDefaultStyle()->CreateMenu(this, myself->GetIdentity());
dvander@1488: 
dvander@1488: 	/* Add the sorted items */
dvander@1488: 	for (size_t i = 0; i < m_SortedCats.size(); i++)
dvander@1488: 	{
dvander@1496: 		if (m_Categories[m_SortedCats[i]]->obj_list.size() == 0)
dvander@1496: 		{
dvander@1496: 			continue;
dvander@1496: 		}
dvander@1488: 		root_menu->AppendItem(m_Categories[m_SortedCats[i]]->obj->name, ItemDrawInfo(""));
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	/* Now we need to handle un-sorted items.  This is slightly trickier, as we need to 
dvander@1488: 	 * pre-render each category name, and cache those names.  Phew!
dvander@1488: 	 */
dvander@1488: 	if (m_UnsortedCats.size())
dvander@1488: 	{
dvander@1488: 		obj_by_name_t *item_list = new obj_by_name_t[m_UnsortedCats.size()];
dvander@1488: 		for (size_t i = 0; i < m_UnsortedCats.size(); i++)
dvander@1488: 		{
dvander@1488: 			obj_by_name_t *temp_obj = &item_list[i];
dvander@1488: 			topmenu_object_t *obj = m_Categories[m_UnsortedCats[i]]->obj;
dvander@1495: 			obj->callbacks->OnTopMenuDrawOption(this,
dvander@1495: 				client, 
dvander@1488: 				obj->object_id,
dvander@1488: 				temp_obj->name,
dvander@1488: 				sizeof(temp_obj->name));
dvander@1488: 			temp_obj->obj_index = m_UnsortedCats[i];
dvander@1488: 		}
dvander@1488: 
dvander@1488: 		/* Sort our temp list */
dvander@1488: 		qsort(item_list, m_UnsortedCats.size(), sizeof(obj_by_name_t), _SortObjectNamesDescending);
dvander@1488: 
dvander@1488: 		/* Add the new sorted categories */
dvander@1497: 		for (size_t i = 0; i < m_UnsortedCats.size(); i++)
dvander@1488: 		{
dvander@1496: 			if (m_Categories[item_list[i].obj_index]->obj_list.size() == 0)
dvander@1496: 			{
dvander@1496: 				continue;
dvander@1496: 			}
dvander@1488: 			root_menu->AppendItem(m_Categories[item_list[i].obj_index]->obj->name, ItemDrawInfo(""));
dvander@1488: 		}
dvander@1488: 
dvander@1488: 		delete [] item_list;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	/* Set the menu's title */
dvander@1488: 	char renderbuf[128];
dvander@1495: 	m_pTitle->OnTopMenuDrawTitle(this, client, 0, renderbuf, sizeof(renderbuf));
dvander@1488: 	root_menu->SetDefaultTitle(renderbuf);
dvander@1488: 
dvander@1488: 	/* The client is now fully updated */
dvander@1488: 	pClient->root = root_menu;
dvander@1488: 	pClient->user_id = pPlayer->GetUserId();
dvander@1488: 	pClient->menu_serial = m_SerialNo;
dvander@1488: 	pClient->last_position = 0;
dvander@1488: 	pClient->last_category = 0;
dvander@1488: 	pClient->last_root_pos = 0;
dvander@1488: }
dvander@1488: 
dvander@1488: void TopMenu::UpdateClientCategory(int client, unsigned int category)
dvander@1488: {
dvander@1488: 	/* Update the client's root menu just in case it needs it.  This 
dvander@1488: 	 * will validate that we have both a valid client and a valid
dvander@1488: 	 * category structure for that client.
dvander@1488: 	 */
dvander@1488: 	UpdateClientRoot(client);
dvander@1488: 
dvander@1488: 	/* Now it's guaranteed that our category tables will be usable */
dvander@1488: 	topmenu_player_t *pClient = &m_clients[client];
dvander@1488: 	topmenu_category_t *cat = m_Categories[category];
dvander@1488: 	topmenu_player_category_t *player_cat = &(pClient->cats[category]);
dvander@1488: 
dvander@1488: 	/* Does the category actually need updating? */
dvander@1488: 	if (player_cat->serial == cat->serial)
dvander@1488: 	{
dvander@1488: 		return;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	/* Destroy any existing menu */
dvander@1488: 	if (player_cat->menu)
dvander@1488: 	{
dvander@1488: 		player_cat->menu->Destroy();
dvander@1488: 		player_cat->menu = NULL;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	if (pClient->last_category == category)
dvander@1488: 	{
dvander@1488: 		pClient->last_position = 0;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	IBaseMenu *cat_menu = menus->GetDefaultStyle()->CreateMenu(this, myself->GetIdentity());
dvander@1488: 	
dvander@1488: 	/* Categories get an "exit back" button */
dvander@1488: 	cat_menu->SetMenuOptionFlags(cat_menu->GetMenuOptionFlags() | MENUFLAG_BUTTON_EXITBACK);
dvander@1488: 
dvander@1488: 	/* Re-sort the category if needed */
dvander@1488: 	SortCategoryIfNeeded(category);
dvander@1488: 
dvander@1488: 	/* Build the menu with the sorted items first */
dvander@1488: 	for (size_t i = 0; i < cat->sorted.size(); i++)
dvander@1488: 	{
dvander@1488: 		cat_menu->AppendItem(cat->sorted[i]->name, ItemDrawInfo(""));
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	/* Now handle unsorted items */
dvander@1488: 	if (cat->unsorted.size())
dvander@1488: 	{
dvander@1488: 		/* Build a list of the item names */
dvander@1488: 		obj_by_name_t *item_list = new obj_by_name_t[cat->unsorted.size()];
dvander@1488: 		for (size_t i = 0; i < cat->unsorted.size(); i++)
dvander@1488: 		{
dvander@1488: 			obj_by_name_t *item = &item_list[i];
dvander@1488: 			topmenu_object_t *obj = cat->unsorted[i];
dvander@1495: 			obj->callbacks->OnTopMenuDrawOption(this,
dvander@1495: 				client,
dvander@1488: 				obj->object_id,
dvander@1488: 				item->name,
dvander@1488: 				sizeof(item->name));
dvander@1488: 			item->obj_index = (unsigned int)i;
dvander@1488: 		}
dvander@1488: 
dvander@1488: 		/* Sort the names */
dvander@1488: 		qsort(item_list, cat->unsorted.size(), sizeof(obj_by_name_t), _SortObjectNamesDescending);
dvander@1488: 
dvander@1488: 		/* Add to the menu */
dvander@1488: 		for (size_t i = 0; i < cat->unsorted.size(); i++)
dvander@1488: 		{
dvander@1488: 			cat_menu->AppendItem(cat->unsorted[item_list[i].obj_index]->name, ItemDrawInfo(""));
dvander@1488: 		}
dvander@1488: 
dvander@1488: 		delete [] item_list;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	/* Set the menu's title */
dvander@1488: 	char renderbuf[128];
dvander@1495: 	cat->obj->callbacks->OnTopMenuDrawTitle(this, 
dvander@1495: 		client, 
dvander@1495: 		cat->obj->object_id, 
dvander@1495: 		renderbuf, 
dvander@1495: 		sizeof(renderbuf));
dvander@1488: 	cat_menu->SetDefaultTitle(renderbuf);
dvander@1488: 
dvander@1488: 	/* We're done! */
dvander@1488: 	player_cat->menu = cat_menu;
dvander@1488: 	player_cat->serial = cat->serial;
dvander@1488: }
dvander@1488: 
dvander@1488: void TopMenu::SortCategoryIfNeeded(unsigned int category)
dvander@1488: {
dvander@1488: 	topmenu_category_t *cat = m_Categories[category];
dvander@1488: 	if (!cat->reorder)
dvander@1488: 	{
dvander@1488: 		return;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	cat->sorted.clear();
dvander@1488: 	cat->unsorted.clear();
dvander@1488: 
dvander@1488: 	if (cat->obj_list.size() == 0)
dvander@1488: 	{
dvander@1488: 		cat->reorder = false;
dvander@1488: 		return;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	CVector<unsigned int> to_sort;
dvander@1488: 	for (size_t i = 0; i < cat->obj_list.size(); i++)
dvander@1488: 	{
dvander@1488: 		to_sort.push_back(i);
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	/* Find a matching category in the configs */
dvander@1488: 	config_category_t *config_cat = NULL; 
dvander@1488: 	for (size_t i = 0; i < m_Config.cats.size(); i++)
dvander@1488: 	{
dvander@1488: 		if (strcmp(m_Config.strings.GetString(m_Config.cats[i]->name), cat->obj->name) == 0)
dvander@1488: 		{
dvander@1488: 			config_cat = m_Config.cats[i];
dvander@1488: 			break;
dvander@1488: 		}
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	/* If there is a matching category, build a pre-sorted item list */
dvander@1488: 	if (config_cat != NULL)
dvander@1488: 	{
dvander@1488: 		/* Find matching objects in this category */
dvander@1488: 		for (size_t i = 0; i < config_cat->commands.size(); i++)
dvander@1488: 		{
dvander@1488: 			const char *config_name = m_Config.strings.GetString(config_cat->commands[i]);
dvander@1488: 			for (size_t j = 0; j < to_sort.size(); j++)
dvander@1488: 			{
dvander@1488: 				if (strcmp(config_name, cat->obj_list[to_sort[j]]->name) == 0)
dvander@1488: 				{
dvander@1488: 					/* Place in the final list, then remove from the temporary list */
dvander@1488: 					cat->sorted.push_back(cat->obj_list[to_sort[j]]);
dvander@1488: 					to_sort.erase(to_sort.iterAt(j));
dvander@1488: 					break;
dvander@1488: 				}
dvander@1488: 			}
dvander@1488: 		}
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	/* Push any remaining items onto the unsorted list */
dvander@1488: 	for (size_t i = 0; i < to_sort.size(); i++)
dvander@1488: 	{
dvander@1488: 		cat->unsorted.push_back(cat->obj_list[to_sort[i]]);
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	cat->reorder = false;
dvander@1488: }
dvander@1488: 
dvander@1488: void TopMenu::SortCategoriesIfNeeded()
dvander@1488: {
dvander@1488: 	if (!m_bCatsNeedResort)
dvander@1488: 	{
dvander@1488: 		return;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	/* Clear sort results */
dvander@1488: 	m_SortedCats.clear();
dvander@1488: 	m_UnsortedCats.clear();
dvander@1488: 
dvander@1488: 	if (m_Categories.size() == 0)
dvander@1488: 	{
dvander@1488: 		m_bCatsNeedResort = false;
dvander@1488: 		return;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	CVector<unsigned int> to_sort;
dvander@1488: 	for (unsigned int i = 0; i < (unsigned int)m_Categories.size(); i++)
dvander@1488: 	{
dvander@1488: 		to_sort.push_back(i);
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	/* If we have any predefined categories, add them in as they appear. */
dvander@1488: 	for (size_t i= 0; i < m_Config.cats.size(); i++)
dvander@1488: 	{
dvander@1488: 		/* Find this category and map it in if we can */
dvander@1488: 		for (size_t j = 0; j < to_sort.size(); j++)
dvander@1488: 		{
dvander@1488: 			if (strcmp(m_Config.strings.GetString(m_Config.cats[i]->name), 
dvander@1488: 					   m_Categories[to_sort[j]]->obj->name) == 0)
dvander@1488: 			{
dvander@1488: 				/* Add to the real list and remove from the temporary */
dvander@1488: 				m_SortedCats.push_back(to_sort[j]);
dvander@1488: 				to_sort.erase(to_sort.iterAt(j));
dvander@1488: 				break;
dvander@1488: 			}
dvander@1488: 		}
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	/* Push any remaining items onto the unsorted list */
dvander@1488: 	for (size_t i = 0; i < to_sort.size(); i++)
dvander@1488: 	{
dvander@1488: 		m_UnsortedCats.push_back(to_sort[i]);
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	m_bCatsNeedResort = false;
dvander@1488: }
dvander@1488: 
dvander@1488: void TopMenu::CreatePlayers(int max_clients)
dvander@1488: {
dvander@1488: 	m_max_clients = max_clients;
dvander@1488: 	m_clients = (topmenu_player_t *)malloc(sizeof(topmenu_player_t) * (max_clients + 1));
dvander@1488: 	memset(m_clients, 0, sizeof(topmenu_player_t) * (max_clients + 1));
dvander@1488: }
dvander@1488: 
dvander@1488: void TopMenu::TearDownClient(topmenu_player_t *player)
dvander@1488: {
dvander@1488: 	if (player->cats != NULL)
dvander@1488: 	{
dvander@1488: 		for (unsigned int i = 0; i < player->cat_count; i++)
dvander@1488: 		{
dvander@1488: 			topmenu_player_category_t *player_cat = &(player->cats[i]);
dvander@1488: 			if (player_cat->menu != NULL)
dvander@1488: 			{
dvander@1488: 				player_cat->menu->Destroy();
dvander@1488: 			}
dvander@1488: 		}
dvander@1488: 		delete [] player->cats;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	if (player->root != NULL)
dvander@1488: 	{
dvander@1488: 		player->root->Destroy();
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	memset(player, 0, sizeof(topmenu_player_t));
dvander@1488: }
dvander@1488: 
dvander@1488: bool TopMenu::LoadConfiguration(const char *file, char *error, size_t maxlength)
dvander@1488: {
dvander@1488: 	SMCParseError err;
dvander@1488: 	unsigned int line = 0, col = 0;
dvander@1488: 
dvander@1488: 	if ((err = textparsers->ParseFile_SMC(file, this, &line, &col))
dvander@1488: 		!= SMCParse_Okay)
dvander@1488: 	{
dvander@1488: 		const char *err_string = textparsers->GetSMCErrorString(err);
dvander@1488: 		if (!err_string)
dvander@1488: 		{
dvander@1488: 			err_string = "Unknown";
dvander@1488: 		}
dvander@1488: 
dvander@1488: 		UTIL_Format(error, maxlength, "%s", err_string);
dvander@1488: 
dvander@1488: 		return false;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	return true;
dvander@1488: }
dvander@1488: 
dvander@1495: bool TopMenu::OnIdentityRemoval(IdentityToken_t *owner)
dvander@1495: {
dvander@1495: 	/* First sweep the categories owned by us */
dvander@1495: 	CVector<unsigned int> obj_list;
dvander@1495: 	for (size_t i = 0; i < m_Categories.size(); i++)
dvander@1495: 	{
dvander@1495: 		if (m_Categories[i]->obj->owner == owner)
dvander@1495: 		{
dvander@1495: 			obj_list.push_back(m_Categories[i]->obj->object_id);
dvander@1495: 		}
dvander@1495: 	}
dvander@1495: 
dvander@1495: 	for (size_t i = 0; i < obj_list.size(); i++)
dvander@1495: 	{
dvander@1495: 		RemoveFromMenu(obj_list[i]);
dvander@1495: 	}
dvander@1495: 
dvander@1495: 	/* Now we can look for actual items */
dvander@1495: 	for (size_t i = 0; i < m_Objects.size(); i++)
dvander@1495: 	{
dvander@1495: 		if (m_Objects[i]->is_free)
dvander@1495: 		{
dvander@1495: 			continue;
dvander@1495: 		}
dvander@1495: 		if (m_Objects[i]->owner == owner)
dvander@1495: 		{
dvander@1495: 			assert(m_Objects[i]->type != TopMenuObject_Category);
dvander@1495: 			RemoveFromMenu(m_Objects[i]->object_id);
dvander@1495: 		}
dvander@1495: 	}
dvander@1495: 
dvander@1495: 	return true;
dvander@1495: }
dvander@1495: 
dvander@1488: #define PARSE_STATE_NONE		0
dvander@1488: #define PARSE_STATE_CATEGORY	1
dvander@1488: unsigned int ignore_parse_level = 0;
dvander@1488: unsigned int current_parse_state = 0;
dvander@1488: config_category_t *cur_cat = NULL;
dvander@1488: 
dvander@1488: void TopMenu::ReadSMC_ParseStart()
dvander@1488: {
dvander@1488: 	current_parse_state = PARSE_STATE_NONE;
dvander@1488: 	ignore_parse_level = 0;
dvander@1488: 	cur_cat = NULL;
dvander@1488: 
dvander@1488: 	/* Reset the old config */
dvander@1488: 	m_Config.strings.Reset();
dvander@1488: 	for (size_t i = 0; i < m_Config.cats.size(); i++)
dvander@1488: 	{
dvander@1488: 		delete m_Config.cats[i];
dvander@1488: 	}
dvander@1488: 	m_Config.cats.clear();
dvander@1488: }
dvander@1488: 
dvander@1488: SMCParseResult TopMenu::ReadSMC_NewSection(const char *name, bool opt_quotes)
dvander@1488: {
dvander@1488: 	if (ignore_parse_level)
dvander@1488: 	{
dvander@1488: 		ignore_parse_level++;
dvander@1488: 	}
dvander@1488: 	else
dvander@1488: 	{
dvander@1488: 		if (current_parse_state == PARSE_STATE_NONE)
dvander@1488: 		{
dvander@1488: 			cur_cat = new config_category_t;
dvander@1488: 			cur_cat->name = m_Config.strings.AddString(name);
dvander@1488: 			m_Config.cats.push_back(cur_cat);
dvander@1488: 			current_parse_state = PARSE_STATE_CATEGORY;
dvander@1488: 		}
dvander@1488: 		else
dvander@1488: 		{
dvander@1488: 			ignore_parse_level = 1;
dvander@1488: 		}
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	return SMCParse_Continue;
dvander@1488: }
dvander@1488: 
dvander@1488: SMCParseResult TopMenu::ReadSMC_KeyValue(const char *key, const char *value, bool key_quotes, bool value_quotes)
dvander@1488: {
dvander@1488: 	if (ignore_parse_level > 0
dvander@1488: 		|| current_parse_state != PARSE_STATE_CATEGORY
dvander@1488: 		|| cur_cat == NULL)
dvander@1488: 	{
dvander@1488: 		return SMCParse_Continue;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	if (strcmp(key, "item") == 0)
dvander@1488: 	{
dvander@1488: 		cur_cat->commands.push_back(m_Config.strings.AddString(value));
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	return SMCParse_Continue;
dvander@1488: }
dvander@1488: 
dvander@1488: SMCParseResult TopMenu::ReadSMC_LeavingSection()
dvander@1488: {
dvander@1488: 	if (ignore_parse_level)
dvander@1488: 	{
dvander@1488: 		ignore_parse_level--;
dvander@1488: 	}
dvander@1488: 	else
dvander@1488: 	{
dvander@1488: 		if (current_parse_state == PARSE_STATE_CATEGORY)
dvander@1488: 		{
dvander@1488: 			cur_cat = NULL;
dvander@1488: 			current_parse_state = PARSE_STATE_NONE;
dvander@1488: 		}
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	return SMCParse_Continue;
dvander@1488: }
dvander@1488: 
dvander@1495: unsigned int TopMenu::FindCategory(const char *name)
dvander@1495: {
dvander@1495: 	topmenu_object_t **p_obj = m_ObjLookup.retrieve(name);
dvander@1495: 	if (!p_obj)
dvander@1495: 	{
dvander@1495: 		return 0;
dvander@1495: 	}
dvander@1495: 
dvander@1495: 	topmenu_object_t *obj = *p_obj;
dvander@1495: 	if (obj->type != TopMenuObject_Category)
dvander@1495: 	{
dvander@1495: 		return 0;
dvander@1495: 	}
dvander@1495: 
dvander@1495: 	return obj->object_id;
dvander@1495: }
dvander@1495: 
dvander@1488: int _SortObjectNamesDescending(const void *ptr1, const void *ptr2)
dvander@1488: {
dvander@1488: 	obj_by_name_t *obj1 = (obj_by_name_t *)ptr1;
dvander@1488: 	obj_by_name_t *obj2 = (obj_by_name_t *)ptr2;
dvander@1488: 	return strcmp(obj1->name, obj2->name);
dvander@1488: }
dvander@1488: 
dvander@1488: unsigned int strncopy(char *dest, const char *src, size_t count)
dvander@1488: {
dvander@1488: 	if (!count)
dvander@1488: 	{
dvander@1488: 		return 0;
dvander@1488: 	}
dvander@1488: 
dvander@1488: 	char *start = dest;
dvander@1488: 	while ((*src) && (--count))
dvander@1488: 	{
dvander@1488: 		*dest++ = *src++;
dvander@1488: 	}
dvander@1488: 	*dest = '\0';
dvander@1488: 
dvander@1488: 	return (dest - start);
dvander@1488: }
dvander@1488: 
dvander@1488: size_t UTIL_Format(char *buffer, size_t maxlength, const char *fmt, ...)
dvander@1488: {
dvander@1488: 	va_list ap;
dvander@1488: 	va_start(ap, fmt);
dvander@1488: 	size_t len = vsnprintf(buffer, maxlength, fmt, ap);
dvander@1488: 	va_end(ap);
dvander@1488: 
dvander@1488: 	if (len >= maxlength)
dvander@1488: 	{
dvander@1488: 		buffer[maxlength - 1] = '\0';
dvander@1488: 		return (maxlength - 1);
dvander@1488: 	}
dvander@1488: 	else
dvander@1488: 	{
dvander@1488: 		return len;
dvander@1488: 	}
dvander@1488: }