SPT-Server-Build/Libraries/SPTarkov.Server.Core/Services/TraderStore.cs

using SPTarkov.DI.Annotations;
using SPTarkov.Server.Core.DI;
using SPTarkov.Server.Core.Models;
using SPTarkov.Server.Core.Models.Common;
using SPTarkov.Server.Core.Models.Eft.Common.Tables;
using SPTarkov.Server.Core.Models.Utils;

namespace SPTarkov.Server.Core.Services;

/// <summary>
/// Source of truth for all default traders as well as any additional trader a server mod may add.
/// </summary>
[Injectable(InjectionType.Singleton, TypePriority = OnLoadOrder.TraderRegistration)]
public class TraderStore(
    DatabaseService databaseService,
    IEnumerable<ITrader> injectedTraders,
    ServerLocalisationService localisationService,
    ISptLogger<TraderStore> logger
) : IOnLoad
{
    private readonly Dictionary<MongoId, ITrader> _traders = new();

    public Task OnLoad()
    {
        logger.Info("Importing traders...");
        var customTraders = 0;

        foreach (var trader in injectedTraders)
        {
            if (trader is ICustomTrader customTrader)
            {
                try
                {
                    var dbTrader = new Trader
                    {
                        Assort = customTrader.GetAssort(),
                        Base = customTrader.GetBase(),
                        QuestAssort = customTrader.GetQuestAssort(),
                        Dialogue = customTrader.GetDialogues(),
                        Suits = customTrader.GetSuits(),
                        Services = customTrader.GetServices(),
                    };
                    databaseService.GetTraders().Add(trader.Id, dbTrader);
                    _traders.Add(trader.Id, trader);
                    logger.Info($"Loaded custom trader: {trader.Name}");
                    customTraders++;
                }
                catch (Exception e)
                {
                    logger.Error(
                        localisationService.GetText(
                            "trader-unable_to_add_custom_trader",
                            new { traderId = trader.Name, error = e.StackTrace }
                        )
                    );
                }
            }
            else
            {
                _traders.Add(trader.Id, trader);
            }
        }

        logger.Info(
            $"Importing traders complete {(customTraders == 0 ? "" : $"[{customTraders} traders loaded]")}"
        );
        return Task.CompletedTask;
    }

    /// <summary>
    /// Returns a trader by given ID.
    /// </summary>
    /// <param name="traderId"></param>
    /// <returns></returns>
    public ITrader? GetTraderById(MongoId traderId)
    {
        return _traders.GetValueOrDefault(traderId);
    }

    /// <summary>
    /// Returns all traders in the game, including custom traders.
    /// </summary>
    /// <returns></returns>
    public IEnumerable<ITrader> GetAllTraders()
    {
        return _traders.Values;
    }
}