tedkulp/Prowlarr
1
0
Fork 0
mirror of https://github.com/Prowlarr/Prowlarr.git synced 2026-04-25 22:59:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Return 429 right away after release search

Browse Source
This commit is contained in:
Bogdan
2023-11-20 02:37:13 +02:00
parent cad42fd005
commit 257c9ab248
2 changed files with 25 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/Prowlarr.Api.V1/Indexers/NewznabController.cs
+18 -5
View File
@@ -146,14 +146,13 @@ namespace NzbDrone.Api.V1.Indexers
var indexer = _indexerFactory.GetInstance(indexerDef);
var blockedIndexerStatus = GetBlockedIndexerStatus(indexer);
var blockedIndexerStatusPre = GetBlockedIndexerStatus(indexer);
if (blockedIndexerStatus?.DisabledTill != null)
if (blockedIndexerStatusPre?.DisabledTill != null)
{
var retryAfterDisabledTill = Convert.ToInt32(blockedIndexerStatus.DisabledTill.Value.ToLocalTime().Subtract(DateTime.Now).TotalSeconds);
AddRetryAfterHeader(retryAfterDisabledTill);
AddRetryAfterHeader(CalculateRetryAfterDisabledTill(blockedIndexerStatusPre.DisabledTill.Value));
return CreateResponse(CreateErrorXML(429, $"Indexer is disabled till {blockedIndexerStatus.DisabledTill.Value.ToLocalTime()} due to recent failures."), statusCode: StatusCodes.Status429TooManyRequests);
return CreateResponse(CreateErrorXML(429, $"Indexer is disabled till {blockedIndexerStatusPre.DisabledTill.Value.ToLocalTime()} due to recent failures."), statusCode: StatusCodes.Status429TooManyRequests);
}
// TODO Optimize this so it's not called here and in ReleaseSearchService (for manual search)
@@ -180,6 +179,15 @@ namespace NzbDrone.Api.V1.Indexers
case "movie":
var results = await _releaseSearchService.Search(request, new List<int> { indexerDef.Id }, false);
var blockedIndexerStatusPost = GetBlockedIndexerStatus(indexer);
if (blockedIndexerStatusPost?.DisabledTill != null)
{
AddRetryAfterHeader(CalculateRetryAfterDisabledTill(blockedIndexerStatusPost.DisabledTill.Value));
return CreateResponse(CreateErrorXML(429, $"Indexer is disabled till {blockedIndexerStatusPost.DisabledTill.Value.ToLocalTime()} due to recent failures."), statusCode: StatusCodes.Status429TooManyRequests);
}
foreach (var result in results.Releases)
{
result.DownloadUrl = result.DownloadUrl.IsNotNullOrWhiteSpace() ? _downloadMappingService.ConvertToProxyLink(new Uri(result.DownloadUrl), request.server, indexerDef.Id, result.Title).AbsoluteUri : null;
@@ -336,5 +344,10 @@ namespace NzbDrone.Api.V1.Indexers
HttpContext.Response.Headers.Add("Retry-After", $"{retryAfterSeconds}");
}
}
private static int CalculateRetryAfterDisabledTill(DateTime disabledTill)
{
return Convert.ToInt32(disabledTill.ToLocalTime().Subtract(DateTime.Now).TotalSeconds);
}
}
}