mirror of
https://github.com/Radarr/Radarr.git
synced 2024-11-05 10:32:35 +01:00
66 lines
2.3 KiB
C#
66 lines
2.3 KiB
C#
using System;
|
|
using Ninject;
|
|
using NLog;
|
|
using NzbDrone.Core.Model;
|
|
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 static readonly Logger Logger = LogManager.GetCurrentClassLogger();
|
|
|
|
[Inject]
|
|
public DownloadProvider(SabProvider sabProvider, HistoryProvider historyProvider,
|
|
EpisodeProvider episodeProvider, ExternalNotificationProvider externalNotificationProvider)
|
|
{
|
|
_sabProvider = sabProvider;
|
|
_historyProvider = historyProvider;
|
|
_episodeProvider = episodeProvider;
|
|
_externalNotificationProvider = externalNotificationProvider;
|
|
}
|
|
|
|
public DownloadProvider()
|
|
{
|
|
}
|
|
|
|
public virtual bool DownloadReport(EpisodeParseResult parseResult)
|
|
{
|
|
if (_sabProvider.IsInQueue(parseResult))
|
|
{
|
|
Logger.Warn("Episode {0} is already in sab's queue. skipping.", parseResult);
|
|
return false;
|
|
}
|
|
|
|
var sabTitle = _sabProvider.GetSabTitle(parseResult);
|
|
var addSuccess = _sabProvider.AddByUrl(parseResult.NzbUrl, sabTitle);
|
|
|
|
if (addSuccess)
|
|
{
|
|
foreach (var episode in _episodeProvider.GetEpisodesByParseResult(parseResult))
|
|
{
|
|
var history = new History();
|
|
history.Date = DateTime.Now;
|
|
history.Indexer = parseResult.Indexer;
|
|
history.IsProper = parseResult.Quality.Proper;
|
|
history.Quality = parseResult.Quality.QualityType;
|
|
history.NzbTitle = parseResult.NzbTitle;
|
|
history.EpisodeId = episode.EpisodeId;
|
|
history.SeriesId = episode.SeriesId;
|
|
|
|
_historyProvider.Add(history);
|
|
_episodeProvider.MarkEpisodeAsFetched(episode.EpisodeId);
|
|
}
|
|
}
|
|
|
|
_externalNotificationProvider.OnGrab(sabTitle);
|
|
|
|
return addSuccess;
|
|
}
|
|
}
|
|
} |