SPT-Server-Build/Libraries/SPTarkov.Server.Core/Services/MatchBotDetailsCacheService.cs

using System.Collections.Concurrent;
using SPTarkov.DI.Annotations;
using SPTarkov.Server.Core.Constants;
using SPTarkov.Server.Core.Models.Eft.Common.Tables;
using SPTarkov.Server.Core.Models.Enums;
using SPTarkov.Server.Core.Models.Spt.Bots;
using SPTarkov.Server.Core.Models.Utils;

namespace SPTarkov.Server.Core.Services;

/// <summary>
///     Cache bots in a dictionary, keyed by the bots ID
/// </summary>
[Injectable(InjectionType.Singleton)]
public class MatchBotDetailsCacheService(ISptLogger<MatchBotDetailsCacheService> logger)
{
    private static readonly HashSet<string> _sidesToCache = [Sides.PmcUsec, Sides.PmcBear];

    protected readonly ConcurrentDictionary<string, BotDetailsForChatMessages> BotDetailsCache =
        new();

    /// <summary>
    ///     Store a bot in the cache, keyed by its ID.
    /// </summary>
    /// <param name="botToCache"> Bot details to cache </param>
    public void CacheBot(BotBase botToCache)
    {
        if (botToCache is null || botToCache.Id is null)
        {
            return;
        }

        if (botToCache.Info?.Nickname is null)
        {
            logger.Warning(
                $"Unable to cache: {botToCache.Info?.Settings?.Role} bot with id: {botToCache.Id} as it lacks a nickname"
            );
            return;
        }

        // If bot isn't a PMC, skip
        if (
            botToCache.Info?.Settings?.Role is null
            || !_sidesToCache.Contains(botToCache.Info.Settings.Role)
        )
        {
            return;
        }

        BotDetailsCache.TryAdd(
            botToCache.Id,
            new BotDetailsForChatMessages()
            {
                Nickname = botToCache.Info.Nickname.Trim(),
                Side = botToCache.Info.Side == Sides.PmcUsec ? DogtagSide.Usec : DogtagSide.Bear,
                Aid = botToCache.Aid,
                Type = botToCache.Info.MemberCategory,
                Level = botToCache.Info.Level,
            }
        );
    }

    /// <summary>
    ///     Clean the cache of all bot details.
    /// </summary>
    public void ClearCache()
    {
        BotDetailsCache.Clear();
    }

    /// <summary>
    ///     Find a bot in the cache by its ID.
    /// </summary>
    /// <param name="id"> ID of bot to find </param>
    /// <returns></returns>
    public BotDetailsForChatMessages? GetBotById(string? id)
    {
        if (id == null)
        {
            return null;
        }

        var botInCache = BotDetailsCache.GetValueOrDefault(id, null);
        if (botInCache is null)
        {
            logger.Warning($"Bot not found in match bot cache: {id}");
            return null;
        }

        return botInCache;
    }
}