mirror of https://github.com/Sonarr/Sonarr
169 lines
5.1 KiB
C#
169 lines
5.1 KiB
C#
using System;
|
|
using System.Linq;
|
|
using System.Xml.Linq;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Core.Indexers.Exceptions;
|
|
using NzbDrone.Core.Parser.Model;
|
|
|
|
namespace NzbDrone.Core.Indexers.Torznab
|
|
{
|
|
public class TorznabRssParser : TorrentRssParser
|
|
{
|
|
public const String ns = "{http://torznab.com/schemas/2015/feed}";
|
|
|
|
protected override bool PreProcess(IndexerResponse indexerResponse)
|
|
{
|
|
var xdoc = XDocument.Parse(indexerResponse.Content);
|
|
var error = xdoc.Descendants("error").FirstOrDefault();
|
|
|
|
if (error == null) return true;
|
|
|
|
var code = Convert.ToInt32(error.Attribute("code").Value);
|
|
var errorMessage = error.Attribute("description").Value;
|
|
|
|
if (code >= 100 && code <= 199) throw new ApiKeyException("Invalid API key");
|
|
|
|
if (!indexerResponse.Request.Url.ToString().Contains("apikey=") && errorMessage == "Missing parameter")
|
|
{
|
|
throw new ApiKeyException("Indexer requires an API key");
|
|
}
|
|
|
|
if (errorMessage == "Request limit reached")
|
|
{
|
|
throw new RequestLimitReachedException("API limit reached");
|
|
}
|
|
|
|
throw new TorznabException("Torznab error detected: {0}", errorMessage);
|
|
}
|
|
|
|
protected override ReleaseInfo ProcessItem(XElement item, ReleaseInfo releaseInfo)
|
|
{
|
|
var torrentInfo = base.ProcessItem(item, releaseInfo) as TorrentInfo;
|
|
|
|
torrentInfo.TvRageId = GetTvRageId(item);
|
|
|
|
return torrentInfo;
|
|
}
|
|
|
|
protected override ReleaseInfo PostProcess(XElement item, ReleaseInfo releaseInfo)
|
|
{
|
|
var enclosureType = item.Element("enclosure").Attribute("type").Value;
|
|
if (enclosureType != "application/x-bittorrent")
|
|
{
|
|
throw new UnsupportedFeedException("Feed contains {0} instead of application/x-bittorrent", enclosureType);
|
|
}
|
|
|
|
return base.PostProcess(item, releaseInfo);
|
|
}
|
|
|
|
|
|
protected override String GetInfoUrl(XElement item)
|
|
{
|
|
return item.TryGetValue("comments").TrimEnd("#comments");
|
|
}
|
|
|
|
protected override String GetCommentUrl(XElement item)
|
|
{
|
|
return item.TryGetValue("comments");
|
|
}
|
|
|
|
protected override Int64 GetSize(XElement item)
|
|
{
|
|
Int64 size;
|
|
|
|
var sizeString = TryGetTorznabAttribute(item, "size");
|
|
if (!sizeString.IsNullOrWhiteSpace() && Int64.TryParse(sizeString, out size))
|
|
{
|
|
return size;
|
|
}
|
|
|
|
size = GetEnclosureLength(item);
|
|
|
|
return size;
|
|
}
|
|
|
|
protected override DateTime GetPublishDate(XElement item)
|
|
{
|
|
return base.GetPublishDate(item);
|
|
}
|
|
|
|
protected override string GetDownloadUrl(XElement item)
|
|
{
|
|
var url = base.GetDownloadUrl(item);
|
|
|
|
if (!Uri.IsWellFormedUriString(url, UriKind.Absolute))
|
|
{
|
|
url = item.Element("enclosure").Attribute("url").Value;
|
|
}
|
|
|
|
return url;
|
|
}
|
|
|
|
protected virtual Int32 GetTvRageId(XElement item)
|
|
{
|
|
var tvRageIdString = TryGetTorznabAttribute(item, "rageid");
|
|
Int32 tvRageId;
|
|
|
|
if (!tvRageIdString.IsNullOrWhiteSpace() && Int32.TryParse(tvRageIdString, out tvRageId))
|
|
{
|
|
return tvRageId;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
protected override String GetInfoHash(XElement item)
|
|
{
|
|
return TryGetTorznabAttribute(item, "infohash");
|
|
}
|
|
|
|
protected override String GetMagnetUrl(XElement item)
|
|
{
|
|
return TryGetTorznabAttribute(item, "magneturl");
|
|
}
|
|
|
|
protected override Int32? GetSeeders(XElement item)
|
|
{
|
|
var seeders = TryGetTorznabAttribute(item, "seeders");
|
|
|
|
if (seeders.IsNotNullOrWhiteSpace())
|
|
{
|
|
return Int32.Parse(seeders);
|
|
}
|
|
|
|
return base.GetSeeders(item);
|
|
}
|
|
|
|
protected override Int32? GetPeers(XElement item)
|
|
{
|
|
var peers = TryGetTorznabAttribute(item, "peers");
|
|
|
|
if (peers.IsNotNullOrWhiteSpace())
|
|
{
|
|
return Int32.Parse(peers);
|
|
}
|
|
|
|
var seeders = TryGetTorznabAttribute(item, "seeders");
|
|
var leechers = TryGetTorznabAttribute(item, "leechers");
|
|
|
|
if (seeders.IsNotNullOrWhiteSpace() && leechers.IsNotNullOrWhiteSpace())
|
|
{
|
|
return Int32.Parse(seeders) + Int32.Parse(leechers);
|
|
}
|
|
|
|
return base.GetPeers(item);
|
|
}
|
|
|
|
protected String TryGetTorznabAttribute(XElement item, String key, String defaultValue = "")
|
|
{
|
|
var attr = item.Elements(ns + "attr").SingleOrDefault(e => e.Attribute("name").Value.Equals(key, StringComparison.CurrentCultureIgnoreCase));
|
|
|
|
if (attr != null)
|
|
{
|
|
return attr.Attribute("value").Value;
|
|
}
|
|
|
|
return defaultValue;
|
|
}
|
|
}
|
|
}
|