73 lines
3.4 KiB
C#
73 lines
3.4 KiB
C#
using EFT;
|
|
using EFT.InventoryLogic;
|
|
using HarmonyLib;
|
|
using SPT.Reflection.Patching;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Reflection;
|
|
|
|
namespace UIFixes;
|
|
|
|
public class PutToolsBackPatch : ModulePatch
|
|
{
|
|
protected override MethodBase GetTargetMethod()
|
|
{
|
|
return AccessTools.Method(R.ItemReceiver.Type, "method_9"); // GClass1855
|
|
}
|
|
|
|
// The patched method can't handle new items that aren't in stash root.
|
|
// Find items that are in subcontainers and handle them first - the patched method will ignore items that have a CurrentAddress
|
|
// This is a subset of the original method - doesn't handle slots, equipment containers, etc.
|
|
[PatchPrefix]
|
|
public static void Prefix(object __instance, ref JsonItem[] newItems, Profile ___profile_0, ItemFactory ___itemFactory)
|
|
{
|
|
Inventory inventory = ___profile_0.Inventory;
|
|
StashClass stash = inventory.Stash;
|
|
if (inventory != null && stash != null)
|
|
{
|
|
// Handled items are either in these top level containers or are nested inside each other (mods, attachments, etc)
|
|
var handledContainerIds = newItems.Select(i => i._id).Concat([inventory.Stash.Id, inventory.Equipment.Id, inventory.QuestRaidItems.Id, inventory.QuestStashItems.Id, inventory.SortingTable.Id]);
|
|
var unhandledItems = newItems.Where(i => !String.IsNullOrEmpty(i.parentId) && !handledContainerIds.Contains(i.parentId));
|
|
|
|
if (!unhandledItems.Any())
|
|
{
|
|
return;
|
|
}
|
|
|
|
// Change the parameter to remove the items handled here
|
|
newItems = newItems.Except(unhandledItems).ToArray();
|
|
|
|
List<Item> stashItems = stash.GetNotMergedItems().ToList();
|
|
|
|
InventoryControllerClass inventoryController = new R.ItemReceiver(__instance).InventoryController;
|
|
|
|
var tree = ___itemFactory.FlatItemsToTree(unhandledItems.ToArray(), true, null);
|
|
foreach (Item item in tree.Items.Values.Where(i => i.CurrentAddress == null))
|
|
{
|
|
var newItem = unhandledItems.First(i => i._id == item.Id);
|
|
if (newItem.parentId != null || newItem.slotId != null)
|
|
{
|
|
// Assuming here that unhandled items are trying to go into containers in the stash - find that container
|
|
Item parent = stashItems.FirstOrDefault(i => i.Id == newItem.parentId);
|
|
if (parent is ContainerCollection containerCollection)
|
|
{
|
|
if (containerCollection.GetContainer(newItem.slotId) is StashGridClass grid)
|
|
{
|
|
LocationInGrid location = LocationJsonParser.CreateItemLocation<LocationInGrid>(newItem.location);
|
|
ItemAddress itemAddress = new GridItemAddress(grid, location);
|
|
|
|
var operation = InteractionsHandlerClass.Add(item, itemAddress, inventoryController, false);
|
|
if (operation.Succeeded)
|
|
{
|
|
operation.Value.RaiseEvents(inventoryController, CommandStatus.Begin);
|
|
operation.Value.RaiseEvents(inventoryController, CommandStatus.Succeed);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|