rewrite of indexer/episode search

2025-08-14 02:37:08 -07:00 · 2013-04-07 00:30:37 -07:00 · 2013-04-07 00:30:37 -07:00 · a6a4932b44
commit a6a4932b44
parent 9ae21cf7a1
114 changed files with 2045 additions and 5577 deletions
--- a/NzbDrone.Core/IndexerSearch/NzbSearchService.cs
+++ b/NzbDrone.Core/IndexerSearch/NzbSearchService.cs
@ -0,0 +1,136 @@
+using System;
+using System.Collections.Generic;
+using System.Threading.Tasks;
+using NLog;
+using NzbDrone.Core.DataAugmentation.Scene;
+using NzbDrone.Core.DecisionEngine;
+using NzbDrone.Core.IndexerSearch.Definitions;
+using NzbDrone.Core.Indexers;
+using NzbDrone.Core.Model;
+using NzbDrone.Core.Tv;
+using System.Linq;
+
+namespace NzbDrone.Core.IndexerSearch
+{
+    public interface ISearchForNzb
+    {
+        List<DownloadDecision> SearchSingle(int seriesId, int seasonNumber, int episodeNumber);
+        List<DownloadDecision> SearchDaily(int seriesId, DateTime airDate);
+        List<DownloadDecision> SearchSeason(int seriesId, int seasonNumber);
+    }
+
+    public class NzbSearchService : ISearchForNzb
+    {
+        private readonly IIndexerService _indexerService;
+        private readonly IFetchFeedFromIndexers _feedFetcher;
+        private readonly ISceneMappingService _sceneMapping;
+        private readonly ISeriesService _seriesService;
+        private readonly IEpisodeService _episodeService;
+        private readonly IMakeDownloadDecision _makeDownloadDecision;
+        private readonly Logger _logger;
+
+        public NzbSearchService(IIndexerService indexerService, IFetchFeedFromIndexers feedFetcher, ISceneMappingService sceneMapping, ISeriesService seriesService, IEpisodeService episodeService, IMakeDownloadDecision makeDownloadDecision, Logger logger)
+        {
+            _indexerService = indexerService;
+            _feedFetcher = feedFetcher;
+            _sceneMapping = sceneMapping;
+            _seriesService = seriesService;
+            _episodeService = episodeService;
+            _makeDownloadDecision = makeDownloadDecision;
+            _logger = logger;
+        }
+
+        public List<DownloadDecision> SearchSingle(int seriesId, int seasonNumber, int episodeNumber)
+        {
+            var searchSpec = Get<SingleEpisodeSearchDefinition>(seriesId, seasonNumber);
+
+            if (_seriesService.GetSeries(seriesId).UseSceneNumbering)
+            {
+                var episode = _episodeService.GetEpisode(seriesId, seasonNumber, episodeNumber);
+                searchSpec.EpisodeNumber = episode.SceneEpisodeNumber;
+                searchSpec.SeasonNumber = episode.SceneSeasonNumber;
+            }
+            else
+            {
+                searchSpec.EpisodeNumber = episodeNumber;
+                searchSpec.SeasonNumber = seasonNumber;
+            }
+
+            return Dispatch(indexer => _feedFetcher.Fetch(indexer, searchSpec), searchSpec);
+        }
+
+        public List<DownloadDecision> SearchDaily(int seriesId, DateTime airDate)
+        {
+            var searchSpec = Get<DailyEpisodeSearchDefinition>(seriesId);
+            searchSpec.Airtime = airDate;
+
+            return Dispatch(indexer => _feedFetcher.Fetch(indexer, searchSpec), searchSpec);
+        }
+
+        public List<DownloadDecision> SearchSeason(int seriesId, int seasonNumber)
+        {
+            var searchSpec = Get<SeasonSearchDefinition>(seriesId, seasonNumber);
+            searchSpec.SeasonNumber = seasonNumber;
+
+            return Dispatch(indexer => _feedFetcher.Fetch(indexer, searchSpec), searchSpec);
+        }
+
+        private List<DownloadDecision> PartialSeasonSearch(SeasonSearchDefinition search)
+        {
+            var episodesNumbers = _episodeService.GetEpisodesBySeason(search.SeriesId, search.SeasonNumber).Select(c => c.EpisodeNumber);
+            var prefixes = episodesNumbers
+                .Select(i => i / 10)
+                .Distinct()
+                .Select(prefix => new PartialSeasonSearchDefinition(search, prefix));
+
+            var result = new List<DownloadDecision>();
+
+            foreach (var partialSeasonSearchSpec in prefixes)
+            {
+                var spec = partialSeasonSearchSpec;
+                result.AddRange(Dispatch(indexer => _feedFetcher.Fetch(indexer, spec), partialSeasonSearchSpec));
+            }
+
+
+            return result;
+        }
+
+        private TSpec Get<TSpec>(int seriesId, int seasonNumber = -1) where TSpec : SearchDefinitionBase, new()
+        {
+            var spec = new TSpec();
+
+            spec.SeriesId = seriesId;
+            spec.SceneTitle = _sceneMapping.GetSceneName(seriesId, seasonNumber);
+
+            return spec;
+        }
+
+        private List<DownloadDecision> Dispatch(Func<IIndexerBase, IEnumerable<EpisodeParseResult>> searchAction, SearchDefinitionBase definitionBase)
+        {
+            var indexers = _indexerService.GetAvailableIndexers();
+            var parseResults = new List<EpisodeParseResult>();
+
+            Parallel.ForEach(indexers, indexer =>
+            {
+                try
+                {
+                    var indexerReports = searchAction(indexer);
+                    lock (indexer)
+                    {
+                        parseResults.AddRange(indexerReports);
+                    }
+                }
+                catch (Exception e)
+                {
+                    _logger.ErrorException(String.Format("An error has occurred while searching for {0} from: {1}", definitionBase, indexer.Name), e);
+                }
+            });
+
+            _logger.Debug("Total of {0} reports were found for {1} in {2} indexers", parseResults.Count, definitionBase, indexers.Count);
+
+            return _makeDownloadDecision.GetSearchDecision(parseResults, definitionBase).ToList();
+        }
+
+
+    }
+}