SPT-Server-Build/Libraries/Core/Services/TraderPurchasePersisterService.cs

using SptCommon.Annotations;
using Core.Helpers;
using Core.Models.Eft.Profile;
using Core.Models.Spt.Config;
using Core.Models.Utils;
using Core.Servers;
using Core.Utils;

namespace Core.Services;

[Injectable(InjectionType.Singleton)]
public class TraderPurchasePersisterService(
    ISptLogger<TraderPurchasePersisterService> _logger,
    RandomUtil _randomUtil,
    TimeUtil _timeUtil,
    ProfileHelper _profileHelper,
    LocalisationService _localisationService,
    ConfigServer _configServer
)
{
    protected TraderConfig _traderConfig = _configServer.GetConfig<TraderConfig>();

    /**
     * Get the purchases made from a trader for this profile before the last trader reset
     * @param sessionId Session id
     * @param traderId Trader to loop up purchases for
     * @returns Dictionary of assort id and count purchased
     */
    public Dictionary<string, TraderPurchaseData>? GetProfileTraderPurchases(string sessionId, string traderId)
    {
        var profile = _profileHelper.GetFullProfile(sessionId);

        if (profile.TraderPurchases is null)
        {
            return null;
        }

        if (profile.TraderPurchases.ContainsKey(traderId))
        {
            return profile.TraderPurchases[traderId];
        }

        return null;
    }

    /**
     * Get a purchase made from a trader for requested profile before the last trader reset
     * @param sessionId Session id
     * @param traderId Trader to loop up purchases for
     * @param assortId Id of assort to get data for
     * @returns TraderPurchaseData
     */
    public TraderPurchaseData? GetProfileTraderPurchase(
        string sessionId,
        string traderId,
        string assortId)
    {
        var profile = _profileHelper.GetFullProfile(sessionId);

        if (profile.TraderPurchases is null)
        {
            return null;
        }

        if (!profile.TraderPurchases.TryGetValue(traderId, out var _))
        {
            profile.TraderPurchases.TryAdd(traderId, new Dictionary<string, TraderPurchaseData>());
        }

        var traderPurchases = profile.TraderPurchases[traderId];

        if (!traderPurchases.TryGetValue(assortId, out var _))
        {
            traderPurchases.TryAdd(assortId, new TraderPurchaseData());
        }

        return traderPurchases[assortId];
    }

    /**
     * Remove all trader purchase records from all profiles that exist
     * @param traderId Traders id
     */
    public void ResetTraderPurchasesStoredInProfile(string traderId)
    {
        // Reset all profiles purchase dictionaries now a trader update has occured;
        var profiles = _profileHelper.GetProfiles();
        foreach (var profile in profiles)
        {
            // Skip if no purchases
            if (profile.Value.TraderPurchases is null)
            {
                continue;
            }

            // Skip if no trader-speicifc purchases
            if (!profile.Value.TraderPurchases.TryGetValue(traderId, out var _))
            {
                continue;
            }

            profile.Value.TraderPurchases[traderId] = new Dictionary<string, TraderPurchaseData>();
        }
    }

    /**
     * Iterate over all server profiles and remove specific trader purchase data that has passed the trader refresh time
     * @param traderId Trader id
     */
    public void RemoveStalePurchasesFromProfiles(string traderId)
    {
        var profiles = _profileHelper.GetProfiles();
        foreach (var profileKvP in profiles)
        {
            var profile = profileKvP.Value;

            // Skip if no purchases or no trader-specific purchases
            var purchasesFromTrader = profile.TraderPurchases?.GetValueOrDefault(traderId, null);
            if (purchasesFromTrader is null)
            {
                continue;
            }

            foreach (var purchaseKvP in purchasesFromTrader)
            {
                var traderUpdateDetails = _traderConfig.UpdateTime.FirstOrDefault((x) => x.TraderId == traderId);
                if (traderUpdateDetails is null)
                {
                    _logger.Error(
                        _localisationService.GetText(
                            "trader-unable_to_delete_stale_purchases",
                            new
                            {
                                profileId = profile.ProfileInfo.ProfileId,
                                traderId = traderId,
                            }
                        )
                    );

                    continue;
                }

                var purchaseDetails = purchaseKvP.Value;
                var resetTimeForItem =
                    purchaseDetails.PurchaseTimestamp +
                    _randomUtil.GetInt((int)traderUpdateDetails.Seconds.Min, (int)traderUpdateDetails.Seconds.Max);
                if (resetTimeForItem < _timeUtil.GetTimeStamp())
                {
                    // Item was purchased far enough in past a trader refresh would have occured, remove purchase record from profile
                    _logger.Debug($"Removed trader: {traderId} purchase: {purchaseKvP} from profile: {profile.ProfileInfo.ProfileId}");

                    profile.TraderPurchases.Remove(purchaseKvP.Key);
                }
            }
        }
    }
}