74 lines
4.1 KiB
C#
74 lines
4.1 KiB
C#
using System.Linq;
|
|
using NLog;
|
|
using NzbDrone.Core.IndexerSearch.Definitions;
|
|
using NzbDrone.Core.Parser.Model;
|
|
using NzbDrone.Core.Profiles.Releases;
|
|
using NzbDrone.Core.Queue;
|
|
|
|
namespace NzbDrone.Core.DecisionEngine.Specifications
|
|
{
|
|
public class QueueSpecification : IDecisionEngineSpecification
|
|
{
|
|
private readonly IQueueService _queueService;
|
|
private readonly UpgradableSpecification _upgradableSpecification;
|
|
private readonly IPreferredWordService _preferredWordServiceCalculator;
|
|
private readonly Logger _logger;
|
|
|
|
public QueueSpecification(IQueueService queueService,
|
|
UpgradableSpecification UpgradableSpecification,
|
|
IPreferredWordService preferredWordServiceCalculator,
|
|
Logger logger)
|
|
{
|
|
_queueService = queueService;
|
|
_upgradableSpecification = UpgradableSpecification;
|
|
_preferredWordServiceCalculator = preferredWordServiceCalculator;
|
|
_logger = logger;
|
|
}
|
|
|
|
public SpecificationPriority Priority => SpecificationPriority.Default;
|
|
public RejectionType Type => RejectionType.Permanent;
|
|
|
|
public Decision IsSatisfiedBy(RemoteEpisode subject, SearchCriteriaBase searchCriteria)
|
|
{
|
|
var queue = _queueService.GetQueue();
|
|
var matchingSeries = queue.Where(q => q.Series.Id == subject.Series.Id);
|
|
var matchingEpisode = matchingSeries.Where(q => q.RemoteEpisode.Episodes.Select(e => e.Id).Intersect(subject.Episodes.Select(e => e.Id)).Any());
|
|
|
|
foreach (var queueItem in matchingEpisode)
|
|
{
|
|
var remoteEpisode = queueItem.RemoteEpisode;
|
|
|
|
_logger.Debug("Checking if existing release in queue meets cutoff. Queued quality is: {0} - {1}", remoteEpisode.ParsedEpisodeInfo.Quality, remoteEpisode.ParsedEpisodeInfo.Language);
|
|
var queuedItemPreferredWordScore = _preferredWordServiceCalculator.Calculate(subject.Series, queueItem.Title);
|
|
|
|
if (!_upgradableSpecification.CutoffNotMet(subject.Series.Profile,
|
|
subject.Series.LanguageProfile,
|
|
remoteEpisode.ParsedEpisodeInfo.Quality,
|
|
remoteEpisode.ParsedEpisodeInfo.Language,
|
|
queuedItemPreferredWordScore,
|
|
subject.ParsedEpisodeInfo.Quality,
|
|
subject.PreferredWordScore))
|
|
{
|
|
return Decision.Reject("Quality for release in queue already meets cutoff: {0} - {1}", remoteEpisode.ParsedEpisodeInfo.Quality, remoteEpisode.ParsedEpisodeInfo.Language);
|
|
}
|
|
|
|
_logger.Debug("Checking if release is higher quality than queued release. Queued quality is: {0} - {1}", remoteEpisode.ParsedEpisodeInfo.Quality, remoteEpisode.ParsedEpisodeInfo.Language);
|
|
|
|
if (!_upgradableSpecification.IsUpgradable(subject.Series.Profile,
|
|
subject.Series.LanguageProfile,
|
|
remoteEpisode.ParsedEpisodeInfo.Quality,
|
|
remoteEpisode.ParsedEpisodeInfo.Language,
|
|
queuedItemPreferredWordScore,
|
|
subject.ParsedEpisodeInfo.Quality,
|
|
subject.ParsedEpisodeInfo.Language,
|
|
subject.PreferredWordScore))
|
|
{
|
|
return Decision.Reject("Quality for release in queue is of equal or higher preference: {0} - {1}", remoteEpisode.ParsedEpisodeInfo.Quality, remoteEpisode.ParsedEpisodeInfo.Language);
|
|
}
|
|
}
|
|
|
|
return Decision.Accept();
|
|
}
|
|
}
|
|
}
|