rewrite of indexer/episode search

NzbDrone.Core/DecisionEngine/Specifications/AcceptableSizeSpecification.cs

@@ -5,7 +5,7 @@ using NzbDrone.Core.Tv;
 
 namespace NzbDrone.Core.DecisionEngine.Specifications
 {
-    public class AcceptableSizeSpecification : IFetchableSpecification
+    public class AcceptableSizeSpecification : IDecisionEngineSpecification
     {
         private readonly IQualitySizeService _qualityTypeProvider;
         private readonly IEpisodeService _episodeService;

NzbDrone.Core/DecisionEngine/Specifications/AllowedReleaseGroupSpecification.cs

@@ -5,7 +5,7 @@ using NzbDrone.Core.Model;
 
 namespace NzbDrone.Core.DecisionEngine.Specifications
 {
-    public class AllowedReleaseGroupSpecification : IFetchableSpecification
+    public class AllowedReleaseGroupSpecification : IDecisionEngineSpecification
     {
         private readonly IConfigService _configService;
         private readonly Logger _logger;

NzbDrone.Core/DecisionEngine/Specifications/CustomStartDateSpecification.cs

@@ -4,7 +4,7 @@ using NzbDrone.Core.Model;
 
 namespace NzbDrone.Core.DecisionEngine.Specifications
 {
-    public class CustomStartDateSpecification : IFetchableSpecification
+    public class CustomStartDateSpecification : IDecisionEngineSpecification
     {
         private readonly Logger _logger;
 

NzbDrone.Core/DecisionEngine/Specifications/LanguageSpecification.cs

@@ -3,7 +3,7 @@ using NzbDrone.Core.Model;
 
 namespace NzbDrone.Core.DecisionEngine.Specifications
 {
-    public class LanguageSpecification : IFetchableSpecification
+    public class LanguageSpecification : IDecisionEngineSpecification
     {
         private readonly Logger _logger;
 

NzbDrone.Core/DecisionEngine/Specifications/MonitoredEpisodeSpecification.cs

@@ -5,7 +5,7 @@ using NzbDrone.Core.Tv;
 
 namespace NzbDrone.Core.DecisionEngine.Specifications
 {
-    public class MonitoredEpisodeSpecification : IFetchableSpecification
+    public class MonitoredEpisodeSpecification : IDecisionEngineSpecification
     {
         private readonly IEpisodeService _episodeService;
         private readonly ISeriesRepository _seriesRepository;

NzbDrone.Core/DecisionEngine/Specifications/NotInQueueSpecification.cs

@@ -4,7 +4,7 @@ using NzbDrone.Core.Model;
 
 namespace NzbDrone.Core.DecisionEngine.Specifications
 {
-    public class NotInQueueSpecification : IFetchableSpecification
+    public class NotInQueueSpecification : IDecisionEngineSpecification
     {
         private readonly IProvideDownloadClient _downloadClientProvider;
 

NzbDrone.Core/DecisionEngine/Specifications/QualityAllowedByProfileSpecification.cs

@@ -3,7 +3,7 @@ using NzbDrone.Core.Model;
 
 namespace NzbDrone.Core.DecisionEngine.Specifications
 {
-    public class QualityAllowedByProfileSpecification : IFetchableSpecification
+    public class QualityAllowedByProfileSpecification : IDecisionEngineSpecification
     {
         private readonly Logger _logger;
 

NzbDrone.Core/DecisionEngine/Specifications/RetentionSpecification.cs

@@ -4,7 +4,7 @@ using NzbDrone.Core.Model;
 
 namespace NzbDrone.Core.DecisionEngine.Specifications
 {
-    public class RetentionSpecification : IFetchableSpecification
+    public class RetentionSpecification : IDecisionEngineSpecification
     {
         private readonly IConfigService _configService;
         private readonly Logger _logger;

NzbDrone.Core/DecisionEngine/Specifications/Search/DailyEpisodeMatchSpecification.cs

@@ -0,0 +1,43 @@
+using NLog;
+using NzbDrone.Core.IndexerSearch.Definitions;
+using NzbDrone.Core.Model;
+using NzbDrone.Core.Tv;
+
+namespace NzbDrone.Core.DecisionEngine.Specifications.Search
+{
+    public class DailyEpisodeMatchSpecification : IDecisionEngineSearchSpecification
+    {
+        private readonly Logger _logger;
+        private readonly IEpisodeService _episodeService;
+
+        public DailyEpisodeMatchSpecification(Logger logger, IEpisodeService episodeService)
+        {
+            _logger = logger;
+            _episodeService = episodeService;
+        }
+
+        public string RejectionReason
+        {
+            get
+            {
+                return "Episode doesn't match";
+            }
+        }
+        public bool IsSatisfiedBy(EpisodeParseResult episodeParseResult, SearchDefinitionBase searchDefinitionBase)
+        {
+            var dailySearchSpec = searchDefinitionBase as DailyEpisodeSearchDefinition;
+
+            if (dailySearchSpec == null) return true;
+
+            var episode = _episodeService.GetEpisode(dailySearchSpec.SeriesId, dailySearchSpec.Airtime);
+
+            if (!episodeParseResult.AirDate.HasValue || episodeParseResult.AirDate.Value != episode.AirDate.Value)
+            {
+                _logger.Trace("Episode AirDate does not match searched episode number, skipping.");
+                return false;
+            }
+
+            return true;
+        }
+    }
+}

NzbDrone.Core/DecisionEngine/Specifications/Search/IDecisionEngineSearchSpecification.cs

@@ -0,0 +1,11 @@
+using NzbDrone.Core.IndexerSearch;
+using NzbDrone.Core.IndexerSearch.Definitions;
+using NzbDrone.Core.Model;
+
+namespace NzbDrone.Core.DecisionEngine.Specifications.Search
+{
+    public interface IDecisionEngineSearchSpecification : IRejectWithReason
+    {
+        bool IsSatisfiedBy(EpisodeParseResult subject, SearchDefinitionBase searchDefinitionBase);
+    }
+}

NzbDrone.Core/DecisionEngine/Specifications/Search/SeasonMatchSpecification.cs

@@ -0,0 +1,38 @@
+using NLog;
+using NzbDrone.Core.IndexerSearch.Definitions;
+using NzbDrone.Core.Model;
+
+namespace NzbDrone.Core.DecisionEngine.Specifications.Search
+{
+    public class SeasonMatchSpecification : IDecisionEngineSearchSpecification
+    {
+        private readonly Logger _logger;
+
+        public SeasonMatchSpecification(Logger logger)
+        {
+            _logger = logger;
+        }
+
+        public string RejectionReason
+        {
+            get
+            {
+                return "Episode doesn't match";
+            }
+        }
+
+        public bool IsSatisfiedBy(EpisodeParseResult episodeParseResult, SearchDefinitionBase searchDefinitionBase)
+        {
+            var singleEpisodeSpec = searchDefinitionBase as SeasonSearchDefinition;
+            if (singleEpisodeSpec == null) return true;
+
+            if (singleEpisodeSpec.SeasonNumber != episodeParseResult.SeasonNumber)
+            {
+                _logger.Trace("Season number does not match searched season number, skipping.");
+                return false;
+            }
+           
+            return true;
+        }
+    }
+}

NzbDrone.Core/DecisionEngine/Specifications/Search/SingleEpisodeMatchSpecification.cs

@@ -0,0 +1,44 @@
+using NLog;
+using NzbDrone.Core.IndexerSearch.Definitions;
+using NzbDrone.Core.Model;
+
+namespace NzbDrone.Core.DecisionEngine.Specifications.Search
+{
+    public class SingleEpisodeMatchSpecification : IDecisionEngineSearchSpecification
+    {
+        private readonly Logger _logger;
+
+        public SingleEpisodeMatchSpecification(Logger logger)
+        {
+            _logger = logger;
+        }
+
+        public string RejectionReason
+        {
+            get
+            {
+                return "Episode doesn't match";
+            }
+        }
+
+        public bool IsSatisfiedBy(EpisodeParseResult episodeParseResult, SearchDefinitionBase searchDefinitionBase)
+        {
+            var singleEpisodeSpec = searchDefinitionBase as SingleEpisodeSearchDefinition;
+            if (singleEpisodeSpec == null) return true;
+
+            if (singleEpisodeSpec.SeasonNumber != episodeParseResult.SeasonNumber)
+            {
+                _logger.Trace("Season number does not match searched season number, skipping.");
+                return false;
+            }
+
+            if (!episodeParseResult.EpisodeNumbers.Contains(singleEpisodeSpec.EpisodeNumber))
+            {
+                _logger.Trace("Episode number does not match searched episode number, skipping.");
+                return false;
+            }
+
+            return true;
+        }
+    }
+}

NzbDrone.Core/DecisionEngine/Specifications/UpgradeDiskSpecification.cs

@@ -6,7 +6,7 @@ using NzbDrone.Core.Tv;
 
 namespace NzbDrone.Core.DecisionEngine.Specifications
 {
-    public class UpgradeDiskSpecification : IFetchableSpecification
+    public class UpgradeDiskSpecification : IDecisionEngineSpecification
     {
         private readonly QualityUpgradableSpecification _qualityUpgradableSpecification;
         private readonly Logger _logger;

NzbDrone.Core/DecisionEngine/Specifications/UpgradeHistorySpecification.cs

@@ -4,7 +4,7 @@ using NzbDrone.Core.Model;
 
 namespace NzbDrone.Core.DecisionEngine.Specifications
 {
-    public class UpgradeHistorySpecification : IFetchableSpecification
+    public class UpgradeHistorySpecification : IDecisionEngineSpecification
     {
         private readonly IHistoryService _historyService;
         private readonly QualityUpgradableSpecification _qualityUpgradableSpecification;