refringe
93 lines
2.9 KiB
C#
93 lines
2.9 KiB
C#
using SPTarkov.DI.Annotations;
|
|
using SPTarkov.Server.Core.DI;
|
|
using SPTarkov.Server.Core.Models;
|
|
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,
|
|
LocalisationService localisationService,
|
|
ISptLogger<TraderStore> logger
|
|
) : IOnLoad
|
|
{
|
|
private readonly Dictionary<string, 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(string traderId)
|
|
{
|
|
if (_traders.TryGetValue(traderId, out var trader))
|
|
{
|
|
return trader;
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns all traders in the game, including custom traders.
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
public IEnumerable<ITrader> GetAllTraders()
|
|
{
|
|
return _traders.Values;
|
|
}
|
|
}
|