76 lines
2.9 KiB
C#
76 lines
2.9 KiB
C#
using Content.Server.Storage.Components;
|
|
using Content.Server.Storage.EntitySystems;
|
|
using Content.Server.Temperature.Systems;
|
|
using Content.Shared.Clothing.Components;
|
|
using Content.Shared.Interaction.Events;
|
|
using Content.Shared.Inventory;
|
|
using Content.Shared.Inventory.Events;
|
|
using Robust.Shared.Containers;
|
|
using InventoryComponent = Content.Shared.Inventory.InventoryComponent;
|
|
|
|
namespace Content.Server.Inventory
|
|
{
|
|
public sealed class ServerInventorySystem : InventorySystem
|
|
{
|
|
[Dependency] private readonly StorageSystem _storageSystem = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<ClothingComponent, UseInHandEvent>(OnUseInHand);
|
|
|
|
SubscribeNetworkEvent<OpenSlotStorageNetworkMessage>(OnOpenSlotStorage);
|
|
}
|
|
|
|
private void OnUseInHand(EntityUid uid, ClothingComponent component, UseInHandEvent args)
|
|
{
|
|
if (args.Handled || !component.QuickEquip)
|
|
return;
|
|
|
|
QuickEquip(uid, component, args);
|
|
}
|
|
|
|
private void OnOpenSlotStorage(OpenSlotStorageNetworkMessage ev, EntitySessionEventArgs args)
|
|
{
|
|
if (args.SenderSession.AttachedEntity is not EntityUid { Valid: true } uid)
|
|
return;
|
|
|
|
if (TryGetSlotEntity(uid, ev.Slot, out var entityUid) && TryComp<ServerStorageComponent>(entityUid, out var storageComponent))
|
|
{
|
|
_storageSystem.OpenStorageUI(entityUid.Value, uid, storageComponent);
|
|
}
|
|
}
|
|
|
|
public void TransferEntityInventories(EntityUid uid, EntityUid target)
|
|
{
|
|
if (TryGetContainerSlotEnumerator(uid, out var enumerator))
|
|
{
|
|
Dictionary<string, EntityUid?> inventoryEntities = new();
|
|
var slots = GetSlots(uid);
|
|
while (enumerator.MoveNext(out var containerSlot))
|
|
{
|
|
//records all the entities stored in each of the target's slots
|
|
foreach (var slot in slots)
|
|
{
|
|
if (TryGetSlotContainer(target, slot.Name, out var conslot, out var _) &&
|
|
conslot.ID == containerSlot.ID)
|
|
{
|
|
inventoryEntities.Add(slot.Name, containerSlot.ContainedEntity);
|
|
}
|
|
}
|
|
//drops everything in the target's inventory on the ground
|
|
containerSlot.EmptyContainer();
|
|
}
|
|
/// This takes the objects we removed and stored earlier
|
|
/// and actually equips all of it to the new entity
|
|
foreach (var item in inventoryEntities)
|
|
{
|
|
if (item.Value != null)
|
|
TryEquip(target, item.Value.Value, item.Key, true);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|