using System; using System.Collections.Generic; using System.Linq; using System.Net; using FluentValidation.Results; using NLog; using NzbDrone.Common; using NzbDrone.Common.Http; using NzbDrone.Core.Configuration; using NzbDrone.Core.Indexers.Exceptions; using NzbDrone.Core.IndexerSearch.Definitions; using NzbDrone.Core.Parser; using NzbDrone.Core.Parser.Model; using NzbDrone.Core.ThingiProvider; namespace NzbDrone.Core.Indexers { public abstract class HttpIndexerBase : IndexerBase where TSettings : IProviderConfig, new() { private const Int32 MaxNumResultsPerQuery = 1000; private readonly IHttpClient _httpClient; public override bool SupportsRss { get { return true; } } public override bool SupportsSearch { get { return true; } } public bool SupportsPaging { get { return PageSize > 0; } } public virtual Int32 PageSize { get { return 0; } } public abstract IIndexerRequestGenerator GetRequestGenerator(); public abstract IParseIndexerResponse GetParser(); public HttpIndexerBase(IHttpClient httpClient, IConfigService configService, IParsingService parsingService, Logger logger) : base(configService, parsingService, logger) { _httpClient = httpClient; } public override IList FetchRecent() { if (!SupportsRss) { return new List(); } var generator = GetRequestGenerator(); return FetchReleases(generator.GetRecentRequests()); } public override IList Fetch(SingleEpisodeSearchCriteria searchCriteria) { if (!SupportsSearch) { return new List(); } var generator = GetRequestGenerator(); return FetchReleases(generator.GetSearchRequests(searchCriteria)); } public override IList Fetch(SeasonSearchCriteria searchCriteria) { if (!SupportsSearch) { return new List(); } var generator = GetRequestGenerator(); return FetchReleases(generator.GetSearchRequests(searchCriteria)); } public override IList Fetch(DailyEpisodeSearchCriteria searchCriteria) { if (!SupportsSearch) { return new List(); } var generator = GetRequestGenerator(); return FetchReleases(generator.GetSearchRequests(searchCriteria)); } public override IList Fetch(AnimeEpisodeSearchCriteria searchCriteria) { if (!SupportsSearch) { return new List(); } var generator = GetRequestGenerator(); return FetchReleases(generator.GetSearchRequests(searchCriteria)); } public override IList Fetch(SpecialEpisodeSearchCriteria searchCriteria) { if (!SupportsSearch) { return new List(); } var generator = GetRequestGenerator(); return FetchReleases(generator.GetSearchRequests(searchCriteria)); } protected virtual IList FetchReleases(IList> pageableRequests) { var releases = new List(); var url = String.Empty; var parser = GetParser(); try { foreach (var pageableRequest in pageableRequests) { var pagedReleases = new List(); foreach (var request in pageableRequest) { url = request.Url.ToString(); var page = FetchPage(request, parser); pagedReleases.AddRange(page); if (!IsFullPage(page) || pagedReleases.Count >= MaxNumResultsPerQuery) { break; } } releases.AddRange(pagedReleases); } } catch (WebException webException) { if (webException.Message.Contains("502") || webException.Message.Contains("503") || webException.Message.Contains("timed out")) { _logger.Warn("{0} server is currently unavailable. {1} {2}", this, url, webException.Message); } else { _logger.Warn("{0} {1} {2}", this, url, webException.Message); } } catch (HttpException httpException) { if ((int)httpException.Response.StatusCode == 429) { _logger.Warn("API Request Limit reached for {0}", this); } _logger.Warn("{0} {1}", this, httpException.Message); } catch (RequestLimitReachedException) { // TODO: Backoff for x period. _logger.Warn("API Request Limit reached for {0}", this); } catch (ApiKeyException) { _logger.Warn("Invalid API Key for {0} {1}", this, url); } catch (Exception feedEx) { feedEx.Data.Add("FeedUrl", url); _logger.ErrorException("An error occurred while processing feed. " + url, feedEx); } return CleanupReleases(releases); } protected virtual Boolean IsFullPage(IList page) { return PageSize != 0 && page.Count >= PageSize; } protected virtual IList FetchPage(IndexerRequest request, IParseIndexerResponse parser) { _logger.Debug("Downloading Feed " + request.Url); var response = new IndexerResponse(request, _httpClient.Execute(request.HttpRequest)); return parser.ParseResponse(response).ToList(); } protected override void Test(List failures) { failures.AddIfNotNull(TestConnection()); } protected virtual ValidationFailure TestConnection() { try { var parser = GetParser(); var generator = GetRequestGenerator(); var releases = FetchPage(generator.GetRecentRequests().First().First(), parser); if (releases.Empty()) { return new ValidationFailure("Url", "No results were returned from your indexer, please check your settings."); } } catch (ApiKeyException) { _logger.Warn("Indexer returned result for RSS URL, API Key appears to be invalid"); return new ValidationFailure("ApiKey", "Invalid API Key"); } catch (RequestLimitReachedException) { _logger.Warn("Request limit reached"); } catch (Exception ex) { _logger.WarnException("Unable to connect to indexer: " + ex.Message, ex); return new ValidationFailure("Url", "Unable to connect to indexer, check the log for more details"); } return null; } } }