/**

 * 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 "sm_stringutil.h"

#include "MenuStyle_Base.h"

#include "PlayerManager.h"

#include "MenuManager.h"

#include "HandleSys.h"

#if defined MENU_DEBUG

#include "Logger.h"

#endif

BaseMenuStyle::BaseMenuStyle() : m_WatchList(256), m_hHandle(BAD_HANDLE)

{

}

Handle_t BaseMenuStyle::GetHandle()

{

	/* Don't create the handle until we need it */

	if (m_hHandle == BAD_HANDLE)

	{

		m_hHandle = g_Menus.CreateStyleHandle(this);

	}

	return m_hHandle;

}

void BaseMenuStyle::AddClientToWatch(int client)

{

#if defined MENU_DEBUG

	g_Logger.LogMessage("[SM_MENU] AddClientToWatch(%d)", client);

#endif

	m_WatchList.push_back(client);

}

void BaseMenuStyle::RemoveClientFromWatch(int client)

{

#if defined MENU_DEBUG

	g_Logger.LogMessage("[SM_MENU] RemoveClientFromWatch(%d)", client);

#endif

	m_WatchList.remove(client);

}

void BaseMenuStyle::_CancelClientMenu(int client, MenuCancelReason reason, bool bAutoIgnore/* =false */)

{

#if defined MENU_DEBUG

	g_Logger.LogMessage("[SM_MENU] _CancelClientMenu() (client %d) (bAutoIgnore %d) (reason %d)", client, bAutoIgnore, reason);

#endif

	CBaseMenuPlayer *player = GetMenuPlayer(client);

	menu_states_t &states = player->states;

	bool bOldIgnore = player->bAutoIgnore;

	if (bAutoIgnore)

	{

		player->bAutoIgnore = true;

	}

	/* Save states */

	IMenuHandler *mh = states.mh;

	IBaseMenu *menu = states.menu;

	/* Clear menu */

	player->bInMenu = false;

	if (player->menuHoldTime)

	{

		RemoveClientFromWatch(client);

	}

	/* Fire callbacks */

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

	/* Only fire end if there's a valid menu */

	if (menu)

	{

		mh->OnMenuEnd(menu, MenuEnd_Cancelled);

	}

	if (bAutoIgnore)

	{

		player->bAutoIgnore = bOldIgnore;

	}

}

void BaseMenuStyle::CancelMenu(CBaseMenu *menu)

{

#if defined MENU_DEBUG

	g_Logger.LogMessage("[SM_MENU] CancelMenu() (menu %p)", menu);

#endif

	int maxClients = g_Players.GetMaxClients();

	for (int i=1; i<=maxClients; i++)

	{

		CBaseMenuPlayer *player = GetMenuPlayer(i);

		if (player->bInMenu)

		{

			menu_states_t &states = player->states;

			if (states.menu == menu)

			{

				_CancelClientMenu(i, MenuCancel_Interrupted);

			}

		}

	}

}

bool BaseMenuStyle::CancelClientMenu(int client, bool autoIgnore)

{

#if defined MENU_DEBUG

	g_Logger.LogMessage("[SM_MENU] CancelClientMenu() (client %d) (bAutoIgnore %d)", client, autoIgnore);

#endif

	if (client < 1 || client > g_Players.MaxClients())

	{

		return false;

	}

	CBaseMenuPlayer *player = GetMenuPlayer(client);

	if (!player->bInMenu)

	{

		return false;

	}

	_CancelClientMenu(client, MenuCancel_Interrupted, autoIgnore);

	return true;

}

MenuSource BaseMenuStyle::GetClientMenu(int client, void **object)

{

	if (client < 1 || client > g_Players.GetMaxClients())

	{

		return MenuSource_None;

	}

	CBaseMenuPlayer *player = GetMenuPlayer(client);

	if (player->bInMenu)

	{

		IBaseMenu *menu;

		if ((menu=player->states.menu) != NULL)

		{

			if (object)

			{

				*object = menu;

			}

			return MenuSource_BaseMenu;

		}

		return MenuSource_Display;

	} else if (player->bInExternMenu) {

		if (player->menuHoldTime != 0

			&& (gpGlobals->curtime > player->menuStartTime + player->menuHoldTime))

		{

			player->bInExternMenu = false;

			return MenuSource_None;

		}

		return MenuSource_External;

	}

	return MenuSource_None;

}

void BaseMenuStyle::OnClientDisconnected(int client)

{

#if defined MENU_DEBUG

	g_Logger.LogMessage("[SM_MENU] OnClientDisconnected(%d)", client);

#endif

	CBaseMenuPlayer *player = GetMenuPlayer(client);

	if (!player->bInMenu)

	{

		return;

	}

	_CancelClientMenu(client, MenuCancel_Disconnected, true);

	player->bInMenu = false;

	player->bInExternMenu = false;

}

static int do_lookup[256];

void BaseMenuStyle::ProcessWatchList()

{

	if (!m_WatchList.size())

	{

		return;

	}

	unsigned int total = 0;

	for (FastLink<int>::iterator iter=m_WatchList.begin(); iter!=m_WatchList.end(); ++iter)

	{

		do_lookup[total++] = (*iter);

	}

#if defined MENU_DEBUG

	if (total)

	{

		g_Logger.LogMessage("[SM_MENU] ProcessWatchList() found %d clients", total);

	}

#endif

	int client;

	CBaseMenuPlayer *player;

	float curTime = gpGlobals->curtime;

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

	{

		client = do_lookup[i];

		player = GetMenuPlayer(client);

#if defined MENU_DEBUG

		g_Logger.LogMessage("[SM_MENU] ProcessWatchList() (client %d) (bInMenu %d) (menuHoldTime %d) (curTime %f) (menuStartTime %f)",

			client,

			player->bInMenu,

			player->menuHoldTime,

			curTime,

			player->menuStartTime);

#endif

		if (!player->bInMenu || !player->menuHoldTime)

		{

#if defined MENU_DEBUG

			g_Logger.LogMessage("[SM_MENU] ProcessWatchList() removing client %d", client);

#endif

			m_WatchList.remove(client);

			continue;

		}

		if (curTime > player->menuStartTime + player->menuHoldTime)

		{

			_CancelClientMenu(client, MenuCancel_Timeout, false);

		}

	}

}

void BaseMenuStyle::ClientPressedKey(int client, unsigned int key_press)

{

#if defined MENU_DEBUG

	g_Logger.LogMessage("[SM_MENU] ClientPressedKey() (client %d) (key_press %d)", client, key_press);

#endif

	CBaseMenuPlayer *player = GetMenuPlayer(client);

	/* First question: Are we in a menu? */

	if (!player->bInMenu)

	{

		return;

	}

	bool cancel = false;

	unsigned int item = 0;

	MenuCancelReason reason = MenuCancel_Exit;

	MenuEndReason end_reason = MenuEnd_Selected;

	menu_states_t &states = player->states;

	assert(states.mh != NULL);

	if (states.menu == NULL)

	{

		item = key_press;

	} else if (key_press < 1 || key_press > GetMaxPageItems()) {

		cancel = true;

	} else {

		ItemSelection type = states.slots[key_press].type;

		/* For navigational items, we're going to redisplay */

		if (type == ItemSel_Back)

		{

			if (!RedoClientMenu(client, ItemOrder_Descending))

			{

				cancel = true;

				reason = MenuCancel_NoDisplay;

				end_reason = MenuEnd_Cancelled;

			} else {

				return;

			}

		} else if (type == ItemSel_Next) {

			if (!RedoClientMenu(client, ItemOrder_Ascending))

			{

				cancel = true;						/* I like Saltines. */

				reason = MenuCancel_NoDisplay;

				end_reason = MenuEnd_Cancelled;

			} else {

				return;

			}

		} else if (type == ItemSel_Exit || type == ItemSel_None) {

			cancel = true;

			reason = MenuCancel_Exit;

			end_reason = MenuEnd_Exit;

		} else if (type == ItemSel_ExitBack) {

			cancel = true;

			reason = MenuCancel_ExitBack;

			end_reason = MenuEnd_ExitBack;

		} else {

			item = states.slots[key_press].item;

		}

	}

	/* Save variables */

	IMenuHandler *mh = states.mh;

	IBaseMenu *menu = states.menu;

	/* Clear states */

	player->bInMenu = false;

	if (player->menuHoldTime)

	{

		RemoveClientFromWatch(client);

	}

	if (cancel)

	{

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

	} else {

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

	}

	/* Only fire end for valid menus */

	if (menu)

	{

		mh->OnMenuEnd(menu, end_reason);

	}

}

bool BaseMenuStyle::DoClientMenu(int client, IMenuPanel *menu, IMenuHandler *mh, unsigned int time)

{

#if defined MENU_DEBUG

	g_Logger.LogMessage("[SM_MENU] DoClientMenu() (client %d) (panel %p) (mh %p) (time %d)",

		client,

		menu,

		mh,

		time);

#endif

	CPlayer *pPlayer = g_Players.GetPlayerByIndex(client);

	if (!pPlayer || pPlayer->IsFakeClient() || !pPlayer->IsInGame())

	{

		return false;

	}

	CBaseMenuPlayer *player = GetMenuPlayer(client);

	if (player->bAutoIgnore)

	{

		return false;

	}

	/* For the duration of this, we are going to totally ignore whether

	* the player is already in a menu or not (except to cancel the old one).

	* Instead, we are simply going to ignore any further menu displays, so

	* this display can't be interrupted.

	*/

	player->bAutoIgnore = true;

	/* Cancel any old menus */

	menu_states_t &states = player->states;

	if (player->bInMenu)

	{

		_CancelClientMenu(client, MenuCancel_Interrupted, true);

	}

	states.firstItem = 0;

	states.lastItem = 0;

	states.menu = NULL;

	states.mh = mh;

	states.apiVers = SMINTERFACE_MENUMANAGER_VERSION;

	player->bInMenu = true;

	player->bInExternMenu = false;

	player->menuStartTime = gpGlobals->curtime;

	player->menuHoldTime = time;

	if (time)

	{

		AddClientToWatch(client);

	}

	/* Draw the display */

	SendDisplay(client, menu);

	/* We can be interrupted again! */

	player->bAutoIgnore = false;

	return true;

}

bool BaseMenuStyle::DoClientMenu(int client, CBaseMenu *menu, IMenuHandler *mh, unsigned int time)

{

#if defined MENU_DEBUG

	g_Logger.LogMessage("[SM_MENU] DoClientMenu() (client %d) (menu %p) (mh %p) (time %d)",

		client,

		menu,

		mh,

		time);

#endif

	mh->OnMenuStart(menu);

	CPlayer *pPlayer = g_Players.GetPlayerByIndex(client);

	if (!pPlayer || pPlayer->IsFakeClient() || !pPlayer->IsInGame())

	{

#if defined MENU_DEBUG

		g_Logger.LogMessage("[SM_MENU] DoClientMenu(): Failed to display to client %d", client);

#endif

		mh->OnMenuCancel(menu, client, MenuCancel_NoDisplay);

		mh->OnMenuEnd(menu, MenuEnd_Cancelled);

		return false;

	}

	CBaseMenuPlayer *player = GetMenuPlayer(client);

	if (player->bAutoIgnore)

	{

#if defined MENU_DEBUG

		g_Logger.LogMessage("[SM_MENU] DoClientMenu(): Client %d is autoIgnoring", client);

#endif

		mh->OnMenuCancel(menu, client, MenuCancel_NoDisplay);

		mh->OnMenuEnd(menu, MenuEnd_Cancelled);

		return false;

	}

	/* For the duration of this, we are going to totally ignore whether

	* the player is already in a menu or not (except to cancel the old one).

	* Instead, we are simply going to ignore any further menu displays, so

	* this display can't be interrupted.

	*/

	player->bAutoIgnore = true;

	/* Cancel any old menus */

	menu_states_t &states = player->states;

	if (player->bInMenu)

	{

#if defined MENU_DEBUG

		g_Logger.LogMessage("[SM_MENU] DoClientMenu(): Cancelling old menu to client %d", client);

#endif

		_CancelClientMenu(client, MenuCancel_Interrupted, true);

	}

	states.firstItem = 0;

	states.lastItem = 0;

	states.menu = menu;

	states.mh = mh;

	states.apiVers = SMINTERFACE_MENUMANAGER_VERSION;

	IMenuPanel *display = g_Menus.RenderMenu(client, states, ItemOrder_Ascending);

	if (!display)

	{

#if defined MENU_DEBUG

		g_Logger.LogMessage("[SM_MENU] DoClientMenu(): Failed to render to client %d", client);

#endif

		player->bAutoIgnore = false;

		player->bInMenu = false;

		mh->OnMenuCancel(menu, client, MenuCancel_NoDisplay);

		mh->OnMenuEnd(menu, MenuEnd_Cancelled);

		return false;

	}

	/* Finally, set our states */

	player->bInMenu = true;

	player->bInExternMenu = false;

	player->menuStartTime = gpGlobals->curtime;

	player->menuHoldTime = time;

	if (time)

	{

		AddClientToWatch(client);

	}

	/* Draw the display */

	SendDisplay(client, display);

	/* Free the display pointer */

	display->DeleteThis();

	/* We can be interrupted again! */

	player->bAutoIgnore = false;

#if defined MENU_DEBUG

	g_Logger.LogMessage("[SM_MENU] DoClientMenu() finished successfully (client %d)", client);

#endif

	return true;

}

bool BaseMenuStyle::RedoClientMenu(int client, ItemOrder order)

{

#if defined MENU_DEBUG

	g_Logger.LogMessage("[SM_MENU] RedoClientMenu() (client %d) (order %d)", client, order);

#endif

	CBaseMenuPlayer *player = GetMenuPlayer(client);

	menu_states_t &states = player->states;

	player->bAutoIgnore = true;

	IMenuPanel *display = g_Menus.RenderMenu(client, states, order);

	if (!display)

	{

#if defined MENU_DEBUG

		g_Logger.LogMessage("[SM_MENU] RedoClientMenu(): Failed to render menu");

#endif

		if (player->menuHoldTime)

		{

			RemoveClientFromWatch(client);

		}

		player->bAutoIgnore = false;

		return false;

	}

	SendDisplay(client, display);

	display->DeleteThis();

	player->bAutoIgnore = false;

#if defined MENU_DEBUG

	g_Logger.LogMessage("[SM_MENU] RedoClientMenu(): Succeeded to client %d", client);

#endif

	return true;

}

CBaseMenu::CBaseMenu(IMenuHandler *pHandler, IMenuStyle *pStyle, IdentityToken_t *pOwner) : 

m_pStyle(pStyle), m_Strings(512), m_Pagination(7), m_ExitButton(true), 

m_bShouldDelete(false), m_bCancelling(false), m_pOwner(pOwner ? pOwner : g_pCoreIdent), 

m_bDeleting(false), m_bWillFreeHandle(false), m_hHandle(BAD_HANDLE), m_pHandler(pHandler),

m_pVoteHandler(NULL), m_ExitBackButton(false)

{

}

CBaseMenu::~CBaseMenu()

{

	g_Menus.ReleaseVoteWrapper(m_pVoteHandler);

}

Handle_t CBaseMenu::GetHandle()

{

	if (!m_hHandle)

	{

		m_hHandle = g_Menus.CreateMenuHandle(this, m_pOwner);

	}

	return m_hHandle;

}

bool CBaseMenu::AppendItem(const char *info, const ItemDrawInfo &draw)

{

	if (m_Pagination == (unsigned)MENU_NO_PAGINATION

		&& m_items.size() >= m_pStyle->GetMaxPageItems())

	{

		return false;

	}

	CItem item;

	item.infoString = m_Strings.AddString(info);

	if (draw.display)

	{

		item.displayString = m_Strings.AddString(draw.display);

	}

	item.style = draw.style;

	m_items.push_back(item);

	return true;

}

bool CBaseMenu::InsertItem(unsigned int position, const char *info, const ItemDrawInfo &draw)

{

	if (m_Pagination == (unsigned)MENU_NO_PAGINATION

		&& m_items.size() >= m_pStyle->GetMaxPageItems())

	{

		return false;

	}

	if (position >= m_items.size())

	{

		return false;

	}

	CItem item;

	item.infoString = m_Strings.AddString(info);

	if (draw.display)

	{

		item.displayString = m_Strings.AddString(draw.display);

	}

	item.style = draw.style;

	CVector<CItem>::iterator iter = m_items.iterAt(position);

	m_items.insert(iter, item);

	return true;

}

bool CBaseMenu::RemoveItem(unsigned int position)

{

	if (position >= m_items.size())

	{

		return false;

	}

	m_items.erase(m_items.iterAt(position));