Moved source code under src folder - massive change

2026-04-27 22:56:45 -04:00 · 2013-10-02 18:01:32 -07:00
parent 2fc8123d6b
commit 5bf0e197ec
1499 changed files with 1054 additions and 1444 deletions
@@ -0,0 +1,172 @@
+using System;
+using System.Collections.Generic;
+using System.Globalization;
+using System.Threading.Tasks;
+using NLog;
+using NzbDrone.Core.DataAugmentation.Scene;
+using NzbDrone.Core.DecisionEngine;
+using NzbDrone.Core.DecisionEngine.Specifications;
+using NzbDrone.Core.IndexerSearch.Definitions;
+using NzbDrone.Core.Indexers;
+using NzbDrone.Core.Instrumentation;
+using NzbDrone.Core.Parser.Model;
+using NzbDrone.Core.Tv;
+using System.Linq;
+using NzbDrone.Common.TPL;
+
+namespace NzbDrone.Core.IndexerSearch
+{
+    public interface ISearchForNzb
+    {
+        List<DownloadDecision> EpisodeSearch(int episodeId);
+        List<DownloadDecision> SeasonSearch(int seriesId, int seasonNumber);
+    }
+
+    public class NzbSearchService : ISearchForNzb
+    {
+        private readonly IIndexerFactory _indexerFactory;
+        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(IIndexerFactory indexerFactory,
+                                IFetchFeedFromIndexers feedFetcher,
+                                ISceneMappingService sceneMapping,
+                                ISeriesService seriesService,
+                                IEpisodeService episodeService,
+                                IMakeDownloadDecision makeDownloadDecision,
+                                Logger logger)
+        {
+            _indexerFactory = indexerFactory;
+            _feedFetcher = feedFetcher;
+            _sceneMapping = sceneMapping;
+            _seriesService = seriesService;
+            _episodeService = episodeService;
+            _makeDownloadDecision = makeDownloadDecision;
+            _logger = logger;
+        }
+
+        public List<DownloadDecision> EpisodeSearch(int episodeId)
+        {
+            var episode = _episodeService.GetEpisode(episodeId);
+            var series = _seriesService.GetSeries(episode.SeriesId);
+
+            if (series.SeriesType == SeriesTypes.Daily)
+            {
+                if (string.IsNullOrWhiteSpace(episode.AirDate))
+                {
+                    throw new InvalidOperationException("Daily episode is missing AirDate. Try to refresh series info.");
+                }
+
+                return SearchDaily(series, episode);
+            }
+
+            return SearchSingle(series, episode);
+        }
+
+        private List<DownloadDecision> SearchSingle(Series series, Episode episode)
+        {
+            var searchSpec = Get<SingleEpisodeSearchCriteria>(series, new List<Episode>{episode});
+
+            if (series.UseSceneNumbering)
+            {
+                if (episode.SceneSeasonNumber > 0 && episode.SceneEpisodeNumber > 0)
+                {
+                    searchSpec.EpisodeNumber = episode.SceneEpisodeNumber;
+                    searchSpec.SeasonNumber = episode.SceneSeasonNumber;
+                }
+
+                else
+                {
+                    searchSpec.EpisodeNumber = episode.EpisodeNumber;
+                    searchSpec.SeasonNumber = episode.SeasonNumber;
+                }
+            }
+            else
+            {
+                searchSpec.EpisodeNumber = episode.EpisodeNumber;
+                searchSpec.SeasonNumber = episode.SeasonNumber;
+            }
+
+            return Dispatch(indexer => _feedFetcher.Fetch(indexer, searchSpec), searchSpec);
+        }
+
+        private List<DownloadDecision> SearchDaily(Series series, Episode episode)
+        {
+            var airDate = DateTime.ParseExact(episode.AirDate, Episode.AIR_DATE_FORMAT, CultureInfo.InvariantCulture);
+            var searchSpec = Get<DailyEpisodeSearchCriteria>(series, new List<Episode>{ episode });
+            searchSpec.Airtime = airDate;
+
+            return Dispatch(indexer => _feedFetcher.Fetch(indexer, searchSpec), searchSpec);
+        }
+
+        public List<DownloadDecision> SeasonSearch(int seriesId, int seasonNumber)
+        {
+            var series = _seriesService.GetSeries(seriesId);
+            var episodes = _episodeService.GetEpisodesBySeason(seriesId, seasonNumber);
+
+            var searchSpec = Get<SeasonSearchCriteria>(series, episodes);
+            searchSpec.SeasonNumber = seasonNumber;
+
+            return Dispatch(indexer => _feedFetcher.Fetch(indexer, searchSpec), searchSpec);
+        }
+
+        private TSpec Get<TSpec>(Series series, List<Episode> episodes) where TSpec : SearchCriteriaBase, new()
+        {
+            var spec = new TSpec();
+
+            spec.Series = series;
+            spec.SceneTitle = _sceneMapping.GetSceneName(series.TvdbId);
+            spec.Episodes = episodes;
+
+            if (string.IsNullOrWhiteSpace(spec.SceneTitle))
+            {
+                spec.SceneTitle = series.Title;
+            }
+
+            return spec;
+        }
+
+        private List<DownloadDecision> Dispatch(Func<IIndexer, IEnumerable<ReleaseInfo>> searchAction, SearchCriteriaBase criteriaBase)
+        {
+            var indexers = _indexerFactory.GetAvailableProviders().ToList();
+            var reports = new List<ReleaseInfo>();
+
+            _logger.ProgressInfo("Searching {0} indexers for {1}", indexers.Count, criteriaBase);
+
+            var taskList = new List<Task>();
+            var taskFactory = new TaskFactory(TaskCreationOptions.LongRunning, TaskContinuationOptions.None);
+
+            foreach (var indexer in indexers)
+            {
+                var indexerLocal = indexer;
+
+                taskList.Add(taskFactory.StartNew(() =>
+                {
+                    try
+                    {
+                        var indexerReports = searchAction(indexerLocal);
+
+                        lock (reports)
+                        {
+                            reports.AddRange(indexerReports);
+                        }
+                    }
+                    catch (Exception e)
+                    {
+                        _logger.ErrorException("Error while searching for " + criteriaBase, e);
+                    }
+                }).LogExceptions());
+            }
+
+            Task.WaitAll(taskList.ToArray());
+
+            _logger.Debug("Total of {0} reports were found for {1} from {2} indexers", reports.Count, criteriaBase, indexers.Count);
+
+            return _makeDownloadDecision.GetSearchDecision(reports, criteriaBase).ToList();
+        }
+    }
+}