Fix multiple custom weapon

Let's say if you equip AWP and Knife custom weapon, only AWP got equipped. This commit fixes it.

Author: Cruze03

Store/src/item/items/customweapon.cs

@@ -1,5 +1,6 @@
-using CounterStrikeSharp.API;
+using CounterStrikeSharp.API;
 using CounterStrikeSharp.API.Core;
+using CounterStrikeSharp.API.Modules.Entities;
 using CounterStrikeSharp.API.Modules.Memory;
 using CounterStrikeSharp.API.Modules.Utils;
 using System.Runtime.InteropServices;
@@ -80,50 +81,67 @@ public static void OnEntityCreated(CEntityInstance entity)
             return;
         }
 
-        CBasePlayerWeapon weapon = entity.As<CBasePlayerWeapon>();
-
         Server.NextWorldUpdate(() =>
         {
-            if (weapon.OriginalOwnerXuidLow == 0)
-            {
-                return;
-            }
+            var weapon = new CBasePlayerWeapon(entity.Handle);
+			
+            if (!weapon.IsValid) return;
+            
+            if (weapon.OriginalOwnerXuidLow <= 0) return;
 
-            CCSPlayerController? player = Utilities.GetPlayerFromSteamId(weapon.OriginalOwnerXuidLow);
+            SteamID? _steamid = null;
 
-            if (player == null)
-            {
-                return;
-            }
+            if (weapon.OriginalOwnerXuidLow > 0)
+                _steamid = new(weapon.OriginalOwnerXuidLow);
 
-            Store_Equipment? playerequipment = Item.GetPlayerEquipments(player).FirstOrDefault(p => p.SteamID == player.SteamID && p.Type == "customweapon");
+            CCSPlayerController? player = null;
 
-            if (playerequipment == null)
+            if (_steamid != null && _steamid.IsValid())
             {
-                return;
-            }
+                player = Utilities.GetPlayers().FirstOrDefault(p => p.IsValid && p.SteamID == _steamid.SteamId64);
 
-            Dictionary<string, string>? itemdata = Item.GetItem(playerequipment.Type, playerequipment.UniqueId);
-
-            if (itemdata == null)
-            {
-                return;
+                if (player == null)
+                    player = Utilities.GetPlayerFromSteamId(weapon.OriginalOwnerXuidLow);
             }
-
-            if (!weapon.DesignerName.Contains(itemdata["weapon"]))
+            else
             {
-                return;
+                CCSWeaponBaseGun gun = weapon.As<CCSWeaponBaseGun>();
+                player = Utilities.GetPlayerFromIndex((int)weapon.OwnerEntity.Index) ?? Utilities.GetPlayerFromIndex((int)gun.OwnerEntity.Value!.Index);
             }
 
-            CBasePlayerWeapon? activeweapon = player.PlayerPawn.Value?.WeaponServices?.ActiveWeapon.Value;
+            if (string.IsNullOrEmpty(player?.PlayerName)) return;
 
-            if (activeweapon != null && weapon == activeweapon)
-            {
-                Weapon.UpdateModel(player, activeweapon, itemdata["uniqueid"], true);
-            }
-            else
+            var playerequipments = Item.GetPlayerEquipments(player).Where(p => p.SteamID == player.SteamID && p.Type == "customweapon").ToList();
+
+            foreach(var playerequipment in playerequipments)
             {
-                Weapon.UpdateModel(player, weapon, itemdata["uniqueid"], false);
+                Dictionary<string, string>? itemdata = Item.GetItem(playerequipment.Type, playerequipment.UniqueId);
+
+                if (itemdata == null) continue;
+
+                string classname = weapon.DesignerName;
+
+                if (classname.Contains("bayonet"))
+                {
+                    classname = "weapon_knife";
+                }
+                
+                if (!classname.Contains(itemdata["weapon"]))
+                {
+                    continue;
+                }
+
+                CBasePlayerWeapon? activeweapon = player.PlayerPawn.Value?.WeaponServices?.ActiveWeapon.Value;
+
+                if (activeweapon != null && weapon == activeweapon)
+                {
+                    Weapon.UpdateModel(player, activeweapon, itemdata["uniqueid"], true);
+                }
+                else
+                {
+                    Weapon.UpdateModel(player, weapon, itemdata["uniqueid"], false);
+                }
+                break;
             }
         });
     }
@@ -204,4 +222,4 @@ public static void ResetWeapon(CCSPlayerController player, CBasePlayerWeapon wea
             SetViewModel(player, globalnamedata[0]);
         }
     }
-}
+}