1
0
Fork 0
mirror of https://github.com/lidarr/Lidarr synced 2025-02-25 07:12:40 +00:00

Handle ratelimit api response for newznab caps endpoint on certain newznab indexers that have caps behind the apikey

This commit is contained in:
Taloth Saldono 2020-09-24 11:46:23 +02:00 committed by Qstick
parent 72d93e54ed
commit 2d8657a77f
3 changed files with 24 additions and 12 deletions

View file

@ -301,15 +301,15 @@ protected virtual ValidationFailure TestConnection()
return new ValidationFailure(string.Empty, "Query successful, but no results were returned from your indexer. This may be an issue with the indexer or your indexer category settings."); return new ValidationFailure(string.Empty, "Query successful, but no results were returned from your indexer. This may be an issue with the indexer or your indexer category settings.");
} }
} }
catch (ApiKeyException) catch (ApiKeyException ex)
{ {
_logger.Warn("Indexer returned result for RSS URL, API Key appears to be invalid"); _logger.Warn("Indexer returned result for RSS URL, API Key appears to be invalid: " + ex.Message);
return new ValidationFailure("ApiKey", "Invalid API Key"); return new ValidationFailure("ApiKey", "Invalid API Key");
} }
catch (RequestLimitReachedException) catch (RequestLimitReachedException ex)
{ {
_logger.Warn("Request limit reached"); _logger.Warn("Request limit reached: " + ex.Message);
} }
catch (CloudFlareCaptchaException ex) catch (CloudFlareCaptchaException ex)
{ {

View file

@ -67,12 +67,15 @@ private NewznabCapabilities FetchCapabilities(NewznabSettings indexerSettings)
} }
catch (XmlException ex) catch (XmlException ex)
{ {
ex.WithData(response, 128 * 1024);
_logger.Trace("Unexpected Response content ({0} bytes): {1}", response.ResponseData.Length, response.Content);
_logger.Debug(ex, "Failed to parse newznab api capabilities for {0}", indexerSettings.BaseUrl); _logger.Debug(ex, "Failed to parse newznab api capabilities for {0}", indexerSettings.BaseUrl);
ex.WithData(response);
throw; throw;
} }
catch (Exception ex) catch (Exception ex)
{ {
ex.WithData(response, 128 * 1024);
_logger.Trace("Unexpected Response content ({0} bytes): {1}", response.ResponseData.Length, response.Content);
_logger.Error(ex, "Failed to determine newznab api capabilities for {0}, using the defaults instead till Lidarr restarts", indexerSettings.BaseUrl); _logger.Error(ex, "Failed to determine newznab api capabilities for {0}, using the defaults instead till Lidarr restarts", indexerSettings.BaseUrl);
} }
@ -87,14 +90,16 @@ private NewznabCapabilities ParseCapabilities(HttpResponse response)
if (xDoc == null) if (xDoc == null)
{ {
throw new XmlException("Invalid XML"); throw new XmlException("Invalid XML").WithData(response);
} }
NewznabRssParser.CheckError(xDoc, new IndexerResponse(new IndexerRequest(response.Request), response));
var xmlRoot = xDoc.Element("caps"); var xmlRoot = xDoc.Element("caps");
if (xmlRoot == null) if (xmlRoot == null)
{ {
throw new XmlException("Unexpected XML"); throw new XmlException("Unexpected XML").WithData(response);
} }
var xmlLimits = xmlRoot.Element("limits"); var xmlLimits = xmlRoot.Element("limits");

View file

@ -18,14 +18,13 @@ public NewznabRssParser()
UseEnclosureUrl = true; UseEnclosureUrl = true;
} }
protected override bool PreProcess(IndexerResponse indexerResponse) public static void CheckError(XDocument xdoc, IndexerResponse indexerResponse)
{ {
var xdoc = LoadXmlDocument(indexerResponse);
var error = xdoc.Descendants("error").FirstOrDefault(); var error = xdoc.Descendants("error").FirstOrDefault();
if (error == null) if (error == null)
{ {
return true; return;
} }
var code = Convert.ToInt32(error.Attribute("code").Value); var code = Convert.ToInt32(error.Attribute("code").Value);
@ -33,8 +32,7 @@ protected override bool PreProcess(IndexerResponse indexerResponse)
if (code >= 100 && code <= 199) if (code >= 100 && code <= 199)
{ {
_logger.Warn("Invalid API Key: {0}", errorMessage); throw new ApiKeyException(errorMessage);
throw new ApiKeyException("Invalid API key");
} }
if (!indexerResponse.Request.Url.FullUri.Contains("apikey=") && (errorMessage == "Missing parameter" || errorMessage.Contains("apikey"))) if (!indexerResponse.Request.Url.FullUri.Contains("apikey=") && (errorMessage == "Missing parameter" || errorMessage.Contains("apikey")))
@ -50,6 +48,15 @@ protected override bool PreProcess(IndexerResponse indexerResponse)
throw new NewznabException(indexerResponse, errorMessage); throw new NewznabException(indexerResponse, errorMessage);
} }
protected override bool PreProcess(IndexerResponse indexerResponse)
{
var xdoc = LoadXmlDocument(indexerResponse);
CheckError(xdoc, indexerResponse);
return true;
}
protected override bool PostProcess(IndexerResponse indexerResponse, List<XElement> items, List<ReleaseInfo> releases) protected override bool PostProcess(IndexerResponse indexerResponse, List<XElement> items, List<ReleaseInfo> releases)
{ {
var enclosureTypes = items.SelectMany(GetEnclosures).Select(v => v.Type).Distinct().ToArray(); var enclosureTypes = items.SelectMany(GetEnclosures).Select(v => v.Type).Distinct().ToArray();