#include <amxmodx>
#include <engine>
#include <fakemeta>
#include <fun>
#include <hamsandwich>
#include <xs>
#include <cstrike>

#define ENG_NULLENT      -1
#define EV_INT_WEAPONKEY   EV_INT_impulse
#define hk416ex_WEAPONKEY   807
#define MAX_PLAYERS             32
#define IsValidUser(%1) (1 <= %1 <= g_MaxPlayers)

const USE_STOPPED = 0
const OFFSET_ACTIVE_ITEM = 373
const OFFSET_WEAPONOWNER = 41
const OFFSET_LINUX = 5
const OFFSET_LINUX_WEAPONS = 4

#define WEAP_LINUX_XTRA_OFF         4
#define m_fKnown            44
#define m_flNextPrimaryAttack          46
#define m_flTimeWeaponIdle         48
#define m_iClip               51
#define m_fInReload            54
#define PLAYER_LINUX_XTRA_OFF         5
#define m_flNextAttack            83

#define hk416ex_RELOAD_TIME 3.0
#define write_coord_f(%1)   engfunc(EngFunc_WriteCoord,%1)

new const Fire_Sounds[][] = { "cso/hk416ex.wav" }

new hk416ex_V_MODEL[64] = "models/cso/v_hk416ex.mdl"
new hk416ex_P_MODEL[64] = "models/cso/p_hk416ex.mdl"
new hk416ex_W_MODEL[64] = "models/cso/w_hk416ex.mdl"

new const GUNSHOT_DECALS[] = { 41, 42, 43, 44, 45 }

new cvar_dmg_hk416ex, cvar_recoil_hk416ex, cvar_clip_hk416ex, cvar_hk416ex_ammo, cvar_spd_hk416ex
new g_has_hk416ex[33]
new g_MaxPlayers, g_orig_event_hk416ex, g_clip_ammo[33]
new Float:cl_pushangle[MAX_PLAYERS + 1][3], m_iBlood[2]
new g_hk416ex_TmpClip[33] , oldweap[33]

const PRIMARY_WEAPONS_BIT_SUM = (1<<CSW_SCOUT)|(1<<CSW_XM1014)|(1<<CSW_MAC10)|(1<<CSW_AUG)|(1<<CSW_UMP45)|(1<<CSW_SG550)|(1<<CSW_GALIL)|(1<<CSW_FAMAS)|(1<<CSW_AWP)|(1<<CSW_ MP5NAVY)|(1<<CSW_M249)|(1<<CSW_M3)|(1<<CSW_M4A1)|(1<<CSW_TMP)|(1<<CSW_G3SG1)|(1<<CSW_SG552)|(1<<CSW_AK47)|(1<<CSW_P90)
new const WEAPONENTNAMES[][] = { "", "weapon_p228", "", "weapon_scout", "weapon_hegrenade", "weapon_xm1014", "weapon_c4", "weapon_mac10",
          "weapon_aug", "weapon_smokegrenade", "weapon_elite", "weapon_fiveseven", "weapon_ump45", "weapon_sg550",
          "weapon_galil", "weapon_famas", "weapon_usp", "weapon_glock18", "weapon_awp", "weapon_mp5navy", "weapon_m249",
          "weapon_m3", "weapon_m4a1", "weapon_tmp", "weapon_g3sg1", "weapon_flashbang", "weapon_deagle", "weapon_sg552",
          "weapon_ak47", "weapon_knife", "weapon_p90" }

public plugin_init()
{
    register_plugin(" [ZP]Weapon: Hk416ex", "1.0", "Crock")
    register_message(get_user_msgid("DeathMsg"), "message_DeathMsg")
    register_event("CurWeapon","CurrentWeapon","be","1=1")
    RegisterHam(Ham_Item_AddToPlayer, "weapon_m4a1", "fw_hk416ex_AddToPlayer")
    RegisterHam(Ham_Use, "func_tank", "fw_UseStationary_Post", 1)
    RegisterHam(Ham_Use, "func_tankmortar", "fw_UseStationary_Post", 1)
    RegisterHam(Ham_Use, "func_tankrocket", "fw_UseStationary_Post", 1)
    RegisterHam(Ham_Use, "func_tanklaser", "fw_UseStationary_Post", 1)
    for (new i = 1; i < sizeof WEAPONENTNAMES; i++)
       if (WEAPONENTNAMES[i][0]) RegisterHam(Ham_Item_Deploy, WEAPONENTNAMES[i], "fw_Item_Deploy_Post", 1)
    RegisterHam(Ham_Weapon_PrimaryAttack, "weapon_m4a1", "fw_hk416ex_PrimaryAttack")
    RegisterHam(Ham_Weapon_PrimaryAttack, "weapon_m4a1", "fw_hk416ex_PrimaryAttack_Post", 1)
    RegisterHam(Ham_Item_PostFrame, "weapon_m4a1", "hk416ex_ItemPostFrame");
    RegisterHam(Ham_Weapon_Reload, "weapon_m4a1", "hk416ex_Reload");
    RegisterHam(Ham_Weapon_Reload, "weapon_m4a1", "hk416ex_Reload_Post", 1);
    RegisterHam(Ham_TakeDamage, "player", "fw_TakeDamage")
    register_forward(FM_SetModel, "fw_SetModel")
    register_forward(FM_UpdateClientData, "fw_UpdateClientData_Post", 1)
    register_forward(FM_PlaybackEvent, "fwPlaybackEvent")

    RegisterHam(Ham_TraceAttack, "worldspawn", "fw_TraceAttack", 1)
    RegisterHam(Ham_TraceAttack, "func_breakable", "fw_TraceAttack", 1)
    RegisterHam(Ham_TraceAttack, "func_wall", "fw_TraceAttack", 1)
    RegisterHam(Ham_TraceAttack, "func_door", "fw_TraceAttack", 1)
    RegisterHam(Ham_TraceAttack, "func_door_rotating", "fw_TraceAttack", 1)
    RegisterHam(Ham_TraceAttack, "func_plat", "fw_TraceAttack", 1)
    RegisterHam(Ham_TraceAttack, "func_rotating", "fw_TraceAttack", 1)
     

    register_cvar("cs_set_user_money", "6000")
    cvar_dmg_hk416ex = register_cvar("cs_hk416ex_dmg", "0.8")
    cvar_recoil_hk416ex = register_cvar("cs_hk416ex_recoil", "0.2")
    cvar_clip_hk416ex = register_cvar("cs_hk416ex_clip", "30")
    cvar_spd_hk416ex = register_cvar("cs_hk416ex_spd", "0.4")
    cvar_hk416ex_ammo = register_cvar("cs_hk416ex_ammo", "90")
     

     
    register_clcmd("say /lal", "give_hk416ex")
     
    g_MaxPlayers = get_maxplayers()
}

public plugin_precache()
{
    precache_model(hk416ex_V_MODEL)
    precache_model(hk416ex_P_MODEL)
    precache_model(hk416ex_W_MODEL)
    for(new i = 0; i < sizeof Fire_Sounds; i++)
    precache_sound(Fire_Sounds[i])    
    m_iBlood[0] = precache_model("sprites/blood.spr")
    m_iBlood[1] = precache_model("sprites/bloodspray.spr")
    register_forward(FM_PrecacheEvent, "fwPrecacheEvent_Post", 1)
}

public fw_TraceAttack(iEnt, iAttacker, Float:flDamage, Float:fDir[3], ptr, iDamageType)
{
    if(!is_user_alive(iAttacker))
       return;

    new g_currentweapon = get_user_weapon(iAttacker)

    if(g_currentweapon != CSW_M4A1) return
     
    if(!g_has_hk416ex[iAttacker]) return

    static Float:flEnd[3]
    get_tr2(ptr, TR_vecEndPos, flEnd)
     
    if(iEnt)
    {
       // Put decal on an entity
       message_begin(MSG_BROADCAST, SVC_TEMPENTITY)
       write_byte(TE_DECAL)
       write_coord_f(flEnd[0])
       write_coord_f(flEnd[1])
       write_coord_f(flEnd[2])
       write_byte(GUNSHOT_DECALS[random_num ( 0, sizeof GUNSHOT_DECALS -1 ) ] )
       write_short(iEnt)
       message_end()
    }
    else
    {
       // Put decal on "world" (a wall)
       message_begin(MSG_BROADCAST, SVC_TEMPENTITY)
       write_byte(TE_WORLDDECAL)
       write_coord_f(flEnd[0])
       write_coord_f(flEnd[1])
       write_coord_f(flEnd[2])
       write_byte(GUNSHOT_DECALS[random_num ( 0, sizeof GUNSHOT_DECALS -1 ) ] )
       message_end()
    }
     
    // Show sparcles
    message_begin(MSG_BROADCAST, SVC_TEMPENTITY)
    write_byte(TE_GUNSHOTDECAL)
    write_coord_f(flEnd[0])
    write_coord_f(flEnd[1])
    write_coord_f(flEnd[2])
    write_short(iAttacker)
    write_byte(GUNSHOT_DECALS[random_num ( 0, sizeof GUNSHOT_DECALS -1 ) ] )
    message_end()
}

public plugin_natives ()
{
    register_native("give_weapon_hk416ex", "native_give_weapon_add", 1)
}
public native_give_weapon_add(id)
{
    give_hk416ex(id)
}
public fwPrecacheEvent_Post(type, const name[])
{
    if (equal("events/m4a1.sc", name))
    {
       g_orig_event_hk416ex = get_orig_retval()
       return FMRES_HANDLED
    }
     
    return FMRES_IGNORED
}

public client_connect(id)
{
    g_has_hk416ex [id]= false
}

public client_disconnect(id)
{
    g_has_hk416ex [id]= false
}

public fw_SetModel(entity, model[])
{
    if(!is_valid_ent(entity))
       return FMRES_IGNORED;
     
    static szClassName[33]
    entity_get_string(entity, EV_SZ_classname, szClassName, charsmax(szClassName))
        
    if(!equal(szClassName, "weaponbox"))
       return FMRES_IGNORED;
     
    static iOwner
     
    iOwner = entity_get_edict(entity, EV_ENT_owner)
     
    if(equal(model, "models/w_m4a1.mdl"))
    {
       static iStoredSVDID
        
       iStoredSVDID = find_ent_by_owner(ENG_NULLENT, "weapon_m4a1", entity)
     
       if(!is_valid_ent(iStoredSVDID))
          return FMRES_IGNORED;
     
       if(g_has_hk416ex[iOwner])
       {
          entity_set_int(iStoredSVDID, EV_INT_WEAPONKEY, hk416ex_WEAPONKEY)
          g_has_hk416ex [iOwner]= false
           
          entity_set_model(entity, hk416ex_W_MODEL)
           
          return FMRES_SUPERCEDE;
       }
    }
     
     
    return FMRES_IGNORED;
}
public give_hk416ex(id)
{
    drop_weapons(id, 1);
    new iWep2 = give_item(id,"weapon_m4a1")
    if( iWep2 > 0 )
    {
       cs_set_weapon_ammo(iWep2, get_pcvar_num(cvar_clip_hk416ex))
       cs_set_user_money (id, CSW_M4A1, get_pcvar_num(cvar_hk416ex_ammo))    
       set_pdata_float(id, m_flNextAttack, 1.0, PLAYER_LINUX_XTRA_OFF)
    }
    g_has_hk416ex [id]= true;
}

public fw_hk416ex_AddToPlayer(hk416ex, id)
{
    if(!is_valid_ent(hk416ex) || !is_user_connected(id))
       return HAM_IGNORED;
     
    if(entity_get_int(hk416ex, EV_INT_WEAPONKEY) == hk416ex_WEAPONKEY)
    {
       g_has_hk416ex [id]= true
        
       entity_set_int(hk416ex, EV_INT_WEAPONKEY, 0)
        
       return HAM_HANDLED;
    }
     
    return HAM_IGNORED;
}

public fw_UseStationary_Post(entity, caller, activator, use_type)
{
    if (use_type == USE_STOPPED && is_user_connected(caller))
       replace_weapon_models(caller, get_user_weapon(caller))
}

public fw_Item_Deploy_Post(weapon_ent)
{
    static owner
    owner = fm_cs_get_weapon_ent_owner(weapon_ent)
     
    static weaponid
    weaponid = cs_get_weapon_id(weapon_ent)
     
    replace_weapon_models(owner, weaponid)
}

public CurrentWeapon(id)
{
    replace_weapon_models(id, read_data(2))

    if(read_data(2) != CSW_M4A1 || !g_has_hk416ex[id])
           return
       
         static Float:iSpeed
         if(g_has_hk416ex[id])
           iSpeed = get_pcvar_float(cvar_spd_hk416ex)
       
         static weapon[32],Ent
         get_weaponname(read_data(2),weapon,31)
         Ent = find_ent_by_owner(-1,weapon,id)
         if(Ent)
         {
           static Float:Delay
           Delay = get_pdata_float( Ent, 46, 4) * iSpeed
           if (Delay > 0.0)
           {
                set_pdata_float(Ent, 46, Delay, 4)
           }
      }
}

replace_weapon_models(id, weaponid)
{
    switch (weaponid)
    {
       case CSW_M4A1:
       {
                 
          if(g_has_hk416ex[id])
          {
             set_pev(id, pev_viewmodel2, hk416ex_V_MODEL)
             set_pev(id, pev_weaponmodel2, hk416ex_P_MODEL)
             if(oldweap [id]!= CSW_M4A1)  
             {
                set_pdata_float(id, m_flNextAttack, 1.0, PLAYER_LINUX_XTRA_OFF)
             }
          }
       }
    }
    oldweap [id]= weaponid
}

public fw_UpdateClientData_Post(Player, SendWeapons, CD_Handle)
{
         if(!is_user_alive(Player) || (get_user_weapon(Player) != CSW_M4A1) || !g_has_hk416ex[Player])
         return FMRES_IGNORED
     
    set_cd(CD_Handle, CD_flNextAttack, halflife_time () + 0.001)
    return FMRES_HANDLED
}

public fw_hk416ex_PrimaryAttack(Weapon)
{
    new Player = get_pdata_cbase(Weapon, 41, 4)
     
    if (!g_has_hk416ex[Player])
       return;
     
    pev(Player,pev_punchangle,cl_pushangle[Player])
     
    g_clip_ammo [Player]= cs_get_weapon_ammo(Weapon)
}

public fwPlaybackEvent(flags, invoker, eventid, Float:delay, Float:origin[3], Float:angles[3], Float:fparam1, Float:fparam2, iParam1, iParam2, bParam1, bParam2)
{
    if ((eventid != g_orig_event_hk416ex))
       return FMRES_IGNORED
    if (!(1 <= invoker <= g_MaxPlayers))
       return FMRES_IGNORED

    playback_event(flags | FEV_HOSTONLY, invoker, eventid, delay, origin, angles, fparam1, fparam2, iParam1, iParam2, bParam1, bParam2)
    return FMRES_SUPERCEDE
}

public fw_hk416ex_PrimaryAttack_Post(Weapon)
{
    new Player = get_pdata_cbase(Weapon, 41, 4)
     
    new szClip, szAmmo
    get_user_weapon(Player, szClip, szAmmo)
     
    if(!is_user_alive(Player))
       return;

    if(g_has_hk416ex[Player])
    {
       if (!g_clip_ammo[Player])
          return

       new Float:push[3]
       pev(Player,pev_punchangle,push)
       xs_vec_sub(push,cl_pushangle[Player],push)
        
       xs_vec_mul_scalar(push,get_pcvar_float(cvar_recoil_hk416ex),push)
       xs_vec_add(push,cl_pushangle[Player],push)
       set_pev(Player,pev_punchangle,push)
           
       emit_sound(Player, CHAN_WEAPON, Fire_Sounds[0], VOL_NORM, ATTN_NORM, 0, PITCH_NORM)
       UTIL_PlayWeaponAnimation(Player, cs_get_weapon_silen(Weapon) ? 1 : 8)
    }
}

public fw_TakeDamage(victim, inflictor, attacker, Float:damage)
{
    if (victim != attacker && is_user_connected(attacker))
    {
       if(get_user_weapon(attacker) == CSW_M4A1)
       {
          if(g_has_hk416ex[attacker])
             SetHamParamFloat(4, damage * get_pcvar_float(cvar_dmg_hk416ex))
       }
    }
}

public message_DeathMsg(msg_id, msg_dest, id)
{
    static szTruncatedWeapon[33], iAttacker, iVictim
     
    get_msg_arg_string(4, szTruncatedWeapon, charsmax(szTruncatedWeapon))
     
    iAttacker = get_msg_arg_int(1)
    iVictim = get_msg_arg_int(2)
     
    if(!is_user_connected(iAttacker) || iAttacker == iVictim)
       return PLUGIN_CONTINUE
     
    if(equal(szTruncatedWeapon, "m4a1") && get_user_weapon(iAttacker) == CSW_M4A1)
    {
       if(g_has_hk416ex[iAttacker])
          set_msg_arg_string(4, "m4a1")
    }
        
    return PLUGIN_CONTINUE
}

stock fm_cs_get_current_weapon_ent(id)
{
    return get_pdata_cbase(id, OFFSET_ACTIVE_ITEM, OFFSET_LINUX);
}

stock fm_cs_get_weapon_ent_owner(ent)
{
    return get_pdata_cbase(ent, OFFSET_WEAPONOWNER, OFFSET_LINUX_WEAPONS);
}

stock UTIL_PlayWeaponAnimation(const Player, const Sequence)
{
    set_pev(Player, pev_weaponanim, Sequence)
     
    message_begin(MSG_ONE_UNRELIABLE, SVC_WEAPONANIM, .player = Player)
    write_byte(Sequence)
    write_byte(pev(Player, pev_body))
    message_end()
}

public hk416ex_ItemPostFrame(weapon_entity) {
    new id = pev(weapon_entity, pev_owner)
    if (!is_user_connected(id))
       return HAM_IGNORED;

    if (!g_has_hk416ex[id])
       return HAM_IGNORED;

    new Float:flNextAttack = get_pdata_float(id, m_flNextAttack, PLAYER_LINUX_XTRA_OFF)

    new iBpAmmo = cs_get_user_bpammo(id, CSW_M4A1);
    new iClip = get_pdata_int(weapon_entity, m_iClip, WEAP_LINUX_XTRA_OFF)

    new fInReload = get_pdata_int(weapon_entity, m_fInReload, WEAP_LINUX_XTRA_OFF)  

    if( fInReload && flNextAttack <= 0.0 )
    {
       new j = min(get_pcvar_num(cvar_clip_hk416ex) - iClip, iBpAmmo)
     
       set_pdata_int(weapon_entity, m_iClip, iClip + j, WEAP_LINUX_XTRA_OFF)
       cs_set_user_bpammo(id, CSW_M4A1, iBpAmmo-j);
        
       set_pdata_int(weapon_entity, m_fInReload, 0, WEAP_LINUX_XTRA_OFF)
       fInReload = 0
    }

    return HAM_IGNORED;
}

public hk416ex_Reload(weapon_entity) {
    new id = pev(weapon_entity, pev_owner)
    if (!is_user_connected(id))
       return HAM_IGNORED;

    if (!g_has_hk416ex[id])
       return HAM_IGNORED;

    g_hk416ex_TmpClip [id]= -1;

    new iBpAmmo = cs_get_user_bpammo(id, CSW_M4A1);
    new iClip = get_pdata_int(weapon_entity, m_iClip, WEAP_LINUX_XTRA_OFF)

    if (iBpAmmo <= 0)
       return HAM_SUPERCEDE;

    if (iClip >= get_pcvar_num(cvar_clip_hk416ex))
       return HAM_SUPERCEDE;

    g_hk416ex_TmpClip [id]= iClip;

    return HAM_IGNORED;
}

public hk416ex_Reload_Post(weapon_entity) {
    new id = pev(weapon_entity, pev_owner)
    if (!is_user_connected(id))
       return HAM_IGNORED;

    if (!g_has_hk416ex[id])
       return HAM_IGNORED;

    if (g_hk416ex_TmpClip [id]== -1)
       return HAM_IGNORED;

    set_pdata_int(weapon_entity, m_iClip, g_hk416ex_TmpClip[id], WEAP_LINUX_XTRA_OFF)

    set_pdata_float(weapon_entity, m_flTimeWeaponIdle, hk416ex_RELOAD_TIME, WEAP_LINUX_XTRA_OFF)

    set_pdata_float(id, m_flNextAttack, hk416ex_RELOAD_TIME, PLAYER_LINUX_XTRA_OFF)

    set_pdata_int(weapon_entity, m_fInReload, 1, WEAP_LINUX_XTRA_OFF)

    UTIL_PlayWeaponAnimation(id, cs_get_weapon_silen(weapon_entity) ? 4 : 11)

    return HAM_IGNORED;
}

stock drop_weapons(id, dropwhat)
{
      static weapons[32], num, i, weaponid
      num = 0
      get_user_weapons(id, weapons, num)
       
      for (i = 0; i < num; i++)
      {
           weaponid = weapons
            
           [i]if (dropwhat == 1 && ((1<<weaponid) & PRIMARY_WEAPONS_BIT_SUM))
           {
                static wname[32]
                get_weaponname(weaponid, wname, sizeof wname - 1)
                engclient_cmd(id, "drop", wname)
           }
      }
}