discord-bot/CompatBot/Database/Providers/TitleUpdateInfoProvider.cs

using System.Xml.Serialization;
using Microsoft.EntityFrameworkCore;
using PsnClient.POCOs;

namespace CompatBot.Database.Providers;

public static class TitleUpdateInfoProvider
{
    private static readonly PsnClient.Client Client = new();
    private static readonly XmlSerializer XmlSerializer = new(typeof(TitlePatch));

    public static async ValueTask<TitlePatch?> GetAsync(string? productId, CancellationToken cancellationToken)
        => await GetCachedAsync(productId, false).ConfigureAwait(false)
           ?? await GetFromApiAsync(productId, cancellationToken).ConfigureAwait(false);

    private static async ValueTask<TitlePatch?> GetFromApiAsync(string? productId, CancellationToken cancellationToken)
    {
        if (string.IsNullOrEmpty(productId))
            return default;

        productId = productId.ToUpper();
        var (update, xml) = await Client.GetTitleUpdatesAsync(productId, cancellationToken).ConfigureAwait(false);
        if (xml is {Length: > 10})
        {
            var xmlChecksum = xml.GetStableHash();
            await using var wdb = await ThumbnailDb.OpenWriteAsync().ConfigureAwait(false);
            var updateInfo = wdb.GameUpdateInfo.FirstOrDefault(ui => ui.ProductCode == productId);
            if (updateInfo is null)
                wdb.GameUpdateInfo.Add(new() {ProductCode = productId, MetaHash = xmlChecksum, MetaXml = xml, Timestamp = DateTime.UtcNow.Ticks});
            else if (updateInfo.MetaHash != xmlChecksum && update?.Tag?.Packages is {Length: >0})
            {
                updateInfo.MetaHash = xmlChecksum;
                updateInfo.MetaXml = xml;
                updateInfo.Timestamp = DateTime.UtcNow.Ticks;
            }
            else if (updateInfo.MetaHash == xmlChecksum)
                updateInfo.Timestamp = DateTime.UtcNow.Ticks;
            await wdb.SaveChangesAsync(cancellationToken).ConfigureAwait(false);
        }
        if (update is not {Tag.Packages.Length: >0})
            return await GetCachedAsync(productId, true).ConfigureAwait(false);
        return update;
    }

    private static async ValueTask<TitlePatch?> GetCachedAsync(string? productId, bool returnStale = false)
    {
        await using var db = await ThumbnailDb.OpenReadAsync().ConfigureAwait(false);
        var updateInfo = db.GameUpdateInfo
            .AsNoTracking()
            .FirstOrDefault(ui => ui.ProductCode == productId);
        if (updateInfo is null
            || (!returnStale && updateInfo.Timestamp < DateTime.UtcNow.AddDays(-1).Ticks))
            return null;

        await using var memStream = Config.MemoryStreamManager.GetStream(Encoding.UTF8.GetBytes(updateInfo.MetaXml));
        var update = (TitlePatch?)XmlSerializer.Deserialize(memStream);
        if (update is null)
            return null;
        
        update.OfflineCacheTimestamp = updateInfo.Timestamp.AsUtc();
        return update;
    }

    public static async Task RefreshGameUpdateInfoAsync(CancellationToken cancellationToken)
    {
        await using var db = await ThumbnailDb.OpenReadAsync().ConfigureAwait(false);
        do
        {
            var productCodeList = await db.Thumbnail.AsNoTracking().Select(t => t.ProductCode).ToListAsync(cancellationToken).ConfigureAwait(false);
            foreach (var titleId in productCodeList)
            {
                var updateInfo = db.GameUpdateInfo.AsNoTracking().FirstOrDefault(ui => ui.ProductCode == titleId);
                if (!cancellationToken.IsCancellationRequested
                    && (updateInfo?.Timestamp is null or 0L || updateInfo.Timestamp.AsUtc() < DateTime.UtcNow.AddMonths(-1)))
                {
                    await GetAsync(titleId, cancellationToken).ConfigureAwait(false);
                    await Task.Delay(TimeSpan.FromSeconds(1), cancellationToken).ConfigureAwait(false);
                }
            }
            await Task.Delay(TimeSpan.FromDays(1), cancellationToken).ConfigureAwait(false);
        } while (!cancellationToken.IsCancellationRequested);
    }
}