crystall-punk-14/Content.Client/Inventory/StrippableBoundUserInterface.cs

using Content.Client.Strip;
using Content.Shared.IdentityManagement;
using Content.Shared.Strip.Components;
using JetBrains.Annotations;
using Robust.Client.GameObjects;

namespace Content.Client.Inventory
{
    [UsedImplicitly]
    public sealed class StrippableBoundUserInterface : BoundUserInterface
    {
        public Dictionary<(string ID, string Name), string>? Inventory { get; private set; }
        public Dictionary<string, string>? Hands { get; private set; }
        public Dictionary<EntityUid, string>? Handcuffs { get; private set; }
        public Dictionary<EntityUid, string>? Ensnare { get; private set; }

        [ViewVariables]
        private StrippingMenu? _strippingMenu;

        public StrippableBoundUserInterface(ClientUserInterfaceComponent owner, Enum uiKey) : base(owner, uiKey)
        {
        }

        protected override void Open()
        {
            base.Open();

            var entMan = IoCManager.Resolve<IEntityManager>();
            _strippingMenu = new StrippingMenu($"{Loc.GetString("strippable-bound-user-interface-stripping-menu-title", ("ownerName", Identity.Name(Owner.Owner, entMan)))}");

            _strippingMenu.OnClose += Close;
            _strippingMenu.OpenCenteredLeft();
            UpdateMenu();
        }

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

            _strippingMenu?.Dispose();
        }

        private void UpdateMenu()
        {
            if (_strippingMenu == null) return;

            _strippingMenu.ClearButtons();

            if (Inventory != null)
            {
                foreach (var (slot, name) in Inventory)
                {
                    _strippingMenu.AddButton(slot.Name, name, (ev) =>
                    {
                        SendMessage(new StrippingInventoryButtonPressed(slot.ID));
                    });
                }
            }

            if (Hands != null)
            {
                foreach (var (hand, name) in Hands)
                {
                    _strippingMenu.AddButton(hand, name, (ev) =>
                    {
                        SendMessage(new StrippingHandButtonPressed(hand));
                    });
                }
            }

            if (Handcuffs != null)
            {
                foreach (var (id, name) in Handcuffs)
                {
                    _strippingMenu.AddButton(Loc.GetString("strippable-bound-user-interface-stripping-menu-handcuffs-button"), name, (ev) =>
                    {
                        SendMessage(new StrippingHandcuffButtonPressed(id));
                    });
                }
            }

            if (Ensnare != null)
            {
                foreach (var (id, name) in Ensnare)
                {
                    _strippingMenu.AddButton(Loc.GetString("strippable-bound-user-interface-stripping-menu-ensnare-button"), name, (ev) =>
                    {
                        SendMessage(new StrippingEnsnareButtonPressed(id));
                    });
                }
            }
        }

        protected override void UpdateState(BoundUserInterfaceState state)
        {
            base.UpdateState(state);

            if (state is not StrippingBoundUserInterfaceState stripState) return;

            Inventory = stripState.Inventory;
            Hands = stripState.Hands;
            Handcuffs = stripState.Handcuffs;
            Ensnare = stripState.Ensnare;

            UpdateMenu();
        }
    }
}