StarHorizon_Public/Content.Client/_NF/Cargo/BUI/CargoOrderConsoleNFBoundUserInterface.cs

using Content.Client.Cargo.UI;
using Content.Shared.Cargo.Events;
using Content.Shared.Cargo.Prototypes;
using Content.Shared.IdentityManagement;
using Robust.Client.GameObjects;
using Robust.Client.Player;
using Robust.Shared.Utility;
using Robust.Shared.Prototypes;
using Content.Shared._NF.Cargo.BUI;
using Content.Shared._NF.Cargo;
using Content.Client._NF.Cargo.UI;

namespace Content.Client._NF.Cargo.BUI;

// Suffixed to avoid BUI collisions (see RT#5648)
public sealed class CargoOrderConsoleNFBoundUserInterface : BoundUserInterface
{
    [ViewVariables]
    private NFCargoConsoleMenu? _menu;

    /// <summary>
    /// This is the separate popup window for individual orders.
    /// </summary>
    [ViewVariables]
    private NFCargoConsoleOrderMenu? _orderMenu;

    [ViewVariables]
    public string? AccountName { get; private set; }

    [ViewVariables]
    public int BankBalance { get; private set; }

    [ViewVariables]
    public int OrderCapacity { get; private set; }

    [ViewVariables]
    public int OrderCount { get; private set; }

    /// <summary>
    /// Currently selected product
    /// </summary>
    [ViewVariables]
    private CargoProductPrototype? _product;

    public CargoOrderConsoleNFBoundUserInterface(EntityUid owner, Enum uiKey) : base(owner, uiKey)
    {
    }

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

        var spriteSystem = EntMan.System<SpriteSystem>();
        var dependencies = IoCManager.Instance!;
        _menu = new NFCargoConsoleMenu(Owner, EntMan, dependencies.Resolve<IPrototypeManager>(), spriteSystem);
        var localPlayer = dependencies.Resolve<IPlayerManager>().LocalEntity;
        var description = new FormattedMessage();

        string orderRequester;

        if (EntMan.TryGetComponent<MetaDataComponent>(localPlayer, out var metadata))
            orderRequester = Identity.Name(localPlayer.Value, EntMan);
        else
            orderRequester = string.Empty;

        _orderMenu = new NFCargoConsoleOrderMenu(OrderCapacity);

        _menu.OnClose += Close;

        _menu.OnItemSelected += (args) =>
        {
            if (args.Button.Parent is not CargoProductRow row)
                return;

            description.Clear();
            description.PushColor(Color.White); // Rich text default color is grey
            if (row.MainButton.ToolTip != null)
                description.AddText(row.MainButton.ToolTip);

            _orderMenu.Description.SetMessage(description);
            _product = row.Product;
            _orderMenu.ProductName.Text = row.ProductName.Text;
            _orderMenu.PointCost.Text = row.PointCost.Text;
            _orderMenu.Requester.Text = orderRequester;
            _orderMenu.Notes.Text = "";
            _orderMenu.Amount.Value = 1;

            _orderMenu.OpenCentered();
        };
        _orderMenu.SubmitButton.OnPressed += (_) =>
        {
            if (AddOrder())
            {
                _orderMenu.Close();
            }
        };

        _menu.OpenCentered();
    }

    private void Populate(List<NFCargoOrderData> orders)
    {
        if (_menu == null) return;

        _menu.PopulateProducts();
        _menu.PopulateCategories();
        _menu.PopulateOrders(orders, OrderCapacity);
    }

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

        if (state is not NFCargoConsoleInterfaceState cState)
            return;

        OrderCapacity = cState.Capacity;
        OrderCount = cState.Count;
        BankBalance = cState.Balance;

        AccountName = cState.Name;

        Populate(cState.Orders);
        _menu?.UpdateBankData(AccountName, BankBalance);
        _orderMenu?.SetOrderCapacity(OrderCapacity);
    }

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

        if (!disposing) return;

        _menu?.Dispose();
        _orderMenu?.Dispose();
    }

    private bool AddOrder()
    {
        var orderAmt = _orderMenu?.Amount.Value ?? 0;
        if (orderAmt < 1)
        {
            return false;
        }

        orderAmt = int.Min(orderAmt, OrderCapacity);

        SendMessage(new CargoConsoleAddOrderMessage(
            _orderMenu?.Requester.Text ?? "",
            _orderMenu?.Notes.Text ?? "",
            _product?.ID ?? "",
            orderAmt));

        return true;
    }
}