Jackett/src/Jackett.Common/Indexers/TorrentBytes.cs

210 lines
10 KiB
C#

using System;
using System.Collections.Generic;
using System.Collections.Specialized;
using System.Diagnostics.CodeAnalysis;
using System.Globalization;
using System.Text;
using System.Threading.Tasks;
using AngleSharp.Html.Parser;
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.Linq;
using NLog;
namespace Jackett.Common.Indexers
{
[ExcludeFromCodeCoverage]
public class TorrentBytes : BaseWebIndexer
{
private string LoginUrl => SiteLink + "takelogin.php";
private string SearchUrl => SiteLink + "browse.php";
private new ConfigurationDataBasicLogin configData => (ConfigurationDataBasicLogin)base.configData;
public TorrentBytes(IIndexerConfigurationService configService, WebClient wc, Logger l, IProtectionService ps,
ICacheService cs)
: base(id: "torrentbytes",
name: "TorrentBytes",
description: "A decade of TorrentBytes",
link: "https://www.torrentbytes.net/",
caps: new TorznabCapabilities
{
TvSearchParams = new List<TvSearchParam>
{
TvSearchParam.Q, TvSearchParam.Season, TvSearchParam.Ep, TvSearchParam.ImdbId
},
MovieSearchParams = new List<MovieSearchParam>
{
MovieSearchParam.Q, MovieSearchParam.ImdbId
},
MusicSearchParams = new List<MusicSearchParam>
{
MusicSearchParam.Q
}
},
configService: configService,
client: wc,
logger: l,
p: ps,
cacheService: cs,
configData: new ConfigurationDataBasicLogin("For best results, change the 'Torrents per page' setting to 100 in your profile on the TorrentBytes webpage."))
{
Encoding = Encoding.GetEncoding("iso-8859-1");
Language = "en-US";
Type = "private";
AddCategoryMapping(23, TorznabCatType.TVAnime, "Anime");
AddCategoryMapping(52, TorznabCatType.PCMac, "Apple/All");
AddCategoryMapping(22, TorznabCatType.PC, "Apps/misc");
AddCategoryMapping(1, TorznabCatType.PC, "Apps/PC");
AddCategoryMapping(28, TorznabCatType.TVForeign, "Foreign Titles");
AddCategoryMapping(50, TorznabCatType.Console, "Games/Consoles");
AddCategoryMapping(42, TorznabCatType.PCGames, "Games/Pack");
AddCategoryMapping(4, TorznabCatType.PCGames, "Games/PC");
AddCategoryMapping(51, TorznabCatType.PC, "Linux/All");
AddCategoryMapping(31, TorznabCatType.OtherMisc, "Misc");
AddCategoryMapping(20, TorznabCatType.MoviesDVD, "Movies/DVD-R");
AddCategoryMapping(12, TorznabCatType.MoviesBluRay, "Movies/Full Blu-ray");
AddCategoryMapping(5, TorznabCatType.MoviesHD, "Movies/HD");
AddCategoryMapping(40, TorznabCatType.Movies, "Movies/Pack");
AddCategoryMapping(19, TorznabCatType.MoviesSD, "Movies/SD");
AddCategoryMapping(49, TorznabCatType.MoviesUHD, "Movies/UHD");
AddCategoryMapping(25, TorznabCatType.Audio, "Music/DVDR");
AddCategoryMapping(48, TorznabCatType.AudioLossless, "Music/Flac");
AddCategoryMapping(6, TorznabCatType.AudioMP3, "Music/MP3");
AddCategoryMapping(43, TorznabCatType.Audio, "Music/Pack");
AddCategoryMapping(34, TorznabCatType.AudioVideo, "Music/Videos");
AddCategoryMapping(45, TorznabCatType.MoviesBluRay, "NonScene/BRrip");
AddCategoryMapping(46, TorznabCatType.MoviesHD, "NonScene/x264");
AddCategoryMapping(44, TorznabCatType.MoviesSD, "NonScene/Xvid");
AddCategoryMapping(37, TorznabCatType.TVHD, "TV/BRrip");
AddCategoryMapping(38, TorznabCatType.TVHD, "TV/HD");
AddCategoryMapping(41, TorznabCatType.TV, "TV/Pack");
AddCategoryMapping(33, TorznabCatType.TVSD, "TV/SD");
AddCategoryMapping(32, TorznabCatType.TVUHD, "TV/UHD");
AddCategoryMapping(39, TorznabCatType.XXXx264, "XXX/HD");
AddCategoryMapping(24, TorznabCatType.XXXImageSet, "XXX/IMGSET");
AddCategoryMapping(21, TorznabCatType.XXXPack, "XXX/Pack");
AddCategoryMapping(9, TorznabCatType.XXXXviD, "XXX/SD");
AddCategoryMapping(29, TorznabCatType.XXX, "XXX/Web");
}
public override async Task<IndexerConfigurationStatus> ApplyConfiguration(JToken configJson)
{
LoadValuesFromJson(configJson);
var pairs = new Dictionary<string, string>
{
{"username", configData.Username.Value},
{"password", configData.Password.Value},
{"returnto", "/"},
{"login", "Log in!"}
};
var loginPage = await RequestWithCookiesAsync(SiteLink, string.Empty);
var result = await RequestLoginAndFollowRedirect(LoginUrl, pairs, loginPage.Cookies, true, SiteLink, SiteLink);
await ConfigureIfOK(
result.Cookies, result.ContentString?.Contains("my.php") == true, () =>
{
var parser = new HtmlParser();
var dom = parser.ParseDocument(result.ContentString);
var messageEl = dom.QuerySelector("td.embedded");
var errorMessage = messageEl != null ? messageEl.TextContent : result.ContentString;
throw new ExceptionWithConfigData(errorMessage, configData);
});
return IndexerConfigurationStatus.RequiresTesting;
}
protected override async Task<IEnumerable<ReleaseInfo>> PerformQuery(TorznabQuery query)
{
var releases = new List<ReleaseInfo>();
var qc = new NameValueCollection
{
{"incldead", "1"}
};
if (query.IsImdbQuery)
{
qc.Add("search", query.ImdbID);
qc.Add("sc", "2"); // search in description
}
else
{
qc.Add("search", query.GetQueryString());
qc.Add("sc", "1"); // search in title
}
foreach (var cat in MapTorznabCapsToTrackers(query))
qc.Add("c" + cat, "1");
var searchUrl = SearchUrl + "?" + qc.GetQueryString();
var response = await RequestWithCookiesAndRetryAsync(searchUrl, referer: SearchUrl);
if (response.IsRedirect) // re-login
{
await ApplyConfiguration(null);
response = await RequestWithCookiesAndRetryAsync(searchUrl, referer: SearchUrl);
}
try
{
var parser = new HtmlParser();
var dom = parser.ParseDocument(response.ContentString);
var rows = dom.QuerySelectorAll("table > tbody:has(tr > td.colhead) > tr:not(:has(td.colhead))");
foreach (var row in rows)
{
var release = new ReleaseInfo();
var link = row.QuerySelector("td:nth-of-type(2) a:nth-of-type(2)");
release.Guid = new Uri(SiteLink + link.GetAttribute("href"));
release.Details = release.Guid;
release.Title = link.GetAttribute("title");
// There isn't a title attribute if the release name isn't truncated.
if (string.IsNullOrWhiteSpace(release.Title))
release.Title = link.FirstChild.TextContent.Trim();
release.Description = release.Title;
// If we search an get no results, we still get a table just with no info.
if (string.IsNullOrWhiteSpace(release.Title))
break;
// Check if the release has been assigned a category
var qCat = row.QuerySelector("td:nth-of-type(1) a");
if (qCat != null)
{
var cat = qCat.GetAttribute("href").Substring(15);
release.Category = MapTrackerCatToNewznab(cat);
}
var qLink = row.QuerySelector("td:nth-of-type(2) a");
release.Link = new Uri(SiteLink + qLink.GetAttribute("href"));
var added = row.QuerySelector("td:nth-of-type(5)").TextContent.Trim();
release.PublishDate = DateTime.ParseExact(added, "yyyy-MM-ddHH:mm:ss", CultureInfo.InvariantCulture);
var sizeStr = row.QuerySelector("td:nth-of-type(7)").TextContent.Trim();
release.Size = ReleaseInfo.GetBytes(sizeStr);
release.Seeders = ParseUtil.CoerceInt(row.QuerySelector("td:nth-of-type(9)").TextContent.Trim());
release.Peers = ParseUtil.CoerceInt(row.QuerySelector("td:nth-of-type(10)").TextContent.Trim()) +
release.Seeders;
var files = row.QuerySelector("td:nth-child(3)").TextContent;
release.Files = ParseUtil.CoerceInt(files);
var grabs = row.QuerySelector("td:nth-child(8)").TextContent;
if (grabs != "----")
release.Grabs = ParseUtil.CoerceInt(grabs);
release.DownloadVolumeFactor =
row.QuerySelector("font[color=\"green\"]:contains(\"F\"):contains(\"L\")") != null ? 0 : 1;
release.UploadVolumeFactor = 1;
releases.Add(release);
}
}
catch (Exception ex)
{
OnParseError(response.ContentString, ex);
}
return releases;
}
}
}