wwdpublic/Content.Client/Storage/StorageBoundUserInterface.cs

using Content.Client.Storage.Systems;
using Content.Shared.Storage;
using JetBrains.Annotations;

namespace Content.Client.Storage;

[UsedImplicitly]
public sealed class StorageBoundUserInterface : BoundUserInterface
{
    [Dependency] private readonly IEntityManager _entManager = default!;

    private readonly StorageSystem _storage;

    public StorageBoundUserInterface(EntityUid owner, Enum uiKey) : base(owner, uiKey)
    {
        IoCManager.InjectDependencies(this);
        _storage = _entManager.System<StorageSystem>();
    }

    protected override void Dispose(bool disposing)
    {
        base.Dispose(disposing);
        if (!disposing)
            return;

        _storage.CloseStorageWindow(Owner);
    }

    protected override void ReceiveMessage(BoundUserInterfaceMessage message)
    {
        base.ReceiveMessage(message);

        if (message is StorageModifyWindowMessage)
        {
            if (_entManager.TryGetComponent<StorageComponent>(Owner, out var comp))
                _storage.OpenStorageWindow((Owner, comp));
        }
    }
}