using SPTarkov.DI.Annotations; using SPTarkov.Server.Core.Models.Eft.Common.Tables; using SPTarkov.Server.Core.Models.Spt.Bots; using SPTarkov.Server.Core.Models.Spt.Config; using SPTarkov.Server.Core.Models.Utils; using SPTarkov.Server.Core.Servers; using SPTarkov.Server.Core.Services; using SPTarkov.Server.Core.Utils; using SPTarkov.Server.Core.Utils.Cloners; namespace SPTarkov.Server.Core.Helpers; [Injectable] public class BotDifficultyHelper( ISptLogger _logger, DatabaseService _databaseService, RandomUtil _randomUtil, ServerLocalisationService _serverLocalisationService, BotHelper _botHelper, ConfigServer _configServer, ICloner _cloner ) { protected readonly PmcConfig _pmcConfig = _configServer.GetConfig(); /// /// Get difficulty settings for desired bot type, if not found use assault bot types /// /// bot type to retrieve difficulty of /// difficulty to get settings for (easy/normal etc) /// bots from database /// Difficulty object public DifficultyCategories GetBotDifficultySettings( string type, string desiredDifficulty, Bots botDb ) { var desiredType = _botHelper.IsBotPmc(type) ? _botHelper.GetPmcSideByRole(type).ToLowerInvariant() : type.ToLowerInvariant(); if (!botDb.Types.ContainsKey(desiredType)) { // No bot found, get fallback difficulty values _logger.Warning( _serverLocalisationService.GetText( "bot-unable_to_get_bot_fallback_to_assault", type ) ); botDb.Types[desiredType] = _cloner.Clone(botDb.Types["assault"]); } // Get settings from raw bot json template file var botTemplate = _botHelper.GetBotTemplate(desiredType); botTemplate.BotDifficulty.TryGetValue(desiredDifficulty, out var difficultySettings); if (difficultySettings is null) { // No bot settings found, use 'assault' bot difficulty instead _logger.Warning( _serverLocalisationService.GetText( "bot-unable_to_get_bot_difficulty_fallback_to_assault", new { botType = desiredType, difficulty = desiredDifficulty } ) ); botDb.Types[desiredType].BotDifficulty[desiredDifficulty] = _cloner.Clone( botDb.Types["assault"].BotDifficulty[desiredDifficulty] ); } return _cloner.Clone(difficultySettings); } /// /// Get difficulty settings for a PMC /// /// "usec" / "bear" /// what difficulty to retrieve /// Difficulty object protected DifficultyCategories GetDifficultySettings(string type, string difficulty) { var difficultySetting = string.Equals( _pmcConfig.Difficulty, "asonline", StringComparison.OrdinalIgnoreCase ) ? difficulty : _pmcConfig.Difficulty.ToLowerInvariant(); difficultySetting = ConvertBotDifficultyDropdownToBotDifficulty(difficultySetting); return _cloner.Clone( _databaseService.GetBots().Types[type].BotDifficulty[difficultySetting] ); } /// /// Translate chosen value from pre-raid difficulty dropdown into bot difficulty value /// /// Dropdown difficulty value to convert /// bot difficulty public string ConvertBotDifficultyDropdownToBotDifficulty(string dropDownDifficulty) { switch (dropDownDifficulty.ToLowerInvariant()) { case "medium": return "normal"; case "random": return ChooseRandomDifficulty(); default: return dropDownDifficulty.ToLowerInvariant(); } } /// /// Choose a random difficulty from - easy/normal/hard/impossible /// /// random difficulty public string ChooseRandomDifficulty() { return _randomUtil.GetArrayValue(["easy", "normal", "hard", "impossible"]); } }