2021-12-22 04:48:20 +00:00
|
|
|
using System.Collections.Generic;
|
2019-12-22 22:08:53 +00:00
|
|
|
using System.Linq;
|
2019-07-01 01:50:01 +00:00
|
|
|
using NLog;
|
2019-12-22 22:08:53 +00:00
|
|
|
using NzbDrone.Common.Cache;
|
2021-12-22 04:48:20 +00:00
|
|
|
using NzbDrone.Core.Download.Clients;
|
2019-12-22 22:08:53 +00:00
|
|
|
using NzbDrone.Core.Indexers;
|
2013-04-01 06:22:16 +00:00
|
|
|
|
|
|
|
namespace NzbDrone.Core.Download
|
|
|
|
{
|
|
|
|
public interface IProvideDownloadClient
|
|
|
|
{
|
2021-12-22 04:48:20 +00:00
|
|
|
IDownloadClient GetDownloadClient(DownloadProtocol downloadProtocol, int indexerId = 0);
|
2014-04-19 15:09:22 +00:00
|
|
|
IEnumerable<IDownloadClient> GetDownloadClients();
|
2014-12-03 00:47:45 +00:00
|
|
|
IDownloadClient Get(int id);
|
2013-04-01 06:22:16 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
public class DownloadClientProvider : IProvideDownloadClient
|
|
|
|
{
|
2019-07-01 01:50:01 +00:00
|
|
|
private readonly Logger _logger;
|
2014-02-14 05:31:49 +00:00
|
|
|
private readonly IDownloadClientFactory _downloadClientFactory;
|
2019-07-01 01:50:01 +00:00
|
|
|
private readonly IDownloadClientStatusService _downloadClientStatusService;
|
2021-12-22 04:48:20 +00:00
|
|
|
private readonly IIndexerFactory _indexerFactory;
|
2019-07-01 01:50:01 +00:00
|
|
|
private readonly ICached<int> _lastUsedDownloadClient;
|
2013-04-01 06:22:16 +00:00
|
|
|
|
2021-12-22 04:48:20 +00:00
|
|
|
public DownloadClientProvider(IDownloadClientStatusService downloadClientStatusService,
|
|
|
|
IDownloadClientFactory downloadClientFactory,
|
|
|
|
IIndexerFactory indexerFactory,
|
|
|
|
ICacheManager cacheManager,
|
|
|
|
Logger logger)
|
2013-04-01 06:22:16 +00:00
|
|
|
{
|
2019-07-01 01:50:01 +00:00
|
|
|
_logger = logger;
|
2014-02-14 05:31:49 +00:00
|
|
|
_downloadClientFactory = downloadClientFactory;
|
2019-07-01 01:50:01 +00:00
|
|
|
_downloadClientStatusService = downloadClientStatusService;
|
2021-12-22 04:48:20 +00:00
|
|
|
_indexerFactory = indexerFactory;
|
2019-07-01 01:50:01 +00:00
|
|
|
_lastUsedDownloadClient = cacheManager.GetCache<int>(GetType(), "lastDownloadClientId");
|
2013-04-01 06:22:16 +00:00
|
|
|
}
|
|
|
|
|
2021-12-22 04:48:20 +00:00
|
|
|
public IDownloadClient GetDownloadClient(DownloadProtocol downloadProtocol, int indexerId = 0)
|
2013-04-01 06:22:16 +00:00
|
|
|
{
|
2019-07-01 01:50:01 +00:00
|
|
|
var availableProviders = _downloadClientFactory.GetAvailableProviders().Where(v => v.Protocol == downloadProtocol).ToList();
|
|
|
|
|
2019-12-22 22:08:53 +00:00
|
|
|
if (!availableProviders.Any())
|
|
|
|
{
|
|
|
|
return null;
|
|
|
|
}
|
2019-07-01 01:50:01 +00:00
|
|
|
|
2021-12-22 04:48:20 +00:00
|
|
|
if (indexerId > 0)
|
|
|
|
{
|
|
|
|
var indexer = _indexerFactory.Find(indexerId);
|
|
|
|
|
|
|
|
if (indexer != null && indexer.DownloadClientId > 0)
|
|
|
|
{
|
|
|
|
var client = availableProviders.SingleOrDefault(d => d.Definition.Id == indexer.DownloadClientId);
|
|
|
|
|
|
|
|
return client ?? throw new DownloadClientUnavailableException($"Indexer specified download client is not available");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-07-01 01:50:01 +00:00
|
|
|
var blockedProviders = new HashSet<int>(_downloadClientStatusService.GetBlockedProviders().Select(v => v.ProviderId));
|
|
|
|
|
|
|
|
if (blockedProviders.Any())
|
|
|
|
{
|
|
|
|
var nonBlockedProviders = availableProviders.Where(v => !blockedProviders.Contains(v.Definition.Id)).ToList();
|
|
|
|
|
|
|
|
if (nonBlockedProviders.Any())
|
|
|
|
{
|
|
|
|
availableProviders = nonBlockedProviders;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
_logger.Trace("No non-blocked Download Client available, retrying blocked one.");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Use the first priority clients first
|
|
|
|
availableProviders = availableProviders.GroupBy(v => (v.Definition as DownloadClientDefinition).Priority)
|
|
|
|
.OrderBy(v => v.Key)
|
|
|
|
.First().OrderBy(v => v.Definition.Id).ToList();
|
|
|
|
|
|
|
|
var lastId = _lastUsedDownloadClient.Find(downloadProtocol.ToString());
|
|
|
|
|
|
|
|
var provider = availableProviders.FirstOrDefault(v => v.Definition.Id > lastId) ?? availableProviders.First();
|
|
|
|
|
|
|
|
_lastUsedDownloadClient.Set(downloadProtocol.ToString(), provider.Definition.Id);
|
|
|
|
|
|
|
|
return provider;
|
2014-04-19 15:09:22 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
public IEnumerable<IDownloadClient> GetDownloadClients()
|
|
|
|
{
|
2019-06-14 03:54:25 +00:00
|
|
|
return _downloadClientFactory.GetAvailableProviders();
|
2013-04-01 06:22:16 +00:00
|
|
|
}
|
2015-01-31 08:24:09 +00:00
|
|
|
|
2014-12-03 00:47:45 +00:00
|
|
|
public IDownloadClient Get(int id)
|
|
|
|
{
|
|
|
|
return _downloadClientFactory.GetAvailableProviders().Single(d => d.Definition.Id == id);
|
|
|
|
}
|
2013-04-01 06:22:16 +00:00
|
|
|
}
|
2014-12-19 00:26:42 +00:00
|
|
|
}
|