mirror of https://github.com/Jackett/Jackett
420 lines
19 KiB
C#
420 lines
19 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Diagnostics.CodeAnalysis;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using Jackett.Common.Extensions;
|
|
using Jackett.Common.Models;
|
|
using Jackett.Common.Models.IndexerConfig.Bespoke;
|
|
using Jackett.Common.Services.Interfaces;
|
|
using Jackett.Common.Utils;
|
|
using Jackett.Common.Utils.Clients;
|
|
using Newtonsoft.Json;
|
|
using Newtonsoft.Json.Linq;
|
|
using NLog;
|
|
|
|
namespace Jackett.Common.Indexers
|
|
{
|
|
[ExcludeFromCodeCoverage]
|
|
public class BeyondHDAPI : IndexerBase
|
|
{
|
|
public override string Id => "beyond-hd-api";
|
|
public override string Name => "Beyond-HD (API)";
|
|
public override string Description => "Without BeyondHD, your HDTV is just a TV";
|
|
public override string SiteLink { get; protected set; } = "https://beyond-hd.me/";
|
|
public override string Language => "en-US";
|
|
public override string Type => "private";
|
|
|
|
public override bool SupportsPagination => true;
|
|
|
|
public override TorznabCapabilities TorznabCaps => SetCapabilities();
|
|
|
|
private readonly string APIBASE = "https://beyond-hd.me/api/torrents/";
|
|
|
|
private new ConfigurationDataBeyondHDApi configData
|
|
{
|
|
get => (ConfigurationDataBeyondHDApi)base.configData;
|
|
set => base.configData = value;
|
|
}
|
|
|
|
public BeyondHDAPI(IIndexerConfigurationService configService, WebClient wc, Logger l,
|
|
IProtectionService ps, ICacheService cs)
|
|
: base(configService: configService,
|
|
client: wc,
|
|
logger: l,
|
|
p: ps,
|
|
cacheService: cs,
|
|
configData: new ConfigurationDataBeyondHDApi("Find the API and RSS keys under your security settings (your profile picture -> my security)"))
|
|
{
|
|
}
|
|
|
|
private TorznabCapabilities SetCapabilities()
|
|
{
|
|
var caps = new TorznabCapabilities
|
|
{
|
|
LimitsDefault = 100,
|
|
LimitsMax = 100,
|
|
TvSearchParams = new List<TvSearchParam>
|
|
{
|
|
TvSearchParam.Q, TvSearchParam.Season, TvSearchParam.Ep, TvSearchParam.ImdbId, TvSearchParam.TmdbId
|
|
},
|
|
MovieSearchParams = new List<MovieSearchParam>
|
|
{
|
|
MovieSearchParam.Q, MovieSearchParam.ImdbId, MovieSearchParam.TmdbId
|
|
}
|
|
};
|
|
|
|
caps.Categories.AddCategoryMapping(1, TorznabCatType.Movies, "Movies");
|
|
caps.Categories.AddCategoryMapping(2, TorznabCatType.TV, "TV");
|
|
|
|
return caps;
|
|
}
|
|
protected virtual int ApiKeyLength => 32;
|
|
protected virtual int RSSKeyLength => 32;
|
|
public override async Task<IndexerConfigurationStatus> ApplyConfiguration(JToken configJson)
|
|
{
|
|
LoadValuesFromJson(configJson);
|
|
|
|
if (configData.ApiKey.Value.IsNullOrWhiteSpace())
|
|
{
|
|
throw new Exception("Missing API Key.");
|
|
}
|
|
if (configData.ApiKey.Value.Length != ApiKeyLength)
|
|
{
|
|
throw new Exception($"Invalid API Key configured: expected length: {ApiKeyLength}, got {configData.ApiKey.Value.Length}");
|
|
}
|
|
if (configData.RSSKey.Value.IsNullOrWhiteSpace())
|
|
{
|
|
throw new Exception("Missing RSS Key.");
|
|
}
|
|
if (configData.RSSKey.Value.Length != RSSKeyLength)
|
|
{
|
|
throw new Exception($"Invalid RSS Key configured: expected length: {RSSKeyLength}, got {configData.RSSKey.Value.Length}");
|
|
}
|
|
|
|
IsConfigured = false;
|
|
try
|
|
{
|
|
var results = await PerformQuery(new TorznabQuery());
|
|
|
|
if (!results.Any())
|
|
{
|
|
throw new Exception("Testing returned no results!");
|
|
}
|
|
|
|
IsConfigured = true;
|
|
SaveConfig();
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
throw new ExceptionWithConfigData(e.Message, configData);
|
|
}
|
|
|
|
return IndexerConfigurationStatus.Completed;
|
|
}
|
|
|
|
protected override async Task<IEnumerable<ReleaseInfo>> PerformQuery(TorznabQuery query)
|
|
{
|
|
var apiKey = configData.ApiKey.Value;
|
|
var apiUrl = $"{APIBASE}{apiKey}";
|
|
|
|
var postData = new Dictionary<string, object>
|
|
{
|
|
{ BHDParams.action, "search" },
|
|
{ BHDParams.rsskey, configData.RSSKey.Value },
|
|
{ BHDParams.search, query.GetQueryString() },
|
|
};
|
|
|
|
if (configData.FilterFreeleech.Value)
|
|
{
|
|
postData.Add(BHDParams.freeleech, 1);
|
|
}
|
|
|
|
if (configData.FilterLimited.Value)
|
|
{
|
|
postData.Add(BHDParams.limited, 1);
|
|
}
|
|
|
|
if (configData.FilterRefund.Value)
|
|
{
|
|
postData.Add(BHDParams.refund, 1);
|
|
}
|
|
|
|
if (configData.FilterRewind.Value)
|
|
{
|
|
postData.Add(BHDParams.rewind, 1);
|
|
}
|
|
|
|
if (configData.SearchTypes.Values.Any())
|
|
{
|
|
postData.Add(BHDParams.types, configData.SearchTypes.Values.ToArray());
|
|
}
|
|
|
|
var categories = MapTorznabCapsToTrackers(query);
|
|
|
|
if (categories.Any())
|
|
{
|
|
postData.Add(BHDParams.categories, categories.Select(int.Parse).ToArray());
|
|
}
|
|
|
|
if (query.IsImdbQuery)
|
|
{
|
|
postData.Add(BHDParams.imdb_id, query.ImdbIDShort);
|
|
}
|
|
else if (query.IsTmdbQuery)
|
|
{
|
|
postData.Add(BHDParams.tmdb_id, query.TmdbID.ToString());
|
|
}
|
|
|
|
if (query.Limit > 0 && query.Offset > 0)
|
|
{
|
|
var page = (query.Offset / query.Limit) + 1;
|
|
postData.Add("page", page.ToString());
|
|
}
|
|
|
|
var bhdResponse = await GetBHDResponse(apiUrl, postData);
|
|
var releaseInfos = bhdResponse.results.Select(MapToReleaseInfo);
|
|
|
|
return releaseInfos;
|
|
}
|
|
|
|
private ReleaseInfo MapToReleaseInfo(BHDResult bhdResult)
|
|
{
|
|
var downloadUri = new Uri(bhdResult.download_url);
|
|
|
|
var title = GetReleaseTitle(bhdResult);
|
|
|
|
var releaseInfo = new ReleaseInfo
|
|
{
|
|
Title = title,
|
|
Seeders = bhdResult.seeders,
|
|
Guid = new Uri(bhdResult.url),
|
|
Details = new Uri(bhdResult.url),
|
|
Link = downloadUri,
|
|
InfoHash = bhdResult.info_hash,
|
|
Peers = bhdResult.leechers + bhdResult.seeders,
|
|
Grabs = bhdResult.times_completed,
|
|
PublishDate = bhdResult.created_at,
|
|
Size = bhdResult.size,
|
|
Category = MapTrackerCatDescToNewznab(bhdResult.category)
|
|
};
|
|
|
|
if (bhdResult.imdb_id.IsNotNullOrWhiteSpace())
|
|
{
|
|
releaseInfo.Imdb = ParseUtil.GetImdbId(bhdResult.imdb_id);
|
|
}
|
|
|
|
if (bhdResult.tmdb_id.IsNotNullOrWhiteSpace() && ParseUtil.TryCoerceLong(bhdResult.tmdb_id.Split('/')[1], out var tmdbResult))
|
|
{
|
|
releaseInfo.TMDb = tmdbResult;
|
|
}
|
|
|
|
releaseInfo.DownloadVolumeFactor = 1;
|
|
releaseInfo.UploadVolumeFactor = 1;
|
|
|
|
if (bhdResult.freeleech == 1 || bhdResult.limited == 1)
|
|
{
|
|
releaseInfo.DownloadVolumeFactor = 0;
|
|
}
|
|
|
|
if (bhdResult.promo25 == 1)
|
|
{
|
|
releaseInfo.DownloadVolumeFactor = .75;
|
|
}
|
|
|
|
if (bhdResult.promo50 == 1)
|
|
{
|
|
releaseInfo.DownloadVolumeFactor = .50;
|
|
}
|
|
|
|
if (bhdResult.promo75 == 1)
|
|
{
|
|
releaseInfo.DownloadVolumeFactor = .25;
|
|
}
|
|
|
|
return releaseInfo;
|
|
}
|
|
|
|
private string GetReleaseTitle(BHDResult bhdResult)
|
|
{
|
|
var title = bhdResult.name.Trim();
|
|
|
|
if (!configData.AddHybridFeaturesToTitle.Value)
|
|
{
|
|
return title;
|
|
}
|
|
|
|
var features = new List<string>();
|
|
|
|
if (bhdResult.dv == 1)
|
|
{
|
|
features.Add("Dolby Vision");
|
|
}
|
|
|
|
if (bhdResult.hdr10 == 1)
|
|
{
|
|
features.Add("HDR10");
|
|
}
|
|
|
|
if (bhdResult.hdr10plus == 1)
|
|
{
|
|
features.Add("HDR10+");
|
|
}
|
|
|
|
if (bhdResult.hlg == 1)
|
|
{
|
|
features.Add("HLG");
|
|
}
|
|
|
|
if (features.Count > 1)
|
|
{
|
|
title += $" ({string.Join(" / ", features)})";
|
|
}
|
|
|
|
return title;
|
|
}
|
|
|
|
private async Task<BHDResponse> GetBHDResponse(string apiUrl, Dictionary<string, object> postData)
|
|
{
|
|
var request = new WebRequest
|
|
{
|
|
Url = apiUrl,
|
|
Type = RequestType.POST,
|
|
Headers = new Dictionary<string, string>
|
|
{
|
|
{ "Accept", "application/json" },
|
|
{ "Content-Type", "application/json" }
|
|
},
|
|
RawBody = JsonConvert.SerializeObject(postData)
|
|
};
|
|
|
|
var response = await webclient.GetResultAsync(request);
|
|
if (response != null && response.ContentString.StartsWith("<"))
|
|
{
|
|
// the response was not JSON, likely a HTML page for a server outage
|
|
logger.Warn(response.ContentString);
|
|
throw new Exception("The response was not JSON");
|
|
}
|
|
|
|
var bhdresponse = JsonConvert.DeserializeObject<BHDResponse>(response.ContentString);
|
|
|
|
if (bhdresponse.status_code == 0)
|
|
{
|
|
throw new Exception(bhdresponse.status_message);
|
|
}
|
|
|
|
return bhdresponse;
|
|
}
|
|
|
|
internal class BHDParams
|
|
{
|
|
internal const string action = "action"; // string - The torrents endpoint action you wish to perform. (search)
|
|
internal const string rsskey = "rsskey"; // string - Your personal RSS key (RID) if you wish for results to include the uploaded_by and download_url fields
|
|
internal const string page = "page"; // int - The page number of the results. Only if the result set has more than 100 total matches.
|
|
|
|
internal const string search = "search"; // string - The torrent name. It does support !negative searching. Example: Christmas Movie
|
|
internal const string info_hash = "info_hash"; // string - The torrent info_hash. This is an exact match.
|
|
internal const string folder_name = "folder_name"; // string - The torrent folder name. This is an exact match.file_name string The torrent included file names. This is an exact match.
|
|
internal const string size = "size"; // int - The torrent size. This is an exact match.
|
|
internal const string uploaded_by = "uploaded_by"; // string - The uploaders username. Only non anonymous results will be returned.
|
|
internal const string imdb_id = "imdb_id"; // int - The ID of the matching IMDB page.
|
|
internal const string tmdb_id = "tmdb_id"; // int - The ID of the matching TMDB page.
|
|
internal const string categories = "categories"; // string - Any categories separated by comma(s). TV, Movies)
|
|
internal const string types = "types"; // string - Any types separated by comma(s). BD Remux, 1080p, etc.)
|
|
internal const string sources = "sources"; // string - Any sources separated by comma(s). Blu-ray, WEB, DVD, etc.)
|
|
internal const string genres = "genres"; // string - Any genres separated by comma(s). Action, Anime, StandUp, Western, etc.)
|
|
internal const string groups = "groups"; // string - Any internal release groups separated by comma(s).FraMeSToR, BHDStudio, BeyondHD, RPG, iROBOT, iFT, ZR, MKVULTRA
|
|
internal const string freeleech = "freeleech"; // int - The torrent freeleech status. 1 = Must match.
|
|
internal const string limited = "limited"; // int - The torrent limited UL promo. 1 = Must match.
|
|
internal const string promo25 = "promo25"; // int - The torrent 25% promo. 1 = Must match.
|
|
internal const string promo50 = "promo50"; // int - The torrent 50% promo. 1 = Must match.
|
|
internal const string promo75 = "promo75"; // int - The torrent 75% promo. 1 = Must match.
|
|
internal const string refund = "refund"; // int - The torrent refund promo. 1 = Must match.
|
|
internal const string rescue = "rescue"; // int - The torrent rescue promo. 1 = Must match.
|
|
internal const string rewind = "rewind"; // int - The torrent rewind promo. 1 = Must match.
|
|
internal const string stream = "stream"; // int - The torrent Stream Optimized flag. 1 = Must match.
|
|
internal const string sd = "sd"; // int - The torrent SD flag. 1 = Must match.
|
|
internal const string pack = "pack"; // int - The torrent TV pack flag. 1 = Must match.
|
|
internal const string h264 = "h264"; // int - The torrent x264/h264 codec flag. 1 = Must match.
|
|
internal const string h265 = "h265"; // int - The torrent x265/h265 codec flag. 1 = Must match.
|
|
internal const string alive = "alive"; // int - The torrent has at least 1 seeder. 1 = Must match.
|
|
internal const string dying = "dying"; // int - The torrent has less than 3 seeders. 1 = Must match.
|
|
internal const string dead = "dead"; // int - The torrent has no seeders. 1 = Must match.
|
|
internal const string reseed = "reseed"; // int - The torrent has no seeders and an active reseed request. 1 = Must match.
|
|
internal const string seeding = "seeding"; // int - The torrent is seeded by you. 1 = Must match.
|
|
internal const string leeching = "leeching"; // int - The torrent is being leeched by you. 1 = Must match.
|
|
internal const string completed = "completed"; // int - The torrent has been completed by you. 1 = Must match.
|
|
internal const string incomplete = "incomplete"; // int - The torrent has not been completed by you. 1 = Must match.
|
|
internal const string notdownloaded = "notdownloaded"; // int - The torrent has not been downloaded you. 1 = Must match.
|
|
internal const string min_bhd = "min_bhd"; // int - The minimum BHD rating.
|
|
internal const string vote_bhd = "vote_bhd"; // int - The minimum number of BHD votes.
|
|
internal const string min_imdb = "min_imdb"; // int - The minimum IMDb rating.
|
|
internal const string vote_imdb = "vote_imdb"; // int - The minimum number of IMDb votes.
|
|
internal const string min_tmdb = "min_tmdb"; // int - The minimum TMDb rating.
|
|
internal const string vote_tmdb = "vote_tmdb"; // int - The minimum number of TDMb votes.
|
|
internal const string min_year = "min_year"; // int - The earliest release year.
|
|
internal const string max_year = "max_year"; // int - The latest release year.
|
|
internal const string sort = "sort"; // string - Field to sort results by. (bumped_at, created_at, seeders, leechers, times_completed, size, name, imdb_rating, tmdb_rating, bhd_rating). Default is bumped_at
|
|
internal const string order = "order"; // string - The direction of the sort of results. (asc, desc). Default is desc
|
|
|
|
// Most of the comma separated fields are OR searches.
|
|
internal const string features = "features"; // string - Any features separated by comma(s). DV, HDR10, HDR10P, Commentary)
|
|
internal const string countries = "countries"; // string - Any production countries separated by comma(s). France, Japan, etc.)
|
|
internal const string languages = "languages"; // string - Any spoken languages separated by comma(s). French, English, etc.)
|
|
internal const string audios = "audios"; // string - Any audio tracks separated by comma(s). English, Japanese,etc.)
|
|
internal const string subtitles = "subtitles"; // string - Any subtitles separated by comma(s). Dutch, Finnish, Swedish, etc.)
|
|
|
|
}
|
|
|
|
class BHDResponse
|
|
{
|
|
public int status_code { get; set; } // The status code of the post request. (0 = Failed and 1 = Success)
|
|
public string status_message { get; set; } // If status code=0 then there will be an explanation
|
|
public int page { get; set; } // The current page of results that you're on.
|
|
public int total_pages { get; set; } // int The total number of pages of results matching your query.
|
|
public int total_results { get; set; } // The total number of results matching your query.
|
|
public bool success { get; set; } // The status of the call. (True = Success, False = Error)
|
|
public BHDResult[] results { get; set; } // The results that match your query.
|
|
}
|
|
|
|
class BHDResult
|
|
{
|
|
public int id { get; set; }
|
|
public string name { get; set; }
|
|
public string folder_name { get; set; }
|
|
public string info_hash { get; set; }
|
|
public long size { get; set; }
|
|
public string uploaded_by { get; set; }
|
|
public string category { get; set; }
|
|
public string type { get; set; }
|
|
public int seeders { get; set; }
|
|
public int leechers { get; set; }
|
|
public int times_completed { get; set; }
|
|
public string imdb_id { get; set; }
|
|
public string tmdb_id { get; set; }
|
|
public decimal bhd_rating { get; set; }
|
|
public decimal tmdb_rating { get; set; }
|
|
public decimal imdb_rating { get; set; }
|
|
public int tv_pack { get; set; }
|
|
public int promo25 { get; set; }
|
|
public int promo50 { get; set; }
|
|
public int promo75 { get; set; }
|
|
public int freeleech { get; set; }
|
|
public int rewind { get; set; }
|
|
public int refund { get; set; }
|
|
public int limited { get; set; }
|
|
public int rescue { get; set; }
|
|
public DateTime bumped_at { get; set; }
|
|
public DateTime created_at { get; set; }
|
|
public string url { get; set; }
|
|
public string download_url { get; set; }
|
|
public int dv { get; set; }
|
|
public int hdr10 { get; set; }
|
|
[JsonProperty("hdr10+")]
|
|
public int hdr10plus { get; set; }
|
|
public int hlg { get; set; }
|
|
public int commentary { get; set; }
|
|
}
|
|
}
|
|
}
|