96 lines
3.2 KiB
C#
96 lines
3.2 KiB
C#
using Content.Shared.Containers.ItemSlots;
|
|
using Content.Shared.Interaction;
|
|
using Content.Shared.Nutrition.Components;
|
|
using Content.Shared.Nutrition.EntitySystems;
|
|
using Robust.Shared.Containers;
|
|
using System.Diagnostics.CodeAnalysis;
|
|
|
|
namespace Content.Shared.Cabinet;
|
|
|
|
/// <summary>
|
|
/// Controls ItemCabinet slot locking and visuals.
|
|
/// </summary>
|
|
public sealed class ItemCabinetSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly ItemSlotsSystem _slots = default!;
|
|
[Dependency] private readonly OpenableSystem _openable = default!;
|
|
[Dependency] private readonly SharedAppearanceSystem _appearance = default!;
|
|
|
|
/// <inheritdoc/>
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<ItemCabinetComponent, ComponentStartup>(OnStartup);
|
|
SubscribeLocalEvent<ItemCabinetComponent, MapInitEvent>(OnMapInit);
|
|
SubscribeLocalEvent<ItemCabinetComponent, EntInsertedIntoContainerMessage>(OnContainerModified);
|
|
SubscribeLocalEvent<ItemCabinetComponent, EntRemovedFromContainerMessage>(OnContainerModified);
|
|
SubscribeLocalEvent<ItemCabinetComponent, OpenableOpenedEvent>(OnOpened);
|
|
SubscribeLocalEvent<ItemCabinetComponent, OpenableClosedEvent>(OnClosed);
|
|
}
|
|
|
|
private void OnStartup(Entity<ItemCabinetComponent> ent, ref ComponentStartup args)
|
|
{
|
|
UpdateAppearance(ent);
|
|
}
|
|
|
|
private void OnMapInit(Entity<ItemCabinetComponent> ent, ref MapInitEvent args)
|
|
{
|
|
// update at mapinit to avoid copy pasting locked: true and locked: false for each closed/open prototype
|
|
SetSlotLock(ent, !_openable.IsOpen(ent));
|
|
}
|
|
|
|
private void UpdateAppearance(Entity<ItemCabinetComponent> ent)
|
|
{
|
|
_appearance.SetData(ent, ItemCabinetVisuals.ContainsItem, HasItem(ent));
|
|
}
|
|
|
|
private void OnContainerModified(EntityUid uid, ItemCabinetComponent component, ContainerModifiedMessage args)
|
|
{
|
|
if (args.Container.ID == component.Slot)
|
|
UpdateAppearance((uid, component));
|
|
}
|
|
|
|
private void OnOpened(Entity<ItemCabinetComponent> ent, ref OpenableOpenedEvent args)
|
|
{
|
|
SetSlotLock(ent, false);
|
|
}
|
|
|
|
private void OnClosed(Entity<ItemCabinetComponent> ent, ref OpenableClosedEvent args)
|
|
{
|
|
SetSlotLock(ent, true);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Tries to get the cabinet's item slot.
|
|
/// </summary>
|
|
public bool TryGetSlot(Entity<ItemCabinetComponent> ent, [NotNullWhen(true)] out ItemSlot? slot)
|
|
{
|
|
slot = null;
|
|
if (!TryComp<ItemSlotsComponent>(ent, out var slots))
|
|
return false;
|
|
|
|
return _slots.TryGetSlot(ent, ent.Comp.Slot, out slot, slots);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns true if the cabinet contains an item.
|
|
/// </summary>
|
|
public bool HasItem(Entity<ItemCabinetComponent> ent)
|
|
{
|
|
return TryGetSlot(ent, out var slot) && slot.HasItem;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Lock or unlock the underlying item slot.
|
|
/// </summary>
|
|
public void SetSlotLock(Entity<ItemCabinetComponent> ent, bool closed)
|
|
{
|
|
if (!TryComp<ItemSlotsComponent>(ent, out var slots))
|
|
return;
|
|
|
|
if (_slots.TryGetSlot(ent, ent.Comp.Slot, out var slot, slots))
|
|
_slots.SetLock(ent, slot, closed, slots);
|
|
}
|
|
}
|