AI захватит мир

Code/Inventory/Usable/AmmoUseableBase.cs

@@ -2,35 +2,83 @@
 
 public abstract class AmmoUseableBase : UseableBase
 {
+	private WeaponItemDefinition _cachedWeaponDef;
+	private InventoryItem _cachedAmmoItem;
+	
 	protected InventoryItem AmmoItem => FindAmmoItem();
 
 	private InventoryItem FindAmmoItem()
 	{
-		//По типу патрон поиск + енум типа патрон
-		return Dedugan.Local.Inventory.Items.FirstOrDefault( i => i.Definition.Name == "Pistol Ammo" );
+		var player = Dedugan.Local;
+		if (player?.Inventory == null) 
+		{
+			return null;
+		}
+
+		// Кэшируем WeaponDefinition для избежания повторных вызовов
+		if (_cachedWeaponDef == null)
+		{
+			_cachedWeaponDef = GetWeaponDefinition();
+			if (_cachedWeaponDef == null) 
+			{
+				return null;
+			}
+		}
+
+		// Проверяем кэшированный результат
+		if (_cachedAmmoItem != null && _cachedAmmoItem.Count > 0)
+		{
+			return _cachedAmmoItem;
+		}
+
+		// Ищем патроны
+		foreach (var item in player.Inventory.Items)
+		{
+			if (item.Definition is AmmoItemDefinition ammoDef && ammoDef.IsCompatibleWith(_cachedWeaponDef.AmmoType))
+			{
+				_cachedAmmoItem = item;
+				return item;
+			}
+		}
+
+		_cachedAmmoItem = null;
+		return null;
+	}
+
+	protected virtual WeaponItemDefinition GetWeaponDefinition()
+	{
+		// Переопределите в наследниках для возврата определения оружия
+		return null;
 	}
 
 	public override bool CanUse()
 	{
 		var ammo = AmmoItem;
-		return base.CanUse() && ammo != null && ammo.Count > 0;
+		var baseCanUse = base.CanUse();
+		var hasAmmo = ammo != null && ammo.Count > 0;
+		
+		return baseCanUse && hasAmmo;
 	}
 
 	public override void Use()
 	{
-		if ( !CanUse() )
+		if (!CanUse())
 			return;
 
 		OnUse();
 
 		var ammo = AmmoItem;
-		if ( ammo != null )
+		if (ammo != null)
 		{
-			ammo.Count--;
-
-			if ( ammo.Count <= 0 )
+			// Уменьшаем количество патронов
+			if (ammo.TryRemoveCount(1))
 			{
-				Dedugan.Local.Inventory.RemoveItem( ammo );
+				// Если патроны закончились, удаляем предмет из инвентаря
+				if (ammo.Count <= 0)
+				{
+					Dedugan.Local.Inventory.RemoveItem(ammo);
+					_cachedAmmoItem = null; // Очищаем кэш
+				}
 			}
 		}
 	}