using System; using System.Linq; using NLog; using NzbDrone.Common; using NzbDrone.Core.Configuration; using NzbDrone.Core.Datastore; using NzbDrone.Core.Download; using NzbDrone.Core.Messaging.Commands; using NzbDrone.Core.Messaging.Events; using NzbDrone.Core.Tv.Events; namespace NzbDrone.Core.Blacklisting { public interface IBlacklistService { bool Blacklisted(int seriesId,string sourceTitle, DateTime publishedDate); PagingSpec Paged(PagingSpec pagingSpec); void Delete(int id); } public class BlacklistService : IBlacklistService, IExecute, IHandle, IHandle { private readonly IBlacklistRepository _blacklistRepository; private readonly IRedownloadFailedDownloads _redownloadFailedDownloadService; public BlacklistService(IBlacklistRepository blacklistRepository, IRedownloadFailedDownloads redownloadFailedDownloadService) { _blacklistRepository = blacklistRepository; _redownloadFailedDownloadService = redownloadFailedDownloadService; } public bool Blacklisted(int seriesId, string sourceTitle, DateTime publishedDate) { var blacklisted = _blacklistRepository.Blacklisted(seriesId, sourceTitle); return blacklisted.Any(item => HasSamePublishedDate(item, publishedDate)); } public PagingSpec Paged(PagingSpec pagingSpec) { return _blacklistRepository.GetPaged(pagingSpec); } public void Delete(int id) { _blacklistRepository.Delete(id); } private bool HasSamePublishedDate(Blacklist item, DateTime publishedDate) { if (!item.PublishedDate.HasValue) return true; return item.PublishedDate.Value.AddDays(-2) <= publishedDate && item.PublishedDate.Value.AddDays(2) >= publishedDate; } public void Execute(ClearBlacklistCommand message) { _blacklistRepository.Purge(); } public void Handle(DownloadFailedEvent message) { var blacklist = new Blacklist { SeriesId = message.SeriesId, EpisodeIds = message.EpisodeIds, SourceTitle = message.SourceTitle, Quality = message.Quality, Date = DateTime.UtcNow, PublishedDate = DateTime.Parse(message.Data.GetValueOrDefault("publishedDate", null)) }; _blacklistRepository.Insert(blacklist); _redownloadFailedDownloadService.Redownload(message.SeriesId, message.EpisodeIds); } public void Handle(SeriesDeletedEvent message) { var blacklisted = _blacklistRepository.BlacklistedBySeries(message.Series.Id); _blacklistRepository.DeleteMany(blacklisted); } } }