#include <sourcemod>
#include <sdktools>

public OnPluginStart()
{
    HookEvent("upgrade_pack_added", Event_UpgradePackAdded);
}

public Action:Event_UpgradePackAdded(Handle:event, const String:name[], bool:dontBroadcast)
{
    new i = GetClientOfUserId(GetEventInt(event, "userid"));
    new iWeapon = GetPlayerWeaponSlot(i, 0);

    int AmmoType = GetEntProp(iWeapon, Prop_Data, "m_iPrimaryAmmoType");
    int Clip = GetEntProp(iWeapon, Prop_Send, "m_iClip1");
    int Ammo = GetEntProp(i, Prop_Data, "m_iAmmo", _, AmmoType);

    char sWeapon[32];
    GetEntityClassname(iWeapon, sWeapon, sizeof(sWeapon));

    if(bIsSurvivor(i))
    {
        int num;
        if(StrEqual(sWeapon, "weapon_hunting_rifle"))
        {
            num = 120 - Ammo;
            SetEntProp(iWeapon , Prop_Send, "m_iClip1", Clip);
            SetEntProp(i, Prop_Send, "m_iAmmo", num, _, AmmoType);
            SetEntProp(iWeapon, Prop_Send, "m_nUpgradedPrimaryAmmoLoaded", 0);
            SetEntProp(iWeapon, Prop_Send, "m_upgradeBitVec", 0);
        }
        else if(StrEqual(sWeapon, "weapon_rifle_ak47"))
        {
            SetEntProp(iWeapon , Prop_Send, "m_iClip1", 0);
            SetEntProp(i, Prop_Send, "m_iAmmo", 120 + Clip, _, AmmoType);
            SetEntProp(iWeapon, Prop_Send, "m_nUpgradedPrimaryAmmoLoaded", 0);
            SetEntProp(iWeapon, Prop_Send, "m_upgradeBitVec", 0);
        }
        else if(StrEqual(sWeapon, "weapon_pistol"))
        {
            SetEntProp(iWeapon , Prop_Send, "m_iClip1", 0);
            SetEntProp(i, Prop_Send, "m_iAmmo", 50, _, AmmoType);
            SetEntProp(iWeapon, Prop_Send, "m_nUpgradedPrimaryAmmoLoaded", 0);
            SetEntProp(iWeapon, Prop_Send, "m_upgradeBitVec", 0);
        }
        else if(StrEqual(sWeapon, "weapon_rifle"))
        {
            SetEntProp(iWeapon , Prop_Send, "m_iClip1", 0);
            SetEntProp(i, Prop_Send, "m_iAmmo", 160 + Clip, _, AmmoType);
            SetEntProp(iWeapon, Prop_Send, "m_nUpgradedPrimaryAmmoLoaded", 0);
            SetEntProp(iWeapon, Prop_Send, "m_upgradeBitVec", 0);
        }
        else if(StrEqual(sWeapon, "weapon_pistol_magnum"))
        {
            SetEntProp(iWeapon , Prop_Send, "m_iClip1", 0);
            SetEntProp(i, Prop_Send, "m_iAmmo", 50, _, AmmoType);
            SetEntProp(iWeapon, Prop_Send, "m_nUpgradedPrimaryAmmoLoaded", 0);
            SetEntProp(iWeapon, Prop_Send, "m_upgradeBitVec", 0);
        }
        else if(StrEqual(sWeapon, "weapon_rifle_desert"))
        {
            SetEntProp(iWeapon , Prop_Send, "m_iClip1", 0);
            SetEntProp(i, Prop_Send, "m_iAmmo", 160 + Clip, _, AmmoType);
            SetEntProp(iWeapon, Prop_Send, "m_nUpgradedPrimaryAmmoLoaded", 0);
            SetEntProp(iWeapon, Prop_Send, "m_upgradeBitVec", 0);
        }
        else if(StrEqual(sWeapon, "weapon_rifle_sg552"))
        {
            SetEntProp(iWeapon , Prop_Send, "m_iClip1", 0);
            SetEntProp(i, Prop_Send, "m_iAmmo", 160 + Clip, _, AmmoType);
            SetEntProp(iWeapon, Prop_Send, "m_nUpgradedPrimaryAmmoLoaded", 0);
            SetEntProp(iWeapon, Prop_Send, "m_upgradeBitVec", 0);
        }
        else if(StrEqual(sWeapon, "weapon_smg"))
        {
            SetEntProp(iWeapon , Prop_Send, "m_iClip1", 0);
            SetEntProp(i, Prop_Send, "m_iAmmo", 350 + Clip, _, AmmoType);
            SetEntProp(iWeapon, Prop_Send, "m_nUpgradedPrimaryAmmoLoaded", 0);
            SetEntProp(iWeapon, Prop_Send, "m_upgradeBitVec", 0);
        }
        else if(StrEqual(sWeapon, "weapon_smg_mp5"))
        {
            SetEntProp(iWeapon , Prop_Send, "m_iClip1", 0);
            SetEntProp(i, Prop_Send, "m_iAmmo", 350 + Clip, _, AmmoType);
            SetEntProp(iWeapon, Prop_Send, "m_nUpgradedPrimaryAmmoLoaded", 0);
            SetEntProp(iWeapon, Prop_Send, "m_upgradeBitVec", 0);
        }
        else if(StrEqual(sWeapon, "weapon_sniper_awp"))
        {
            SetEntProp(iWeapon , Prop_Send, "m_iClip1", 0);
            SetEntProp(i, Prop_Send, "m_iAmmo", 40 + Clip, _, AmmoType);
            SetEntProp(iWeapon, Prop_Send, "m_nUpgradedPrimaryAmmoLoaded", 0);
            SetEntProp(iWeapon, Prop_Send, "m_upgradeBitVec", 0);
        }
        else if(StrEqual(sWeapon, "weapon_sniper_military"))
        {
            SetEntProp(iWeapon , Prop_Send, "m_iClip1", 0);
            SetEntProp(i, Prop_Send, "m_iAmmo", 80 + Clip, _, AmmoType);
            SetEntProp(iWeapon, Prop_Send, "m_nUpgradedPrimaryAmmoLoaded", 0);
            SetEntProp(iWeapon, Prop_Send, "m_upgradeBitVec", 0);
        }
        else if(StrEqual(sWeapon, "weapon_sniper_scout"))
        {
            SetEntProp(iWeapon , Prop_Send, "m_iClip1", 0);
            SetEntProp(i, Prop_Send, "m_iAmmo", 80 + Clip, _, AmmoType);
            SetEntProp(iWeapon, Prop_Send, "m_nUpgradedPrimaryAmmoLoaded", 0);
            SetEntProp(iWeapon, Prop_Send, "m_upgradeBitVec", 0);
        }
        else if(StrEqual(sWeapon, "weapon_grenade_launcher"))
        {
            SetEntProp(iWeapon , Prop_Send, "m_iClip1", 0);
            SetEntProp(i, Prop_Send, "m_iAmmo", 8 + Clip, _, AmmoType);
            SetEntProp(iWeapon, Prop_Send, "m_nUpgradedPrimaryAmmoLoaded", 0);
            SetEntProp(iWeapon, Prop_Send, "m_upgradeBitVec", 0);
        }
        else if(StrEqual(sWeapon, "weapon_rifle_m60"))
        {
            SetEntProp(iWeapon , Prop_Send, "m_iClip1", 0);
            SetEntProp(i, Prop_Send, "m_iAmmo", 200 + Clip, _, AmmoType);
            SetEntProp(iWeapon, Prop_Send, "m_nUpgradedPrimaryAmmoLoaded", 0);
            SetEntProp(iWeapon, Prop_Send, "m_upgradeBitVec", 0);
        }
        else if(StrEqual(sWeapon, "weapon_smg_silenced "))
        {
            SetEntProp(iWeapon , Prop_Send, "m_iClip1", 0);
            SetEntProp(i, Prop_Send, "m_iAmmo", 200 + Clip, _, AmmoType);
            SetEntProp(iWeapon, Prop_Send, "m_nUpgradedPrimaryAmmoLoaded", 0);
            SetEntProp(iWeapon, Prop_Send, "m_upgradeBitVec", 0);
        }
    }
}

stock bool bIsSurvivor(int client)
{
    return client > 0 && client <= MaxClients && IsClientInGame(client) && GetClientTeam(client) == 2 && !IsClientInKickQueue(client) && IsPlayerAlive(client);
}