using Cleanuparr.Domain.Enums;
using Cleanuparr.Infrastructure.Events;
using Cleanuparr.Infrastructure.Events.Interfaces;
using Cleanuparr.Infrastructure.Features.Files;
using Cleanuparr.Infrastructure.Features.ItemStriker;
using Cleanuparr.Infrastructure.Features.MalwareBlocker;
using Cleanuparr.Infrastructure.Helpers;
using Cleanuparr.Infrastructure.Http;
using Cleanuparr.Infrastructure.Interceptors;
using Cleanuparr.Infrastructure.Services.Interfaces;
using Cleanuparr.Persistence.Models.Configuration;
using Microsoft.Extensions.Caching.Memory;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
using DelugeService = Cleanuparr.Infrastructure.Features.DownloadClient.Deluge.DelugeService;
using QBitService = Cleanuparr.Infrastructure.Features.DownloadClient.QBittorrent.QBitService;
using TransmissionService = Cleanuparr.Infrastructure.Features.DownloadClient.Transmission.TransmissionService;
using UTorrentService = Cleanuparr.Infrastructure.Features.DownloadClient.UTorrent.UTorrentService;
namespace Cleanuparr.Infrastructure.Features.DownloadClient;
///
/// Factory responsible for creating download client service instances
///
public sealed class DownloadServiceFactory : IDownloadServiceFactory
{
private readonly ILogger _logger;
private readonly IServiceProvider _serviceProvider;
public DownloadServiceFactory(
ILogger logger,
IServiceProvider serviceProvider
)
{
_serviceProvider = serviceProvider;
_logger = logger;
}
///
/// Creates a download service using the specified client configuration
///
/// The client configuration to use
/// An implementation of IDownloadService or null if the client is not available
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),
DownloadClientTypeName.uTorrent => CreateUTorrentService(downloadClientConfig),
_ => throw new NotSupportedException($"Download client type {downloadClientConfig.TypeName} is not supported")
};
}
private QBitService CreateQBitService(DownloadClientConfig downloadClientConfig)
{
var logger = _serviceProvider.GetRequiredService>();
var cache = _serviceProvider.GetRequiredService();
var filenameEvaluator = _serviceProvider.GetRequiredService();
var striker = _serviceProvider.GetRequiredService();
var dryRunInterceptor = _serviceProvider.GetRequiredService();
var hardLinkFileService = _serviceProvider.GetRequiredService();
var httpClientProvider = _serviceProvider.GetRequiredService();
var eventPublisher = _serviceProvider.GetRequiredService();
var blocklistProvider = _serviceProvider.GetRequiredService();
var ruleEvaluator = _serviceProvider.GetRequiredService();
var ruleManager = _serviceProvider.GetRequiredService();
// Create the QBitService instance
QBitService service = new(
logger, cache, filenameEvaluator, striker, dryRunInterceptor,
hardLinkFileService, httpClientProvider, eventPublisher, blocklistProvider, downloadClientConfig, ruleEvaluator, ruleManager
);
return service;
}
private DelugeService CreateDelugeService(DownloadClientConfig downloadClientConfig)
{
var logger = _serviceProvider.GetRequiredService>();
var filenameEvaluator = _serviceProvider.GetRequiredService();
var cache = _serviceProvider.GetRequiredService();
var striker = _serviceProvider.GetRequiredService();
var dryRunInterceptor = _serviceProvider.GetRequiredService();
var hardLinkFileService = _serviceProvider.GetRequiredService();
var httpClientProvider = _serviceProvider.GetRequiredService();
var eventPublisher = _serviceProvider.GetRequiredService();
var blocklistProvider = _serviceProvider.GetRequiredService();
var ruleEvaluator = _serviceProvider.GetRequiredService();
var ruleManager = _serviceProvider.GetRequiredService();
// Create the DelugeService instance
DelugeService service = new(
logger, cache, filenameEvaluator, striker, dryRunInterceptor,
hardLinkFileService, httpClientProvider, eventPublisher, blocklistProvider, downloadClientConfig, ruleEvaluator, ruleManager
);
return service;
}
private TransmissionService CreateTransmissionService(DownloadClientConfig downloadClientConfig)
{
var logger = _serviceProvider.GetRequiredService>();
var cache = _serviceProvider.GetRequiredService();
var filenameEvaluator = _serviceProvider.GetRequiredService();
var striker = _serviceProvider.GetRequiredService();
var dryRunInterceptor = _serviceProvider.GetRequiredService();
var hardLinkFileService = _serviceProvider.GetRequiredService();
var httpClientProvider = _serviceProvider.GetRequiredService();
var eventPublisher = _serviceProvider.GetRequiredService();
var blocklistProvider = _serviceProvider.GetRequiredService();
var ruleEvaluator = _serviceProvider.GetRequiredService();
var ruleManager = _serviceProvider.GetRequiredService();
// Create the TransmissionService instance
TransmissionService service = new(
logger, cache, filenameEvaluator, striker, dryRunInterceptor,
hardLinkFileService, httpClientProvider, eventPublisher, blocklistProvider, downloadClientConfig, ruleEvaluator, ruleManager
);
return service;
}
private UTorrentService CreateUTorrentService(DownloadClientConfig downloadClientConfig)
{
var logger = _serviceProvider.GetRequiredService>();
var cache = _serviceProvider.GetRequiredService();
var filenameEvaluator = _serviceProvider.GetRequiredService();
var striker = _serviceProvider.GetRequiredService();
var dryRunInterceptor = _serviceProvider.GetRequiredService();
var hardLinkFileService = _serviceProvider.GetRequiredService();
var httpClientProvider = _serviceProvider.GetRequiredService();
var eventPublisher = _serviceProvider.GetRequiredService();
var blocklistProvider = _serviceProvider.GetRequiredService();
var loggerFactory = _serviceProvider.GetRequiredService();
var ruleEvaluator = _serviceProvider.GetRequiredService();
var ruleManager = _serviceProvider.GetRequiredService();
// Create the UTorrentService instance
UTorrentService service = new(
logger, cache, filenameEvaluator, striker, dryRunInterceptor,
hardLinkFileService, httpClientProvider, eventPublisher, blocklistProvider, downloadClientConfig, loggerFactory, ruleEvaluator, ruleManager
);
return service;
}
}