Radarr/src/NzbDrone.Core/Download/DownloadClientProvider.cs

46 lines
1.5 KiB
C#
Raw Normal View History

2014-12-07 05:55:17 +00:00
using System.Linq;
using System.Collections.Generic;
using NzbDrone.Core.Indexers;
2013-04-01 06:22:16 +00:00
namespace NzbDrone.Core.Download
{
public interface IProvideDownloadClient
{
IDownloadClient GetDownloadClient(DownloadProtocol downloadProtocol);
IEnumerable<IDownloadClient> GetDownloadClients();
IDownloadClient Get(int id);
2013-04-01 06:22:16 +00:00
}
public class DownloadClientProvider : IProvideDownloadClient
{
private readonly IDownloadClientFactory _downloadClientFactory;
2013-04-01 06:22:16 +00:00
public DownloadClientProvider(IDownloadClientFactory downloadClientFactory)
2013-04-01 06:22:16 +00:00
{
_downloadClientFactory = downloadClientFactory;
2013-04-01 06:22:16 +00:00
}
public IDownloadClient GetDownloadClient(DownloadProtocol downloadProtocol)
2013-04-01 06:22:16 +00:00
{
return _downloadClientFactory.GetAvailableProviders().FirstOrDefault(v => v.Protocol == downloadProtocol);
}
public IEnumerable<IDownloadClient> GetDownloadClients()
{
2015-01-31 08:24:09 +00:00
return _downloadClientFactory.GetAvailableProviders();//.Select(MapDownloadClient);
2013-04-01 06:22:16 +00:00
}
2015-01-31 08:24:09 +00:00
public IDownloadClient Get(int id)
{
return _downloadClientFactory.GetAvailableProviders().Single(d => d.Definition.Id == id);
}
2015-01-31 08:24:09 +00:00
public IDownloadClient MapDownloadClient(IDownloadClient downloadClient)
{
_downloadClientFactory.SetProviderCharacteristics(downloadClient, (DownloadClientDefinition)downloadClient.Definition);
2015-01-31 08:24:09 +00:00
return downloadClient;
}
2013-04-01 06:22:16 +00:00
}
}