mirror of https://github.com/Radarr/Radarr
112 lines
3.5 KiB
C#
112 lines
3.5 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Net;
|
|
using Newtonsoft.Json;
|
|
using Newtonsoft.Json.Linq;
|
|
using NzbDrone.Common.Http;
|
|
using NzbDrone.Core.Indexers.Exceptions;
|
|
using NzbDrone.Core.Parser.Model;
|
|
|
|
namespace NzbDrone.Core.Indexers.HDBits
|
|
{
|
|
public class HDBitsParser : IParseIndexerResponse
|
|
{
|
|
private readonly HDBitsSettings _settings;
|
|
|
|
public HDBitsParser(HDBitsSettings settings)
|
|
{
|
|
_settings = settings;
|
|
}
|
|
|
|
public IList<ReleaseInfo> ParseResponse(IndexerResponse indexerResponse)
|
|
{
|
|
var torrentInfos = new List<ReleaseInfo>();
|
|
|
|
if (indexerResponse.HttpResponse.StatusCode != HttpStatusCode.OK)
|
|
{
|
|
throw new IndexerException(indexerResponse,
|
|
"Unexpected response status {0} code from API request",
|
|
indexerResponse.HttpResponse.StatusCode);
|
|
}
|
|
|
|
var jsonResponse = JsonConvert.DeserializeObject<HDBitsResponse>(indexerResponse.Content);
|
|
|
|
if (jsonResponse.Status != StatusCode.Success)
|
|
{
|
|
throw new IndexerException(indexerResponse,
|
|
"HDBits API request returned status code {0}: {1}",
|
|
jsonResponse.Status,
|
|
jsonResponse.Message ?? string.Empty);
|
|
}
|
|
|
|
if (jsonResponse.Data is not JArray responseData)
|
|
{
|
|
throw new IndexerException(indexerResponse,
|
|
"Indexer API call response missing result data");
|
|
}
|
|
|
|
var queryResults = responseData.ToObject<TorrentQueryResponse[]>();
|
|
|
|
foreach (var result in queryResults)
|
|
{
|
|
var id = result.Id;
|
|
|
|
torrentInfos.Add(new HDBitsInfo
|
|
{
|
|
Guid = $"HDBits-{id}",
|
|
Title = result.Name,
|
|
Size = result.Size,
|
|
InfoHash = result.Hash,
|
|
DownloadUrl = GetDownloadUrl(id),
|
|
InfoUrl = GetInfoUrl(id),
|
|
Seeders = result.Seeders,
|
|
Peers = result.Leechers + result.Seeders,
|
|
PublishDate = result.Added.ToUniversalTime(),
|
|
ImdbId = result.ImdbInfo?.Id ?? 0,
|
|
IndexerFlags = GetIndexerFlags(result)
|
|
});
|
|
}
|
|
|
|
return torrentInfos.ToArray();
|
|
}
|
|
|
|
private static IndexerFlags GetIndexerFlags(TorrentQueryResponse item)
|
|
{
|
|
IndexerFlags flags = 0;
|
|
|
|
if (item.FreeLeech == "yes")
|
|
{
|
|
flags |= IndexerFlags.G_Freeleech;
|
|
}
|
|
|
|
if (item.TypeOrigin == 1)
|
|
{
|
|
flags |= IndexerFlags.G_Internal;
|
|
}
|
|
|
|
return flags;
|
|
}
|
|
|
|
private string GetDownloadUrl(string torrentId)
|
|
{
|
|
var url = new HttpUri(_settings.BaseUrl)
|
|
.CombinePath("/download.php")
|
|
.AddQueryParam("id", torrentId)
|
|
.AddQueryParam("passkey", _settings.ApiKey);
|
|
|
|
return url.FullUri;
|
|
}
|
|
|
|
private string GetInfoUrl(string torrentId)
|
|
{
|
|
var url = new HttpUri(_settings.BaseUrl)
|
|
.CombinePath("/details.php")
|
|
.AddQueryParam("id", torrentId);
|
|
|
|
return url.FullUri;
|
|
}
|
|
|
|
public Action<IDictionary<string, string>, DateTime?> CookiesUpdater { get; set; }
|
|
}
|
|
}
|