mirror of https://github.com/Radarr/Radarr
60 lines
1.9 KiB
C#
60 lines
1.9 KiB
C#
![]() |
using System;
|
|||
|
using System.Linq;
|
|||
|
using NLog;
|
|||
|
using NzbDrone.Common.Eventing;
|
|||
|
using NzbDrone.Core.Configuration;
|
|||
|
using NzbDrone.Core.Model;
|
|||
|
|
|||
|
namespace NzbDrone.Core.Download
|
|||
|
{
|
|||
|
public interface IDownloadService
|
|||
|
{
|
|||
|
bool DownloadReport(EpisodeParseResult parseResult);
|
|||
|
}
|
|||
|
|
|||
|
public class DownloadService : IDownloadService
|
|||
|
{
|
|||
|
private readonly IProvideDownloadClient _downloadClientProvider;
|
|||
|
private readonly IConfigService _configService;
|
|||
|
private readonly IEventAggregator _eventAggregator;
|
|||
|
private readonly Logger _logger;
|
|||
|
|
|||
|
|
|||
|
public DownloadService(IProvideDownloadClient downloadClientProvider, IConfigService configService,
|
|||
|
IEventAggregator eventAggregator, Logger logger)
|
|||
|
{
|
|||
|
_downloadClientProvider = downloadClientProvider;
|
|||
|
_configService = configService;
|
|||
|
_eventAggregator = eventAggregator;
|
|||
|
_logger = logger;
|
|||
|
}
|
|||
|
|
|||
|
|
|||
|
public bool DownloadReport(EpisodeParseResult parseResult)
|
|||
|
{
|
|||
|
var downloadTitle = parseResult.OriginalString;
|
|||
|
if (!_configService.DownloadClientUseSceneName)
|
|||
|
{
|
|||
|
downloadTitle = parseResult.GetDownloadTitle();
|
|||
|
}
|
|||
|
|
|||
|
var provider = _downloadClientProvider.GetDownloadClient();
|
|||
|
var recentEpisode = ContainsRecentEpisode(parseResult);
|
|||
|
|
|||
|
bool success = provider.DownloadNzb(parseResult.NzbUrl, downloadTitle, recentEpisode);
|
|||
|
|
|||
|
if (success)
|
|||
|
{
|
|||
|
_logger.Trace("Download added to Queue: {0}", downloadTitle);
|
|||
|
_eventAggregator.Publish(new EpisodeGrabbedEvent(parseResult));
|
|||
|
}
|
|||
|
|
|||
|
return success;
|
|||
|
}
|
|||
|
|
|||
|
private static bool ContainsRecentEpisode(EpisodeParseResult parseResult)
|
|||
|
{
|
|||
|
return parseResult.Episodes.Any(e => e.AirDate >= DateTime.Today.AddDays(-7));
|
|||
|
}
|
|||
|
}
|
|||
|
}
|