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

using SPTarkov.DI.Annotations;
using SPTarkov.Server.Core.Generators;
using SPTarkov.Server.Core.Helpers;
using SPTarkov.Server.Core.Models.Eft.Weather;
using SPTarkov.Server.Core.Models.Enums;
using SPTarkov.Server.Core.Models.Spt.Config;
using SPTarkov.Server.Core.Servers;
using SPTarkov.Server.Core.Utils;

namespace SPTarkov.Server.Core.Services;

[Injectable(InjectionType.Singleton)]
public class RaidWeatherService(
    TimeUtil timeUtil,
    WeatherGenerator weatherGenerator,
    SeasonalEventService seasonalEventService,
    WeightedRandomHelper weightedRandomHelper,
    ConfigServer configServer
)
{
    protected readonly WeatherConfig WeatherConfig = configServer.GetConfig<WeatherConfig>();
    protected readonly List<Weather> WeatherForecast = [];

    /// <summary>
    ///     Generate 24 hours of weather data starting from midnight today
    /// </summary>
    public void GenerateWeather(Season currentSeason)
    {
        // When to start generating weather from in milliseconds
        var staringTimestamp = timeUtil.GetTodayMidnightTimeStamp();

        // How far into future do we generate weather
        var futureTimestampToReach = staringTimestamp + timeUtil.GetHoursAsSeconds(WeatherConfig.Weather.GenerateWeatherAmountHours ?? 1);

        // Keep adding new weather until we have reached desired future date
        var nextTimestamp = staringTimestamp;
        while (nextTimestamp <= futureTimestampToReach)
        {
            var newWeatherToAddToCache = weatherGenerator.GenerateWeather(currentSeason, nextTimestamp);

            // Add generated weather for time period to cache
            WeatherForecast.Add(newWeatherToAddToCache);

            // Increment timestamp so next loop can begin at correct time
            nextTimestamp += GetWeightedWeatherTimePeriod();
        }
    }

    /// <summary>
    ///     Get a time period to increment by, e.g. 15 or 30 minutes as milliseconds
    /// </summary>
    /// <returns>milliseconds</returns>
    protected long GetWeightedWeatherTimePeriod()
    {
        var chosenTimePeriodMinutes = weightedRandomHelper
            .WeightedRandom(WeatherConfig.Weather.TimePeriod.Values, WeatherConfig.Weather.TimePeriod.Weights)
            .Item;

        return chosenTimePeriodMinutes * 60;
    }

    /// <summary>
    ///     Find the first matching weather object that applies to the current time
    /// </summary>
    public Weather GetCurrentWeather()
    {
        var currentSeason = seasonalEventService.GetActiveWeatherSeason();
        ValidateWeatherDataExists(currentSeason);

        return WeatherForecast.Find(weather => weather.Timestamp >= timeUtil.GetTimeStamp());
    }

    /// <summary>
    ///     Find all matching weather objects that applies to the current time + future
    /// </summary>
    public IEnumerable<Weather> GetUpcomingWeather()
    {
        var currentSeason = seasonalEventService.GetActiveWeatherSeason();
        ValidateWeatherDataExists(currentSeason);

        return WeatherForecast.Where(weather => weather.Timestamp >= timeUtil.GetTimeStamp());
    }

    /// <summary>
    ///     Ensure future weather data exists
    /// </summary>
    protected void ValidateWeatherDataExists(Season currentSeason)
    {
        // Clear expired weather data
        WeatherForecast.RemoveAll(weather => weather.Timestamp < timeUtil.GetTimeStamp());

        // Check data exists for current time
        var result = WeatherForecast.Where(weather => weather.Timestamp >= timeUtil.GetTimeStamp());
        if (!result.Any())
        {
            GenerateWeather(currentSeason);
        }
    }
}