Fixed: Correctly handle Repack Releases

2026-04-20 21:54:25 -04:00 · 2019-05-04 23:50:00 -04:00
parent 23316329ed
commit 2f1290d488
22 changed files with 673 additions and 72 deletions
@@ -0,0 +1,67 @@
+using System;
+using NLog;
+using NzbDrone.Common.Extensions;
+using NzbDrone.Core.IndexerSearch.Definitions;
+using NzbDrone.Core.MediaFiles;
+using NzbDrone.Core.Parser.Model;
+
+namespace NzbDrone.Core.DecisionEngine.Specifications
+{
+    public class RepackSpecification : IDecisionEngineSpecification
+    {
+        private readonly IMediaFileService _mediaFileService;
+        private readonly UpgradableSpecification _upgradableSpecification;
+        private readonly Logger _logger;
+
+        public RepackSpecification(IMediaFileService mediaFileService, UpgradableSpecification upgradableSpecification, Logger logger)
+        {
+            _mediaFileService = mediaFileService;
+            _upgradableSpecification = upgradableSpecification;
+            _logger = logger;
+        }
+
+        public SpecificationPriority Priority => SpecificationPriority.Database;
+        public RejectionType Type => RejectionType.Permanent;
+
+        public Decision IsSatisfiedBy(RemoteAlbum subject, SearchCriteriaBase searchCriteria)
+        {
+            if (!subject.ParsedAlbumInfo.Quality.Revision.IsRepack)
+            {
+                return Decision.Accept();
+            }
+
+            foreach (var album in subject.Albums)
+            {
+                var releaseGroup = subject.ParsedAlbumInfo.ReleaseGroup;
+                var trackFiles = _mediaFileService.GetFilesByAlbum(album.Id);
+
+                foreach (var file in trackFiles)
+                {
+                    if (_upgradableSpecification.IsRevisionUpgrade(file.Quality, subject.ParsedAlbumInfo.Quality))
+                    {
+                        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();
+        }
+    }
+}