/**

 * vim: set ts=4 :

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

 * SourceMod

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

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

 *

 * This program is free software; you can redistribute it and/or 

 * modify it under the terms of the GNU General Public License, 

 * version 3.0, as published by the Free Software Foundation.

 * 

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program.  If not, see <http://www.gnu.org/licenses/>.

 *

 * As a special exception, AlliedModders LLC gives you permission to 

 * link the code of this program (as well as its derivative works) to 

 * "Half-Life 2," the "Source Engine," the "SourcePawn JIT," and any 

 * Game MODs that run on software by the Valve Corporation.  You must 

 * obey the GNU General Public License in all respects for all other 

 * code used. Additionally, AlliedModders LLC grants this exception 

 * to all derivative works. AlliedModders LLC defines further 

 * exceptions, found in LICENSE.txt (as of this writing, version 

 * JULY-31-2007), or <http://www.sourcemod.net/license.php>.

 *

 * Version: $Id$

 */

#include "sm_globals.h"

#include <sh_stack.h>

#include "MenuManager.h"

#include "MenuStyle_Valve.h"

#include "MenuStyle_Radio.h"

#include "HandleSys.h"

#include "PluginSys.h"

#include "sm_stringutil.h"

#if defined MENU_DEBUG

#include "Logger.h"

#endif

#if defined CreateMenu

#undef CreateMenu

#endif

/**

 * And God said, "let there be menus," and behold, there were menus.  

 * God saw the menus and they were good.  And the evening and the morning

 * were the third day.

 */

enum MenuAction

{

	MenuAction_Start = (1<<0),		/**< A menu has been started (nothing passed) */

	MenuAction_Display = (1<<1),	/**< A menu is about to be displayed (param1=client, param2=MenuPanel Handle) */

	MenuAction_Select = (1<<2),		/**< An item was selected (param1=client, param2=item) */

	MenuAction_Cancel = (1<<3),		/**< The menu was cancelled (param1=client, param2=item) */

	MenuAction_End = (1<<4),		/**< A menu's display/selection cycle is complete (nothing passed). */

	MenuAction_VoteEnd = (1<<5),	/**< (VOTE ONLY): A vote sequence has ended (param1=chosen item) */

	MenuAction_VoteStart = (1<<6), 	/**< (VOTE ONLY): A vote sequence has started */

	MenuAction_VoteCancel = (1<<7),	/**< (VOTE ONLY): A vote sequence has been cancelled (nothing passed) */

	MenuAction_DrawItem = (1<<8),	/**< A style is being drawn; return the new style (param1=client, param2=item) */

	MenuAction_DisplayItem = (1<<9),	/**< the odd duck */

};

class CPanelHandler : public IMenuHandler

{

	friend class MenuNativeHelpers;

public:

	CPanelHandler()

	{

	}

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

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

private:

	IPluginFunction *m_pFunc;

	IPlugin *m_pPlugin;

};

class CMenuHandler : public IMenuHandler

{

	friend class MenuNativeHelpers;

public:

	CMenuHandler(IPluginFunction *pBasic, int flags);

public:

	void OnMenuStart(IBaseMenu *menu);

	void OnMenuDisplay(IBaseMenu *menu, int client, IMenuPanel *display);

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

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

	void OnMenuEnd(IBaseMenu *menu, MenuEndReason reason);

	void OnMenuDestroy(IBaseMenu *menu);

	void OnMenuVoteStart(IBaseMenu *menu);

	void OnMenuVoteEnd(IBaseMenu *menu,

		unsigned int item, 

		unsigned int winningVotes,

		unsigned int totalVotes);

	void OnMenuVoteCancel(IBaseMenu *menu);

	void OnMenuDrawItem(IBaseMenu *menu, int client, unsigned int item, unsigned int &style);

	unsigned int OnMenuDisplayItem(IBaseMenu *menu, int client, IMenuPanel *panel, unsigned int item, const ItemDrawInfo &dr);

#if 0

	void OnMenuDrawItem(IBaseMenu *menu, int client, unsigned int item, unsigned int &style);

	void OnMenuDisplayItem(IBaseMenu *menu, int client, unsigned int item, const char **display);

#endif

private:

	cell_t DoAction(IBaseMenu *menu, MenuAction action, cell_t param1, cell_t param2, cell_t def_res=0);

private:

	IPluginFunction *m_pBasic;

	int m_Flags;

};

/**

 * GLOBAL CLASS FOR HELPERS

 */

class MenuNativeHelpers : 

	public SMGlobalClass,

	public IHandleTypeDispatch,

	public IPluginsListener

{

public:

	virtual void OnSourceModAllInitialized()

	{

		m_PanelType = g_HandleSys.CreateType("IMenuPanel", this, 0, NULL, NULL, g_pCoreIdent, NULL);

		m_TempPanelType = g_HandleSys.CreateType("TempIMenuPanel", this, m_PanelType, NULL, NULL, g_pCoreIdent, NULL);

		g_PluginSys.AddPluginsListener(this);

	}

	virtual void OnSourceModShutdown()

	{

		g_HandleSys.RemoveType(m_TempPanelType, g_pCoreIdent);

		g_HandleSys.RemoveType(m_PanelType, g_pCoreIdent);

		while (!m_FreePanelHandlers.empty())

		{

			delete m_FreePanelHandlers.front();

			m_FreePanelHandlers.pop();

		}

		while (!m_FreeMenuHandlers.empty())

		{

			delete m_FreeMenuHandlers.front();

			m_FreeMenuHandlers.pop();

		}

	}

	virtual void OnHandleDestroy(HandleType_t type, void *object)

	{

		if (type == m_TempPanelType)

		{

			return;

		}

		IMenuPanel *panel = (IMenuPanel *)object;

		panel->DeleteThis();

	}

	/**

	 *   It is extremely important that unloaded plugins don't crash.

	 * Thus, if a plugin unloads, we run through every handler we have.

	 * This means we do almost no runtime work for keeping track of

	 * our panel handlers (we don't have to store a list of the running 

	 * ones), but when push comes to shove, we have to scan them all

	 * in case any of them are active.

	 */

	virtual void OnPluginUnloaded(IPlugin *plugin)

	{

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

		{

			if (m_PanelHandlers[i]->m_pPlugin == plugin)

			{

				m_PanelHandlers[i]->m_pPlugin = NULL;

				m_PanelHandlers[i]->m_pFunc = NULL;

			}

		}

	}

	inline HandleType_t GetPanelType()

	{

		return m_PanelType;

	}

	inline HandleType_t GetTempPanelType()

	{

		return m_TempPanelType;

	}

	CPanelHandler *GetPanelHandler(IPluginFunction *pFunction)

	{

		CPanelHandler *handler;

		if (m_FreePanelHandlers.empty())

		{

			handler = new CPanelHandler;

			m_PanelHandlers.push_back(handler);

		} else {

			handler = m_FreePanelHandlers.front();

			m_FreePanelHandlers.pop();

		}

		handler->m_pFunc = pFunction;

		handler->m_pPlugin = g_PluginSys.GetPluginByCtx(pFunction->GetParentContext()->GetContext());

		return handler;

	}

	void FreePanelHandler(CPanelHandler *handler)

	{

		handler->m_pFunc = NULL;

		handler->m_pPlugin = NULL;

		m_FreePanelHandlers.push(handler);

	}

	CMenuHandler *GetMenuHandler(IPluginFunction *pFunction, int flags)

	{

		CMenuHandler *handler;

		if (m_FreeMenuHandlers.empty())

		{

			handler = new CMenuHandler(pFunction, flags);

		} else {

			handler = m_FreeMenuHandlers.front();

			m_FreeMenuHandlers.pop();

			handler->m_pBasic = pFunction;

			handler->m_Flags = flags;

		}

		return handler;

	}

	void FreeMenuHandler(CMenuHandler *handler)

	{

		m_FreeMenuHandlers.push(handler);

	}

private:

	HandleType_t m_PanelType;

	HandleType_t m_TempPanelType;

	CStack<CPanelHandler *> m_FreePanelHandlers;

	CStack<CMenuHandler *> m_FreeMenuHandlers;

	CVector<CPanelHandler *> m_PanelHandlers;

} g_MenuHelpers;

/**

 * BASIC PANEL HANDLER WRAPPER

 */

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

{

	if (m_pFunc)

	{

		m_pFunc->PushCell(BAD_HANDLE);

		m_pFunc->PushCell(MenuAction_Cancel);

		m_pFunc->PushCell(client);

		m_pFunc->PushCell(reason);

		m_pFunc->Execute(NULL);

	}

	g_MenuHelpers.FreePanelHandler(this);

}

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

{

	if (m_pFunc)

	{

		m_pFunc->PushCell(BAD_HANDLE);

		m_pFunc->PushCell(MenuAction_Select);

		m_pFunc->PushCell(client);

		m_pFunc->PushCell(item);

		m_pFunc->Execute(NULL);

	}

	g_MenuHelpers.FreePanelHandler(this);

}

static IMenuPanel *s_pCurPanel = NULL;

static unsigned int s_CurPanelReturn = 0;

static const ItemDrawInfo *s_CurDrawInfo = NULL;

/**

 * MENU HANDLER WRAPPER

 */

CMenuHandler::CMenuHandler(IPluginFunction *pBasic, int flags) : 

	m_pBasic(pBasic), m_Flags(flags)

{

	/* :TODO: We can probably cache the handle ahead of time */

}

void CMenuHandler::OnMenuStart(IBaseMenu *menu)

{

	if ((m_Flags & (int)MenuAction_Start) == (int)MenuAction_Start)

	{

		DoAction(menu, MenuAction_Start, 0, 0);

	}

}

void CMenuHandler::OnMenuDisplay(IBaseMenu *menu, int client, IMenuPanel *panel)

{

	if ((m_Flags & (int)MenuAction_Display) == (int)MenuAction_Display)

	{

		HandleSecurity sec;

		sec.pIdentity = g_pCoreIdent;

		sec.pOwner = m_pBasic->GetParentContext()->GetIdentity();

		HandleAccess access;

		g_HandleSys.InitAccessDefaults(NULL, &access);

		access.access[HandleAccess_Delete] = HANDLE_RESTRICT_IDENTITY|HANDLE_RESTRICT_OWNER;

		Handle_t hndl = g_HandleSys.CreateHandleEx(g_MenuHelpers.GetTempPanelType(), panel, &sec, &access, NULL);

		DoAction(menu, MenuAction_Display, client, hndl);

		g_HandleSys.FreeHandle(hndl, &sec);

	}

}

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

{

	DoAction(menu, MenuAction_Select, client, item);

}

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

{

	DoAction(menu, MenuAction_Cancel, client, (cell_t)reason);

}

void CMenuHandler::OnMenuEnd(IBaseMenu *menu, MenuEndReason reason)

{

	DoAction(menu, MenuAction_End, reason, 0);

}

void CMenuHandler::OnMenuDestroy(IBaseMenu *menu)

{

	g_MenuHelpers.FreeMenuHandler(this);

}

void CMenuHandler::OnMenuVoteStart(IBaseMenu *menu)

{

	DoAction(menu, MenuAction_VoteStart, 0, 0);

}

void CMenuHandler::OnMenuVoteEnd(IBaseMenu *menu,

								 unsigned int item,

								 unsigned int winningVotes,

								 unsigned int totalVotes)

{

	DoAction(menu, MenuAction_VoteEnd, item, (totalVotes << 16) | (winningVotes & 0xFFFF));

}

void CMenuHandler::OnMenuVoteCancel(IBaseMenu *menu)

{

	DoAction(menu, MenuAction_VoteCancel, 0, 0);

}

void CMenuHandler::OnMenuDrawItem(IBaseMenu *menu, int client, unsigned int item, unsigned int &style)

{

	if ((m_Flags & (int)MenuAction_DrawItem) == (int)MenuAction_DrawItem)

	{

		cell_t result = DoAction(menu, MenuAction_DrawItem, client, item, style);

		style = (unsigned int)result;

	}

}

unsigned int CMenuHandler::OnMenuDisplayItem(IBaseMenu *menu,

									 int client,

									 IMenuPanel *panel,

									 unsigned int item,

									 const ItemDrawInfo &dr)

{

	if ((m_Flags & (int)MenuAction_DisplayItem) == (int)MenuAction_DisplayItem)

	{

		IMenuPanel *oldpanel = s_pCurPanel;

		unsigned int oldret = s_CurPanelReturn;

		const ItemDrawInfo *oldinfo = s_CurDrawInfo;

		s_pCurPanel = panel;

		s_CurPanelReturn = 0;

		s_CurDrawInfo = &dr;

		cell_t res = DoAction(menu, MenuAction_DisplayItem, client, item, 0);

		if (!res)

		{

			res = s_CurPanelReturn;

		}

		s_pCurPanel = oldpanel;

		s_CurPanelReturn = oldret;

		s_CurDrawInfo = oldinfo;

		return res;

	}

	return 0;

}

cell_t CMenuHandler::DoAction(IBaseMenu *menu, MenuAction action, cell_t param1, cell_t param2, cell_t def_res)

{

#if defined MENU_DEBUG

	g_Logger.LogMessage("[SM_MENU] CMenuHandler::DoAction() (menu %p/%08x) (action %d) (param1 %d) (param2 %d)",

		menu,

		menu->GetHandle(),

		action,

		param1,

		param2);

#endif

	cell_t res = def_res;

	m_pBasic->PushCell(menu->GetHandle());

	m_pBasic->PushCell((cell_t)action);

	m_pBasic->PushCell(param1);

	m_pBasic->PushCell(param2);

	m_pBasic->Execute(&res);

	return res;

}

/**

 * INLINE FUNCTIONS FOR NATIVES

 */

inline Handle_t MakePanelHandle(IMenuPanel *panel, IPluginContext *pContext)

{

	return g_HandleSys.CreateHandle(g_MenuHelpers.GetPanelType(), panel, pContext->GetIdentity(), g_pCoreIdent, NULL);

}

inline HandleError ReadPanelHandle(Handle_t hndl, IMenuPanel **panel)

{

	HandleSecurity sec;

	sec.pIdentity = g_pCoreIdent;

	sec.pOwner = NULL;

	return g_HandleSys.ReadHandle(hndl, g_MenuHelpers.GetPanelType(), &sec, (void **)panel);

}

inline IMenuStyle *GetStyleFromCell(cell_t cell)

{

	enum MenuStyle

	{

		MenuStyle_Default = 0,		/**< The "default" menu style for the mod */

		MenuStyle_Valve = 1,		/**< The Valve provided menu style (Used on HL2DM) */

		MenuStyle_Radio = 2,		/**< The simpler menu style commonly used on CS:S */

	};

	if (cell == MenuStyle_Valve)

	{

		return &g_ValveMenuStyle;

	} else if (cell == MenuStyle_Radio

			   && g_RadioMenuStyle.IsSupported())

	{

		return &g_RadioMenuStyle;

	}

	return g_Menus.GetDefaultStyle();

}

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

 **** NATIVE DEFINITIONS ***********

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

static cell_t CreateMenu(IPluginContext *pContext, const cell_t *params)

{

	IMenuStyle *style = g_Menus.GetDefaultStyle();

	IPluginFunction *pFunction;

	if ((pFunction=pContext->GetFunctionById(params[1])) == NULL)

	{

		return pContext->ThrowNativeError("Function id %x is invalid", params[1]);

	}

	CMenuHandler *handler = g_MenuHelpers.GetMenuHandler(pFunction, params[2]);

	IBaseMenu *menu = style->CreateMenu(handler, pContext->GetIdentity());

	Handle_t hndl = menu->GetHandle();

	if (!hndl)

	{

		menu->Destroy();

		return BAD_HANDLE;

	}

	return hndl;

}

static cell_t DisplayMenu(IPluginContext *pContext, const cell_t *params)

{

	Handle_t hndl = (Handle_t)params[1];

	HandleError err;

	IBaseMenu *menu;

	if ((err=g_Menus.ReadMenuHandle(params[1], &menu)) != HandleError_None)

	{

		return pContext->ThrowNativeError("Menu handle %x is invalid (error %d)", hndl, err);

	}

	return menu->Display(params[2], params[3]) ? 1 : 0;

}

static cell_t VoteMenu(IPluginContext *pContext, const cell_t *params)

{

	Handle_t hndl = (Handle_t)params[1];

	HandleError err;

	IBaseMenu *menu;

	if ((err=g_Menus.ReadMenuHandle(params[1], &menu)) != HandleError_None)

	{

		return pContext->ThrowNativeError("Menu handle %x is invalid (error %d)", hndl, err);

	}

	cell_t *addr;

	pContext->LocalToPhysAddr(params[2], &addr);

	return menu->BroadcastVote(addr, params[3], params[4]) ? 1 : 0;

}

static cell_t AddMenuItem(IPluginContext *pContext, const cell_t *params)

{

	Handle_t hndl = (Handle_t)params[1];

	HandleError err;

	IBaseMenu *menu;

	if ((err=g_Menus.ReadMenuHandle(params[1], &menu)) != HandleError_None)

	{

		return pContext->ThrowNativeError("Menu handle %x is invalid (error %d)", hndl, err);

	}

	char *info;

	ItemDrawInfo dr;

	pContext->LocalToString(params[2], &info);

	pContext->LocalToString(params[3], (char **)&dr.display);

	dr.style = params[4];

	return menu->AppendItem(info, dr) ? 1 : 0;

}

static cell_t InsertMenuItem(IPluginContext *pContext, const cell_t *params)

{

	Handle_t hndl = (Handle_t)params[1];

	HandleError err;

	IBaseMenu *menu;

	if ((err=g_Menus.ReadMenuHandle(params[1], &menu)) != HandleError_None)

	{

		return pContext->ThrowNativeError("Menu handle %x is invalid (error %d)", hndl, err);

	}

	char *info;

	ItemDrawInfo dr;

	pContext->LocalToString(params[3], &info);

	pContext->LocalToString(params[4], (char **)&dr.display);

	dr.style = params[5];

	return menu->InsertItem(params[2], info, dr) ? 1 : 0;

}

static cell_t RemoveMenuItem(IPluginContext *pContext, const cell_t *params)

{

	Handle_t hndl = (Handle_t)params[1];

	HandleError err;

	IBaseMenu *menu;

	if ((err=g_Menus.ReadMenuHandle(params[1], &menu)) != HandleError_None)

	{

		return pContext->ThrowNativeError("Menu handle %x is invalid (error %d)", hndl, err);

	}

	return menu->RemoveItem(params[2]) ? 1 : 0;

}

static cell_t RemoveAllMenuItems(IPluginContext *pContext, const cell_t *params)

{

	Handle_t hndl = (Handle_t)params[1];

	HandleError err;