Fixed: Correctly handle Repacks, restrict to same group

2026-04-21 22:05:43 -04:00 · 2019-07-12 23:56:58 -04:00
parent 7698ae00dd
commit 3bf5476922
8 changed files with 266 additions and 17 deletions
@@ -0,0 +1,65 @@
+using System;
+using System.Linq;
+using NLog;
+using NzbDrone.Common.Extensions;
+using NzbDrone.Core.IndexerSearch.Definitions;
+using NzbDrone.Core.Parser.Model;
+
+namespace NzbDrone.Core.DecisionEngine.Specifications
+{
+    public class RepackSpecification : IDecisionEngineSpecification
+    {
+        private readonly UpgradableSpecification _upgradableSpecification;
+        private readonly Logger _logger;
+
+        public RepackSpecification(UpgradableSpecification upgradableSpecification, Logger logger)
+        {
+            _upgradableSpecification = upgradableSpecification;
+            _logger = logger;
+        }
+
+        public SpecificationPriority Priority => SpecificationPriority.Database;
+        public RejectionType Type => RejectionType.Permanent;
+
+        public Decision IsSatisfiedBy(RemoteMovie subject, SearchCriteriaBase searchCriteria)
+        {
+            if (!subject.ParsedMovieInfo.Quality.Revision.IsRepack)
+            {
+                return Decision.Accept();
+            }
+
+            if (subject.Movie.MovieFileId != 0)
+            {
+                var file = subject.Movie.MovieFile;
+
+                if (_upgradableSpecification.IsRevisionUpgrade(file.Quality, subject.ParsedMovieInfo.Quality))
+                {
+                    var releaseGroup = subject.ParsedMovieInfo.ReleaseGroup;
+                    var fileReleaseGroup = file.ReleaseGroup;
+
+                    if (fileReleaseGroup.IsNullOrWhiteSpace())
+                    {
+                        return Decision.Reject("Unable to determine release group for the existing file");
+                    }
+
+                    if (releaseGroup.IsNullOrWhiteSpace())
+                    {
+                        return Decision.Reject("Unable to determine release group for this release");
+                    }
+
+                    if (!fileReleaseGroup.Equals(releaseGroup, StringComparison.InvariantCultureIgnoreCase))
+                    {
+                        _logger.Debug(
+                            "Release is a repack for a different release group. Release Group: {0}. File release group: {0}",
+                            releaseGroup, fileReleaseGroup);
+                        return Decision.Reject(
+                            "Release is a repack for a different release group. Release Group: {0}. File release group: {0}",
+                            releaseGroup, fileReleaseGroup);
+                    }
+                }
+            }
+
+            return Decision.Accept();
+        }
+    }
+}