Fixed: Refactored the Indexer architecture to support non-rss indexers.

2026-04-23 22:25:09 -04:00 · 2014-09-07 13:44:24 +02:00
parent 22c9bc402f
commit 5e62c2335f
57 changed files with 2196 additions and 1470 deletions
@@ -0,0 +1,106 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using NzbDrone.Common;
+using NzbDrone.Common.Http;
+using NzbDrone.Core.IndexerSearch.Definitions;
+
+namespace NzbDrone.Core.Indexers.Omgwtfnzbs
+{
+    public class OmgwtfnzbsRequestGenerator : IIndexerRequestGenerator
+    {
+        public String BaseUrl { get; set; }
+        public OmgwtfnzbsSettings Settings { get; set; }
+
+        public OmgwtfnzbsRequestGenerator()
+        {
+            BaseUrl = "http://rss.omgwtfnzbs.org/rss-search.php";
+        }
+
+        public virtual IList<IEnumerable<IndexerRequest>> GetRecentRequests()
+        {
+            var pageableRequests = new List<IEnumerable<IndexerRequest>>();
+
+            pageableRequests.AddIfNotNull(GetPagedRequests(null));
+
+            return pageableRequests;
+        }
+
+        public virtual IList<IEnumerable<IndexerRequest>> GetSearchRequests(SingleEpisodeSearchCriteria searchCriteria)
+        {
+            var pageableRequests = new List<IEnumerable<IndexerRequest>>();
+
+            foreach (var queryTitle in searchCriteria.QueryTitles)
+            {
+                pageableRequests.AddIfNotNull(GetPagedRequests(String.Format("{0}+S{1:00}E{2:00}",
+                    queryTitle,
+                    searchCriteria.SeasonNumber,
+                    searchCriteria.EpisodeNumber)));
+            }
+
+            return pageableRequests;
+        }
+
+        public virtual IList<IEnumerable<IndexerRequest>> GetSearchRequests(SeasonSearchCriteria searchCriteria)
+        {
+            var pageableRequests = new List<IEnumerable<IndexerRequest>>();
+
+            foreach (var queryTitle in searchCriteria.QueryTitles)
+            {
+                pageableRequests.AddIfNotNull(GetPagedRequests(String.Format("{0}+S{1:00}",
+                    queryTitle,
+                    searchCriteria.SeasonNumber)));
+            }
+
+            return pageableRequests;
+        }
+
+        public virtual IList<IEnumerable<IndexerRequest>> GetSearchRequests(DailyEpisodeSearchCriteria searchCriteria)
+        {
+            var pageableRequests = new List<IEnumerable<IndexerRequest>>();
+
+            foreach (var queryTitle in searchCriteria.QueryTitles)
+            {
+                pageableRequests.AddIfNotNull(GetPagedRequests(String.Format("{0}+{1:yyyy MM dd}",
+                    queryTitle,
+                    searchCriteria.AirDate)));
+            }
+
+            return pageableRequests;
+        }
+
+        public virtual IList<IEnumerable<IndexerRequest>> GetSearchRequests(AnimeEpisodeSearchCriteria searchCriteria)
+        {
+            return new List<IEnumerable<IndexerRequest>>();
+        }
+
+        public virtual IList<IEnumerable<IndexerRequest>> GetSearchRequests(SpecialEpisodeSearchCriteria searchCriteria)
+        {
+            var pageableRequests = new List<IEnumerable<IndexerRequest>>();
+
+            foreach (var queryTitle in searchCriteria.EpisodeQueryTitles)
+            {
+                var query = queryTitle.Replace('+', ' ');
+                query = System.Web.HttpUtility.UrlEncode(query);
+
+                pageableRequests.AddIfNotNull(GetPagedRequests(query));
+            }
+
+            return pageableRequests;
+        }
+
+        private IEnumerable<IndexerRequest> GetPagedRequests(String query)
+        {
+            var url = new StringBuilder();
+            url.AppendFormat("{0}?catid=19,20&user={1}&api={2}&eng=1", BaseUrl, Settings.Username, Settings.ApiKey);
+
+            if (query.IsNotNullOrWhiteSpace())
+            {
+                url.AppendFormat("&search={0}", query);
+            }
+
+            yield return new IndexerRequest(url.ToString(), HttpAccept.Rss);
+        }
+    }
+}