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

#pragma semicolon 1

#define EF_NODRAW 32

new CustomModel1;
new CustomModel2;

new bool:SpawnCheck[MAXPLAYERS+1];
new ClientVM[MAXPLAYERS+1][2];
new bool:IsCustom[MAXPLAYERS+1];

public OnPluginStart()
{
     HookEvent("player_death", Event_PlayerDeath);
     HookEvent("player_spawn", Event_PlayerSpawn);
      
     for (new client = 1; client <= MaxClients; client++)  
     {  
         if (IsClientInGame(client))  
         {
             SDKHook(client, SDKHook_PostThinkPost, OnPostThinkPost);
              
             ClientVM[client][0] = GetEntPropEnt(client, Prop_Send, "m_hViewModel");
              
             new PVM = -1;
             while ((PVM = FindEntityByClassname(PVM, "predicted_viewmodel")) != -1)
             {
                 if (GetEntPropEnt(PVM, Prop_Send, "m_hOwner") == client)
                 {
                     if (GetEntProp(PVM, Prop_Send, "m_nViewModelIndex") == 1)
                     {
                         ClientVM[client][1] = PVM;
                         break;
                     }
                 }
             }
         }  
     }
}

public OnMapStart()
{
     CustomModel1 = PrecacheModel("models/Weapons/v_smg_p90.mdl");
     CustomModel2 = PrecacheModel("models/Weapons/v_smg_tmp.mdl");
}

public OnClientPutInServer(client)
{
     SDKHook(client, SDKHook_PostThinkPost, OnPostThinkPost);
}

public OnEntityCreated(entity, const String:classname[])
{
     if (StrEqual(classname, "predicted_viewmodel", false))
     {
         SDKHook(entity, SDKHook_Spawn, OnEntitySpawned);
     }
}

public OnEntitySpawned(entity)
{
     new Owner = GetEntPropEnt(entity, Prop_Send, "m_hOwner");
     if ((Owner > 0) && (Owner <= MaxClients))
     {
         if (GetEntProp(entity, Prop_Send, "m_nViewModelIndex") == 0)
         {
             ClientVM[Owner][0] = entity;
         }
         else if (GetEntProp(entity, Prop_Send, "m_nViewModelIndex") == 1)
         {
             ClientVM[Owner][1] = entity;
         }
     }
}

public OnPostThinkPost(client)
{
     static OldWeapon[MAXPLAYERS + 1];
     static OldSequence[MAXPLAYERS + 1];
     static Float:OldCycle[MAXPLAYERS + 1];
      
     decl String:ClassName[30];
     new WeaponIndex;
      
     if (!IsPlayerAlive(client))
     {
         new spec = GetEntPropEnt(client, Prop_Send, "m_hObserverTarget");
         if (spec != -1)
         {
             WeaponIndex = GetEntPropEnt(spec, Prop_Send, "m_hActiveWeapon");
             GetEdictClassname(WeaponIndex, ClassName, sizeof(ClassName));
             if (StrEqual("weapon_ump45", ClassName, false))
             {
                 SetEntProp(ClientVM[client][1], Prop_Send, "m_nModelIndex", CustomModel1);
             }
             else if (StrEqual("weapon_mp5navy", ClassName, false))
             {
                 SetEntProp(ClientVM[client][1], Prop_Send, "m_nModelIndex", CustomModel2);
             }
         }
          
         return;
     }
      
     WeaponIndex = GetEntPropEnt(client, Prop_Send, "m_hActiveWeapon");
     new Sequence = GetEntProp(ClientVM[client][0], Prop_Send, "m_nSequence");
     new Float:Cycle = GetEntPropFloat(ClientVM[client][0], Prop_Data, "m_flCycle");
      
     if (WeaponIndex <= 0)
     {
         new EntEffects = GetEntProp(ClientVM[client][1], Prop_Send, "m_fEffects");
         EntEffects |= EF_NODRAW;
         SetEntProp(ClientVM[client][1], Prop_Send, "m_fEffects", EntEffects);
          
         IsCustom [client]= false;
              
         OldWeapon [client]= WeaponIndex;
         OldSequence [client]= Sequence;
         OldCycle [client]= Cycle;
          
         return;
     }

     if (WeaponIndex != OldWeapon[client])
     {
         GetEdictClassname(WeaponIndex, ClassName, sizeof(ClassName));
         if (StrEqual("weapon_ump45", ClassName, false))
         {
             new EntEffects = GetEntProp(ClientVM[client][0], Prop_Send, "m_fEffects");
             EntEffects |= EF_NODRAW;
             SetEntProp(ClientVM[client][0], Prop_Send, "m_fEffects", EntEffects);
             EntEffects = GetEntProp(ClientVM[client][1], Prop_Send, "m_fEffects");
             EntEffects &= ~EF_NODRAW;
             SetEntProp(ClientVM[client][1], Prop_Send, "m_fEffects", EntEffects);
              
             SetEntProp(ClientVM[client][1], Prop_Send, "m_nModelIndex", CustomModel1);
             SetEntPropEnt(ClientVM[client][1], Prop_Send, "m_hWeapon", GetEntPropEnt(ClientVM[client][0], Prop_Send, "m_hWeapon"));
              
             SetEntProp(ClientVM[client][1], Prop_Send, "m_nSequence", GetEntProp(ClientVM[client][0], Prop_Send, "m_nSequence"));
             SetEntPropFloat(ClientVM[client][1], Prop_Send, "m_flPlaybackRate", GetEntPropFloat(ClientVM[client][0], Prop_Send, "m_flPlaybackRate"));
              
             IsCustom [client]= true;
         }
         else if (StrEqual("weapon_mp5navy", ClassName, false))
         {
             new EntEffects = GetEntProp(ClientVM[client][0], Prop_Send, "m_fEffects");
             EntEffects |= EF_NODRAW;
             SetEntProp(ClientVM[client][0], Prop_Send, "m_fEffects", EntEffects);
              
             EntEffects = GetEntProp(ClientVM[client][1], Prop_Send, "m_fEffects");
             EntEffects &= ~EF_NODRAW;
             SetEntProp(ClientVM[client][1], Prop_Send, "m_fEffects", EntEffects);
              
             SetEntProp(ClientVM[client][1], Prop_Send, "m_nModelIndex", CustomModel2);
             SetEntPropEnt(ClientVM[client][1], Prop_Send, "m_hWeapon", GetEntPropEnt(ClientVM[client][0], Prop_Send, "m_hWeapon"));
              
             SetEntProp(ClientVM[client][1], Prop_Send, "m_nSequence", GetEntProp(ClientVM[client][0], Prop_Send, "m_nSequence"));
             SetEntPropFloat(ClientVM[client][1], Prop_Send, "m_flPlaybackRate", GetEntPropFloat(ClientVM[client][0], Prop_Send, "m_flPlaybackRate"));
              
             IsCustom [client]= true;
         }
         else
         {
             new EntEffects = GetEntProp(ClientVM[client][1], Prop_Send, "m_fEffects");
             EntEffects |= EF_NODRAW;
             SetEntProp(ClientVM[client][1], Prop_Send, "m_fEffects", EntEffects);
              
             IsCustom [client]= false;
         }
     }
     else
     {
         if (IsCustom[client])
         {
             SetEntProp(ClientVM[client][1], Prop_Send, "m_nSequence", GetEntProp(ClientVM[client][0], Prop_Send, "m_nSequence"));
             SetEntPropFloat(ClientVM[client][1], Prop_Send, "m_flPlaybackRate", GetEntPropFloat(ClientVM[client][0], Prop_Send, "m_flPlaybackRate"));
              
             if ((Cycle < OldCycle[client]) && (Sequence == OldSequence[client]))
             {
                 SetEntProp(ClientVM[client][1], Prop_Send, "m_nSequence", 0);
             }
         }
     }
     if (SpawnCheck[client])
     {
         SpawnCheck [client]= false;
         if (IsCustom[client])
         {
             new EntEffects = GetEntProp(ClientVM[client][0], Prop_Send, "m_fEffects");
             EntEffects |= EF_NODRAW;
             SetEntProp(ClientVM[client][0], Prop_Send, "m_fEffects", EntEffects);
         }
     }
      
     OldWeapon [client]= WeaponIndex;
     OldSequence [client]= Sequence;
     OldCycle [client]= Cycle;
}

public Event_PlayerDeath(Handle:event, const String:name[], bool:dontBroadcast)
{
     new UserId = GetEventInt(event, "userid");
     new client = GetClientOfUserId(UserId);
      
     new EntEffects = GetEntProp(ClientVM[client][1], Prop_Send, "m_fEffects");
     EntEffects |= EF_NODRAW;
     SetEntProp(ClientVM[client][1], Prop_Send, "m_fEffects", EntEffects);
}

public Event_PlayerSpawn(Handle:event, const String:name[], bool:dontBroadcast)
{
     new UserId = GetEventInt(event, "userid");
     new client = GetClientOfUserId(UserId);
      
     SpawnCheck [client]= true;
}