/**

 * vim: set ts=4 :

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

 * SourceMod (C)2004-2007 AlliedModders LLC.  All rights reserved.

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

 *

 * This file is not open source and may not be copied without explicit

 * written permission of AlliedModders LLC.  This file may not be redistributed 

 * in whole or significant part.

 * For information, see LICENSE.txt or http://www.sourcemod.net/license.php

 *

 * Version: $Id$

 */

#include <stdarg.h>

#include <time.h>

#include "MenuManager.h"

#include "sm_stringutil.h"

#include "sourcemm_api.h"

#include "PlayerManager.h"

#include "MenuStyle_Valve.h"

#include "ShareSys.h"

MenuManager g_Menus;

/*************************************

 *************************************

 **** BROADCAST HANDLING WRAPPERS ****

 *************************************

 *************************************/

BroadcastHandler::BroadcastHandler(IMenuHandler *handler) : m_pHandler(handler), numClients(0)

{

}

unsigned int BroadcastHandler::GetMenuAPIVersion2()

{

	return m_pHandler->GetMenuAPIVersion2();

}

void BroadcastHandler::OnMenuCancel(IBaseMenu *menu, int client, MenuCancelReason reason)

{

	m_pHandler->OnMenuCancel(menu, client, reason);

}

void BroadcastHandler::OnMenuDisplay(IBaseMenu *menu, int client, IMenuDisplay *display)

{

	numClients++;

	m_pHandler->OnMenuDisplay(menu, client, display);

}

void BroadcastHandler::OnBroadcastEnd(IBaseMenu *menu)

{

	g_Menus.FreeBroadcastHandler(this);

}

void BroadcastHandler::OnMenuSelect(IBaseMenu *menu, int client, unsigned int item)

{

	m_pHandler->OnMenuSelect(menu, client, item);

}

void BroadcastHandler::OnMenuEnd(IBaseMenu *menu)

{

	assert(numClients > 0);

	/* Only fire if all clients have gotten a menu end */

	if (--numClients == 0)

	{

		IMenuHandler *pHandler = m_pHandler;

		OnBroadcastEnd(menu);

		pHandler->OnMenuEnd(menu);

	}

}

VoteHandler::VoteHandler(IMenuVoteHandler *handler) 

: BroadcastHandler(handler), m_pVoteHandler(handler)

{

}

void VoteHandler::Initialize(IBaseMenu *menu)

{

	unsigned int numItems = menu->GetItemCount();

	if (m_counts.size() >= numItems)

	{

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

		{

			m_counts[i] = 0;

		}

	} else {

		for (size_t i=0; i<m_counts.size(); i++)

		{

			m_counts[i] = 0;

		}

		m_counts.resize(numItems, 0);

	}

}

void VoteHandler::OnMenuSelect(IBaseMenu *menu, int client, unsigned int item)

{

	if (item < numItems)

	{

		m_counts[item]++;

	}

	BroadcastHandler::OnMenuSelect(menu, client, item);

}

void VoteHandler::OnBroadcastEnd(IBaseMenu *menu)

{

	m_ties.clear();

	size_t highest = 0;

	for (size_t i=1; i<numItems; i++)

	{

		if (m_counts[i] > m_counts[highest])

		{

			m_ties.clear();

			highest = i;

		} else if (m_counts[i] == m_counts[highest]) {

			m_ties.push_back(i);

		}

	}

	if (m_ties.size())

	{

		m_ties.push_back(highest);

		srand(static_cast<unsigned int>(time(NULL)));

		highest = m_ties[rand() % m_ties.size()];

	}

	m_pVoteHandler->OnMenuVoteEnd(menu, highest);

	g_Menus.FreeVoteHandler(this);

}

/*******************************

 *******************************

 ******** MENU MANAGER *********

 *******************************

 *******************************/

MenuManager::MenuManager()

{

	m_Styles.push_back(&g_ValveMenuStyle);

	SetDefaultStyle(&g_ValveMenuStyle);

}

void MenuManager::OnSourceModAllInitialized()

{

	g_ShareSys.AddInterface(NULL, this);

}

void MenuManager::OnSourceModAllShutdown()

{

	while (!m_BroadcastHandlers.empty())

	{

		delete m_BroadcastHandlers.front();

		m_BroadcastHandlers.pop();

	}

	while (!m_VoteHandlers.empty())

	{

		delete m_VoteHandlers.front();

		m_VoteHandlers.pop();

	}

}

bool MenuManager::SetDefaultStyle(IMenuStyle *style)

{

	if (!style)

	{

		return false;

	}

	m_pDefaultStyle = style;

	return true;

}

IMenuStyle *MenuManager::GetStyle(unsigned int index)

{

	if (index >= GetStyleCount())

	{

		return NULL;

	}

	return m_Styles[index];

}

void MenuManager::AddStyle(IMenuStyle *style)

{

	m_Styles.push_back(style);

}

unsigned int MenuManager::GetStyleCount()

{

	return (unsigned int)m_Styles.size();

}

IMenuStyle *MenuManager::FindStyleByName(const char *name)

{

	unsigned int count = GetStyleCount();

	for (unsigned int i=0; i<count; i++)

	{

		IMenuStyle *ptr = GetStyle(i);

		if (strcasecmp(ptr->GetStyleName(), name) == 0)

		{

			return ptr;

		}

	}

	return NULL;

}

inline bool IsSlotItem(IMenuDisplay *display,

					   unsigned int style)

{

	if (!display->CanDrawItem(style))

	{

	return false;

	}

	if ((style & ITEMDRAW_IGNORE) == ITEMDRAW_IGNORE)

	{

		return false;

	}

	if (style & ITEMDRAW_RAWLINE)

	{

		return false;

	}

	return true;

}

IMenuDisplay *MenuManager::RenderMenu(int client, menu_states_t &md, ItemOrder order)

{

	IBaseMenu *menu = md.menu;

	if (!menu)

	{

		return NULL;

	}

	struct

	{

		unsigned int position;

		ItemDrawInfo draw;

	} drawItems[10];

	/* Figure out how many items to draw */

	IMenuStyle *style = menu->GetDrawStyle();

	unsigned int pgn = menu->GetPagination();

	unsigned int maxItems = style->GetMaxPageItems();

	if (pgn != MENU_NO_PAGINATION)

	{

		maxItems = pgn;

	}

	unsigned int totalItems = menu->GetItemCount();

	unsigned int startItem = 0;

	/* For pagination, find the starting point. */

	if (pgn != MENU_NO_PAGINATION)

	{

		if (order == ItemOrder_Ascending)

		{

			startItem = md.lastItem;

			/* This shouldn't happen with well-coded menus.

			 * If the item is out of bounds, switch the order to

			 * Items_Descending and make us start from the top.

			 */

			if (startItem >= totalItems)

			{

				startItem = totalItems - 1;

				order = ItemOrder_Descending;

			}

		} else if (order == ItemOrder_Descending) {

			startItem = md.firstItem;

			/* This shouldn't happen with well-coded menus.

			 * If searching backwards doesn't give us enough room,

			 * start from the beginning and change to ascending.

			 */

			if (startItem <= maxItems)

			{

				startItem = 0;

				order = ItemOrder_Ascending;

			}

		}

	}

	/* Get our Display pointer and initialize some crap */

	IMenuDisplay *display = menu->CreateDisplay();

	IMenuHandler *mh = md.mh;

	bool foundExtra = false;

	unsigned int extraItem = 0;

	/**

	 * We keep searching until:

	 * 1) There are no more items

	 * 2) We reach one OVER the maximum number of slot items

	 * 3) We have reached maxItems and pagination is MENU_NO_PAGINATION

	 */

	unsigned int i = startItem;

	unsigned int foundItems = 0;

	while (true)

	{

		ItemDrawInfo &dr = drawItems[foundItems].draw;

		/* Is the item valid? */

		if (menu->GetItemInfo(i, &dr) != NULL)

		{

			/* Ask the user to change the style, if necessary */

			mh->OnMenuDrawItem(menu, client, i, dr.style);

			/* Check if it's renderable */

			if (IsSlotItem(display, dr.style))

			{

				/* If we've already found the max number of items,

				 * This means we should just cancel out and log our

				 * "last item."

				 */

				if (foundItems >= maxItems)

				{

					foundExtra = true;

					extraItem = i;

					break;

				}

				drawItems[foundItems++].position = i;

			}

		}

		/* If there's no pagination, stop once the menu is full. */

		if (pgn == MENU_NO_PAGINATION)

		{

			/* If we've filled up, then stop */

			if (foundItems >= maxItems)

			{

				break;

			}

		}

		/* If we're descending and this is the first item, stop */

		if (order == ItemOrder_Descending)

		{

			if (i == 0)

			{

				break;

			}

			i--;

		} 

		/* If we're ascending and this is the last item, stop */

		else if (order == ItemOrder_Ascending)

		{

			if (i >= totalItems - 1)

			{

				break;

			}

			i++;

		}

	}

	/* There were no items to draw! */

	if (!foundItems)

	{

		display->DeleteThis();

		return NULL;

	}

	/* Check initial buttons */

	bool displayPrev = false;

	bool displayNext = false;

	if (foundExtra)

	{

		if (order == ItemOrder_Descending)

		{

			displayPrev = true;

			md.firstItem = extraItem;

		} else if (order == ItemOrder_Ascending) {

			displayNext = true;

			md.lastItem = extraItem;

		}

	}

	/**

	 * If we're paginated, we have to find if there is another page.

	 * Sadly, the only way to do this is to try drawing more items!

	 */

	if (pgn != MENU_NO_PAGINATION)

	{

		unsigned int lastItem = 0;

		ItemDrawInfo dr;

		/* Find the last feasible item to search from. */

		if (order == ItemOrder_Descending)

		{

			lastItem = drawItems[0].position;

			if (lastItem >= totalItems - 1)

			{

				goto skip_search;

			}

			while (++lastItem < totalItems)

			{

				if (menu->GetItemInfo(lastItem, &dr) != NULL)

				{

					mh->OnMenuDrawItem(menu, client, lastItem, dr.style);

					if (IsSlotItem(display, dr.style))

					{

						displayNext = true;

						md.lastItem = lastItem;

						break;

					}

				}

			}

		} else if (order == ItemOrder_Ascending) {

			lastItem = drawItems[0].position;

			if (lastItem == 0)

			{

				goto skip_search;

			}

			lastItem--;

			while (lastItem-- != 0)

			{

				if (menu->GetItemInfo(lastItem, &dr) != NULL)

				{

					mh->OnMenuDrawItem(menu, client, lastItem, dr.style);

					if (IsSlotItem(display, dr.style))

					{

						displayPrev = true;

						md.firstItem = lastItem;

						break;

					}

				}

			}

		}

	}

skip_search:

	/* Draw the item according to the order */

	menu_slots_t *slots = md.slots;

	unsigned int position = 0;			/* Keep track of the last position */

	if (order == ItemOrder_Ascending)

	{

		for (unsigned int i=0; i<foundItems; i++)

		{

			ItemDrawInfo &dr = drawItems[i].draw;

			menu->GetItemInfo(drawItems[i].position, &dr);

			mh->OnMenuDisplayItem(menu, client, drawItems[i].position, &(dr.display));

			if ((position = display->DrawItem(dr)) != 0)

			{

				slots[position].item = drawItems[i].position;

				slots[position].type = ItemSel_Item;

			}

		}

	} else if (order == ItemOrder_Descending) {

		unsigned int i = foundItems;

		/* NOTE: There will always be at least one item because

		 * of the check earlier.

		 */

		while (i--)

		{

			ItemDrawInfo &dr = drawItems[i].draw;

			menu->GetItemInfo(drawItems[i].position, &dr);

			mh->OnMenuDisplayItem(menu, client, drawItems[i].position, &(dr.display));

			if ((position = display->DrawItem(dr)) != 0)

			{

				slots[position].item = drawItems[i].position;

				slots[position].type = ItemSel_Item;

			}

		}

	}

	/* Now, we need to check if we need to add anything extra */

	if (pgn != MENU_NO_PAGINATION)

	{

		bool canDrawDisabled = display->CanDrawItem(ITEMDRAW_DISABLED);

		bool exitButton = menu->GetExitButton();

		char text[50];

		/* Calculate how many items we are allowed for control stuff */

		unsigned int padding = style->GetMaxPageItems() - maxItems;

		/* Add the number of available slots */

		padding += (maxItems - foundItems);

		/* Someday, if we are able to re-enable this, we will be very lucky men. */

#if 0

		if (!style->FeatureExists(MenuStyleFeature_ImplicitExit))

		{

#endif

		/* Even if we don't draw an exit button, we invalidate the slot. */

		padding--;

#if 0

		} else {

			/* Otherwise, we don't draw anything and leave the slot available */

			exitButton = false;

		}

#endif

		/* Subtract two slots for the displayNext/displayPrev padding */

		padding -= 2;

		/**

		 * We allow next/prev to be undrawn if neither exists.

		 * Thus, we only need padding if one of them will be drawn,

		 * or the exit button will be drawn.

		 */

		ItemDrawInfo padItem(NULL, ITEMDRAW_SPACER);

		if (exitButton || (displayNext || displayPrev))

		{

			/* Add spacers so we can pad to the end */

			for (unsigned int i=0; i<padding; i++)

			{

				position = display->DrawItem(padItem);

				slots[position].type = ItemSel_None;

			}

		}

		/* Put a fake spacer before control stuff, if possible */

		{

			ItemDrawInfo draw("", ITEMDRAW_RAWLINE|ITEMDRAW_SPACER);

			display->DrawItem(draw);

		}

		/* PREVIOUS */

		ItemDrawInfo dr(text, 0);

		if (displayPrev || canDrawDisabled)

		{

			CorePlayerTranslate(client, text, sizeof(text), "Back", NULL);

			dr.style = displayPrev ? 0 : ITEMDRAW_DISABLED;

			position = display->DrawItem(dr);

			slots[position].type = ItemSel_Back;

		} else if ((displayNext || canDrawDisabled) || exitButton) {

			/* If we can't display this, 

			 * but there is a "next" or "exit" button, we need to pad!

			 */

			position = display->DrawItem(padItem);

			slots[position].type = ItemSel_None;

		}

		/* NEXT */

		if (displayNext || canDrawDisabled)

		{

			CorePlayerTranslate(client, text, sizeof(text), "Next", NULL);

			dr.style = displayNext ? 0 : ITEMDRAW_DISABLED;

			position = display->DrawItem(dr);

			slots[position].type = ItemSel_Next;

		} else if (exitButton) {

			/* If we can't display this,

			 * but there is an exit button, we need to pad!

			 */

			position = display->DrawItem(padItem);

			slots[position].type = ItemSel_None;

		}

		/* EXIT */

		if (exitButton)

		{

			CorePlayerTranslate(client, text, sizeof(text), "Exit", NULL);

			dr.style = 0;

			position = display->DrawItem(dr);

			slots[position].type = ItemSel_Exit;

		}

	}

	/* Lastly, fill in any slots we could have missed */

	for (unsigned int i = position + 1; i < 10; i++)

	{

		slots[i].type = ItemSel_None;

	}

	/* Do title stuff */

	mh->OnMenuDisplay(menu, client, display);

	display->DrawTitle(menu->GetDefaultTitle(), true);

	return display;

}

unsigned int MenuManager::BroadcastMenu(IBaseMenu *menu, 

										IMenuHandler *handler, 

										int clients[], 

										unsigned int numClients, 

										unsigned int time)

{

	BroadcastHandler *bh;

	if (m_BroadcastHandlers.empty())

	{

		bh = new BroadcastHandler(handler);

	} else {

		bh = m_BroadcastHandlers.front();

		m_BroadcastHandlers.pop();

		bh->m_pHandler = handler;

		bh->numClients = 0;

	}

	handler->OnMenuStart(menu);

	unsigned int total = 0;

	for (unsigned int i=0; i<numClients; i++)