ss14-wl/Content.Client/Changeling/UI/ChangelingTransformBoundUserInterface.cs

using Content.Client.Stylesheets.Palette;
using Content.Client.UserInterface.Controls;
using Content.Shared.Changeling.Components;
using Content.Shared.Changeling.Systems;
using JetBrains.Annotations;
using Robust.Client.UserInterface;
using Robust.Shared.Utility;

namespace Content.Client.Changeling.UI;

[UsedImplicitly]
public sealed partial class ChangelingTransformBoundUserInterface(EntityUid owner, Enum uiKey) : BoundUserInterface(owner, uiKey)
{
    private SimpleRadialMenu? _menu;
    private static readonly Color SelectedOptionBackground = Palettes.Green.Element.WithAlpha(128);
    private static readonly Color DisabledOptionBackground = Palettes.Slate.Element.WithAlpha(128);
    private static readonly Color SelectedOptionHoverBackground = Palettes.Green.HoveredElement.WithAlpha(128);
    private static readonly Color DisabledOptionHoverBackground = Palettes.Slate.HoveredElement.WithAlpha(128);

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

        _menu = this.CreateWindow<SimpleRadialMenu>();
        Update();
        _menu.OpenOverMouseScreenPosition();
    }

    public override void Update()
    {
        if (_menu == null)
            return;

        if (!EntMan.TryGetComponent<ChangelingIdentityComponent>(Owner, out var lingIdentity))
            return;

        var models = ConvertToButtons(lingIdentity.ConsumedIdentities.Keys, lingIdentity?.CurrentIdentity);

        _menu.SetButtons(models);
    }

    private IEnumerable<RadialMenuOptionBase> ConvertToButtons(
        IEnumerable<EntityUid> identities,
        EntityUid? currentIdentity
    )
    {
        var buttons = new List<RadialMenuOptionBase>();
        var dropButtons = new List<RadialMenuOptionBase>();

        foreach (var identity in identities)
        {
            // Options for selecting identities.
            var option = new RadialMenuActionOption<NetEntity>(SendIdentitySelect, EntMan.GetNetEntity(identity))
            {
                IconSpecifier = RadialMenuIconSpecifier.With(identity),
                ToolTip = Loc.GetString("changeling-transform-bui-select-entity", ("entity", identity)),
                BackgroundColor = (currentIdentity == identity) ? SelectedOptionBackground : null, // mark as selected
                HoverBackgroundColor = (currentIdentity == identity) ? SelectedOptionHoverBackground : null
            };
            buttons.Add(option);

            // Options for dropping identities.
            var dropOption = new RadialMenuActionOption<NetEntity>(SendIdentityDrop, EntMan.GetNetEntity(identity))
            {
                IconSpecifier = RadialMenuIconSpecifier.With(identity),
                ToolTip = (currentIdentity == identity)
                    ? Loc.GetString("changeling-transform-bui-drop-identity-cannot-drop")
                    : Loc.GetString("changeling-transform-bui-drop-identity-entity", ("entity", identity)),
                BackgroundColor = (currentIdentity == identity) ? DisabledOptionBackground : null, // cannot drop your current identity
                HoverBackgroundColor = (currentIdentity == identity) ? DisabledOptionHoverBackground : null
            };
            dropButtons.Add(dropOption);
        }

        // Menu category for dropping identities.
        var dropMenuButton = new RadialMenuNestedLayerOption(dropButtons)
        {
            IconSpecifier = RadialMenuIconSpecifier.With(new SpriteSpecifier.Texture(new("/Textures/Interface/VerbIcons/delete.svg.192dpi.png"))),
            ToolTip = Loc.GetString("changeling-transform-bui-drop-identity-menu")
        };
        buttons.Add(dropMenuButton);

        return buttons;
    }

    private void SendIdentitySelect(NetEntity identityId)
    {
        SendPredictedMessage(new ChangelingTransformIdentitySelectMessage(identityId));
    }

    private void SendIdentityDrop(NetEntity identityId)
    {
        SendPredictedMessage(new ChangelingTransformIdentityDropMessage(identityId));
    }
}