New: Setting to prevent download of early releases (#485)

* New: Setting to prevent download of early releases * Fixup! Test and Wording
2026-04-21 22:04:31 -04:00 · 2018-09-20 22:21:26 -04:00
parent 9b0a7c60ed
commit e08f39ebe0
17 changed files with 222 additions and 8 deletions
@@ -0,0 +1,68 @@
+using System.Linq;
+using NLog;
+using NzbDrone.Core.Datastore;
+using NzbDrone.Core.Indexers;
+using NzbDrone.Core.IndexerSearch.Definitions;
+using NzbDrone.Core.Parser.Model;
+
+namespace NzbDrone.Core.DecisionEngine.Specifications
+{
+    public class EarlyReleaseSpecification : IDecisionEngineSpecification
+    {
+        private readonly IIndexerFactory _indexerFactory;
+        private readonly Logger _logger;
+
+        public EarlyReleaseSpecification(IIndexerFactory indexerFactory, Logger logger)
+        {
+            _indexerFactory = indexerFactory;
+            _logger = logger;
+        }
+
+        public SpecificationPriority Priority => SpecificationPriority.Default;
+        public RejectionType Type => RejectionType.Permanent;
+
+        public Decision IsSatisfiedBy(RemoteAlbum subject, SearchCriteriaBase searchCriteria)
+        {
+            var releaseInfo = subject.Release;
+
+            if (releaseInfo == null || releaseInfo.IndexerId == 0)
+            {
+                return Decision.Accept();
+            }
+
+            IndexerDefinition indexer;
+            try
+            {
+                indexer = _indexerFactory.Get(subject.Release.IndexerId);
+            }
+            catch (ModelNotFoundException)
+            {
+                _logger.Debug("Indexer with id {0} does not exist, skipping early release check", subject.Release.IndexerId);
+                return Decision.Accept();
+            }
+
+            var indexerSettings = indexer.Settings as IIndexerSettings;
+
+            if (subject.Albums.Count != 1 || indexerSettings?.EarlyReleaseLimit == null)
+            {
+                return Decision.Accept();
+            }
+
+            var releaseDate = subject.Albums.First().ReleaseDate;
+
+            if (releaseDate == null) return Decision.Accept();
+
+            var isEarly = (releaseDate.Value > subject.Release.PublishDate.AddDays(indexerSettings.EarlyReleaseLimit.Value));
+
+            if (isEarly)
+            {
+                var message = $"Release published date, {subject.Release.PublishDate.ToShortDateString()}, is outside of {indexerSettings.EarlyReleaseLimit.Value} day early grab limit allowed by user";
+
+                _logger.Debug(message);
+                return Decision.Reject(message);
+            }
+
+            return Decision.Accept();
+        }
+    }
+}