Jackett/src/Jackett.Common/Indexers/Abstract/CouchPotatoTracker.cs

129 lines
4.6 KiB
C#
Raw Normal View History

2017-09-15 16:57:43 +00:00
using System;
using System.Collections.Generic;
using System.Collections.Specialized;
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.Linq;
using NLog;
2017-09-15 16:57:43 +00:00
namespace Jackett.Common.Indexers.Abstract
2017-09-15 16:57:43 +00:00
{
public abstract class CouchPotatoTracker : BaseWebIndexer
{
protected string endpoint;
protected string APIUrl { get { return SiteLink + endpoint; } }
private new ConfigurationDataUserPasskey configData
2017-09-15 16:57:43 +00:00
{
get { return (ConfigurationDataUserPasskey)base.configData; }
set { base.configData = value; }
}
public CouchPotatoTracker(IIndexerConfigurationService configService, WebClient client, Logger logger, IProtectionService p, ConfigurationDataUserPasskey configData, string name, string description, string link, string endpoint)
2017-09-15 16:57:43 +00:00
: base(name: name,
description: description,
link: link,
caps: new TorznabCapabilities(),
configService: configService,
client: client,
logger: logger,
p: p,
configData: configData
)
2017-09-15 16:57:43 +00:00
{
this.endpoint = endpoint;
2019-05-11 03:27:25 +00:00
TorznabCaps.SupportsImdbMovieSearch = true;
2017-09-15 16:57:43 +00:00
}
public override async Task<IndexerConfigurationStatus> ApplyConfiguration(JToken configJson)
{
LoadValuesFromJson(configJson);
IsConfigured = true;
SaveConfig();
return await Task.FromResult(IndexerConfigurationStatus.RequiresTesting);
2017-09-15 16:57:43 +00:00
}
2017-09-19 09:32:12 +00:00
protected virtual string GetSearchString(TorznabQuery query)
{
// can be overriden to alter the search string
return query.GetQueryString();
}
2017-09-15 16:57:43 +00:00
protected override async Task<IEnumerable<ReleaseInfo>> PerformQuery(TorznabQuery query)
{
var releases = new List<ReleaseInfo>();
2017-09-19 09:32:12 +00:00
var searchString = GetSearchString(query);
2017-09-15 16:57:43 +00:00
var searchUrl = APIUrl;
var queryCollection = new NameValueCollection();
2020-02-09 02:35:16 +00:00
2017-09-15 16:57:43 +00:00
if (!string.IsNullOrEmpty(query.ImdbID))
2020-02-09 02:35:16 +00:00
{
2018-08-25 16:53:47 +00:00
queryCollection.Add("imdbid", query.ImdbID);
2017-09-15 16:57:43 +00:00
}
if (searchString != null)
{
queryCollection.Add("search", searchString);
}
queryCollection.Add("passkey", configData.Passkey.Value);
queryCollection.Add("user", configData.Username.Value);
searchUrl += "?" + queryCollection.GetQueryString();
var response = await RequestStringWithCookiesAndRetry(searchUrl);
JObject json = null;
try
{
json = JObject.Parse(response.Content);
}
catch (Exception ex)
{
throw new Exception("Error while parsing json: " + response.Content, ex);
}
var error = (string)json["error"];
if (error != null)
throw new Exception(error);
2017-09-19 09:32:12 +00:00
if ((int)json["total_results"] == 0)
return releases;
2017-09-15 16:57:43 +00:00
try
{
foreach (JObject r in json["results"])
{
var release = new ReleaseInfo();
release.Title = (string)r["release_name"];
release.Comments = new Uri((string)r["details_url"]);
release.Link = new Uri((string)r["download_url"]);
release.Guid = release.Link;
release.Imdb = ParseUtil.GetImdbID((string)r["imdb_id"]);
var freeleech = (bool)r["freeleech"];
if (freeleech)
release.DownloadVolumeFactor = 0;
else
release.DownloadVolumeFactor = 1;
release.UploadVolumeFactor = 1;
var type = (string)r["type"];
release.Category = MapTrackerCatToNewznab(type);
release.Size = (long?)r["size"] * 1024 * 1024;
release.Seeders = (int?)r["seeders"];
release.Peers = release.Seeders + (int?)r["leechers"];
release.PublishDate = DateTimeUtil.FromUnknown((string)r["publish_date"]);
releases.Add(release);
}
}
catch (Exception ex)
{
OnParseError(response.Content, ex);
}
return releases;
}
}
}