#include <amxmodx>
#include <hamsandwich>
#include <cstrike>
#include <fakemeta_util>

#define PLUGIN "Weapons after spawn"
#define VERSION "0.4"
#define AUTHOR "Atrocraz"

#define REMEMBER_AMMO

new g_weapons[33][32], wpn_count[33], bool:bKilled[33], iSpawnCount[33]
#if defined REMEMBER_AMMO
, g_ammo[33][32]
#else
, const MAXBPAMMO[] = { -1, 52, -1, 90, 1, 32, 1, 100, 90, 1, 120, 100, 100, 90, 90, 90, 100, 120, 30, 120, 200, 32, 90, 120, 90, 2, 35, 90, 90, -1, 100 }
#endif
 
public plugin_init(){
        RegisterHam(Ham_Killed, "player", "fw_PlayerKilled")
        RegisterHam(Ham_Spawn, "player", "fw_PlayerSpawn", 1)
}

public client_connect(id)	iSpawnCount[id] = 0, bKilled[id] = false
	
public fw_PlayerKilled(id){
        wpn_count[id] = 0
	bKilled[id] = true
	get_user_weapons(id, g_weapons[id], wpn_count[id])
	#if defined REMEMBER_AMMO
		for(new i = 0; i < wpn_count[id]; i++){
	                static weaponid
	                weaponid = g_weapons[id][i]

			if(weaponid == 29) // Knife
				continue;

	                g_ammo[id][weaponid] = cs_get_user_bpammo(id, weaponid)
	        }
	#endif
}
 
public fw_PlayerSpawn(id){
	if(iSpawnCount[id] == 2){
		if(bKilled[id]){
			switch(cs_get_user_team(id)){
				case CS_TEAM_T:	fm_strip_user_gun(id, CSW_GLOCK18)
				case CS_TEAM_CT:	fm_strip_user_gun(id, CSW_USP)
			}
		        for(new i = 0; i < wpn_count[id]; i++){
		                static name[25], weaponid
		                weaponid = g_weapons[id][i]
	
				if(weaponid == 6 || weaponid == 2 || weaponid == 29) // C4, Knife and unused index
					continue;
	
		               	get_weaponname(weaponid,name,sizeof(name))
		               	fm_give_item(id, name)
		
				#if defined REMEMBER_AMMO
					cs_set_user_bpammo(id, weaponid, g_ammo[id][weaponid])
				#else
		               		cs_set_user_bpammo(id, weaponid, MAXBPAMMO[weaponid])
				#endif
				bKilled[id] = false
		        }
		}
	}else iSpawnCount[id]++
}