You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
134 lines
4.9 KiB
134 lines
4.9 KiB
using System;
|
|
using System.Collections.Generic;
|
|
using System.Net;
|
|
using System.Text;
|
|
using Microsoft.AspNetCore.Mvc;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Core.Indexers;
|
|
using NzbDrone.Core.IndexerSearch;
|
|
using NzbDrone.Core.Parser;
|
|
using Prowlarr.Http;
|
|
using Prowlarr.Http.Extensions;
|
|
using Prowlarr.Http.REST;
|
|
|
|
namespace Prowlarr.Api.V1.Indexers
|
|
{
|
|
[V1ApiController]
|
|
public class IndexerController : ProviderControllerBase<IndexerResource, IIndexer, IndexerDefinition>
|
|
{
|
|
private IIndexerFactory _indexerFactory { get; set; }
|
|
private ISearchForNzb _nzbSearchService { get; set; }
|
|
private IDownloadMappingService _downloadMappingService { get; set; }
|
|
private IDownloadService _downloadService { get; set; }
|
|
|
|
public IndexerController(IndexerFactory indexerFactory, ISearchForNzb nzbSearchService, IDownloadMappingService downloadMappingService, IDownloadService downloadService, IndexerResourceMapper resourceMapper)
|
|
: base(indexerFactory, "indexer", resourceMapper)
|
|
{
|
|
_indexerFactory = indexerFactory;
|
|
_nzbSearchService = nzbSearchService;
|
|
_downloadMappingService = downloadMappingService;
|
|
_downloadService = downloadService;
|
|
}
|
|
|
|
protected override void Validate(IndexerDefinition definition, bool includeWarnings)
|
|
{
|
|
if (!definition.Enable)
|
|
{
|
|
return;
|
|
}
|
|
|
|
base.Validate(definition, includeWarnings);
|
|
}
|
|
|
|
[HttpGet("{id:int}/newznab")]
|
|
public IActionResult GetNewznabResponse(int id, [FromQuery] NewznabRequest request)
|
|
{
|
|
var requestType = request.t;
|
|
request.source = UserAgentParser.ParseSource(Request.Headers["User-Agent"]);
|
|
request.server = Request.GetServerUrl();
|
|
|
|
if (requestType.IsNullOrWhiteSpace())
|
|
{
|
|
throw new BadRequestException("Missing Function Parameter");
|
|
}
|
|
|
|
var indexer = _indexerFactory.Get(id);
|
|
|
|
if (indexer == null)
|
|
{
|
|
throw new NotFoundException("Indexer Not Found");
|
|
}
|
|
|
|
var indexerInstance = _indexerFactory.GetInstance(indexer);
|
|
|
|
switch (requestType)
|
|
{
|
|
case "caps":
|
|
return Content(indexerInstance.GetCapabilities().ToXml(), "application/rss+xml");
|
|
case "search":
|
|
case "tvsearch":
|
|
case "music":
|
|
case "book":
|
|
case "movie":
|
|
var results = _nzbSearchService.Search(request, new List<int> { indexer.Id }, false);
|
|
return Content(results.ToXml(indexerInstance.Protocol), "application/rss+xml");
|
|
default:
|
|
throw new BadRequestException("Function Not Available");
|
|
}
|
|
}
|
|
|
|
[HttpGet("{id:int}/download")]
|
|
public object GetDownload(int id, string link, string file)
|
|
{
|
|
var indexerDef = _indexerFactory.Get(id);
|
|
var indexer = _indexerFactory.GetInstance(indexerDef);
|
|
|
|
if (link.IsNullOrWhiteSpace() || file.IsNullOrWhiteSpace())
|
|
{
|
|
throw new BadRequestException("Invalid Prowlarr link");
|
|
}
|
|
|
|
file = WebUtility.UrlDecode(file);
|
|
|
|
if (indexer == null)
|
|
{
|
|
throw new NotFoundException("Indexer Not Found");
|
|
}
|
|
|
|
var source = UserAgentParser.ParseSource(Request.Headers["User-Agent"]);
|
|
|
|
var unprotectedlLink = _downloadMappingService.ConvertToNormalLink(link);
|
|
|
|
// If Indexer is set to download via Redirect then just redirect to the link
|
|
if (indexer.SupportsRedirect && indexerDef.Redirect)
|
|
{
|
|
_downloadService.RecordRedirect(unprotectedlLink, id, source, file);
|
|
return RedirectPermanent(unprotectedlLink);
|
|
}
|
|
|
|
var downloadBytes = Array.Empty<byte>();
|
|
downloadBytes = _downloadService.DownloadReport(unprotectedlLink, id, source, file);
|
|
|
|
// handle magnet URLs
|
|
if (downloadBytes.Length >= 7
|
|
&& downloadBytes[0] == 0x6d
|
|
&& downloadBytes[1] == 0x61
|
|
&& downloadBytes[2] == 0x67
|
|
&& downloadBytes[3] == 0x6e
|
|
&& downloadBytes[4] == 0x65
|
|
&& downloadBytes[5] == 0x74
|
|
&& downloadBytes[6] == 0x3a)
|
|
{
|
|
var magnetUrl = Encoding.UTF8.GetString(downloadBytes);
|
|
return RedirectPermanent(magnetUrl);
|
|
}
|
|
|
|
var contentType = indexer.Protocol == DownloadProtocol.Torrent ? "application/x-bittorrent" : "application/x-nzb";
|
|
var extension = indexer.Protocol == DownloadProtocol.Torrent ? "torrent" : "nzb";
|
|
var filename = $"{file}.{extension}";
|
|
|
|
return File(downloadBytes, contentType, filename);
|
|
}
|
|
}
|
|
}
|