/**
 * HLstatsX Community Edition - SourceMod plugin to generate advanced weapon logging
 * http://www.hlxcommunity.com
 * Copyright (C) 2009 Nicholas Hastings (psychonic)
 * Copyright (C) 2007-2008 TTS Oetzel & Goerz GmbH
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * 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, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */

#pragma semicolon 1
 
#include <sourcemod>
#include <sdktools>

#define NAME "SuperLogs: GES"
#define VERSION "1.1.2"

#define MAX_LOG_WEAPONS 18
#define MAX_WEAPON_LEN 13
#define PREFIX_LEN 7

#define GES


new g_weapon_stats[MAXPLAYERS+1][MAX_LOG_WEAPONS][20];
new const String:g_weapon_list[MAX_LOG_WEAPONS][MAX_WEAPON_LEN] = {
									"pp7",
									"pp7_silenced",
									"dd44",
									"klobb",
									"cmag",
									"kf7",
									"zmg",
									"d5k",
									"d5k_silenced",
									"phantom",
									"AR33",
									"rcp90",
									"shotgun",
									"auto_shotgun",
									"sniper_rifle",
									"golden_gun",
									"silver_pp7",
									"golden_pp7"
								};

new const String:g_weapon_loglist[MAX_LOG_WEAPONS][] = {
									"#GE_PP7",
									"#GE_PP7_SILENCED",
									"#GE_DD44",
									"#GE_Klobb",
									"#GE_CougarMagnum",
									"#GE_KF7Soviet",
									"#GE_ZMG",
									"#GE_D5K",
									"#GE_D5K_SILENCED",
									"#GE_Phantom",
									"#GE_AR33",
									"#GE_RCP90",
									"#GE_Shotgun",
									"#GE_AutoShotgun",
									"#GE_SniperRifle",
									"#GE_GoldenGun",
									"#GE_SilverPP7",
									"#GE_GoldPP7"
								};
								
new Handle:g_cvar_wstats = INVALID_HANDLE;
new Handle:g_cvar_headshots = INVALID_HANDLE;
new Handle:g_cvar_locations = INVALID_HANDLE;
new Handle:g_cvar_actions = INVALID_HANDLE;
new Handle:g_cvar_classchanges = INVALID_HANDLE;

new bool:g_logwstats = true;
new bool:g_logheadshots = true;
new bool:g_loglocations = true;
new bool:g_logactions = true;
new bool:g_logclasschanges = true;

#include <loghelper>
#include <wstatshelper>


public Plugin:myinfo = {
	name = NAME,
	author = "psychonic",
	description = "Advanced logging for GoldenEye: Source. Generates auxilary logging for use with log parsers such as HLstatsX and Psychostats",
	version = VERSION,
	url = "http://www.hlxcommunity.com"
};


public OnPluginStart()
{
	CreatePopulateWeaponTrie();
	
	g_cvar_wstats = CreateConVar("superlogs_wstats", "1", "Enable logging of weapon stats (default on)", 0, true, 0.0, true, 1.0);
	g_cvar_headshots = CreateConVar("superlogs_headshots", "1", "Enable logging of headshot player action (default off)", 0, true, 0.0, true, 1.0);
	g_cvar_locations = CreateConVar("superlogs_locations", "1", "Enable logging of location on player death (default on)", 0, true, 0.0, true, 1.0);
	g_cvar_actions = CreateConVar("superlogs_actions", "1", "Enable logging of actions, such as round winner and awards won (default on)", 0, true, 0.0, true, 1.0);
	g_cvar_classchanges = CreateConVar("superlogs_classchanges", "1", "Enable logging of character changes (default on)", 0, true, 0.0, true, 1.0);
	HookConVarChange(g_cvar_wstats, OnCvarWstatsChange);
	HookConVarChange(g_cvar_headshots, OnCvarHeadshotsChange);
	HookConVarChange(g_cvar_locations, OnCvarLocationsChange);
	HookConVarChange(g_cvar_actions, OnCvarActionsChange);
	HookConVarChange(g_cvar_classchanges, OnCvarClasschangesChange);
	CreateConVar("superlogs_ges_version", VERSION, NAME, FCVAR_SPONLY|FCVAR_REPLICATED|FCVAR_NOTIFY);
		
	hook_wstats();
	HookEvent("player_hurt",  Event_PlayerHurt);
	HookEvent("player_death", Event_PlayerDeathPre, EventHookMode_Pre);
	HookEvent("player_changeident",	Event_RoleChange);
	HookEvent("round_end",    Event_RoundEnd);
		
	CreateTimer(1.0, LogMap);
	
	GetTeams();
}


public OnMapStart()
{
	GetTeams();
}

hook_wstats()
{
	HookEvent("player_death", Event_PlayerDeath);
	HookEvent("player_shoot",  Event_PlayerShoot);
	HookEvent("player_spawn", Event_PlayerSpawn);
	HookEvent("player_disconnect", Event_PlayerDisconnect, EventHookMode_Pre);
}

unhook_wstats()
{
	UnhookEvent("player_death", Event_PlayerDeath);
	UnhookEvent("player_shoot",  Event_PlayerShoot);
	UnhookEvent("player_spawn", Event_PlayerSpawn);
	UnhookEvent("player_disconnect", Event_PlayerDisconnect, EventHookMode_Pre);
}

public OnClientPutInServer(client)
{
	reset_player_stats(client);
}


public Event_PlayerShoot(Handle:event, const String:name[], bool:dontBroadcast)
{
	// "userid"		"local" 	// user ID on server
	// "weapon"	"local" 	// weapon name
	// "mode"		"local" 	// weapon mode [0 normal, 1 aimed]

	new attacker = GetClientOfUserId(GetEventInt(event, "userid"));
	if (g_logwstats && attacker > 0)
	{
		decl String:weapon[MAX_WEAPON_LEN+PREFIX_LEN];
		GetEventString(event, "weapon", weapon, sizeof(weapon));
		new weapon_index = get_weapon_index(weapon[PREFIX_LEN]);
		if (weapon_index > -1)
		{
			g_weapon_stats[attacker][weapon_index][LOG_HIT_SHOTS]++;
		}
	}
}


public Event_PlayerHurt(Handle:event, const String:name[], bool:dontBroadcast)
{
	// "userid"		"local" 	// user ID who was hurt
	// "attacker"	"local" 	// user ID who attacked
	// "weapon"	"local" 	// weapon name attacker used
	// "health"		"local" 	// health remaining
	// "armor"		"local" 	// armor remaining
	// "damage"	"local" 	// how much damage in this attack
	// "hitgroup"	"local" 	// what hitgroup was hit

	new attacker  = GetClientOfUserId(GetEventInt(event, "attacker"));
	new hitgroup  = GetEventInt(event, "hitgroup");
	new bool:headshot = (GetEventInt(event, "health") <= 0 && hitgroup == HITGROUP_HEAD);
	
	if (g_logwstats && attacker > 0)
	{
		decl String: weapon[MAX_WEAPON_LEN+PREFIX_LEN];
		GetEventString(event, "weapon", weapon, sizeof(weapon));

		new weapon_index = get_weapon_index(weapon[PREFIX_LEN]);
		if (weapon_index > -1)
		{
			g_weapon_stats[attacker][weapon_index][LOG_HIT_HITS]++;
			g_weapon_stats[attacker][weapon_index][LOG_HIT_DAMAGE]  += GetEventInt(event, "damage");
			if (hitgroup < 8)
			{
				g_weapon_stats[attacker][weapon_index][hitgroup + LOG_HIT_OFFSET]++;
			}
			if (headshot)
			{
				g_weapon_stats[attacker][weapon_index][LOG_HIT_HEADSHOTS]++;
			}
		}
	}
	if (g_logheadshots && headshot)
	{
		LogPlayerEvent(attacker, "triggered", "headshot");
	}
}

public Action:Event_PlayerDeathPre(Handle:event, const String:name[], bool:dontBroadcast)
{
	LogKillLoc(GetClientOfUserId(GetEventInt(event, "attacker")), GetClientOfUserId(GetEventInt(event, "userid")));
	
	return Plugin_Continue;
}


public Event_PlayerDeath(Handle:event, const String:name[], bool:dontBroadcast)
{
	// "userid"		"short"	// user ID who died
	// "attacker"	"short"	// user ID who killed
	// "weapon"	"string" 	// weapon name killed used
	// "weaponid"	"short"	// GE Weapon ID (for easy comparison)
	// "custom"		"byte"	// Used for achievements

	new victim   = GetClientOfUserId(GetEventInt(event, "userid"));
	new attacker = GetClientOfUserId(GetEventInt(event, "attacker"));
	if (g_logwstats && (victim > 0) && (attacker > 0))
	{
		decl String: weapon[MAX_WEAPON_LEN+PREFIX_LEN];
		GetEventString(event, "weapon", weapon, sizeof(weapon));
		new weapon_index = get_weapon_index(weapon[PREFIX_LEN]);
		if (weapon_index > -1)
		{
			g_weapon_stats[attacker][weapon_index][LOG_HIT_KILLS]++;
			g_weapon_stats[victim][weapon_index][LOG_HIT_DEATHS]++;
			if (GetClientTeam(attacker) == GetClientTeam(victim))
			{
				g_weapon_stats[attacker][weapon_index][LOG_HIT_TEAMKILLS]++;
			}
			dump_player_stats(victim);
		}
	}
}

public Event_PlayerSpawn(Handle:event, const String:name[], bool:dontBroadcast)
{
	// "userid"        "short"         // user ID on server          

	new client = GetClientOfUserId(GetEventInt(event, "userid"));
	if (client > 0)
	{
		reset_player_stats(client);
	}
}

public Event_RoundEnd(Handle:event, const String:name[], bool:dontBroadcast)
{
	if (g_logwstats)
	{
		WstatsDumpAll();
	}
	if (g_logactions)
	{
		new winner = GetEventInt(event, "winnerid");
		if (winner > 0)
		{
			LogPlayerEvent(winner, "triggered", "Round_Win", true);
		}
		else
		{
			winner = GetEventInt(event, "teamid");
			if (winner > 0)
			{
				LogTeamEvent(winner, "triggered", "Round_Win_Team");
			}
		}
		new awards[6];
		awards[0] = GetEventInt(event, "award1_id");
		awards[1] = GetEventInt(event, "award2_id");
		awards[2] = GetEventInt(event, "award3_id");
		awards[3] = GetEventInt(event, "award4_id");
		awards[4] = GetEventInt(event, "award5_id");
		awards[5] = GetEventInt(event, "award6_id");

		new winners[6];
		winners[0] = GetEventInt(event, "award1_winner");
		winners[1] = GetEventInt(event, "award2_winner");
		winners[2] = GetEventInt(event, "award3_winner");
		winners[3] = GetEventInt(event, "award4_winner");
		winners[4] = GetEventInt(event, "award5_winner");
		winners[5] = GetEventInt(event, "award6_winner");

		for (new i = 0; i < 6; i++)
		{
			if (winners[i] > 0)
			{
				switch(awards[i])
				{
					case 0:
						LogPlayerEvent(winners[i], "triggered", "GE_AWARD_DEADLY", true);
					case 1:
						LogPlayerEvent(winners[i], "triggered", "GE_AWARD_HONORABLE", true);
					case 2:
						LogPlayerEvent(winners[i], "triggered", "GE_AWARD_PROFESSIONAL", true);
					case 3:
						LogPlayerEvent(winners[i], "triggered", "GE_AWARD_MARKSMANSHIP", true);
					case 4:
						LogPlayerEvent(winners[i], "triggered", "GE_AWARD_AC10", true);
					case 5:
						LogPlayerEvent(winners[i], "triggered", "GE_AWARD_FRANTIC", true);
					case 6:
						LogPlayerEvent(winners[i], "triggered", "GE_AWARD_WTA", true);
					case 7:
						LogPlayerEvent(winners[i], "triggered", "GE_AWARD_LEMMING", true);
					case 8:
						LogPlayerEvent(winners[i], "triggered", "GE_AWARD_LONGIN", true);
					case 9:
						LogPlayerEvent(winners[i], "triggered", "GE_AWARD_SHORTIN", true);
					case 10:
						LogPlayerEvent(winners[i], "triggered", "GE_AWARD_DISHONORABLE", true);
					case 11:
						LogPlayerEvent(winners[i], "triggered", "GE_AWARD_NOTAC10", true);
					case 12:
						LogPlayerEvent(winners[i], "triggered", "GE_AWARD_MOSTLYHARMLESS", true);
				}
			}
		}
	}
}


public Event_RoleChange(Handle:event, const String:name[], bool:dontBroadcast)
{
	// "playerid"	"short"
	// "ident"		"string"
	
	new client = GetEventInt(event, "playerid");

	if (client > 0)
	{
		decl String:ident[32];
		GetEventString(event, "ident", ident, sizeof(ident));
		LogRoleChange(client, ident);
	}	
}

public Action:Event_PlayerDisconnect(Handle:event, const String:name[], bool:dontBroadcast)
{
	new client = GetClientOfUserId(GetEventInt(event, "userid"));
	OnPlayerDisconnect(client);
	return Plugin_Continue;
}

public Action:LogMap(Handle:timer)
{
	// Called 1 second after OnPluginStart since srcds does not log the first map loaded. Idea from Stormtrooper's "mapfix.sp" for psychostats
	LogMapLoad();
}


public OnCvarWstatsChange(Handle:cvar, const String:oldVal[], const String:newVal[])
{
	new bool:old_value = g_logwstats;
	g_logwstats = GetConVarBool(g_cvar_wstats);
	
	if (old_value != g_logwstats)
	{
		if (g_logwstats)
		{
			hook_wstats();
			if (!g_logheadshots)
			{
				HookEvent("player_hurt", Event_PlayerHurt);
			}
			if (!g_logactions)
			{
				HookEvent("round_end", Event_RoundEnd);
			}
		}
		else
		{
			unhook_wstats();
			if (!g_logheadshots)
			{
				UnhookEvent("player_hurt", Event_PlayerHurt);
			}
			if (!g_logactions)
			{
				UnhookEvent("round_end", Event_RoundEnd);
			}
		}
	}
}



public OnCvarHeadshotsChange(Handle:cvar, const String:oldVal[], const String:newVal[])
{
	new bool:old_value = g_logheadshots;
	g_logheadshots = GetConVarBool(g_cvar_headshots);
	
	if (old_value != g_logheadshots)
	{
		if (g_logheadshots && !g_logwstats)
		{
			HookEvent("player_hurt", Event_PlayerHurt);
		}
		else if (!g_logwstats)
		{
			UnhookEvent("player_hurt", Event_PlayerHurt);
		}
	}
}

public OnCvarLocationsChange(Handle:cvar, const String:oldVal[], const String:newVal[])
{
	new bool:old_value = g_loglocations;
	g_loglocations = GetConVarBool(g_cvar_locations);
	
	if (old_value != g_loglocations)
	{
		if (g_loglocations)
		{
			HookEvent("player_death", Event_PlayerDeathPre, EventHookMode_Pre);
		}
		else
		{
			UnhookEvent("player_death", Event_PlayerDeathPre, EventHookMode_Pre);
		}
	}
}

public OnCvarActionsChange(Handle:cvar, const String:oldVal[], const String:newVal[])
{
	new bool:old_value = g_logactions;
	g_logactions = GetConVarBool(g_cvar_actions);
	
	if (old_value != g_logactions)
	{
		if (g_logactions && !g_logwstats)
		{
			HookEvent("round_end", Event_RoundEnd);
		}
		else if (!g_logactions)
		{
			UnhookEvent("round_end", Event_RoundEnd);
		}
	}
}

public OnCvarClasschangesChange(Handle:cvar, const String:oldVal[], const String:newVal[])
{
	new bool:old_value = g_logclasschanges;
	g_logclasschanges = GetConVarBool(g_cvar_classchanges);
	
	if (old_value != g_logclasschanges)
	{
		if (g_logclasschanges)
		{
			HookEvent("player_changeident",	Event_RoleChange);
		}
		else
		{
			UnhookEvent("player_changeident",	Event_RoleChange);
		}
	}
}