ss14-wl/Content.Shared/ContainerHeld/ContainerHeldSystem.cs

using Robust.Shared.Containers;

using Content.Shared.Item;
using Content.Shared.Storage;
using Content.Shared.Storage.EntitySystems;
using Content.Shared.Toggleable;

namespace Content.Shared.ContainerHeld;

public sealed partial class ContainerHeldSystem : EntitySystem
{
    [Dependency] private SharedItemSystem _item = default!;
    [Dependency] private SharedAppearanceSystem _appearance = default!;
    [Dependency] private SharedStorageSystem _storage = default!;

    public override void Initialize()
    {
        base.Initialize();

        SubscribeLocalEvent<ContainerHeldComponent, EntInsertedIntoContainerMessage>(OnContainerModified);
        SubscribeLocalEvent<ContainerHeldComponent, EntRemovedFromContainerMessage>(OnContainerModified);
    }

    private void OnContainerModified(EntityUid uid, ContainerHeldComponent comp, ContainerModifiedMessage args)
    {
        if (!(HasComp<StorageComponent>(uid)
              && TryComp<AppearanceComponent>(uid, out var appearance)
              && TryComp<ItemComponent>(uid, out var item)))
        {
            return;
        }
        if (_storage.GetCumulativeItemAreas(uid) >= comp.Threshold)
        {
            _item.SetHeldPrefix(uid, "full", component: item);
            _appearance.SetData(uid, ToggleableVisuals.Enabled, true, appearance);
        }
        else
        {
            _item.SetHeldPrefix(uid, "empty", component: item);
            _appearance.SetData(uid, ToggleableVisuals.Enabled, false, appearance);
        }
    }
}