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"]); } }