Radarr/NzbDrone.Core/Providers/DownloadProvider.cs

82 lines
3.0 KiB
C#
Raw Normal View History

2011-05-20 05:52:05 +00:00
using System;
using Ninject;
2011-05-20 05:52:05 +00:00
using NLog;
using NzbDrone.Core.Model;
using NzbDrone.Core.Providers.Core;
2011-05-20 05:52:05 +00:00
using NzbDrone.Core.Repository;
namespace NzbDrone.Core.Providers
{
public class DownloadProvider
{
private readonly SabProvider _sabProvider;
private readonly HistoryProvider _historyProvider;
private readonly EpisodeProvider _episodeProvider;
private readonly ExternalNotificationProvider _externalNotificationProvider;
private readonly ConfigProvider _configProvider;
private readonly BlackholeProvider _blackholeProvider;
2011-05-20 05:52:05 +00:00
private static readonly Logger Logger = LogManager.GetCurrentClassLogger();
[Inject]
public DownloadProvider(SabProvider sabProvider, HistoryProvider historyProvider,
EpisodeProvider episodeProvider, ExternalNotificationProvider externalNotificationProvider,
ConfigProvider configProvider, BlackholeProvider blackholeProvider)
2011-05-20 05:52:05 +00:00
{
_sabProvider = sabProvider;
_historyProvider = historyProvider;
_episodeProvider = episodeProvider;
_externalNotificationProvider = externalNotificationProvider;
_configProvider = configProvider;
_blackholeProvider = blackholeProvider;
2011-05-20 05:52:05 +00:00
}
public DownloadProvider()
{
}
public virtual bool DownloadReport(EpisodeParseResult parseResult)
{
var sabTitle = _sabProvider.GetSabTitle(parseResult);
bool addSuccess = false;
if (_configProvider.DownloadClient == DownloadClientType.Blackhole)
addSuccess = _blackholeProvider.DownloadNzb(parseResult, sabTitle);
if (_configProvider.DownloadClient == DownloadClientType.Sabnzbd)
2011-05-20 05:52:05 +00:00
{
if(_sabProvider.IsInQueue(parseResult))
{
Logger.Warn("Episode {0} is already in sab's queue. skipping.", parseResult);
return false;
}
2011-05-20 05:52:05 +00:00
addSuccess = _sabProvider.AddByUrl(parseResult.NzbUrl, sabTitle);
}
if (addSuccess)
{
Logger.Trace("Download added to Queue: {0}", sabTitle);
2011-06-22 01:12:20 +00:00
foreach (var episode in _episodeProvider.GetEpisodesByParseResult(parseResult))
{
var history = new History();
history.Date = DateTime.Now;
history.Indexer = parseResult.Indexer;
2011-05-28 19:23:35 +00:00
history.IsProper = parseResult.Quality.Proper;
history.Quality = parseResult.Quality.QualityType;
2012-01-20 06:35:10 +00:00
history.NzbTitle = parseResult.OriginalString;
history.EpisodeId = episode.EpisodeId;
history.SeriesId = episode.SeriesId;
_historyProvider.Add(history);
_episodeProvider.MarkEpisodeAsFetched(episode.EpisodeId);
}
}
_externalNotificationProvider.OnGrab(sabTitle);
return addSuccess;
2011-05-20 05:52:05 +00:00
}
}
}