Jackett/src/Jackett/Indexers/IPTorrents.cs

212 lines
7.5 KiB
C#
Raw Normal View History

2015-04-16 01:16:16 +00:00
using CsQuery;
2015-07-19 00:27:41 +00:00
using Jackett.Models;
using Jackett.Utils;
2015-04-16 01:16:16 +00:00
using Newtonsoft.Json.Linq;
2015-07-19 00:27:41 +00:00
using NLog;
2015-04-16 01:16:16 +00:00
using System;
using System.Collections.Generic;
2015-05-05 04:25:28 +00:00
using System.Globalization;
2015-04-16 01:16:16 +00:00
using System.Linq;
using System.Net;
using System.Net.Http;
using System.Text;
using System.Threading.Tasks;
2015-04-18 22:08:36 +00:00
using System.Web;
2015-04-16 01:16:16 +00:00
namespace Jackett.Indexers
{
2015-07-19 13:22:50 +00:00
public class IPTorrents : IIndexer
2015-05-04 04:12:14 +00:00
{
2015-04-16 01:16:16 +00:00
2015-07-19 13:22:50 +00:00
public event Action<IIndexer, JToken> OnSaveConfigurationRequested;
public event Action<IIndexer, string, Exception> OnResultParsingError;
2015-04-16 01:16:16 +00:00
2015-05-04 04:12:14 +00:00
public string DisplayName { get { return "IPTorrents"; } }
2015-04-18 22:08:36 +00:00
2015-05-04 04:12:14 +00:00
public string DisplayDescription { get { return "Always a step ahead"; } }
2015-04-18 22:08:36 +00:00
2015-05-04 04:12:14 +00:00
public Uri SiteLink { get { return new Uri(BaseUrl); } }
2015-04-16 01:16:16 +00:00
public bool RequiresRageIDLookupDisabled { get { return true; } }
2015-05-04 04:12:14 +00:00
public bool IsConfigured { get; private set; }
2015-04-18 22:08:36 +00:00
2015-07-12 17:52:32 +00:00
static string chromeUserAgent = BrowserUtil.ChromeUserAgent;
2015-04-16 01:16:16 +00:00
2015-05-04 04:12:14 +00:00
static string BaseUrl = "https://iptorrents.com";
2015-04-16 01:16:16 +00:00
2015-05-04 04:12:14 +00:00
string SearchUrl = BaseUrl + "/t?q=";
2015-05-04 04:12:14 +00:00
CookieContainer cookies;
HttpClientHandler handler;
HttpClient client;
2015-07-19 00:27:41 +00:00
Logger logger;
2015-07-19 00:27:41 +00:00
public IPTorrents(Logger l)
2015-05-04 04:12:14 +00:00
{
2015-07-19 00:27:41 +00:00
logger = l;
2015-05-04 04:12:14 +00:00
IsConfigured = false;
cookies = new CookieContainer();
handler = new HttpClientHandler
{
CookieContainer = cookies,
AllowAutoRedirect = true,
UseCookies = true,
};
client = new HttpClient(handler);
}
2015-05-04 04:12:14 +00:00
public async Task<ConfigurationData> GetConfigurationForSetup()
{
await client.GetAsync(new Uri(BaseUrl));
var config = new ConfigurationDataBasicLogin();
return (ConfigurationData)config;
}
2015-05-04 04:12:14 +00:00
public async Task ApplyConfiguration(JToken configJson)
{
2015-05-04 04:12:14 +00:00
var config = new ConfigurationDataBasicLogin();
config.LoadValuesFromJson(configJson);
2015-05-04 04:12:14 +00:00
var pairs = new Dictionary<string, string> {
{ "username", config.Username.Value },
{ "password", config.Password.Value }
};
2015-05-04 04:12:14 +00:00
var content = new FormUrlEncodedContent(pairs);
var message = new HttpRequestMessage();
message.Method = HttpMethod.Post;
message.Content = content;
message.RequestUri = new Uri(BaseUrl);
message.Headers.Referrer = new Uri(BaseUrl);
message.Headers.UserAgent.ParseAdd(chromeUserAgent);
var response = await client.SendAsync(message);
var responseContent = await response.Content.ReadAsStringAsync();
if (!responseContent.Contains("/my.php"))
{
CQ dom = responseContent;
var messageEl = dom["body > div"].First();
var errorMessage = messageEl.Text().Trim();
throw new ExceptionWithConfigData(errorMessage, (ConfigurationData)config);
}
else
{
var configSaveData = new JObject();
2015-07-12 16:23:40 +00:00
cookies.DumpToJson(SiteLink, configSaveData);
2015-05-04 04:12:14 +00:00
if (OnSaveConfigurationRequested != null)
OnSaveConfigurationRequested(this, configSaveData);
IsConfigured = true;
}
}
HttpRequestMessage CreateHttpRequest(Uri uri)
{
var message = new HttpRequestMessage();
message.Method = HttpMethod.Get;
message.RequestUri = uri;
message.Headers.UserAgent.ParseAdd(chromeUserAgent);
return message;
}
public void LoadFromSavedConfiguration(Newtonsoft.Json.Linq.JToken jsonConfig)
{
2015-07-19 00:27:41 +00:00
cookies.FillFromJson(new Uri(BaseUrl), jsonConfig, logger);
2015-05-04 04:12:14 +00:00
IsConfigured = true;
}
public async Task<ReleaseInfo[]> PerformQuery(TorznabQuery query)
{
List<ReleaseInfo> releases = new List<ReleaseInfo>();
var searchString = query.SanitizedSearchTerm + " " + query.GetEpisodeSearchString();
var episodeSearchUrl = SearchUrl + HttpUtility.UrlEncode(searchString);
2015-05-04 04:12:14 +00:00
var request = CreateHttpRequest(new Uri(episodeSearchUrl));
var response = await client.SendAsync(request);
var results = await response.Content.ReadAsStringAsync();
2015-05-04 04:12:14 +00:00
try
{
CQ dom = results;
2015-05-04 04:12:14 +00:00
var rows = dom["table.torrents > tbody > tr"];
foreach (var row in rows.Skip(1))
2015-05-04 04:12:14 +00:00
{
var release = new ReleaseInfo();
var qRow = row.Cq();
var qTitleLink = qRow.Find("a.t_title").First();
release.Title = qTitleLink.Text().Trim();
release.Description = release.Title;
release.Guid = new Uri(BaseUrl + qTitleLink.Attr("href"));
release.Comments = release.Guid;
DateTime pubDate;
var descString = qRow.Find(".t_ctime").Text();
var dateString = descString.Split('|').Last().Trim();
dateString = dateString.Split(new string[] { " by " }, StringSplitOptions.None)[0];
var dateValue = ParseUtil.CoerceFloat(dateString.Split(' ')[0]);
var dateUnit = dateString.Split(' ')[1];
if (dateUnit.Contains("minute"))
pubDate = DateTime.Now - TimeSpan.FromMinutes(dateValue);
else if (dateUnit.Contains("hour"))
pubDate = DateTime.Now - TimeSpan.FromHours(dateValue);
else if (dateUnit.Contains("day"))
pubDate = DateTime.Now - TimeSpan.FromDays(dateValue);
else if (dateUnit.Contains("week"))
pubDate = DateTime.Now - TimeSpan.FromDays(7 * dateValue);
else if (dateUnit.Contains("month"))
pubDate = DateTime.Now - TimeSpan.FromDays(30 * dateValue);
else if (dateUnit.Contains("year"))
pubDate = DateTime.Now - TimeSpan.FromDays(365 * dateValue);
else
pubDate = DateTime.MinValue;
release.PublishDate = pubDate;
var qLink = row.ChildElements.ElementAt(3).Cq().Children("a");
release.Link = new Uri(BaseUrl + qLink.Attr("href"));
var sizeStr = row.ChildElements.ElementAt(5).Cq().Text().Trim();
var sizeVal = ParseUtil.CoerceFloat(sizeStr.Split(' ')[0]);
var sizeUnit = sizeStr.Split(' ')[1];
release.Size = ReleaseInfo.GetBytes(sizeUnit, sizeVal);
release.Seeders = ParseUtil.CoerceInt(qRow.Find(".t_seeders").Text().Trim());
release.Peers = ParseUtil.CoerceInt(qRow.Find(".t_leechers").Text().Trim()) + release.Seeders;
releases.Add(release);
2015-05-04 04:12:14 +00:00
}
}
catch (Exception ex)
{
OnResultParsingError(this, results, ex);
throw ex;
2015-05-04 04:12:14 +00:00
}
return releases.ToArray();
}
public async Task<byte[]> Download(Uri link)
{
var request = CreateHttpRequest(link);
var response = await client.SendAsync(request);
var bytes = await response.Content.ReadAsByteArrayAsync();
return bytes;
}
}
2015-04-16 01:16:16 +00:00
}