clodanSPT
94 lines
2.9 KiB
C#
94 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 : IOnLoad
|
|
{
|
|
private readonly DatabaseService _databaseService;
|
|
private readonly IEnumerable<ITrader> _injectedTraders;
|
|
private readonly ISptLogger<TraderStore> _logger;
|
|
|
|
private readonly Dictionary<string, ITrader> _traders = new();
|
|
|
|
public TraderStore(DatabaseService databaseService,
|
|
IEnumerable<ITrader> injectedTraders,
|
|
ISptLogger<TraderStore> logger)
|
|
{
|
|
_databaseService = databaseService;
|
|
_injectedTraders = injectedTraders;
|
|
_logger = logger;
|
|
}
|
|
|
|
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("Failed to load custom trader", e);
|
|
}
|
|
}
|
|
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;
|
|
}
|
|
}
|