mirror/Jackett
1
0
Fork 0
mirror of https://github.com/Jackett/Jackett synced 2025-01-01 12:46:23 +00:00
Code Issues Releases Wiki Activity

HDBits with passkey login and assorted improvements. (#6980)

Browse source
This commit is contained in:
Jonas Stendahl 2020-01-20 22:15:22 +01:00 committed by garfield69
parent 67491cb4f6
commit b60c6d132c
1 changed files with 203 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

203
src/Jackett.Common/Indexers/HDBitsApi.cs Normal file
View file

@ -0,0 +1,203 @@
using System;
using System.Collections.Generic;
using System.Collections.Specialized;
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;
using static Jackett.Common.Models.IndexerConfig.ConfigurationData;
namespace Jackett.Common.Indexers
{
public class HDBitsApi : BaseWebIndexer
{
private string APIUrl { get { return SiteLink + "api/"; } }
private new ConfigurationDataUserPasskey configData
{
get { return (ConfigurationDataUserPasskey)base.configData; }
set { base.configData = value; }
}
public HDBitsApi(IIndexerConfigurationService configService, WebClient wc, Logger l, IProtectionService ps)
: base(name: "HDBits (API)",
description: "The HighDefinition Bittorrent Community",
link: "https://hdbits.org/",
caps: new TorznabCapabilities(),
configService: configService,
client: wc,
logger: l,
p: ps,
configData: new ConfigurationDataUserPasskey())
{
Encoding = Encoding.UTF8;
Language = "en-us";
Type = "private";
TorznabCaps.SupportsImdbMovieSearch = true;
AddCategoryMapping(6, TorznabCatType.Audio, "Audio Track");
AddCategoryMapping(3, TorznabCatType.TVDocumentary, "Documentary");
AddCategoryMapping(8, TorznabCatType.Other, "Misc/Demo");
AddCategoryMapping(1, TorznabCatType.Movies, "Movie");
AddCategoryMapping(4, TorznabCatType.Audio, "Music");
AddCategoryMapping(5, TorznabCatType.TVSport, "Sport");
AddCategoryMapping(2, TorznabCatType.TV, "TV");
AddCategoryMapping(7, TorznabCatType.XXX, "XXX");
}
public override async Task<IndexerConfigurationStatus> ApplyConfiguration(JToken configJson)
{
LoadValuesFromJson(configJson);
dynamic requestData = new JObject();
try
{
await MakeApiRequest("test", requestData);
}
catch (Exception e)
{
throw new ExceptionWithConfigData(e.Message, configData);
}
IsConfigured = true;
SaveConfig();
return IndexerConfigurationStatus.Completed;
}
protected override async Task<IEnumerable<ReleaseInfo>> PerformQuery(TorznabQuery query)
{
dynamic requestData = new JObject();
var queryString = query.GetQueryString();
int? imdbId = ParseUtil.GetImdbID(query.ImdbID);
if (imdbId != null)
{
requestData["imdb"] = new JObject();
requestData["imdb"]["id"] = imdbId;
}
else if (!string.IsNullOrWhiteSpace(queryString))
{
requestData["search"] = queryString;
}
var categories = MapTorznabCapsToTrackers(query);
if (categories.Count > 0)
{
requestData["category"] = new JArray();
foreach (var cat in categories)
{
requestData["category"].Add(new JValue(cat));
}
}
requestData["limit"] = 100;
var response = await MakeApiRequest("torrents", requestData);
var releases = new List<ReleaseInfo>();
foreach (JObject r in response["data"])
{
var release = new ReleaseInfo();
release.Title = (string)r["name"];
release.Comments = new Uri(SiteLink+"details.php?id="+(string)r["id"]);
release.Link = new Uri(SiteLink+"download.php/"+(string)r["filename"]+"?id="+(string)r["id"]+"&passkey="+configData.Passkey.Value);
release.Guid = release.Link;
if (r.ContainsKey("imdb"))
{
release.Imdb = ParseUtil.GetImdbID((string)r["imdb"]["id"]);
}
if (r.ContainsKey("tvdb"))
{
release.TVDBId = (long)r["tvdb"]["id"];
}
release.UploadVolumeFactor = 1;
int[] mediumsFor50 = {1,5,4};
// 100% Neutral Leech: all XXX content.
if ((int)r["type_category"] == 7)
{
release.DownloadVolumeFactor = 0;
release.UploadVolumeFactor = 0;
}
// 100% Free Leech: all blue torrents.
else if ((string)r["freeleech"] == "yes")
{
release.DownloadVolumeFactor = 0;
}
// 50% Free Leech: all full discs, remuxes, caps and all internal encodes.
else if (mediumsFor50.Contains((int)r["type_medium"]) || (int)r["type_origin"] == 1)
{
release.DownloadVolumeFactor = 0.5;
}
// 25% Free Leech: all TV content that is not an internal encode.
else if ((int)r["type_category"] == 2 && (int)r["type_origin"] != 1)
{
release.DownloadVolumeFactor = 0.75;
}
// 0% Free Leech: all the content not matching any of the above.
else
{
release.DownloadVolumeFactor = 1;
}
release.Category = MapTrackerCatToNewznab((string)r["type_category"]);
release.Size = (long)r["size"];
release.Files = (long)r["numfiles"];
release.Grabs = (long)r["times_completed"];
release.Seeders = (int)r["seeders"];
release.Peers = release.Seeders + (int)r["leechers"];
release.PublishDate = DateTimeUtil.UnixTimestampToDateTime((int)r["utadded"]);
releases.Add(release);
}
return releases;
}
private async Task<JObject> MakeApiRequest(string url, JObject requestData)
{
requestData["username"] = configData.Username.Value;
requestData["passkey"] = configData.Passkey.Value;
JObject json = null;
var response = await PostDataWithCookiesAndRetry(APIUrl + url, null, null, null, new Dictionary<string, string>()
{
{"Accept", "application/json"},
{"Content-Type", "application/json"}
}, requestData.ToString(), false);
CheckTrackerDown(response);
try
{
json = JObject.Parse(response.Content);
}
catch (Exception ex)
{
throw new Exception("Error while parsing json: " + response.Content, ex);
}
if ((int)json["status"] != 0)
{
throw new Exception("HDBits returned an error with status code " + (int)json["status"] + ": " + (string)json["message"]);
}
return json;
}
}
}