Cj
51 lines
1.6 KiB
C#
51 lines
1.6 KiB
C#
using SPTarkov.DI.Annotations;
|
|
using SPTarkov.Server.Core.Helpers;
|
|
using SPTarkov.Server.Core.Models.Common;
|
|
using SPTarkov.Server.Core.Models.Spt.Presets;
|
|
using SPTarkov.Server.Core.Models.Utils;
|
|
using SPTarkov.Server.Core.Services;
|
|
|
|
namespace SPTarkov.Server.Core.Controllers;
|
|
|
|
[Injectable]
|
|
public class PresetController(
|
|
ISptLogger<PresetController> logger,
|
|
PresetHelper presetHelper,
|
|
DatabaseService databaseService
|
|
)
|
|
{
|
|
/// <summary>
|
|
/// Keyed by item tpl, value = collection of preset ids
|
|
/// </summary>
|
|
public void Initialize()
|
|
{
|
|
var presets = databaseService.GetGlobals().ItemPresets;
|
|
var result = new Dictionary<MongoId, PresetCacheDetails>();
|
|
foreach (var (presetId, preset) in presets)
|
|
{
|
|
if (presetId != preset.Id)
|
|
{
|
|
logger.Error(
|
|
$"Preset for template tpl: '{preset.Items.FirstOrDefault()?.Template} {preset.Name}' has invalid key: ({presetId} != {preset.Id}). Skipping"
|
|
);
|
|
|
|
continue;
|
|
}
|
|
|
|
// Get root items tpl
|
|
var tpl = preset.Items.FirstOrDefault()?.Template;
|
|
result.TryAdd(tpl.Value, new PresetCacheDetails { PresetIds = [] });
|
|
|
|
result.TryGetValue(tpl.Value, out var details);
|
|
details.PresetIds.Add(presetId);
|
|
if (preset.Encyclopedia is not null)
|
|
{
|
|
// Flag this preset as being the default for the weapon
|
|
details.DefaultId = preset.Id;
|
|
}
|
|
}
|
|
|
|
presetHelper.HydratePresetStore(result);
|
|
}
|
|
}
|