beyond-hd: add new indexer based on API (#11481)

Thanks to: PNWebster
This commit is contained in:
Webster 2021-05-05 10:24:05 +02:00 committed by GitHub
parent 921093934f
commit 381e674ac4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 290 additions and 0 deletions

View File

@ -0,0 +1,276 @@
using System;
using System.Collections.Generic;
using System.Diagnostics.CodeAnalysis;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Jackett.Common.Models;
using Jackett.Common.Models.IndexerConfig;
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 : BaseWebIndexer
{
private readonly string APIBASE = "https://beyond-hd.me/api/torrents/";
private new ConfigurationDataAPIKeyAndRSSKey configData
{
get => (ConfigurationDataAPIKeyAndRSSKey)base.configData;
set => base.configData = value;
}
public BeyondHDAPI(IIndexerConfigurationService configService, WebClient wc, Logger l,
IProtectionService ps, ICacheService cs)
: base(id: "beyond-hd-api",
name: "Beyond-HD (API)",
description: "Without BeyondHD, your HDTV is just a TV",
link: "https://beyond-hd.me/",
caps: new TorznabCapabilities
{
LimitsDefault = 100,
LimitsMax = 100,
TvSearchParams = new List<TvSearchParam>
{
TvSearchParam.Q, TvSearchParam.Season, TvSearchParam.Ep, TvSearchParam.ImdbId
},
MovieSearchParams = new List<MovieSearchParam>
{
MovieSearchParam.Q, MovieSearchParam.ImdbId, MovieSearchParam.TmdbId
}
},
configService: configService,
client: wc,
logger: l,
p: ps,
cacheService: cs,
configData: new ConfigurationDataAPIKeyAndRSSKey())
{
Encoding = Encoding.UTF8;
Language = "en-us";
Type = "private";
AddCategoryMapping("Movies", TorznabCatType.Movies);
AddCategoryMapping("TV", TorznabCatType.TV);
}
public override async Task<IndexerConfigurationStatus> ApplyConfiguration(JToken configJson)
{
LoadValuesFromJson(configJson);
IsConfigured = false;
try
{
var results = await PerformQuery(new TorznabQuery());
if (results.Count() == 0)
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}";
Dictionary<string, string> postData = new Dictionary<string, string>
{
{ BHDParams.action, "search" },
{ BHDParams.rsskey, configData.RSSKey.Value },
{ BHDParams.search, query.SanitizedSearchTerm },
};
if (query.IsTVSearch)
{
postData.Add(BHDParams.categories, "TV");
if (query.Season != 0)
postData[BHDParams.search] = $"{query.SanitizedSearchTerm} {query.GetEpisodeSearchString()}";
}
else if (query.IsMovieSearch)
{
postData.Add(BHDParams.categories, "Movies");
}
var imdbId = ParseUtil.GetImdbID(query.ImdbID);
if (imdbId != null)
postData.Add(BHDParams.imdb_id, imdbId.ToString());
if (query.IsTmdbQuery)
postData.Add(BHDParams.tmdb_id, query.TmdbID.Value.ToString());
var bhdResponse = await GetBHDResponse(apiUrl, postData);
var releaseInfos = bhdResponse.results.Select(mapToReleaseInfo);
return releaseInfos;
}
private ReleaseInfo mapToReleaseInfo(BHDResult bhdResult)
{
var uri = new Uri(bhdResult.url);
var downloadUri = new Uri(bhdResult.download_url);
var releaseInfo = new ReleaseInfo
{
Title = bhdResult.name,
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 = MapTrackerCatToNewznab(bhdResult.category)
};
if (!string.IsNullOrEmpty(bhdResult.imdb_id))
releaseInfo.Imdb = ParseUtil.GetImdbID(bhdResult.imdb_id);
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 async Task<BHDResponse> GetBHDResponse(string apiUrl, Dictionary<string, string> postData)
{
var request = new WebRequest
{
PostData = postData,
Type = RequestType.POST,
Url = apiUrl
};
var response = await webclient.GetResultAsync(request);
var bhdresponse = JsonConvert.DeserializeObject<BHDResponse>(response.ContentString);
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 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; }
}
}
}

View File

@ -0,0 +1,14 @@
namespace Jackett.Common.Models.IndexerConfig
{
public class ConfigurationDataAPIKeyAndRSSKey : ConfigurationData
{
public StringConfigurationItem ApiKey { get; private set; }
public StringConfigurationItem RSSKey { get; private set; }
public ConfigurationDataAPIKeyAndRSSKey()
{
ApiKey = new StringConfigurationItem("API Key");
RSSKey = new StringConfigurationItem("RSS Key");
}
}
}