Restrict nzbs based on release group, server side

New: Only grab NZBs if release group is wanted (configurable)

NzbDrone.Core/Model/ReportRejectionType.cs

@@ -17,5 +17,6 @@ namespace NzbDrone.Core.Model
         DownloadClientFailure = 10,
         Skipped = 11,
         Failure = 12,
+        ReleaseGroupNotWanted = 13
     }
 }

NzbDrone.Core/NzbDrone.Core.csproj

@@ -289,6 +289,7 @@
     <Compile Include="Model\Xbmc\ErrorResult.cs" />
     <Compile Include="Model\Xbmc\IconType.cs" />
     <Compile Include="Providers\BannerProvider.cs" />
+    <Compile Include="Providers\DecisionEngine\AllowedReleaseGroupSpecification.cs" />
     <Compile Include="Providers\Indexer\NzbClub.cs" />
     <Compile Include="Providers\Indexer\NzbIndex.cs" />
     <Compile Include="Providers\Indexer\FileSharingTalk.cs" />

NzbDrone.Core/Providers/Core/ConfigProvider.cs

@@ -514,6 +514,12 @@ namespace NzbDrone.Core.Providers.Core
             set { SetValue("MetadataUseBanners", value); }
         }
 
+        public virtual string AllowedReleaseGroups
+        {
+            get { return GetValue("AllowedReleaseGroups"); }
+            set { SetValue("AllowedReleaseGroups", value); }
+        }
+
         private string GetValue(string key)
         {
             return GetValue(key, String.Empty);

NzbDrone.Core/Providers/DecisionEngine/AllowedDownloadSpecification.cs

@@ -13,18 +13,21 @@ namespace NzbDrone.Core.Providers.DecisionEngine
         private readonly AcceptableSizeSpecification _acceptableSizeSpecification;
         private readonly AlreadyInQueueSpecification _alreadyInQueueSpecification;
         private readonly RetentionSpecification _retentionSpecification;
+        private readonly AllowedReleaseGroupSpecification _allowedReleaseGroupSpecification;
         private static readonly Logger logger = LogManager.GetCurrentClassLogger();
 
         [Inject]
         public AllowedDownloadSpecification(QualityAllowedByProfileSpecification qualityAllowedByProfileSpecification,
             UpgradeDiskSpecification upgradeDiskSpecification, AcceptableSizeSpecification acceptableSizeSpecification,
-            AlreadyInQueueSpecification alreadyInQueueSpecification, RetentionSpecification retentionSpecification)
+            AlreadyInQueueSpecification alreadyInQueueSpecification, RetentionSpecification retentionSpecification,
+            AllowedReleaseGroupSpecification allowedReleaseGroupSpecification)
         {
             _qualityAllowedByProfileSpecification = qualityAllowedByProfileSpecification;
             _upgradeDiskSpecification = upgradeDiskSpecification;
             _acceptableSizeSpecification = acceptableSizeSpecification;
             _alreadyInQueueSpecification = alreadyInQueueSpecification;
             _retentionSpecification = retentionSpecification;
+            _allowedReleaseGroupSpecification = allowedReleaseGroupSpecification;
         }
 
         public AllowedDownloadSpecification()
@@ -37,8 +40,9 @@ namespace NzbDrone.Core.Providers.DecisionEngine
             if (!_upgradeDiskSpecification.IsSatisfiedBy(subject)) return ReportRejectionType.ExistingQualityIsEqualOrBetter;
             if (!_retentionSpecification.IsSatisfiedBy(subject)) return ReportRejectionType.Retention;
             if (!_acceptableSizeSpecification.IsSatisfiedBy(subject)) return ReportRejectionType.Size;
+            if (!_allowedReleaseGroupSpecification.IsSatisfiedBy(subject)) return ReportRejectionType.ReleaseGroupNotWanted;
             if (_alreadyInQueueSpecification.IsSatisfiedBy(subject)) return ReportRejectionType.AlreadyInQueue;
-
+            
             logger.Debug("Episode {0} is needed", subject);
             return ReportRejectionType.None;
         }

NzbDrone.Core/Providers/DecisionEngine/AllowedReleaseGroupSpecification.cs

@@ -0,0 +1,48 @@
+using System;
+using System.Linq;
+using NLog;
+using Ninject;
+using NzbDrone.Core.Model;
+using NzbDrone.Core.Providers.Core;
+
+namespace NzbDrone.Core.Providers.DecisionEngine
+{
+    public class AllowedReleaseGroupSpecification
+    {
+        private readonly ConfigProvider _configProvider;
+        private static readonly Logger logger = LogManager.GetCurrentClassLogger();
+
+        [Inject]
+        public AllowedReleaseGroupSpecification(ConfigProvider configProvider)
+        {
+            _configProvider = configProvider;
+        }
+
+        public AllowedReleaseGroupSpecification()
+        {
+            
+        }
+
+        public virtual bool IsSatisfiedBy(EpisodeParseResult subject)
+        {
+            logger.Trace("Beginning release group check for: {0}", subject);
+
+            var allowed = _configProvider.AllowedReleaseGroups;
+
+            if (string.IsNullOrWhiteSpace(allowed))
+                return true;
+
+            foreach(var group in allowed.Trim(',', ' ').Split(','))
+            {
+                if (subject.ReleaseGroup.Equals(group.Trim(' '), StringComparison.CurrentCultureIgnoreCase))
+                {
+                    logger.Trace("Item: {0}'s release group is wanted: {1}", subject, subject.ReleaseGroup);
+                    return true;
+                }
+            }
+            
+            logger.Trace("Item: {0}'s release group is not wanted: {1}", subject, subject.ReleaseGroup);
+            return false;
+        }    
+    }
+}