2017-08-14 02:58:42 +00:00
|
|
|
|
using System;
|
2015-02-26 16:13:15 +00:00
|
|
|
|
using System.Linq;
|
|
|
|
|
using NLog;
|
|
|
|
|
using NzbDrone.Core.IndexerSearch.Definitions;
|
|
|
|
|
using NzbDrone.Core.Parser.Model;
|
|
|
|
|
using NzbDrone.Core.Queue;
|
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.DecisionEngine.Specifications
|
|
|
|
|
{
|
|
|
|
|
public class QueueSpecification : IDecisionEngineSpecification
|
|
|
|
|
{
|
|
|
|
|
private readonly IQueueService _queueService;
|
|
|
|
|
private readonly QualityUpgradableSpecification _qualityUpgradableSpecification;
|
|
|
|
|
private readonly Logger _logger;
|
|
|
|
|
|
|
|
|
|
public QueueSpecification(IQueueService queueService,
|
|
|
|
|
QualityUpgradableSpecification qualityUpgradableSpecification,
|
|
|
|
|
Logger logger)
|
|
|
|
|
{
|
|
|
|
|
_queueService = queueService;
|
|
|
|
|
_qualityUpgradableSpecification = qualityUpgradableSpecification;
|
|
|
|
|
_logger = logger;
|
|
|
|
|
}
|
|
|
|
|
|
2016-12-09 06:54:15 +00:00
|
|
|
|
public RejectionType Type => RejectionType.Permanent;
|
2015-02-26 16:13:15 +00:00
|
|
|
|
|
2017-08-14 02:58:42 +00:00
|
|
|
|
public Decision IsSatisfiedBy(RemoteAlbum subject, SearchCriteriaBase searchCriteria)
|
2015-02-26 16:13:15 +00:00
|
|
|
|
{
|
|
|
|
|
var queue = _queueService.GetQueue()
|
2017-08-14 02:58:42 +00:00
|
|
|
|
.Select(q => q.RemoteAlbum).ToList();
|
2015-02-26 16:13:15 +00:00
|
|
|
|
|
2017-08-14 02:58:42 +00:00
|
|
|
|
var matchingArtist = queue.Where(q => q.Artist.Id == subject.Artist.Id);
|
|
|
|
|
var matchingAlbum = matchingArtist.Where(q => q.Albums.Select(e => e.Id).Intersect(subject.Albums.Select(e => e.Id)).Any());
|
2015-02-26 16:13:15 +00:00
|
|
|
|
|
2017-08-14 02:58:42 +00:00
|
|
|
|
foreach (var remoteAlbum in matchingAlbum)
|
2015-02-26 16:13:15 +00:00
|
|
|
|
{
|
2017-08-14 02:58:42 +00:00
|
|
|
|
_logger.Debug("Checking if existing release in queue meets cutoff. Queued quality is: {0}", remoteAlbum.ParsedAlbumInfo.Quality);
|
2015-02-26 16:13:15 +00:00
|
|
|
|
|
2017-08-14 02:58:42 +00:00
|
|
|
|
if (!_qualityUpgradableSpecification.CutoffNotMet(subject.Artist.Profile, remoteAlbum.ParsedAlbumInfo.Quality, subject.ParsedAlbumInfo.Quality))
|
2015-02-26 16:13:15 +00:00
|
|
|
|
{
|
2017-08-14 02:58:42 +00:00
|
|
|
|
return Decision.Reject("Quality for release in queue already meets cutoff: {0}", remoteAlbum.ParsedAlbumInfo.Quality);
|
2015-02-26 16:13:15 +00:00
|
|
|
|
}
|
|
|
|
|
|
2017-08-14 02:58:42 +00:00
|
|
|
|
_logger.Debug("Checking if release is higher quality than queued release. Queued quality is: {0}", remoteAlbum.ParsedAlbumInfo.Quality);
|
2015-02-26 16:13:15 +00:00
|
|
|
|
|
2017-08-14 02:58:42 +00:00
|
|
|
|
if (!_qualityUpgradableSpecification.IsUpgradable(subject.Artist.Profile, remoteAlbum.ParsedAlbumInfo.Quality, subject.ParsedAlbumInfo.Quality))
|
2015-02-26 16:13:15 +00:00
|
|
|
|
{
|
2017-08-14 02:58:42 +00:00
|
|
|
|
return Decision.Reject("Quality for release in queue is of equal or higher preference: {0}", remoteAlbum.ParsedAlbumInfo.Quality);
|
2015-02-26 16:13:15 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return Decision.Accept();
|
2017-08-14 02:58:42 +00:00
|
|
|
|
|
2015-02-26 16:13:15 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|