#include <sdktools>
#include <sdkhooks>

#pragma newdecls required
#pragma semicolon 1
#pragma tabsize 0

bool lateLoad;

float ForAllMelees = 3.0;
float KatanaAndTopor = 8.0;
float Machete = 3.0;
float Crowbar = 3.0;
float knife = 4.0;

float ForAllMeleesT = 0.0;
float KatanaAndToporT = 3.0;
float MacheteT = 3.0;
float CrowbarT = 3.0;
float knifeT = 1.0;

float ForAllMeleesC = 0.0;
float KatanaAndToporC = 3.0;
float MacheteC = 3.0;
float CrowbarC = 3.0;
float knifeC = 1.0;

float ForAllMeleesHead = 4.0;
float KatanaAndToporHead = 10.0;
float MacheteHead = 3.0;
float CrowbarHead = 3.0;
float knifeHead = 5.0;

float ForAllMeleesTHead = 4.0;
float KatanaAndToporTHead = 20.0;
float MacheteTHead = 3.0;
float CrowbarTHead = 3.0;
float knifeTHead = 8.0;

float ForAllMeleesCHead = 1.0;
float KatanaAndToporCHead = 5.0;
float MacheteCHead = 3.0;
float CrowbarCHead = 3.0;
float knifeCHead = 2.0;

public APLRes AskPluginLoad2(Handle plugin, bool late, char[] erro, int errMax)
{
    lateLoad = late;
    return APLRes_Success;
}

public void OnPluginStart()
{
    if (lateLoad)
    {
        for (int client = 1; client <= MaxClients; client++)
        {
            if (IsClientInGame(client))
            {
                OnClientPutInServer(client);
            }
        }
    }
}

public void OnClientPutInServer(int client)
{
    SDKHook(client, SDKHook_TraceAttack, OnTakeDamage);
}

public void OnClientDisconnect(int client)
{
    SDKUnhook(client, SDKHook_TraceAttack, OnTakeDamage);
}

public Action OnTakeDamage(int victim, int &attacker, int &inflictor, float &damage, int &damagetype, int &ammotype, int hitbox, int hitgroup)
{
    if (!(damagetype & DMG_BURN || damagetype & DMG_BLAST))
    {
        if (IsValidClient(attacker) && GetClientTeam(attacker) == 2)
        {
            if(IsValidClient(victim) && GetClientTeam(victim) == 3)
            {
                char classname[64];
                GetClientWeapon(attacker, classname, sizeof(classname));
                if(StrEqual(classname, "weapon_melee"))
                {
                    GetEntPropString(GetPlayerWeaponSlot(attacker, 1), Prop_Data, "m_strMapSetScriptName", classname, sizeof(classname));
                }
                else return Plugin_Continue;
                int vClass = GetEntProp(victim, Prop_Send, "m_zombieClass");
                switch(vClass)
                {
                    case 6:
                    {
                        if(StrEqual(classname, "knife"))
                        {
                            damage = knifeC;
                            if(hitgroup == 1) damage = knifeCHead;
                            return Plugin_Changed;
                        }
                        else if(StrEqual(classname, "katana") || StrEqual(classname, "fireaxe"))
                        {
                            damage = KatanaAndToporC;
                            if(hitgroup == 1) damage = KatanaAndToporCHead;
                            return Plugin_Changed;
                        }
                        else if(StrEqual(classname, "machete"))
                        {
                            damage = MacheteC;
                            if(hitgroup == 1) damage = MacheteCHead;
                            return Plugin_Changed;
                        }
                        else if(StrEqual(classname, "crowbar"))
                        {
                            damage = CrowbarC;
                            if(hitgroup == 1) damage = CrowbarCHead;
                            return Plugin_Changed;
                        }
                        else
                        {
                            damage = ForAllMeleesC;
                            if(hitgroup == 1) damage = ForAllMeleesCHead;
                            return Plugin_Changed;
                        }
                    }
                    case 8:
                    {
                        if(StrEqual(classname, "knife"))
                        {
                            damage = knifeT;
                            if(hitgroup == 1) damage = knifeTHead;
                            return Plugin_Changed;
                        }
                        else if(StrEqual(classname, "katana") || StrEqual(classname, "fireaxe"))
                        {
                            damage = KatanaAndToporT;
                            if(hitgroup == 1) damage = KatanaAndToporTHead;
                            return Plugin_Changed;
                        }
                        else if(StrEqual(classname, "machete"))
                        {
                            damage = MacheteT;
                            if(hitgroup == 1) damage = MacheteTHead;
                            return Plugin_Changed;
                        }
                        else if(StrEqual(classname, "crowbar"))
                        {
                            damage = CrowbarT;
                            if(hitgroup == 1) damage = CrowbarTHead;
                            return Plugin_Changed;
                        }
                        else
                        {
                            damage = ForAllMeleesT;
                            if(hitgroup == 1) damage = ForAllMeleesTHead;
                            return Plugin_Changed;
                        }
                    }
                    default:
                    {
                        if(StrEqual(classname, "knife"))
                        {
                            damage = knife;
                            if(hitgroup == 1) damage = knifeHead;
                            return Plugin_Changed;
                        }
                        else if(StrEqual(classname, "katana") || StrEqual(classname, "fireaxe"))
                        {
                            damage = KatanaAndTopor;
                            if(hitgroup == 1) damage = KatanaAndToporHead;
                            return Plugin_Changed;
                        }
                        else if(StrEqual(classname, "machete"))
                        {
                            damage = Machete;
                            if(hitgroup == 1) damage =  MacheteHead;
                            return Plugin_Changed;
                        }
                        else if(StrEqual(classname, "crowbar"))
                        {
                            damage = Crowbar;
                            if(hitgroup == 1) damage = CrowbarHead;
                            return Plugin_Changed;
                        }
                        else
                        {
                            damage =  ForAllMelees;
                            if(hitgroup == 1) damage = ForAllMeleesHead;
                            return Plugin_Changed;
                        }
                    }
                }
            }
        }
    }
    return Plugin_Continue;
}

public bool IsValidClient(int client)
{
    if (client <= 0)
        return false;

    if (client > MaxClients)
        return false;

    if (!IsClientInGame(client))
        return false;

    if (!IsPlayerAlive(client))
        return false;

    return true;
}