/**

  * vim: set ts=4 :

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

  * SourceMod Rock The Vote Plugin

  * Creates a map vote when the required number of players have requested one.

  *

  * SourceMod (C)2004-2008 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 <sourcemod>

 #include <mapchooser>

 #pragma semicolon 1

 public Plugin:myinfo =

 {

 	name = "Map Nominations",

 	author = "AlliedModders LLC",

 	description = "Provides Map Nominations",

 	version = SOURCEMOD_VERSION,

 	url = "http://www.sourcemod.net/"

 };

 new Handle:g_Cvar_ExcludeOld = INVALID_HANDLE;

 new Handle:g_Cvar_ExcludeCurrent = INVALID_HANDLE;

 new Handle:g_MapList = INVALID_HANDLE;

 new Handle:g_MapMenu = INVALID_HANDLE;

 new g_mapFileSerial = -1;

 #define MAPSTATUS_ENABLED (1<<0)

 #define MAPSTATUS_DISABLED (1<<1)

 #define MAPSTATUS_EXCLUDE_CURRENT (1<<2)

 #define MAPSTATUS_EXCLUDE_PREVIOUS (1<<3)

 #define MAPSTATUS_EXCLUDE_NOMINATED (1<<4)

 new Handle:g_mapTrie;

 public OnPluginStart()

 {

 	LoadTranslations("common.phrases");

 	LoadTranslations("nominations.phrases");

 	new arraySize = ByteCountToCells(33);	

 	g_MapList = CreateArray(arraySize);

 	g_Cvar_ExcludeOld = CreateConVar("sm_nominate_excludeold", "1", "Specifies if the current map should be excluded from the Nominations list", 0, true, 0.00, true, 1.0);

 	g_Cvar_ExcludeCurrent = CreateConVar("sm_nominate_excludecurrent", "1", "Specifies if the MapChooser excluded maps should also be excluded from Nominations", 0, true, 0.00, true, 1.0);

 	RegConsoleCmd("sm_nominate", Command_Nominate);

 	RegAdminCmd("sm_nominate_addmap", Command_Addmap, ADMFLAG_CHANGEMAP, "sm_nominate_addmap <mapname> - Forces a map to be on the next mapvote.");

 	g_mapTrie = CreateTrie();

 }

 public OnConfigsExecuted()

 {

 	if (ReadMapList(g_MapList,

 					g_mapFileSerial,

 					"nominations",

 					MAPLIST_FLAG_CLEARARRAY|MAPLIST_FLAG_MAPSFOLDER)

 		== INVALID_HANDLE)

 	{

 		if (g_mapFileSerial == -1)

 		{

 			SetFailState("Unable to create a valid map list.");

 		}

 	}

 	BuildMapMenu();

 }

 public OnNominationRemoved(const String:map[], owner)

 {

 	new status;

 	/* Is the map in our list? */

 	if (!GetTrieValue(g_mapTrie, map, status))

 	{

 		return;	

 	}

 	/* Was the map disabled due to being nominated */

 	if ((status & MAPSTATUS_EXCLUDE_NOMINATED) != MAPSTATUS_EXCLUDE_NOMINATED)

 	{

 		return;

 	}

 	SetTrieValue(g_mapTrie, map, MAPSTATUS_ENABLED);	

 }

 public Action:Command_Addmap(client, args)

 {

 	if (args < 1)

 	{

 		ReplyToCommand(client, "[SM] Usage: sm_nominate_addmap <mapname>");

 		return Plugin_Handled;

 	}

 	decl String:mapname[64];

 	GetCmdArg(1, mapname, sizeof(mapname));

 	new status;

 	if (!GetTrieValue(g_mapTrie, mapname, status))

 	{

 		ReplyToCommand(client, "%t", "Map was not found", mapname);

 		return Plugin_Handled;		

 	}

 	new NominateResult:result = NominateMap(mapname, true, 0);

 	if (result > Nominate_Replaced)

 	{

 		/* We assume already in vote is the casue because the maplist does a Map Validity check and we forced, so it can't be full */

 		ReplyToCommand(client, "%t", "Map Already In Vote", mapname);

 		return Plugin_Handled;	

 	}

 	SetTrieValue(g_mapTrie, mapname, MAPSTATUS_DISABLED|MAPSTATUS_EXCLUDE_NOMINATED);

 	ReplyToCommand(client, "%t", "Map Inserted", mapname);

 	LogAction(client, -1, "\"%L\" inserted map \"%s\".", client, mapname);

 	return Plugin_Handled;		

 }

 public OnClientSayCommand_Post(client, const String:command[], const String:sArgs[])

 {

 	if (!client)

 	{

 		return;

 	}

 	if (strcmp(sArgs, "nominate", false) == 0)

 	{

 		new ReplySource:old = SetCmdReplySource(SM_REPLY_TO_CHAT);

 		AttemptNominate(client);

 		SetCmdReplySource(old);

 	}

 }

 public Action:Command_Nominate(client, args)

 {

 	if (!client)

 	{

 		return Plugin_Handled;

 	}

 	if (args == 0)

 	{

 		AttemptNominate(client);

 		return Plugin_Handled;

 	}

 	decl String:mapname[64];

 	GetCmdArg(1, mapname, sizeof(mapname));

 	new status;

 	if (!GetTrieValue(g_mapTrie, mapname, status))

 	{

 		ReplyToCommand(client, "%t", "Map was not found", mapname);

 		return Plugin_Handled;		

 	}

 	if ((status & MAPSTATUS_DISABLED) == MAPSTATUS_DISABLED)

 	{

 		if ((status & MAPSTATUS_EXCLUDE_CURRENT) == MAPSTATUS_EXCLUDE_CURRENT)

 		{

 			ReplyToCommand(client, "[SM] %t", "Can't Nominate Current Map");

 		}

 		if ((status & MAPSTATUS_EXCLUDE_PREVIOUS) == MAPSTATUS_EXCLUDE_PREVIOUS)

 		{

 			ReplyToCommand(client, "[SM] %t", "Map in Exclude List");

 		}

 		if ((status & MAPSTATUS_EXCLUDE_NOMINATED) == MAPSTATUS_EXCLUDE_NOMINATED)

 		{

 			ReplyToCommand(client, "[SM] %t", "Map Already Nominated");

 		}

 		return Plugin_Handled;

 	}

 	new NominateResult:result = NominateMap(mapname, false, client);

 	if (result > Nominate_Replaced)

 	{

 		if (result == Nominate_AlreadyInVote)

 		{

 			ReplyToCommand(client, "%t", "Map Already In Vote", mapname);

 		}

 		else

 		{

 			ReplyToCommand(client, "[SM] %t", "Map Already Nominated");

 		}

 		return Plugin_Handled;	

 	}

 	/* Map was nominated! - Disable the menu item and update the trie */

 	SetTrieValue(g_mapTrie, mapname, MAPSTATUS_DISABLED|MAPSTATUS_EXCLUDE_NOMINATED);

 	decl String:name[64];

 	GetClientName(client, name, sizeof(name));

 	PrintToChatAll("[SM] %t", "Map Nominated", name, mapname);

 	return Plugin_Continue;

 }

 AttemptNominate(client)

 {

 	SetMenuTitle(g_MapMenu, "%T", "Nominate Title", client);

 	DisplayMenu(g_MapMenu, client, MENU_TIME_FOREVER);

 	return;

 }

 BuildMapMenu()

 {

 	if (g_MapMenu != INVALID_HANDLE)

 	{

 		CloseHandle(g_MapMenu);

 		g_MapMenu = INVALID_HANDLE;

 	}

 	ClearTrie(g_mapTrie);

 	g_MapMenu = CreateMenu(Handler_MapSelectMenu, MENU_ACTIONS_DEFAULT|MenuAction_DrawItem|MenuAction_DisplayItem);

 	decl String:map[64];

 	new Handle:excludeMaps = INVALID_HANDLE;

 	decl String:currentMap[32];

 	if (GetConVarBool(g_Cvar_ExcludeOld))

 	{	

 		excludeMaps = CreateArray(ByteCountToCells(33));

 		GetExcludeMapList(excludeMaps);

 	}

 	if (GetConVarBool(g_Cvar_ExcludeCurrent))

 	{

 		GetCurrentMap(currentMap, sizeof(currentMap));

 	}

 	for (new i = 0; i < GetArraySize(g_MapList); i++)

 	{

 		new status = MAPSTATUS_ENABLED;

 		GetArrayString(g_MapList, i, map, sizeof(map));

 		if (GetConVarBool(g_Cvar_ExcludeCurrent))

 		{

 			if (StrEqual(map, currentMap))

 			{

 				status = MAPSTATUS_DISABLED|MAPSTATUS_EXCLUDE_CURRENT;

 			}

 		}

 		/* Dont bother with this check if the current map check passed */

 		if (GetConVarBool(g_Cvar_ExcludeOld) && status == MAPSTATUS_ENABLED)

 		{

 			if (FindStringInArray(excludeMaps, map) != -1)

 			{

 				status = MAPSTATUS_DISABLED|MAPSTATUS_EXCLUDE_PREVIOUS;

 			}

 		}

 		AddMenuItem(g_MapMenu, map, map);

 		SetTrieValue(g_mapTrie, map, status);

 	}

 	SetMenuExitButton(g_MapMenu, true);

 	if (excludeMaps != INVALID_HANDLE)

 	{

 		CloseHandle(excludeMaps);

 	}

 }

 public Handler_MapSelectMenu(Handle:menu, MenuAction:action, param1, param2)

 {

 	switch (action)

 	{

 		case MenuAction_Select:

 		{

 			decl String:map[64], String:name[64];

 			GetMenuItem(menu, param2, map, sizeof(map));		

 			GetClientName(param1, name, 64);

 			new NominateResult:result = NominateMap(map, false, param1);

 			/* Don't need to check for InvalidMap because the menu did that already */

 			if (result == Nominate_AlreadyInVote)

 			{

 				PrintToChat(param1, "[SM] %t", "Map Already Nominated");

 				return 0;

 			}

 			else if (result == Nominate_VoteFull)

 			{

 				PrintToChat(param1, "[SM] %t", "Max Nominations");

 				return 0;

 			}

 			SetTrieValue(g_mapTrie, map, MAPSTATUS_DISABLED|MAPSTATUS_EXCLUDE_NOMINATED);

 			if (result == Nominate_Replaced)

 			{

 				PrintToChatAll("[SM] %t", "Map Nomination Changed", name, map);

 				return 0;	

 			}

 			PrintToChatAll("[SM] %t", "Map Nominated", name, map);

 		}

 		case MenuAction_DrawItem:

 		{

 			decl String:map[64];

 			GetMenuItem(menu, param2, map, sizeof(map));

 			new status;

 			if (!GetTrieValue(g_mapTrie, map, status))

 			{

 				LogError("Menu selection of item not in trie. Major logic problem somewhere.");

 				return ITEMDRAW_DEFAULT;

 			}

 			if ((status & MAPSTATUS_DISABLED) == MAPSTATUS_DISABLED)

 			{

 				return ITEMDRAW_DISABLED;	

 			}

 			return ITEMDRAW_DEFAULT;

 		}

 		case MenuAction_DisplayItem:

 		{

 			decl String:map[64];

 			GetMenuItem(menu, param2, map, sizeof(map));

 			new status;

 			if (!GetTrieValue(g_mapTrie, map, status))

 			{

 				LogError("Menu selection of item not in trie. Major logic problem somewhere.");

 				return 0;

 			}

 			decl String:display[100];

 			if ((status & MAPSTATUS_DISABLED) == MAPSTATUS_DISABLED)

 			{

 				if ((status & MAPSTATUS_EXCLUDE_CURRENT) == MAPSTATUS_EXCLUDE_CURRENT)

 				{

 					Format(display, sizeof(display), "%s (%T)", map, "Current Map", param1);

 					return RedrawMenuItem(display);

 				}

 				if ((status & MAPSTATUS_EXCLUDE_PREVIOUS) == MAPSTATUS_EXCLUDE_PREVIOUS)

 				{

 					Format(display, sizeof(display), "%s (%T)", map, "Recently Played", param1);

 					return RedrawMenuItem(display);

 				}

 				if ((status & MAPSTATUS_EXCLUDE_NOMINATED) == MAPSTATUS_EXCLUDE_NOMINATED)

 				{

 					Format(display, sizeof(display), "%s (%T)", map, "Nominated", param1);

 					return RedrawMenuItem(display);

 				}

 			}

 			return 0;

 		}

 	}

 	return 0;

 }