Cleanuparr/code/Infrastructure/Verticals/DownloadClient/DownloadServiceFactory.cs

using Common.Configuration;
using Common.Enums;
using Infrastructure.Events;
using Infrastructure.Http;
using Infrastructure.Interceptors;
using Infrastructure.Verticals.ContentBlocker;
using Infrastructure.Verticals.DownloadClient.Deluge;
using Infrastructure.Verticals.DownloadClient.QBittorrent;
using Infrastructure.Verticals.DownloadClient.Transmission;
using Infrastructure.Verticals.Files;
using Infrastructure.Verticals.ItemStriker;
using Microsoft.Extensions.Caching.Memory;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;

namespace Infrastructure.Verticals.DownloadClient;

/// <summary>
/// Factory responsible for creating download client service instances
/// </summary>
public sealed class DownloadServiceFactory
{
    private readonly IServiceProvider _serviceProvider;
    private readonly ILogger<DownloadServiceFactory> _logger;
    
    public DownloadServiceFactory(
        IServiceProvider serviceProvider, 
        ILogger<DownloadServiceFactory> logger)
    {
        _serviceProvider = serviceProvider;
        _logger = logger;
    }

    // /// <summary>
    // /// Creates a download service using the specified client ID
    // /// </summary>
    // /// <param name="clientId">The client ID to create a service for</param>
    // /// <returns>An implementation of IDownloadService or null if the client is not available</returns>
    // public IDownloadService? GetDownloadService(Guid clientId)
    // {
    //     var config = _configManager.GetConfiguration<DownloadClientConfig>();
    //     var clientConfig = config.GetClientConfig(clientId);
    //     
    //     if (clientConfig == null)
    //     {
    //         _logger.LogWarning("No download client configuration found for ID {clientId}", clientId);
    //         return null;
    //     }
    //     
    //     if (!clientConfig.Enabled)
    //     {
    //         _logger.LogWarning("Download client {clientId} is disabled", clientId);
    //         return null;
    //     }
    //     
    //     return GetDownloadService(clientConfig);
    // }
    
    /// <summary>
    /// Creates a download service using the specified client configuration
    /// </summary>
    /// <param name="downloadClientConfig">The client configuration to use</param>
    /// <returns>An implementation of IDownloadService or null if the client is not available</returns>
    public IDownloadService GetDownloadService(DownloadClientConfig downloadClientConfig)
    {
        if (!downloadClientConfig.Enabled)
        {
            _logger.LogWarning("Download client {clientId} is disabled, but a service was requested", downloadClientConfig.Id);
        }
        
        return downloadClientConfig.TypeName switch
        {
            DownloadClientTypeName.QBittorrent => CreateQBitService(downloadClientConfig),
            DownloadClientTypeName.Deluge => CreateDelugeService(downloadClientConfig),
            DownloadClientTypeName.Transmission => CreateTransmissionService(downloadClientConfig),
            _ => throw new NotSupportedException($"Download client type {downloadClientConfig.TypeName} is not supported")
        };
    }
    
    private QBitService CreateQBitService(DownloadClientConfig downloadClientConfig)
    {
        var logger = _serviceProvider.GetRequiredService<ILogger<QBitService>>();
        var cache = _serviceProvider.GetRequiredService<IMemoryCache>();
        var filenameEvaluator = _serviceProvider.GetRequiredService<IFilenameEvaluator>();
        var striker = _serviceProvider.GetRequiredService<IStriker>();
        var dryRunInterceptor = _serviceProvider.GetRequiredService<IDryRunInterceptor>();
        var hardLinkFileService = _serviceProvider.GetRequiredService<IHardLinkFileService>();
        var httpClientProvider = _serviceProvider.GetRequiredService<IDynamicHttpClientProvider>();
        var eventPublisher = _serviceProvider.GetRequiredService<EventPublisher>();
        var blocklistProvider = _serviceProvider.GetRequiredService<BlocklistProvider>();
        
        // Create the QBitService instance
        QBitService service = new(
            logger, cache, filenameEvaluator, striker, dryRunInterceptor,
            hardLinkFileService, httpClientProvider, eventPublisher, blocklistProvider, downloadClientConfig
        );
        
        return service;
    }
    
    private DelugeService CreateDelugeService(DownloadClientConfig downloadClientConfig)
    {
        var logger = _serviceProvider.GetRequiredService<ILogger<DelugeService>>();
        var filenameEvaluator = _serviceProvider.GetRequiredService<IFilenameEvaluator>();
        var cache = _serviceProvider.GetRequiredService<IMemoryCache>();
        var striker = _serviceProvider.GetRequiredService<IStriker>();
        var dryRunInterceptor = _serviceProvider.GetRequiredService<IDryRunInterceptor>();
        var hardLinkFileService = _serviceProvider.GetRequiredService<IHardLinkFileService>();
        var httpClientProvider = _serviceProvider.GetRequiredService<IDynamicHttpClientProvider>();
        var eventPublisher = _serviceProvider.GetRequiredService<EventPublisher>();
        var blocklistProvider = _serviceProvider.GetRequiredService<BlocklistProvider>();
        
        // Create the DelugeService instance
        DelugeService service = new(
            logger, cache, filenameEvaluator, striker, dryRunInterceptor,
            hardLinkFileService, httpClientProvider, eventPublisher, blocklistProvider, downloadClientConfig
        );
        
        return service;
    }
    
    private TransmissionService CreateTransmissionService(DownloadClientConfig downloadClientConfig)
    {
        var logger = _serviceProvider.GetRequiredService<ILogger<TransmissionService>>();
        var cache = _serviceProvider.GetRequiredService<IMemoryCache>();
        var filenameEvaluator = _serviceProvider.GetRequiredService<IFilenameEvaluator>();
        var striker = _serviceProvider.GetRequiredService<IStriker>();
        var dryRunInterceptor = _serviceProvider.GetRequiredService<IDryRunInterceptor>();
        var hardLinkFileService = _serviceProvider.GetRequiredService<IHardLinkFileService>();
        var httpClientProvider = _serviceProvider.GetRequiredService<IDynamicHttpClientProvider>();
        var eventPublisher = _serviceProvider.GetRequiredService<EventPublisher>();
        var blocklistProvider = _serviceProvider.GetRequiredService<BlocklistProvider>();
        
        // Create the TransmissionService instance
        TransmissionService service = new(
            logger, cache, filenameEvaluator, striker, dryRunInterceptor,
            hardLinkFileService, httpClientProvider, eventPublisher, blocklistProvider, downloadClientConfig
        );
        
        return service;
    }
}