Jackett/src/Jackett.Common/Indexers/ImmortalSeed.cs

using System;
using System.Collections.Generic;
using System.Diagnostics.CodeAnalysis;
using System.Globalization;
using System.Linq;
using System.Net;
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 Newtonsoft.Json.Linq;
using NLog;

namespace Jackett.Common.Indexers
{
    [ExcludeFromCodeCoverage]
    public class ImmortalSeed : BaseWebIndexer
    {
        private string BrowsePage => SiteLink + "browse.php";
        private string LoginUrl => SiteLink + "takelogin.php";
        private string QueryString => "?do=search&keywords={0}&search_type=t_name&category=0&include_dead_torrents=no";

        public override string[] LegacySiteLinks { get; protected set; } = {
            "http://immortalseed.me/"
        };

        private new ConfigurationDataBasicLogin configData
        {
            get => (ConfigurationDataBasicLogin)base.configData;
            set => base.configData = value;
        }

        public ImmortalSeed(IIndexerConfigurationService configService, Utils.Clients.WebClient wc, Logger l,
            IProtectionService ps, ICacheService cs)
            : base(id: "immortalseed",
                   name: "ImmortalSeed",
                   description: "ImmortalSeed (iS) is a Private Torrent Tracker for MOVIES / TV / GENERAL",
                   link: "https://immortalseed.me/",
                   caps: new TorznabCapabilities
                   {
                       TvSearchParams = new List<TvSearchParam>
                       {
                           TvSearchParam.Q, TvSearchParam.Season, TvSearchParam.Ep
                       },
                       MovieSearchParams = new List<MovieSearchParam>
                       {
                           MovieSearchParam.Q
                       },
                       MusicSearchParams = new List<MusicSearchParam>
                       {
                           MusicSearchParam.Q
                       },
                       BookSearchParams = new List<BookSearchParam>
                       {
                           BookSearchParam.Q
                       }
                   },
                   configService: configService,
                   client: wc,
                   logger: l,
                   p: ps,
                   cacheService: cs,
                   configData: new ConfigurationDataBasicLogin())
        {
            Encoding = Encoding.UTF8;
            Language = "en-us";
            Type = "private";

            AddCategoryMapping(3, TorznabCatType.Other, "Nuked");
            AddCategoryMapping(32, TorznabCatType.TVAnime, "Anime");
            AddCategoryMapping(23, TorznabCatType.PC, "Apps");
            AddCategoryMapping(35, TorznabCatType.AudioAudiobook, "Audiobooks");
            AddCategoryMapping(31, TorznabCatType.TV, "Childrens/Cartoons");
            AddCategoryMapping(54, TorznabCatType.TVDocumentary, "Documentary - HD");
            AddCategoryMapping(41, TorznabCatType.BooksComics, "Comics");
            AddCategoryMapping(25, TorznabCatType.PCGames, "Games");
            AddCategoryMapping(29, TorznabCatType.ConsoleXBox, "Games Xbox");
            AddCategoryMapping(27, TorznabCatType.PCGames, "Games-PC Rips");
            AddCategoryMapping(28, TorznabCatType.ConsolePS4, "Games-PSx");
            AddCategoryMapping(49, TorznabCatType.PCMobileOther, "Mobile");
            AddCategoryMapping(59, TorznabCatType.MoviesUHD, "Movies-4k");
            AddCategoryMapping(60, TorznabCatType.MoviesForeign, "Non-English 4k Movies");
            AddCategoryMapping(16, TorznabCatType.MoviesHD, "Movies HD");
            AddCategoryMapping(18, TorznabCatType.MoviesForeign, "Movies HD Non-English");
            AddCategoryMapping(17, TorznabCatType.MoviesSD, "TS/CAM/PPV");
            AddCategoryMapping(34, TorznabCatType.MoviesForeign, "Movies Low Def Non-English");
            AddCategoryMapping(14, TorznabCatType.MoviesSD, "Movies-SD");
            AddCategoryMapping(33, TorznabCatType.MoviesForeign, "Movies SD Non-English");
            AddCategoryMapping(30, TorznabCatType.AudioOther, "Music");
            AddCategoryMapping(37, TorznabCatType.AudioLossless, "FLAC");
            AddCategoryMapping(36, TorznabCatType.AudioMP3, "MP3");
            AddCategoryMapping(39, TorznabCatType.AudioOther, "Music Other");
            AddCategoryMapping(38, TorznabCatType.AudioVideo, "Music Video");
            AddCategoryMapping(45, TorznabCatType.Other, "Other");
            AddCategoryMapping(7, TorznabCatType.TVSport, "Sports Tv");
            AddCategoryMapping(44, TorznabCatType.TVSport, "Sports Fitness-Instructional");
            AddCategoryMapping(58, TorznabCatType.TVSport, "Olympics");
            AddCategoryMapping(47, TorznabCatType.TVSD, "TV - 480p");
            AddCategoryMapping(8, TorznabCatType.TVHD, "TV - High Definition");
            AddCategoryMapping(48, TorznabCatType.TVSD, "TV - Standard Definition - x264");
            AddCategoryMapping(9, TorznabCatType.TVSD, "TV - Standard Definition - XviD");
            AddCategoryMapping(4, TorznabCatType.TVHD, "TV Season Packs - HD");
            AddCategoryMapping(6, TorznabCatType.TVSD, "TV Season Packs - SD");
            AddCategoryMapping(22, TorznabCatType.BooksEBook, "Ebooks");
            AddCategoryMapping(26, TorznabCatType.PCGames, "Games-PC ISO");
            AddCategoryMapping(46, TorznabCatType.BooksMags, "Magazines");
            AddCategoryMapping(50, TorznabCatType.PCMobileiOS, "IOS");
            AddCategoryMapping(51, TorznabCatType.PCMobileAndroid, "Android");
            AddCategoryMapping(52, TorznabCatType.PC0day, "Windows");
            AddCategoryMapping(53, TorznabCatType.TVDocumentary, "Documentary - SD");
        }

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

            var pairs = new Dictionary<string, string> {
                { "username", configData.Username.Value },
                { "password", configData.Password.Value }
            };

            var response = await RequestLoginAndFollowRedirect(LoginUrl, pairs, null, true, null, LoginUrl);

            await ConfigureIfOK(response.Cookies, response.ContentString.Contains("You have successfully logged in"), () =>
            {
                var errorMessage = response.ContentString;
                throw new ExceptionWithConfigData(errorMessage, configData);
            });

            return IndexerConfigurationStatus.RequiresTesting;
        }

        protected override async Task<IEnumerable<ReleaseInfo>> PerformQuery(TorznabQuery query)
        {
            var releases = new List<ReleaseInfo>();
            var searchUrl = BrowsePage;

            if (!string.IsNullOrWhiteSpace(query.GetQueryString()))
                searchUrl += string.Format(QueryString, WebUtility.UrlEncode(query.GetQueryString()));

            var results = await RequestWithCookiesAndRetryAsync(searchUrl);

            // Occasionally the cookies become invalid, login again if that happens
            if (results.ContentString.Contains("You do not have permission to access this page."))
            {
                await ApplyConfiguration(null);
                results = await RequestWithCookiesAndRetryAsync(searchUrl);
            }

            try
            {
                var parser = new HtmlParser();
                var dom = parser.ParseDocument(results.ContentString);

                var rows = dom.QuerySelectorAll("#sortabletable tr:has(a[href*=\"details.php?id=\"])");
                foreach (var row in rows)
                {
                    var release = new ReleaseInfo();

                    var qDetails = row.QuerySelector("div > a[href*=\"details.php?id=\"]"); // details link, release name get's shortened if it's to long
                    // use Title from tooltip or fallback to Details link if there's no tooltip
                    var qTitle = row.QuerySelector(".tooltip-content > div:nth-of-type(1)") ?? qDetails;
                    release.Title = qTitle.TextContent;

                    var qDesciption = row.QuerySelectorAll(".tooltip-content > div");
                    if (qDesciption.Any())
                        release.Description = qDesciption[1].TextContent.Trim();

                    var qLink = row.QuerySelector("a[href*=\"download.php\"]");
                    release.Link = new Uri(qLink.GetAttribute("href"));
                    release.Guid = release.Link;
                    release.Details = new Uri(qDetails.GetAttribute("href"));

                    // 07-22-2015 11:08 AM
                    var dateString = row.QuerySelectorAll("td:nth-of-type(2) div").Last().LastChild.TextContent.Trim();
                    release.PublishDate = DateTime.ParseExact(dateString, "MM-dd-yyyy hh:mm tt", CultureInfo.InvariantCulture);

                    var sizeStr = row.QuerySelector("td:nth-of-type(5)").TextContent.Trim();
                    release.Size = ReleaseInfo.GetBytes(sizeStr);

                    release.Seeders = ParseUtil.CoerceInt(row.QuerySelector("td:nth-of-type(7)").TextContent.Trim());
                    release.Peers = ParseUtil.CoerceInt(row.QuerySelector("td:nth-of-type(8)").TextContent.Trim()) + release.Seeders;

                    var catLink = row.QuerySelector("td:nth-of-type(1) a").GetAttribute("href");
                    var catSplit = catLink.IndexOf("category=");
                    if (catSplit > -1)
                        catLink = catLink.Substring(catSplit + 9);

                    release.Category = MapTrackerCatToNewznab(catLink);

                    var grabs = row.QuerySelector("td:nth-child(6)").TextContent;
                    release.Grabs = ParseUtil.CoerceInt(grabs);

                    if (row.QuerySelector("img[title^=\"Free Torrent\"]") != null)
                        release.DownloadVolumeFactor = 0;
                    else if (row.QuerySelector("img[title^=\"Silver Torrent\"]") != null)
                        release.DownloadVolumeFactor = 0.5;
                    else
                        release.DownloadVolumeFactor = 1;


                    if (row.QuerySelector("img[title^=\"x2 Torrent\"]") != null)
                        release.UploadVolumeFactor = 2;
                    else
                        release.UploadVolumeFactor = 1;

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

            return releases;
        }
    }
}