void.cat/VoidCat/Services/Files/FileInfoManager.cs

using VoidCat.Model;
using VoidCat.Model.User;
using VoidCat.Services.Abstractions;

namespace VoidCat.Services.Files;

/// <summary>
/// Main interface for getting file info to serve to clients.
/// This interface should wrap all stores and return the combined result
/// </summary>
public sealed class FileInfoManager
{
    private readonly IFileMetadataStore _metadataStore;
    private readonly IPaymentStore _paymentStore;
    private readonly IStatsReporter _statsReporter;
    private readonly IUserStore _userStore;
    private readonly IVirusScanStore _virusScanStore;
    private readonly IUserUploadsStore _userUploadsStore;

    public FileInfoManager(IFileMetadataStore metadataStore, IPaymentStore paymentStore, IStatsReporter statsReporter,
        IUserStore userStore, IVirusScanStore virusScanStore, IUserUploadsStore userUploadsStore)
    {
        _metadataStore = metadataStore;
        _paymentStore = paymentStore;
        _statsReporter = statsReporter;
        _userStore = userStore;
        _virusScanStore = virusScanStore;
        _userUploadsStore = userUploadsStore;
    }

    /// <summary>
    /// Get all metadata for a single file
    /// </summary>
    /// <param name="id"></param>
    /// <returns></returns>
    public ValueTask<PublicVoidFile?> Get(Guid id)
    {
        return Get<PublicVoidFile, FileMeta>(id);
    }

    /// <summary>
    /// Get all private metadata for a single file
    /// </summary>
    /// <param name="id"></param>
    /// <returns></returns>
    public ValueTask<PrivateVoidFile?> GetPrivate(Guid id)
    {
        return Get<PrivateVoidFile, SecretFileMeta>(id);
    }

    /// <summary>
    /// Get all metadata for multiple files
    /// </summary>
    /// <param name="ids"></param>
    /// <returns></returns>
    public async ValueTask<IReadOnlyList<PublicVoidFile>> Get(Guid[] ids)
    {
        var ret = new List<PublicVoidFile>();
        foreach (var id in ids)
        {
            var v = await Get(id);
            if (v != default)
            {
                ret.Add(v);
            }
        }

        return ret;
    }

    /// <summary>
    /// Deletes all file metadata
    /// </summary>
    /// <param name="id"></param>
    /// <returns></returns>
    public async ValueTask Delete(Guid id)
    {
        await _metadataStore.Delete(id);
        await _paymentStore.Delete(id);
        await _statsReporter.Delete(id);
        await _virusScanStore.Delete(id);
    }

    private async ValueTask<TFile?> Get<TFile, TMeta>(Guid id)
        where TMeta : FileMeta where TFile : VoidFile<TMeta>, new()
    {
        var meta = _metadataStore.Get<TMeta>(id);
        var payment = _paymentStore.Get(id);
        var bandwidth = _statsReporter.GetBandwidth(id);
        var virusScan = _virusScanStore.GetByFile(id);
        var uploader = _userUploadsStore.Uploader(id);
        await Task.WhenAll(meta.AsTask(), payment.AsTask(), bandwidth.AsTask(), virusScan.AsTask(), uploader.AsTask());

        if (meta.Result == default) return default;
        var user = uploader.Result.HasValue ? await _userStore.Get<PublicUser>(uploader.Result.Value) : null;

        return new TFile()
        {
            Id = id,
            Metadata = meta.Result,
            Payment = payment.Result,
            Bandwidth = bandwidth.Result,
            Uploader = user?.Flags.HasFlag(UserFlags.PublicProfile) == true ? user : null,
            VirusScan = virusScan.Result
        };
    }
}