You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
170 lines
5.4 KiB
170 lines
5.4 KiB
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Core.Datastore;
|
|
using NzbDrone.Core.Download;
|
|
using NzbDrone.Core.Indexers;
|
|
using NzbDrone.Core.Messaging.Commands;
|
|
using NzbDrone.Core.Messaging.Events;
|
|
using NzbDrone.Core.Music.Events;
|
|
using NzbDrone.Core.Parser.Model;
|
|
|
|
namespace NzbDrone.Core.Blocklisting
|
|
{
|
|
public interface IBlocklistService
|
|
{
|
|
bool Blocklisted(int artistId, ReleaseInfo release);
|
|
PagingSpec<Blocklist> Paged(PagingSpec<Blocklist> pagingSpec);
|
|
void Delete(int id);
|
|
void Delete(List<int> ids);
|
|
}
|
|
|
|
public class BlocklistService : IBlocklistService,
|
|
|
|
IExecute<ClearBlocklistCommand>,
|
|
IHandle<DownloadFailedEvent>,
|
|
IHandleAsync<ArtistsDeletedEvent>
|
|
{
|
|
private readonly IBlocklistRepository _blocklistRepository;
|
|
|
|
public BlocklistService(IBlocklistRepository blocklistRepository)
|
|
{
|
|
_blocklistRepository = blocklistRepository;
|
|
}
|
|
|
|
public bool Blocklisted(int artistId, ReleaseInfo release)
|
|
{
|
|
var blocklistedByTitle = _blocklistRepository.BlocklistedByTitle(artistId, release.Title);
|
|
|
|
if (release.DownloadProtocol == DownloadProtocol.Torrent)
|
|
{
|
|
var torrentInfo = release as TorrentInfo;
|
|
|
|
if (torrentInfo == null)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (torrentInfo.InfoHash.IsNullOrWhiteSpace())
|
|
{
|
|
return blocklistedByTitle.Where(b => b.Protocol == DownloadProtocol.Torrent)
|
|
.Any(b => SameTorrent(b, torrentInfo));
|
|
}
|
|
|
|
var blocklistedByTorrentInfohash = _blocklistRepository.BlocklistedByTorrentInfoHash(artistId, torrentInfo.InfoHash);
|
|
|
|
return blocklistedByTorrentInfohash.Any(b => SameTorrent(b, torrentInfo));
|
|
}
|
|
|
|
return blocklistedByTitle.Where(b => b.Protocol == DownloadProtocol.Usenet)
|
|
.Any(b => SameNzb(b, release));
|
|
}
|
|
|
|
public PagingSpec<Blocklist> Paged(PagingSpec<Blocklist> pagingSpec)
|
|
{
|
|
return _blocklistRepository.GetPaged(pagingSpec);
|
|
}
|
|
|
|
public void Delete(int id)
|
|
{
|
|
_blocklistRepository.Delete(id);
|
|
}
|
|
|
|
public void Delete(List<int> ids)
|
|
{
|
|
_blocklistRepository.DeleteMany(ids);
|
|
}
|
|
|
|
private bool SameNzb(Blocklist item, ReleaseInfo release)
|
|
{
|
|
if (item.PublishedDate == release.PublishDate)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
if (!HasSameIndexer(item, release.Indexer) &&
|
|
HasSamePublishedDate(item, release.PublishDate) &&
|
|
HasSameSize(item, release.Size))
|
|
{
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
private bool SameTorrent(Blocklist item, TorrentInfo release)
|
|
{
|
|
if (release.InfoHash.IsNotNullOrWhiteSpace())
|
|
{
|
|
return release.InfoHash.Equals(item.TorrentInfoHash);
|
|
}
|
|
|
|
return HasSameIndexer(item, release.Indexer);
|
|
}
|
|
|
|
private bool HasSameIndexer(Blocklist item, string indexer)
|
|
{
|
|
if (item.Indexer.IsNullOrWhiteSpace())
|
|
{
|
|
return true;
|
|
}
|
|
|
|
return item.Indexer.Equals(indexer, StringComparison.InvariantCultureIgnoreCase);
|
|
}
|
|
|
|
private bool HasSamePublishedDate(Blocklist item, DateTime publishedDate)
|
|
{
|
|
if (!item.PublishedDate.HasValue)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
return item.PublishedDate.Value.AddMinutes(-2) <= publishedDate &&
|
|
item.PublishedDate.Value.AddMinutes(2) >= publishedDate;
|
|
}
|
|
|
|
private bool HasSameSize(Blocklist item, long size)
|
|
{
|
|
if (!item.Size.HasValue)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
var difference = Math.Abs(item.Size.Value - size);
|
|
|
|
return difference <= 2.Megabytes();
|
|
}
|
|
|
|
public void Execute(ClearBlocklistCommand message)
|
|
{
|
|
_blocklistRepository.Purge();
|
|
}
|
|
|
|
public void Handle(DownloadFailedEvent message)
|
|
{
|
|
var blocklist = new Blocklist
|
|
{
|
|
ArtistId = message.ArtistId,
|
|
AlbumIds = message.AlbumIds,
|
|
SourceTitle = message.SourceTitle,
|
|
Quality = message.Quality,
|
|
Date = DateTime.UtcNow,
|
|
PublishedDate = DateTime.Parse(message.Data.GetValueOrDefault("publishedDate")),
|
|
Size = long.Parse(message.Data.GetValueOrDefault("size", "0")),
|
|
Indexer = message.Data.GetValueOrDefault("indexer"),
|
|
Protocol = (DownloadProtocol)Convert.ToInt32(message.Data.GetValueOrDefault("protocol")),
|
|
Message = message.Message,
|
|
TorrentInfoHash = message.Data.GetValueOrDefault("torrentInfoHash")
|
|
};
|
|
|
|
_blocklistRepository.Insert(blocklist);
|
|
}
|
|
|
|
public void HandleAsync(ArtistsDeletedEvent message)
|
|
{
|
|
_blocklistRepository.DeleteForArtists(message.Artists.Select(x => x.Id).ToList());
|
|
}
|
|
}
|
|
}
|