103 lines
3.9 KiB
C#
103 lines
3.9 KiB
C#
using Comfort.Common;
|
|
using EFT.InventoryLogic;
|
|
using EFT.UI;
|
|
using HarmonyLib;
|
|
using SPT.Reflection.Patching;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Reflection;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace UIFixes
|
|
{
|
|
public static class UnloadAmmoPatches
|
|
{
|
|
public static void Enable()
|
|
{
|
|
new TradingPlayerPatch().Enable();
|
|
new TransferPlayerPatch().Enable();
|
|
new UnloadScavTransferPatch().Enable();
|
|
new NoScavStashPatch().Enable();
|
|
}
|
|
|
|
public class TradingPlayerPatch : ModulePatch
|
|
{
|
|
protected override MethodBase GetTargetMethod()
|
|
{
|
|
return AccessTools.DeclaredProperty(R.TradingInteractions.Type, "AvailableInteractions").GetMethod;
|
|
}
|
|
|
|
[PatchPostfix]
|
|
public static void Postfix(ref IEnumerable<EItemInfoButton> __result)
|
|
{
|
|
var list = __result.ToList();
|
|
list.Insert(list.IndexOf(EItemInfoButton.Repair), EItemInfoButton.UnloadAmmo);
|
|
__result = list;
|
|
}
|
|
}
|
|
|
|
public class TransferPlayerPatch : ModulePatch
|
|
{
|
|
protected override MethodBase GetTargetMethod()
|
|
{
|
|
return AccessTools.DeclaredProperty(R.TransferInteractions.Type, "AvailableInteractions").GetMethod;
|
|
}
|
|
|
|
[PatchPostfix]
|
|
public static void Postfix(ref IEnumerable<EItemInfoButton> __result)
|
|
{
|
|
var list = __result.ToList();
|
|
list.Insert(list.IndexOf(EItemInfoButton.Fold), EItemInfoButton.UnloadAmmo);
|
|
__result = list;
|
|
}
|
|
}
|
|
|
|
// The scav inventory screen has two inventory controllers, the player's and the scav's. Unload always uses the player's, which causes issues
|
|
// because the bullets are never marked as "known" by the scav, so if you click back/next they show up as unsearched, with no way to search
|
|
// This patch forces unload to use the controller of whoever owns the magazine.
|
|
public class UnloadScavTransferPatch : ModulePatch
|
|
{
|
|
protected override MethodBase GetTargetMethod()
|
|
{
|
|
return AccessTools.DeclaredMethod(typeof(InventoryControllerClass), nameof(InventoryControllerClass.UnloadMagazine));
|
|
}
|
|
|
|
[PatchPrefix]
|
|
public static bool Prefix(InventoryControllerClass __instance, MagazineClass magazine, ref Task<IResult> __result)
|
|
{
|
|
if (ItemUiContext.Instance.ContextType != EItemUiContextType.ScavengerInventoryScreen)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
if (magazine.Owner == __instance || magazine.Owner is not InventoryControllerClass ownerInventoryController)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
__result = ownerInventoryController.UnloadMagazine(magazine);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
// Because of the above patch, unload uses the scav's inventory controller, which provides locations to unload ammo: equipment and stash. Why do scavs have a stash?
|
|
// If the equipment is full, the bullets would go to the scav stash, aka a black hole, and are never seen again.
|
|
// Remove the scav's stash
|
|
public class NoScavStashPatch : ModulePatch
|
|
{
|
|
protected override MethodBase GetTargetMethod()
|
|
{
|
|
Type type = typeof(ScavengerInventoryScreen).GetNestedTypes().Single(t => t.GetField("ScavController") != null); // ScavengerInventoryScreen.GClass3156
|
|
return AccessTools.GetDeclaredConstructors(type).Single();
|
|
}
|
|
|
|
[PatchPrefix]
|
|
public static void Prefix(InventoryContainerClass scavController)
|
|
{
|
|
scavController.Inventory.Stash = null;
|
|
}
|
|
}
|
|
}
|
|
}
|