Dalamud/Dalamud/Interface/Internal/Windows/Data/Widgets/MarketBoardWidget.cs

using System.Collections.Concurrent;
using System.Globalization;

using Dalamud.Game.MarketBoard;
using Dalamud.Game.Network.Structures;
using Dalamud.Interface.Utility;
using Dalamud.Interface.Utility.Raii;
using ImGuiNET;

namespace Dalamud.Interface.Internal.Windows.Data.Widgets;

/// <summary>
/// Widget to display market board events.
/// </summary>
internal class MarketBoardWidget : IDataWindowWidget
{
    private readonly ConcurrentQueue<(IMarketBoardHistory MarketBoardHistory, IMarketBoardHistoryListing Listing)> marketBoardHistoryQueue = new();
    private readonly ConcurrentQueue<(IMarketBoardCurrentOfferings MarketBoardCurrentOfferings, IMarketBoardItemListing Listing)> marketBoardCurrentOfferingsQueue = new();
    private readonly ConcurrentQueue<IMarketBoardPurchase> marketBoardPurchasesQueue = new();
    private readonly ConcurrentQueue<IMarketBoardPurchaseHandler> marketBoardPurchaseRequestsQueue = new();
    private readonly ConcurrentQueue<IMarketTaxRates> marketTaxRatesQueue = new();

    private bool trackMarketBoard;
    private int trackedEvents;

    /// <summary> Finalizes an instance of the <see cref="MarketBoardWidget"/> class. </summary>
    ~MarketBoardWidget()
    {
        if (this.trackMarketBoard)
        {
            this.trackMarketBoard = false;
            var marketBoard = Service<MarketBoard>.GetNullable();
            if (marketBoard != null)
            {
                marketBoard.HistoryReceived -= this.MarketBoardHistoryReceived;
                marketBoard.OfferingsReceived -= this.MarketBoardOfferingsReceived;
                marketBoard.ItemPurchased -= this.MarketBoardItemPurchased;
                marketBoard.PurchaseRequested -= this.MarketBoardPurchaseRequested;
                marketBoard.TaxRatesReceived -= this.TaxRatesReceived;
            }
        }
    }

    /// <inheritdoc/>
    public string[]? CommandShortcuts { get; init; } = { "marketboard" };

    /// <inheritdoc/>
    public string DisplayName { get; init; } = "Market Board";

    /// <inheritdoc/>
    public bool Ready { get; set; }

    /// <inheritdoc/>
    public void Load()
    {
        this.trackMarketBoard = false;
        this.trackedEvents = 0;
        this.marketBoardHistoryQueue.Clear();
        this.marketBoardPurchaseRequestsQueue.Clear();
        this.marketBoardPurchasesQueue.Clear();
        this.marketTaxRatesQueue.Clear();
        this.marketBoardCurrentOfferingsQueue.Clear();
        this.Ready = true;
    }

    /// <inheritdoc/>
    public void Draw()
    {
        var marketBoard = Service<MarketBoard>.Get();
        if (ImGui.Checkbox("Track MarketBoard Events", ref this.trackMarketBoard))
        {
            if (this.trackMarketBoard)
            {
                marketBoard.HistoryReceived += this.MarketBoardHistoryReceived;
                marketBoard.OfferingsReceived += this.MarketBoardOfferingsReceived;
                marketBoard.ItemPurchased += this.MarketBoardItemPurchased;
                marketBoard.PurchaseRequested += this.MarketBoardPurchaseRequested;
                marketBoard.TaxRatesReceived += this.TaxRatesReceived;
            }
            else
            {
                marketBoard.HistoryReceived -= this.MarketBoardHistoryReceived;
                marketBoard.OfferingsReceived -= this.MarketBoardOfferingsReceived;
                marketBoard.ItemPurchased -= this.MarketBoardItemPurchased;
                marketBoard.PurchaseRequested -= this.MarketBoardPurchaseRequested;
                marketBoard.TaxRatesReceived -= this.TaxRatesReceived;
            }
        }

        ImGui.SetNextItemWidth(ImGui.GetContentRegionAvail().X / 2);
        if (ImGui.DragInt("Stored Number of Events", ref this.trackedEvents, 0.1f, 1, 512))
        {
            this.trackedEvents = Math.Clamp(this.trackedEvents, 1, 512);
        }

        if (ImGui.Button("Clear Stored Events"))
        {
            this.marketBoardHistoryQueue.Clear();
        }

        using (var tabBar = ImRaii.TabBar("marketTabs"))
        {
            if (tabBar)
            {
                using (var tabItem = ImRaii.TabItem("History"))
                {
                    if (tabItem)
                    {
                        ImGuiTable.DrawTable(string.Empty, this.marketBoardHistoryQueue, this.DrawMarketBoardHistory, ImGuiTableFlags.SizingFixedFit | ImGuiTableFlags.RowBg, "Item ID", "Quantity", "Is HQ?", "Sale Price", "Buyer Name", "Purchase Time");
                    }
                }

                using (var tabItem = ImRaii.TabItem("Offerings"))
                {
                    if (tabItem)
                    {
                        ImGuiTable.DrawTable(string.Empty, this.marketBoardCurrentOfferingsQueue, this.DrawMarketBoardCurrentOfferings, ImGuiTableFlags.SizingFixedFit | ImGuiTableFlags.RowBg, "Item ID", "Quantity", "Is HQ?", "Price Per Unit", "Buyer Name", "Retainer Name", "Last Review Time");
                    }
                }

                using (var tabItem = ImRaii.TabItem("Purchases"))
                {
                    if (tabItem)
                    {
                        ImGuiTable.DrawTable(string.Empty, this.marketBoardPurchasesQueue, this.DrawMarketBoardPurchases, ImGuiTableFlags.SizingFixedFit | ImGuiTableFlags.RowBg, "Item ID", "Quantity");
                    }
                }

                using (var tabItem = ImRaii.TabItem("Purchase Requests"))
                {
                    if (tabItem)
                    {
                        ImGuiTable.DrawTable(string.Empty, this.marketBoardPurchaseRequestsQueue, this.DrawMarketBoardPurchaseRequests, ImGuiTableFlags.SizingFixedFit | ImGuiTableFlags.RowBg, "Item ID", "Quantity", "Price Per Unit", "Listing ID", "Retainer ID");
                    }
                }

                using (var tabItem = ImRaii.TabItem("Taxes"))
                {
                    if (tabItem)
                    {
                        ImGuiTable.DrawTable(string.Empty, this.marketTaxRatesQueue, this.DrawMarketTaxRates, ImGuiTableFlags.SizingFixedFit | ImGuiTableFlags.RowBg, "Uldah", "Limsa Lominsa", "Gridania", "Ishgard", "Kugane", "Crystarium", "Sharlayan");
                    }
                }
            }
        }
    }

    private void TaxRatesReceived(IMarketTaxRates marketTaxRates)
    {
        this.marketTaxRatesQueue.Enqueue(marketTaxRates);

        while (this.marketTaxRatesQueue.Count > this.trackedEvents)
        {
            this.marketTaxRatesQueue.TryDequeue(out _);
        }
    }

    private void MarketBoardPurchaseRequested(IMarketBoardPurchaseHandler marketBoardPurchaseHandler)
    {
        this.marketBoardPurchaseRequestsQueue.Enqueue(marketBoardPurchaseHandler);

        while (this.marketBoardPurchaseRequestsQueue.Count > this.trackedEvents)
        {
            this.marketBoardPurchaseRequestsQueue.TryDequeue(out _);
        }
    }

    private void MarketBoardItemPurchased(IMarketBoardPurchase marketBoardPurchase)
    {
        this.marketBoardPurchasesQueue.Enqueue(marketBoardPurchase);

        while (this.marketBoardPurchasesQueue.Count > this.trackedEvents)
        {
            this.marketBoardPurchasesQueue.TryDequeue(out _);
        }
    }

    private void MarketBoardOfferingsReceived(IMarketBoardCurrentOfferings marketBoardCurrentOfferings)
    {
        foreach (var listing in marketBoardCurrentOfferings.ItemListings)
        {
            this.marketBoardCurrentOfferingsQueue.Enqueue((marketBoardCurrentOfferings, listing));
        }

        while (this.marketBoardCurrentOfferingsQueue.Count > this.trackedEvents)
        {
            this.marketBoardCurrentOfferingsQueue.TryDequeue(out _);
        }
    }

    private void MarketBoardHistoryReceived(IMarketBoardHistory marketBoardHistory)
    {
        foreach (var listing in marketBoardHistory.HistoryListings)
        {
            this.marketBoardHistoryQueue.Enqueue((marketBoardHistory, listing));
        }

        while (this.marketBoardHistoryQueue.Count > this.trackedEvents)
        {
            this.marketBoardHistoryQueue.TryDequeue(out _);
        }
    }

    private void DrawMarketBoardHistory((IMarketBoardHistory History, IMarketBoardHistoryListing Listing) data)
    {
        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.History.ItemId.ToString());

        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.Listing.Quantity.ToString());

        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.Listing.IsHq.ToString());

        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.Listing.SalePrice.ToString());

        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.Listing.BuyerName);

        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.Listing.PurchaseTime.ToString(CultureInfo.InvariantCulture));
    }

    private void DrawMarketBoardCurrentOfferings((IMarketBoardCurrentOfferings MarketBoardCurrentOfferings, IMarketBoardItemListing Listing) data)
    {
        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.Listing.ItemId.ToString());

        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.Listing.ItemQuantity.ToString());

        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.Listing.IsHq.ToString());

        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.Listing.PricePerUnit.ToString());

        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.Listing.PlayerName);

        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.Listing.RetainerName);

        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.Listing.LastReviewTime.ToString(CultureInfo.InvariantCulture));
    }

    private void DrawMarketBoardPurchases(IMarketBoardPurchase data)
    {
        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.CatalogId.ToString());

        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.ItemQuantity.ToString());
    }

    private void DrawMarketBoardPurchaseRequests(IMarketBoardPurchaseHandler data)
    {
        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.CatalogId.ToString());

        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.ItemQuantity.ToString());

        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.PricePerUnit.ToString());

        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.ListingId.ToString());

        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.RetainerId.ToString());
    }

    private void DrawMarketTaxRates(IMarketTaxRates data)
    {
        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.UldahTax.ToString());

        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.LimsaLominsaTax.ToString());

        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.GridaniaTax.ToString());

        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.IshgardTax.ToString());

        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.KuganeTax.ToString());

        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.CrystariumTax.ToString());

        ImGui.TableNextColumn();
        ImGui.TextUnformatted(data.SharlayanTax.ToString());
    }
}