mirror of https://github.com/Radarr/Radarr
96 lines
4.1 KiB
C#
96 lines
4.1 KiB
C#
using System;
|
|
using NLog;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Core.Configuration;
|
|
using NzbDrone.Core.CustomFormats;
|
|
using NzbDrone.Core.History;
|
|
using NzbDrone.Core.IndexerSearch.Definitions;
|
|
using NzbDrone.Core.Parser.Model;
|
|
|
|
namespace NzbDrone.Core.DecisionEngine.Specifications.RssSync
|
|
{
|
|
public class HistorySpecification : IDecisionEngineSpecification
|
|
{
|
|
private readonly IHistoryService _historyService;
|
|
private readonly UpgradableSpecification _upgradableSpecification;
|
|
private readonly ICustomFormatCalculationService _formatService;
|
|
private readonly IConfigService _configService;
|
|
private readonly Logger _logger;
|
|
|
|
public HistorySpecification(IHistoryService historyService,
|
|
UpgradableSpecification upgradableSpecification,
|
|
ICustomFormatCalculationService formatService,
|
|
IConfigService configService,
|
|
Logger logger)
|
|
{
|
|
_historyService = historyService;
|
|
_upgradableSpecification = upgradableSpecification;
|
|
_formatService = formatService;
|
|
_configService = configService;
|
|
_logger = logger;
|
|
}
|
|
|
|
public SpecificationPriority Priority => SpecificationPriority.Database;
|
|
public RejectionType Type => RejectionType.Permanent;
|
|
|
|
public virtual Decision IsSatisfiedBy(RemoteMovie subject, SearchCriteriaBase searchCriteria)
|
|
{
|
|
if (searchCriteria != null)
|
|
{
|
|
_logger.Debug("Skipping history check during search");
|
|
return Decision.Accept();
|
|
}
|
|
|
|
var cdhEnabled = _configService.EnableCompletedDownloadHandling;
|
|
|
|
_logger.Debug("Performing history status check on report");
|
|
_logger.Debug("Checking current status of movie [{0}] in history", subject.Movie.Id);
|
|
var mostRecent = _historyService.MostRecentForMovie(subject.Movie.Id);
|
|
|
|
if (mostRecent != null && mostRecent.EventType == MovieHistoryEventType.Grabbed)
|
|
{
|
|
var customFormats = _formatService.ParseCustomFormat(mostRecent);
|
|
|
|
var cutoffUnmet = _upgradableSpecification.CutoffNotMet(subject.Movie.Profile,
|
|
mostRecent.Quality,
|
|
customFormats,
|
|
subject.ParsedMovieInfo.Quality);
|
|
|
|
var upgradeable = _upgradableSpecification.IsUpgradable(subject.Movie.Profile,
|
|
mostRecent.Quality,
|
|
customFormats,
|
|
subject.ParsedMovieInfo.Quality,
|
|
subject.CustomFormats);
|
|
|
|
var recent = mostRecent.Date.After(DateTime.UtcNow.AddHours(-12));
|
|
if (!recent && cdhEnabled)
|
|
{
|
|
return Decision.Accept();
|
|
}
|
|
|
|
if (!cutoffUnmet)
|
|
{
|
|
if (recent)
|
|
{
|
|
return Decision.Reject("Recent grab event in history already meets cutoff: {0}", mostRecent.Quality);
|
|
}
|
|
|
|
return Decision.Reject("CDH is disabled and grab event in history already meets cutoff: {0}", mostRecent.Quality);
|
|
}
|
|
|
|
if (!upgradeable)
|
|
{
|
|
if (recent)
|
|
{
|
|
return Decision.Reject("Recent grab event in history is of equal or higher quality: {0}", mostRecent.Quality);
|
|
}
|
|
|
|
return Decision.Reject("CDH is disabled and grab event in history is of equal or higher quality: {0}", mostRecent.Quality);
|
|
}
|
|
}
|
|
|
|
return Decision.Accept();
|
|
}
|
|
}
|
|
}
|