[WIP] Support for upcoming SPT 3.10, using EFT 0.15.2.32757

ConsoleCommands/BaseTemplateCommand.cs

@@ -1,35 +1,11 @@
 using System;
 using System.Collections.Generic;
 using System.Linq;
-using Comfort.Common;
-using EFT.InventoryLogic;
-
 #nullable enable
 
 namespace EFT.Trainer.ConsoleCommands;
 
 internal abstract class BaseTemplateCommand : ConsoleCommandWithArgument
 {
 	public override string Pattern => RequiredArgumentPattern;
-
-	protected static ItemTemplate[] FindTemplates(string searchShortNameOrTemplateId)
-	{
-		if (!Singleton<ItemFactory>.Instantiated)
-			return [];
-
-		var templates = Singleton<ItemFactory>
-			.Instance
-			.ItemTemplates;
-
-		// Match by TemplateId
-		if (templates.TryGetValue(searchShortNameOrTemplateId, out var template))
-			return [template];
-
-		// Match by short name(s)
-		return templates
-			.Values
-			.Where(t => t.ShortNameLocalizationKey.Localized().IndexOf(searchShortNameOrTemplateId, StringComparison.OrdinalIgnoreCase) >= 0
-						|| t.NameLocalizationKey.Localized().IndexOf(searchShortNameOrTemplateId, StringComparison.OrdinalIgnoreCase) >= 0)
-			.ToArray();
-	}
 }

ConsoleCommands/Spawn.cs

@@ -1,4 +1,5 @@
 using System;
+using System.Collections.Generic;
 using System.Linq;
 using System.Text.RegularExpressions;
 using System.Threading.Tasks;
@@ -10,7 +11,6 @@
 using EFT.Trainer.Features;
 using EFT.Trainer.Properties;
 using JetBrains.Annotations;
-using UnityEngine;
 using Random = UnityEngine.Random;
 
 #nullable enable
@@ -33,7 +33,7 @@ public override void Execute(Match match)
 			return;
 
 		var search = matchGroup.Value;
-		var templates = FindTemplates(search);
+		var templates = TemplateHelper.FindTemplates(search);
 
 		switch (templates.Length)
 		{
@@ -54,7 +54,7 @@ public override void Execute(Match match)
 
 	internal static void SpawnTemplate(string template, Player player, ConsoleCommand command, Func<ItemTemplate, bool> filter)
 	{
-		var result = FindTemplates(template)
+		var result = TemplateHelper.FindTemplates(template)
 			.FirstOrDefault(filter);
 
 		if (result == null)
@@ -79,21 +79,19 @@ private static void SpawnTemplate(ItemTemplate template, Player player, ConsoleC
 					}
 					else
 					{
-						var itemFactory = Singleton<ItemFactory>.Instance;
+						var itemFactory = Singleton<ItemFactoryClass>.Instance;
 						var item = itemFactory.CreateItem(MongoID.Generate(), template._id, null);
 						if (item == null)
 						{
 							command.AddConsoleLog(Strings.ErrorFailedToCreateItem.Red());
 						}
 						else
 						{
-							item.SpawnedInSession = true; // found in raid
-
 							_ = new TraderControllerClass(item, item.Id, item.ShortName);
 							var go = poolManager.CreateLootPrefab(item, ECameraType.Default);
 
 							go.SetActive(value: true);
-							var lootItem = Singleton<GameWorld>.Instance.CreateLootWithRigidbody(go, item, item.ShortName, Singleton<GameWorld>.Instance, randomRotation: false, null, out _);
+							var lootItem = Singleton<GameWorld>.Instance.CreateLootWithRigidbody(go, item, item.ShortName, randomRotation: false, null, out _, true);
 
 							var transform = player.Transform;
 							var position = transform.position
@@ -103,11 +101,67 @@ private static void SpawnTemplate(ItemTemplate template, Player player, ConsoleC
 
 							lootItem.transform.SetPositionAndRotation(position, transform.rotation);
 							lootItem.LastOwner = player;
+
+							// setup after loot item is created, else we are hitting issues with weapon
+							SetupItem(itemFactory, item);
 						}
 					}
 				});
 
 				return Task.CompletedTask;
 			});
 	}
+
+	private static void SetupItem(ItemFactoryClass itemFactory, Item item)
+	{
+		item.SpawnedInSession = true; // found in raid
+
+		if (item.TryGetItemComponent<DogtagComponent>(out var dogtag))
+		{
+			dogtag.AccountId = Random.Range(0, int.MaxValue).ToString();
+			dogtag.ProfileId = Random.Range(0, int.MaxValue).ToString();
+			dogtag.Nickname = $"Rambo{Random.Range(1, 256)}";
+			dogtag.Side = Enum.GetValues(typeof(EPlayerSide)).Cast<EPlayerSide>().Random();
+			dogtag.Level = Random.Range(1, 69);
+			dogtag.Time = DateTime.Now;
+			dogtag.Status = "died";
+			dogtag.KillerAccountId = Random.Range(0, int.MaxValue).ToString();
+			dogtag.KillerProfileId = Random.Range(0, int.MaxValue).ToString();
+			dogtag.KillerName = "";
+			dogtag.WeaponName = "";
+		}
+
+		if (item.TryGetItemComponent<ArmorHolderComponent>(out var armorHolder))
+			FillSlots(itemFactory, armorHolder.ArmorSlots);
+
+		if (item.TryGetItemComponent<RepairableComponent>(out var repairable))
+		{
+			repairable.MaxDurability = repairable.TemplateDurability;
+			repairable.Durability = repairable.MaxDurability;
+		}
+
+		if (item is CompoundItem compound)
+			FillSlots(itemFactory, compound.AllSlots);
+	}
+
+	private static void FillSlots(ItemFactoryClass itemFactory, IEnumerable<Slot> slots)
+	{
+		foreach (var slot in slots)
+		{
+			if (slot.Items.Any())
+				continue;
+
+			var filter = slot
+				.Filters.FirstOrDefault()?
+				.Filter.Random();
+
+			if (filter == null)
+				continue;
+
+			var item = itemFactory.CreateItem(MongoID.Generate(), filter, null);
+			SetupItem(itemFactory, item);
+
+			slot.AddWithoutRestrictions(item);
+		}
+	}
 }

ConsoleCommands/SpawnHideoutItems.cs

@@ -0,0 +1,40 @@
+using System.Collections.Generic;
+using System.Linq;
+using EFT.Trainer.Extensions;
+using EFT.Trainer.Features;
+using EFT.Trainer.Properties;
+using HarmonyLib;
+using JetBrains.Annotations;
+
+#nullable enable
+
+namespace EFT.Trainer.ConsoleCommands;
+
+[UsedImplicitly]
+internal class SpawnHideoutItems : ConsoleCommandWithoutArgument
+{
+	public override string Name => Strings.CommandSpawnHideoutItems;
+
+	public override void Execute()
+	{
+		var player = GameState.Current?.LocalPlayer;
+		if (!player.IsValid())
+			return;
+
+		var manager = player.Profile?.WishlistManager;
+		if (manager == null)
+			return;
+
+		// Find the obfuscated method that returns the computed hidout items
+		// We need to have the auto-add hideout items enabled in EFT settings
+		var method = AccessTools
+			.GetDeclaredMethods(manager.GetType())
+			.FirstOrDefault(m => m.ReturnType == typeof(IEnumerable<MongoID>));
+
+		if (method?.Invoke(manager, []) is not IEnumerable<MongoID> templates)
+			return;
+
+		foreach (var template in templates)
+			Spawn.SpawnTemplate(template, player, this, i => true);
+	}
+}

ConsoleCommands/Template.cs

@@ -2,6 +2,7 @@
 using System.Text.RegularExpressions;
 using Comfort.Common;
 using EFT.Trainer.Extensions;
+using EFT.Trainer.Features;
 using EFT.Trainer.Properties;
 using JetBrains.Annotations;
 
@@ -20,12 +21,12 @@ public override void Execute(Match match)
 		if (matchGroup is not { Success: true })
 			return;
 
-		if (!Singleton<ItemFactory>.Instantiated)
+		if (!Singleton<ItemFactoryClass>.Instantiated)
 			return;
 
 		var search = matchGroup.Value;
 
-		var templates = FindTemplates(search).ToArray();
+		var templates = TemplateHelper.FindTemplates(search);
 
 		foreach (var template in templates)
 			AddConsoleLog(string.Format(Strings.CommandTemplateEnumerateFormat, template._id, template.ShortNameLocalizationKey.Localized().Green(), template.NameLocalizationKey.Localized()));

ConsoleCommands/TrackList.cs

@@ -23,7 +23,7 @@ internal static void ShowTrackList(ConsoleCommand command, Features.LootItems lo
 			command.AddConsoleLog(Strings.CommandTrackListUpdated);
 
 		foreach (var templateId in lootItems.Wishlist)
-			command.AddConsoleLog(string.Format(Strings.CommandTrackListWishListEnumerateFormat, templateId.LocalizedShortName()));
+			command.AddConsoleLog(string.Format(Strings.CommandTrackListWishListEnumerateFormat, ((MongoID)templateId).LocalizedShortName()));
 
 		foreach (var item in lootItems.TrackedNames)
 		{

Extensions/ItemExtensions.cs

@@ -21,10 +21,10 @@ public static bool IsFiltered(this Item item)
 		if (ItemViewFactory.IsSecureContainer(item))
 			return true;
 
-		if (item.CurrentAddress?.Container is { ParentItem.TemplateId: KnownTemplateIds.BossContainer })
+		if (item.CurrentAddress?.Container?.ParentItem?.TemplateId.ToString() == KnownTemplateIds.BossContainer)
 			return true;
 
-		return item.TemplateId switch
+		return item.TemplateId.ToString() switch
 		{
 			KnownTemplateIds.DefaultInventory or KnownTemplateIds.Pockets => true,
 			_ => false

Features/AirDrop.cs

@@ -0,0 +1,31 @@
+using System.Linq;
+using Comfort.Common;
+using EFT.InventoryLogic;
+using EFT.Trainer.Properties;
+using JetBrains.Annotations;
+using UnityEngine;
+
+#nullable enable
+
+namespace EFT.Trainer.Features;
+
+[UsedImplicitly]
+internal class AirDrop : TriggerFeature
+{
+	public override string Name => Strings.FeatureAirDropName;
+	public override string Description => Strings.FeatureAirDropDescription;
+
+	public override KeyCode Key { get; set; } = KeyCode.None;
+
+	protected override void UpdateOnceWhenTriggered()
+	{
+		var player = GameState.Current?.LocalPlayer;
+		if (player == null)
+			return;
+
+		if (TemplateHelper.FindTemplates(KnownTemplateIds.RedSignalFlare).FirstOrDefault() is not AmmoTemplate template)
+			return;
+
+		player.HandleFlareSuccessEvent(player.Transform.position, template);
+	}
+}

Features/Ammunition.cs

@@ -62,11 +62,11 @@ private static void ShootPostfix(object shot)
 
 	private static Item CreateAmmo(Item ammo)
 	{
-		var instantiated = Singleton<ItemFactory>.Instantiated;
+		var instantiated = Singleton<ItemFactoryClass>.Instantiated;
 		if (!instantiated)
 			return ammo;
 
-		var instance = Singleton<ItemFactory>.Instance;
+		var instance = Singleton<ItemFactoryClass>.Instance;
 		var itemId = Guid.NewGuid().ToString("N").Substring(0, 24);
 		return instance.CreateItem(itemId, ammo.TemplateId, null) ?? ammo;
 	}

Features/Commands.cs

@@ -42,6 +42,10 @@ protected override void Update()
 		if (!PreloaderUI.Instantiated)
 			return;
 
+#if DEBUG
+		ConsoleScreen.IAmDevShowMeLogs = true;
+#endif
+
 		RegisterPropertyDisplays();
 		RegisterCommands();
 	}

Features/Examine.cs

@@ -1,6 +1,7 @@
 using EFT.InventoryLogic;
 using EFT.Trainer.Properties;
 using JetBrains.Annotations;
+using static EFT.Player;
 
 #nullable enable
 
@@ -22,32 +23,30 @@ protected static bool ExaminedPrefix(ref bool __result)
 			return true; // keep using original code, we are not enabled
 
 		__result = true;
-		return false;  // skip the original code and all other prefix methods 
+		return false; // skip the original code and all other prefix methods 
 	}
 
+#pragma warning disable IDE0060
 	[UsedImplicitly]
-	protected static bool GetSearchStatePrefix(SearchableItemClass __instance)
+	protected static bool SinglePlayerInventoryControllerConstructorPrefix(Player player, Profile profile, ref bool examined)
 	{
 		var feature = FeatureFactory.GetFeature<Examine>();
 		if (feature == null || !feature.Enabled)
 			return true; // keep using original code, we are not enabled
 
-		var player = GameState.Current?.LocalPlayer;
-		if (player == null)
-			return true;
-
-		__instance.UncoverAll(player.ProfileId);
+		// this will make the game use the passthrough type implementing IPlayerSearchController, ISearchController with all items known and searched
+		examined = true;
 		return true;
 	}
-
+#pragma warning restore IDE0060
 
 	protected override void UpdateWhenEnabled()
 	{
 		HarmonyPatchOnce(harmony =>
 		{
 			HarmonyPrefix(harmony, typeof(Profile), nameof(Profile.Examined), nameof(ExaminedPrefix), [typeof(string)]);
 			HarmonyPrefix(harmony, typeof(Profile), nameof(Profile.Examined), nameof(ExaminedPrefix), [typeof(Item)]);
-			HarmonyPrefix(harmony, typeof(SearchableItemClass), nameof(SearchableItemClass.GetSearchState), nameof(GetSearchStatePrefix));
+			HarmonyConstructorPrefix(harmony, typeof(SinglePlayerInventoryController), nameof(SinglePlayerInventoryControllerConstructorPrefix), [typeof(Player), typeof(Profile), typeof(bool)]);
 		});
 	}
 }