#pragma semicolon 1

#include <sourcemod>
#include <sdktools>
#include <sdkhooks>

#define MAX_FRAMECHECK 10

bool bPistol[MAXPLAYERS+1] = {true, ...};
bool bPistolFix[MAXPLAYERS+1] = {true, ...};

public void OnPluginStart()
{
    HookEvent("player_use", Event_Ammo_Pickup);
}

public void OnClientPostAdminCheck(int client)
{
    if (IsFakeClient(client)) return;

    bPistol[client] = true;
    bPistolFix[client] = true;

    SDKHook(client, SDKHook_WeaponEquip, Hook_WeaponEquip);
}

public Action Hook_WeaponEquip(int client, int weapon)
{
    if (!IsSurvivor(client) || !bPistolFix[client])
        return;

    bPistolFix[client] = false;
}

/*[----->|BOT PISTOL AMMO|<-----]*/
public void OnGameFrame()
{
    int iFrameskip = 0;
    iFrameskip = (iFrameskip + 1) % MAX_FRAMECHECK;
    if(iFrameskip != 0 || !IsServerProcessing())
        return;

    for (int client = 1; client <= MaxClients; client++)
    {
        if (!IsSurvivor(client) || !IsPlayerAlive(client) || !IsFakeClient(client))
            continue;


        int Weapon = GetPlayerWeaponSlot(client, 1);
        if (Weapon == -1) continue;

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

        int AmmoType = GetEntProp(Weapon, Prop_Data, "m_iPrimaryAmmoType");
        int Ammo = GetEntProp(client, Prop_Send, "m_iAmmo", _, AmmoType);

        if (StrEqual(sWeapon, "weapon_pistol"))
        {
            if (AmmoType == -1) continue;

            if (Ammo == 0)
                SetEntProp(client, Prop_Send, "m_iAmmo", 60, _, AmmoType);
        }
        if (StrEqual(sWeapon, "weapon_pistol", false) && GetEntProp(Weapon, Prop_Send, "m_hasDualWeapons"))
        {
            if (AmmoType == -1) continue;

            if (Ammo == 0)
            {
                SetEntProp(client, Prop_Send, "m_iAmmo", 120, _, AmmoType);
            }
        }
        if  (StrEqual(sWeapon, "weapon_pistol_magnum", false))
        {
            SetEntProp(client, Prop_Send, "m_iAmmo", 40, _, AmmoType);
        }
    }
}

public Action Event_Ammo_Pickup(Event event, const char[] name, bool dontBroadcast)
{
    int client = GetClientOfUserId(event.GetInt("userid"));
    if (!IsSurvivor(client) || !IsPlayerAlive(client)) return;

    int AmmoPile = event.GetInt("targetid");
    if (!IsValidEntity(AmmoPile)) return;

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

    if (!StrEqual(sWeapon, "weapon_ammo_spawn", false))
        return;

    int Weapon = GetPlayerWeaponSlot(client, 1);
    if (Weapon == -1) return;

    GetEntityClassname(Weapon, sWeapon, sizeof(sWeapon));

    if (!(StrEqual(sWeapon, "weapon_pistol", false) || StrEqual(sWeapon, "weapon_pistol_magnum", false)))
        return;

    float cPos[3];
    GetEntPropVector(client, Prop_Data, "m_vecAbsOrigin", cPos);

    float aPos[3];
    GetEntPropVector(AmmoPile, Prop_Data, "m_vecAbsOrigin", aPos);

    if (GetVectorDistance(cPos, aPos) <= 100)
    {
        bPistol[client] = true; //FIXES WEAPON EMPTY MESSAGE IN AMMO LOCK TIMER

        int AmmoType = GetEntProp(Weapon, Prop_Data, "m_iPrimaryAmmoType");
        if (AmmoType == -1) return;

        int Clip = GetEntProp(Weapon, Prop_Send, "m_iClip1");

        if (StrEqual(sWeapon, "weapon_pistol", false) && GetEntProp(Weapon, Prop_Send, "m_hasDualWeapons"))
        {
            SetEntProp(Weapon, Prop_Send, "m_iClip1", Clip);
            SetEntProp(client, Prop_Send, "m_iAmmo", 30, _, AmmoType);
        }
        else if (StrEqual(sWeapon, "weapon_pistol", false))
        {

            SetEntProp(Weapon, Prop_Send, "m_iClip1", Clip);
            SetEntProp(client, Prop_Send, "m_iAmmo", 15, _, AmmoType);
        }

        if (StrEqual(sWeapon, "weapon_pistol_magnum", false))
        {
            SetEntProp(Weapon, Prop_Send, "m_iClip1", Clip);
            SetEntProp(client, Prop_Send, "m_iAmmo", 16, _, AmmoType);
        }
    }
}

stock bool IsValidAdmin(int client)
{
    if (GetUserFlagBits(client) & ADMFLAG_ROOT) return true;
    return false;
}

stock bool IsValidClient(int client)
{
    if (client > 0 && client <= MaxClients && IsClientInGame(client)) return true;
    return false;
}

stock bool IsSpectator(int client)
{
    if (client > 0 && client <= MaxClients && IsClientInGame(client) && GetClientTeam(client) == 1) return true;
    return false;
}

stock bool IsSurvivor(int client)
{
    if (client > 0 && client <= MaxClients && IsClientInGame(client) && GetClientTeam(client) == 2) return true;
    return false;
}

stock bool IsInfected(int client)
{
    if (client > 0 && client <= MaxClients && IsClientInGame(client) && GetClientTeam(client) == 3) return true;
    return false;
}