85 lines
3.0 KiB
C#
85 lines
3.0 KiB
C#
using Aki.Reflection.Patching;
|
|
using Aki.Reflection.Utils;
|
|
using EFT.InventoryLogic;
|
|
using EFT.UI;
|
|
using EFT.UI.DragAndDrop;
|
|
using HarmonyLib;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Reflection;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace UIFixes
|
|
{
|
|
public class LoadAmmoInRaidPatches
|
|
{
|
|
public static void Enable()
|
|
{
|
|
new EnableContextMenuPatch().Enable();
|
|
new SlowLoadingPatch().Enable();
|
|
}
|
|
|
|
public class EnableContextMenuPatch : ModulePatch
|
|
{
|
|
protected override MethodBase GetTargetMethod()
|
|
{
|
|
Type type = PatchConstants.EftTypes.Single(t => t.GetProperty("IsOwnedByPlayer") != null);
|
|
return AccessTools.Method(type, "IsActive");
|
|
}
|
|
|
|
[PatchPrefix]
|
|
public static bool Prefix(EItemInfoButton button, ref bool __result, Item ___item_0)
|
|
{
|
|
if (button != EItemInfoButton.LoadAmmo || !Plugin.InRaid() || !Settings.EnableLoadAmmo.Value)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
__result = MagazineBuildClass.TryFindPresetSource(___item_0).Succeeded;
|
|
return false;
|
|
}
|
|
}
|
|
|
|
public class SlowLoadingPatch : ModulePatch
|
|
{
|
|
protected override MethodBase GetTargetMethod()
|
|
{
|
|
return AccessTools.Method(typeof(ItemUiContext), nameof(ItemUiContext.LoadAmmoByType));
|
|
}
|
|
|
|
// This code is a mix of ItemUiContext.LoadAmmoByType, but then switching over to GridView.AcceptItem
|
|
[PatchPrefix]
|
|
public static bool Prefix(ItemUiContext __instance, MagazineClass magazine, string ammoTemplateId, ref Task __result)
|
|
{
|
|
if (!Plugin.InRaid() || !Settings.EnableLoadAmmo.Value)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
InventoryControllerClass inventoryController = __instance.R().InventoryController;
|
|
EquipmentClass equipment = inventoryController.Inventory.Equipment;
|
|
|
|
List<BulletClass> ammo = [];
|
|
equipment.GetAllAssembledItems(ammo);
|
|
|
|
// Just do the first stack
|
|
BulletClass bullets = ammo.Where(a => a.TemplateId == ammoTemplateId && a.Parent.Container is not Slot)
|
|
.OrderBy(a => a.SpawnedInSession)
|
|
.ThenBy(a => a.StackObjectsCount)
|
|
.FirstOrDefault();
|
|
|
|
if (bullets != null)
|
|
{
|
|
int count = GridView.smethod_0(magazine, bullets);
|
|
inventoryController.LoadMagazine(bullets, magazine, count, false).HandleExceptions();
|
|
}
|
|
|
|
// The calling code in this instance doesn't do anything with the task, but it does await it, so if we don't return a Task it nullrefs
|
|
__result = Task.CompletedTask;
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
}
|