Chomp
87 lines
2.7 KiB
C#
87 lines
2.7 KiB
C#
using Microsoft.Extensions.Logging;
|
|
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;
|
|
}
|
|
}
|