#include <sourcemod>
#include <sdktools>

#pragma newdecls required
#pragma semicolon 1

static bool ignoreNextDamageDealt[MAXPLAYERS+1] = false;

float FotAllMelee = 1.0;
float KatanaAndFireAxe = 1.0;
float MacheteAndCrowbar = 1.0;
float Knife = 1.0;

public void OnPluginStart()
{
    HookEvent("infected_hurt", iInfectedHurt);
}

public Action iInfectedHurt(Handle event, const char[] name, bool dontBroadcast)
{
    char weaponname[64];

    int attacker = GetClientOfUserId(GetEventInt(event, "attacker"));
    int entity = (GetEventInt(event, "entityid"));

    if (!attacker || !IsValidEntity(entity) || ignoreNextDamageDealt[attacker]) return;

    char entname[32];
    GetEdictClassname(entity, entname, sizeof(entname));

    int dmg_health = GetEventInt(event,"amount");
    int eventhealth = GetEntProp(entity, Prop_Data, "m_iHealth");
    int damagedelta;

    if (dmg_health < 1 || eventhealth < 1) return;

    GetClientWeapon(attacker, weaponname, sizeof(weaponname));

    if (StrEqual(weaponname, "weapon_melee"))
    {
        GetEntPropString(GetPlayerWeaponSlot(attacker, 1), Prop_Data, "m_strMapSetScriptName", weaponname, sizeof(weaponname));
    }
    else return;

    if(StrEqual(weaponname, "weapon_knife"))
    {
        damagedelta = RoundToNearest((dmg_health * Knife) - dmg_health);
    }
    else if(StrEqual(weaponname, "weapon_katana") || StrEqual(weaponname, "weapon_fireaxe"))
    {
        damagedelta = RoundToNearest((dmg_health * KatanaAndFireAxe) - dmg_health);
    }
    else if(StrEqual(weaponname, "weapon_machete") || StrEqual(weaponname, "weapon_crowbar"))
    {
        damagedelta = RoundToNearest((dmg_health * MacheteAndCrowbar) - dmg_health);
    }
    else
    {
        damagedelta = RoundToNearest((dmg_health * FotAllMelee) - dmg_health);
    }

    switch (damagedelta > 0)
    {
        case true:
        {
            applyDamage(damagedelta, entity, attacker);
        }
        case false:
        {
            int health = eventhealth - damagedelta;

            if (health < 1)
            {
                damagedelta += (health - 1);
                health = 1;
            }

            SetEntProp(entity, Prop_Data, "m_iHealth", health);
            SetEventInt(event, "amount", damagedelta);
        }
    }
}

static void applyDamage(int damage, int victim, int attacker)
{
    Handle dataPack = CreateDataPack();
    WritePackCell(dataPack, damage);
    WritePackCell(dataPack, victim);
    WritePackCell(dataPack, attacker);

    CreateTimer(0.1, timer_stock_applyDamage, dataPack);
    ignoreNextDamageDealt[attacker] = true;
    CreateTimer(0.2, timer_resetStop, attacker);
}

public Action timer_resetStop(Handle timer, any client)
{
    ignoreNextDamageDealt[client] = false;
}

public Action timer_stock_applyDamage(Handle timer, Handle dataPack)
{
    ResetPack(dataPack);
    int damage = ReadPackCell(dataPack);
    int victim = ReadPackCell(dataPack);
    int attacker = ReadPackCell(dataPack);
    CloseHandle(dataPack);

    float victimPos[3]; char strDamage[16], strDamageTarget[16];

    if (victim < 20 && IsClientInGame(victim))
    {
        GetClientEyePosition(victim, victimPos);
    }
    else if (IsValidEntity(victim))
    {
        GetEntityAbsOrigin(victim, victimPos);
    }
    else return;

    IntToString(damage, strDamage, sizeof(strDamage));
    Format(strDamageTarget, sizeof(strDamageTarget), "hurtme%d", victim);

    int entPointHurt = CreateEntityByName("point_hurt");
    if(!entPointHurt) return;

    DispatchKeyValue(victim, "targetname", strDamageTarget);
    DispatchKeyValue(entPointHurt, "DamageTarget", strDamageTarget);
    DispatchKeyValue(entPointHurt, "Damage", strDamage);
    DispatchKeyValue(entPointHurt, "DamageType", "0");
    DispatchSpawn(entPointHurt);

    TeleportEntity(entPointHurt, victimPos, NULL_VECTOR, NULL_VECTOR);
    AcceptEntityInput(entPointHurt, "Hurt", (attacker && attacker < 20 && IsClientInGame(attacker)) ? attacker : -1);

    DispatchKeyValue(entPointHurt, "classname", "point_hurt");
    DispatchKeyValue(victim, "targetname", "null");
    RemoveEdict(entPointHurt);
}

stock void GetEntityAbsOrigin( int entity, float origin[3])
{
    if (entity > 0 && IsValidEntity(entity))
    {
        float mins[3], maxs[3];
        GetEntPropVector(entity,Prop_Send,"m_vecOrigin",origin);
        GetEntPropVector(entity,Prop_Send,"m_vecMins",mins);
        GetEntPropVector(entity,Prop_Send,"m_vecMaxs",maxs);

        origin[0] += (mins[0] + maxs[0]) * 0.5;
        origin[1] += (mins[1] + maxs[1]) * 0.5;
        origin[2] += (mins[2] + maxs[2]) * 0.5;
    }
}