Jackett/src/Jackett.Common/Indexers/TorrentHeaven.cs

using System;
using System.Collections.Generic;
using System.Collections.Specialized;
using System.Globalization;
using System.Linq;
using System.Text;
using System.Text.RegularExpressions;
using System.Threading.Tasks;
using CsQuery;
using Jackett.Models;
using Jackett.Models.IndexerConfig;
using Jackett.Services.Interfaces;
using Jackett.Utils;
using Jackett.Utils.Clients;
using Newtonsoft.Json.Linq;
using NLog;

namespace Jackett.Indexers
{
    public class TorrentHeaven : BaseWebIndexer
    {
        private string IndexUrl { get { return SiteLink + "index.php"; } }
        private string LoginCompleteUrl { get { return SiteLink + "index.php?strWebValue=account&strWebAction=login_complete&ancestry=verify"; } }
        private static readonly string certificateHash = "6F5CE30D578C2A7AECFB919D0D013976D395055F";

        private new ConfigurationDataCaptchaLogin configData
        {
            get { return (ConfigurationDataCaptchaLogin)base.configData; }
            set { base.configData = value; }
        }

        public TorrentHeaven(IIndexerConfigurationService configService, WebClient wc, Logger l, IProtectionService ps)
            : base(name: "TorrentHeaven",
                   description: "A German general tracker.",
                   link: "https://torrentheaven.myfqdn.info/",
                   caps: TorznabUtil.CreateDefaultTorznabTVCaps(),
                   configService: configService,
                   client: wc,
                   logger: l,
                   p: ps,
                   configData: new ConfigurationDataCaptchaLogin())
        {
            Encoding = Encoding.GetEncoding("iso-8859-1");
            Language = "de-de";
            Type = "private";

            AddCategoryMapping(1, TorznabCatType.PCGames); // GAMES/PC
            AddCategoryMapping(3, TorznabCatType.Console); // GAMES/Sonstige
            AddCategoryMapping(59, TorznabCatType.ConsolePS4); // GAMES/PlayStation
            AddCategoryMapping(60, TorznabCatType.ConsolePSP); // GAMES/PSP
            AddCategoryMapping(63, TorznabCatType.ConsoleWii); // GAMES/Wii
            AddCategoryMapping(67, TorznabCatType.ConsoleXbox360); // GAMES/XBOX 360
            AddCategoryMapping(68, TorznabCatType.PCPhoneOther); // GAMES/PDA / Handy
            AddCategoryMapping(72, TorznabCatType.ConsoleNDS); // GAMES/NDS

            AddCategoryMapping(7, TorznabCatType.MoviesDVD); // MOVIES/DVD
            AddCategoryMapping(8, TorznabCatType.MoviesSD); // MOVIES/SD
            AddCategoryMapping(37, TorznabCatType.MoviesDVD); // MOVIES/DVD Spezial
            AddCategoryMapping(41, TorznabCatType.MoviesForeign); // MOVIES/International
            AddCategoryMapping(101, TorznabCatType.MoviesHD); // MOVIES/720p
            AddCategoryMapping(102, TorznabCatType.MoviesHD); // MOVIES/1080p
            AddCategoryMapping(103, TorznabCatType.MoviesHD); // MOVIES/AVCHD
            AddCategoryMapping(104, TorznabCatType.MoviesBluRay); // MOVIES/Bluray
            AddCategoryMapping(106, TorznabCatType.Movies3D); // MOVIES/3D

            AddCategoryMapping(14, TorznabCatType.Audio); // AUDIO/Musik
            AddCategoryMapping(15, TorznabCatType.AudioAudiobook); // AUDIO/Hörbücher
            AddCategoryMapping(16, TorznabCatType.AudioAudiobook); // AUDIO/Hörspiele
            AddCategoryMapping(36, TorznabCatType.AudioLossless); // AUDIO/Flac
            AddCategoryMapping(42, TorznabCatType.AudioOther); // AUDIO/Soundtracks
            AddCategoryMapping(58, TorznabCatType.AudioVideo); // AUDIO/Musikvideos

            AddCategoryMapping(18, TorznabCatType.TVSD); // TV/Serien SD
            AddCategoryMapping(19, TorznabCatType.TVHD); // TV/Serien HD 720p
            AddCategoryMapping(20, TorznabCatType.TVHD); // TV/Serien HD 1080p
            AddCategoryMapping(49, TorznabCatType.TVSD); // TV/Serien DVD
            AddCategoryMapping(51, TorznabCatType.TVDocumentary); // TV/Doku SD
            AddCategoryMapping(52, TorznabCatType.TVDocumentary); // TV/Doku HD
            AddCategoryMapping(53, TorznabCatType.TV); // TV/Serien Complete Packs
            AddCategoryMapping(54, TorznabCatType.TVSport); // TV/Sport
            AddCategoryMapping(66, TorznabCatType.TVFOREIGN); // TV/International

            AddCategoryMapping(22, TorznabCatType.Books); // MISC/EBooks
            AddCategoryMapping(24, TorznabCatType.Other); // MISC/Sonstiges
            AddCategoryMapping(25, TorznabCatType.Other); // MISC/Tonspuren
            AddCategoryMapping(108, TorznabCatType.TVAnime); // MISC/Anime

            AddCategoryMapping(28, TorznabCatType.PC); // APPLICATIONS/PC
            AddCategoryMapping(29, TorznabCatType.PCPhoneOther); // APPLICATIONS/Mobile
            AddCategoryMapping(30, TorznabCatType.PC); // APPLICATIONS/Sonstige
            AddCategoryMapping(70, TorznabCatType.PC); // APPLICATIONS/Linux
            AddCategoryMapping(71, TorznabCatType.PCMac); // APPLICATIONS/Mac

            webclient.AddTrustedCertificate(new Uri(SiteLink).Host, certificateHash);
        }

        public override async Task<ConfigurationData> GetConfigurationForSetup()
        {
            var loginPage = await RequestStringWithCookies(IndexUrl, string.Empty);
            CQ dom = loginPage.Content;
            CQ qCaptchaImg = dom.Find("td.tablea > img").First();
            if (qCaptchaImg.Length == 1)
            {
                var CaptchaUrl = SiteLink + qCaptchaImg.Attr("src");
                var captchaImage = await RequestBytesWithCookies(CaptchaUrl, loginPage.Cookies);
                configData.CaptchaImage.Value = captchaImage.Content;
            }
            else
            {
                configData.CaptchaImage.Value = new byte[0];
            }
            configData.CaptchaCookie.Value = loginPage.Cookies;
            return configData;
        }

        public override async Task<IndexerConfigurationStatus> ApplyConfiguration(JToken configJson)
        {
            LoadValuesFromJson(configJson);

            var pairs = new Dictionary<string, string>
            {
                { "strWebAction", "login" },
                { "strWebValue", "account" },
                { "jsenabled", "1" },
                { "screenwidth", "2560" },
                { "username", configData.Username.Value },
                { "password", configData.Password.Value }
            };

            if (!string.IsNullOrWhiteSpace(configData.CaptchaText.Value))
            {
                pairs.Add("proofcode", configData.CaptchaText.Value);
            }

            var result = await RequestLoginAndFollowRedirect(IndexUrl, pairs, configData.CaptchaCookie.Value, true, null, IndexUrl, true);
            if (result.Content == null || (!result.Content.Contains("login_complete") && !result.Content.Contains("index.php?strWebValue=account&strWebAction=logout")))
            {
                CQ dom = result.Content;
                var errorMessage = dom["table > tbody > tr > td[valign=top][width=100%]"].Html();
                if (errorMessage.Length == 0)
                    errorMessage = result.Content;
                throw new ExceptionWithConfigData(errorMessage, configData);
            }

            var result2 = await RequestStringWithCookies(LoginCompleteUrl, result.Cookies);
            await ConfigureIfOK(result2.Cookies, result2.Cookies != null && result2.Cookies.Contains("pass"), () =>
                {
                    var errorMessage = "Didn't get a user/pass cookie";
                    throw new ExceptionWithConfigData(errorMessage, configData);
                });
            return IndexerConfigurationStatus.RequiresTesting;
        }

        protected override async Task<IEnumerable<ReleaseInfo>> PerformQuery(TorznabQuery query)
        {
            TimeZoneInfo.TransitionTime startTransition = TimeZoneInfo.TransitionTime.CreateFloatingDateRule(new DateTime(1, 1, 1, 3, 0, 0), 3, 5, DayOfWeek.Sunday);
            TimeZoneInfo.TransitionTime endTransition = TimeZoneInfo.TransitionTime.CreateFloatingDateRule(new DateTime(1, 1, 1, 4, 0, 0), 10, 5, DayOfWeek.Sunday);
            TimeSpan delta = new TimeSpan(1, 0, 0);
            TimeZoneInfo.AdjustmentRule adjustment = TimeZoneInfo.AdjustmentRule.CreateAdjustmentRule(new DateTime(1999, 10, 1), DateTime.MaxValue.Date, delta, startTransition, endTransition);
            TimeZoneInfo.AdjustmentRule[] adjustments = { adjustment };
            TimeZoneInfo germanyTz = TimeZoneInfo.CreateCustomTimeZone("W. Europe Standard Time", new TimeSpan(1, 0, 0), "(GMT+01:00) W. Europe Standard Time", "W. Europe Standard Time", "W. Europe DST Time", adjustments);

            var releases = new List<ReleaseInfo>();

            var searchString = query.GetQueryString();
            var searchUrl = IndexUrl;
            var queryCollection = new NameValueCollection();
            queryCollection.Add("strWebValue", "torrent");
            queryCollection.Add("strWebAction", "search");
            queryCollection.Add("sort", "torrent_added");
            queryCollection.Add("by", "d");
            queryCollection.Add("type", "0");
            queryCollection.Add("do_search", "suchen");
            queryCollection.Add("time", "0");
            queryCollection.Add("details", "title");

            if (!string.IsNullOrWhiteSpace(searchString))
            {
                queryCollection.Add("searchstring", searchString);
            }

            foreach (var cat in MapTorznabCapsToTrackers(query))
            {
                queryCollection.Add("dirs" + cat, "1");
            }
            searchUrl += "?" + queryCollection.GetQueryString();

            var response = await RequestStringWithCookies(searchUrl);
            var results = response.Content;
            var TitleRegexp = new Regex(@"^return buildTable\('(.*?)',\s+");
            try
            {
                CQ dom = results;
                var rows = dom["table.torrenttable > tbody > tr"];

                foreach (var row in rows.Skip(1))
                {
                    var release = new ReleaseInfo();
                    release.MinimumRatio = 0.8;
                    release.MinimumSeedTime = 0;
                    var qRow = row.Cq();

                    var qDetailsLink = qRow.Find("a[href^=index.php?strWebValue=torrent&strWebAction=details]").First();
                    release.Title = TitleRegexp.Match(qDetailsLink.Attr("onmouseover")).Groups[1].Value;

                    var qCatLink = qRow.Find("a[href^=index.php?strWebValue=torrent&strWebAction=search&dir=]").First();
                    var qDLLink = qRow.Find("a[href^=index.php?strWebValue=torrent&strWebAction=download&id=]").First();
                    var qSeeders = qRow.Find("td.column1:eq(3)");
                    var qLeechers = qRow.Find("td.column2:eq(3)");
                    var qDateStr = qRow.Find("font:has(a)").First();
                    var qSize = qRow.Find("td.column2[align=center]").First();

                    var catStr = qCatLink.Attr("href").Split('=')[3].Split('#')[0];
                    release.Category = MapTrackerCatToNewznab(catStr);

                    release.Link = new Uri(SiteLink + qDLLink.Attr("href"));
                    release.Comments = new Uri(SiteLink + qDetailsLink.Attr("href"));
                    release.Guid = release.Link;

                    var sizeStr = qSize.Text();
                    release.Size = ReleaseInfo.GetBytes(sizeStr);

                    release.Seeders = ParseUtil.CoerceInt(qSeeders.Text());
                    release.Peers = ParseUtil.CoerceInt(qLeechers.Text()) + release.Seeders;

                    var dateStr = qDateStr.Text().Trim();
                    var dateStrParts = dateStr.Split();
                    DateTime dateGerman;
                    if (dateStrParts[0] == "Heute")
                        dateGerman = DateTime.SpecifyKind(DateTime.UtcNow.Date, DateTimeKind.Unspecified) + TimeSpan.Parse(dateStrParts[1]);
                    else if (dateStrParts[0] == "Gestern")
                        dateGerman = DateTime.SpecifyKind(DateTime.UtcNow.Date, DateTimeKind.Unspecified) + TimeSpan.Parse(dateStrParts[1]) - TimeSpan.FromDays(1);
                    else
                        dateGerman = DateTime.SpecifyKind(DateTime.ParseExact(dateStrParts[0] + dateStrParts[1], "dd.MM.yyyyHH:mm", CultureInfo.InvariantCulture), DateTimeKind.Unspecified);

                    DateTime pubDateUtc = TimeZoneInfo.ConvertTimeToUtc(dateGerman, germanyTz);
                    release.PublishDate = pubDateUtc.ToLocalTime();

                    var grabs = qRow.Find("td:nth-child(7)").Text();
                    release.Grabs = ParseUtil.CoerceInt(grabs);

                    if (qRow.Find("img[src=\"themes/images/freeleech.png\"]").Length >= 1)
                        release.DownloadVolumeFactor = 0;
                    else if (qRow.Find("img[src=\"themes/images/DL50.png\"]").Length >= 1)
                        release.DownloadVolumeFactor = 0.5;
                    else
                        release.DownloadVolumeFactor = 1;

                    release.UploadVolumeFactor = 1;

                    releases.Add(release);
                }
            }
            catch (Exception ex)
            {
                OnParseError(results, ex);
            }

            return releases;
        }
    }
}