You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
155 lines
5.5 KiB
155 lines
5.5 KiB
7 years ago
|
using System;
|
||
|
using System.Collections.Generic;
|
||
|
using System.Xml;
|
||
|
using System.Xml.Linq;
|
||
|
using NLog;
|
||
|
using NzbDrone.Common.Cache;
|
||
|
using NzbDrone.Common.Extensions;
|
||
|
using NzbDrone.Common.Http;
|
||
|
using NzbDrone.Common.Serializer;
|
||
|
using NzbDrone.Core.Indexers.Newznab;
|
||
|
|
||
|
namespace NzbDrone.Core.Indexers.Headphones
|
||
|
{
|
||
|
public interface IHeadphonesCapabilitiesProvider
|
||
|
{
|
||
|
HeadphonesCapabilities GetCapabilities(HeadphonesSettings settings);
|
||
|
}
|
||
|
|
||
|
public class HeadphonesCapabilitiesProvider : IHeadphonesCapabilitiesProvider
|
||
|
{
|
||
|
private readonly ICached<HeadphonesCapabilities> _capabilitiesCache;
|
||
|
private readonly IHttpClient _httpClient;
|
||
|
private readonly Logger _logger;
|
||
|
|
||
|
public HeadphonesCapabilitiesProvider(ICacheManager cacheManager, IHttpClient httpClient, Logger logger)
|
||
|
{
|
||
|
_capabilitiesCache = cacheManager.GetCache<HeadphonesCapabilities>(GetType());
|
||
|
_httpClient = httpClient;
|
||
|
_logger = logger;
|
||
|
}
|
||
|
|
||
|
public HeadphonesCapabilities GetCapabilities(HeadphonesSettings indexerSettings)
|
||
|
{
|
||
|
var key = indexerSettings.ToJson();
|
||
|
var capabilities = _capabilitiesCache.Get(key, () => FetchCapabilities(indexerSettings), TimeSpan.FromDays(7));
|
||
|
|
||
|
return capabilities;
|
||
|
}
|
||
|
|
||
|
private HeadphonesCapabilities FetchCapabilities(HeadphonesSettings indexerSettings)
|
||
|
{
|
||
|
var capabilities = new HeadphonesCapabilities();
|
||
|
|
||
|
var url = string.Format("{0}{1}?t=caps", indexerSettings.BaseUrl.TrimEnd('/'), indexerSettings.ApiPath.TrimEnd('/'));
|
||
|
|
||
|
if (indexerSettings.ApiKey.IsNotNullOrWhiteSpace())
|
||
|
{
|
||
|
url += "&apikey=" + indexerSettings.ApiKey;
|
||
|
}
|
||
|
|
||
|
var request = new HttpRequest(url, HttpAccept.Rss);
|
||
|
|
||
|
request.AddBasicAuthentication(indexerSettings.Username, indexerSettings.Password);
|
||
|
|
||
|
HttpResponse response;
|
||
|
|
||
|
try
|
||
|
{
|
||
|
response = _httpClient.Get(request);
|
||
|
}
|
||
|
catch (Exception ex)
|
||
|
{
|
||
|
_logger.Debug(ex, "Failed to get headphones api capabilities from {0}", indexerSettings.BaseUrl);
|
||
|
throw;
|
||
|
}
|
||
|
|
||
|
try
|
||
|
{
|
||
|
capabilities = ParseCapabilities(response);
|
||
|
}
|
||
|
catch (XmlException ex)
|
||
|
{
|
||
|
_logger.Debug(ex, "Failed to parse headphones api capabilities for {0}", indexerSettings.BaseUrl);
|
||
|
ex.WithData(response);
|
||
|
throw;
|
||
|
}
|
||
|
catch (Exception ex)
|
||
|
{
|
||
|
_logger.Error(ex, "Failed to determine headphones api capabilities for {0}, using the defaults instead till Lidarr restarts", indexerSettings.BaseUrl);
|
||
|
}
|
||
|
|
||
|
return capabilities;
|
||
|
}
|
||
|
|
||
|
private HeadphonesCapabilities ParseCapabilities(HttpResponse response)
|
||
|
{
|
||
|
var capabilities = new HeadphonesCapabilities();
|
||
|
|
||
|
var xDoc = XDocument.Parse(response.Content);
|
||
|
|
||
|
if (xDoc == null)
|
||
|
{
|
||
|
throw new XmlException("Invalid XML");
|
||
|
}
|
||
|
|
||
|
var xmlRoot = xDoc.Element("caps");
|
||
|
|
||
|
if (xmlRoot == null)
|
||
|
{
|
||
|
throw new XmlException("Unexpected XML");
|
||
|
}
|
||
|
|
||
|
var xmlLimits = xmlRoot.Element("limits");
|
||
|
if (xmlLimits != null)
|
||
|
{
|
||
|
capabilities.DefaultPageSize = int.Parse(xmlLimits.Attribute("default").Value);
|
||
|
capabilities.MaxPageSize = int.Parse(xmlLimits.Attribute("max").Value);
|
||
|
}
|
||
|
|
||
|
var xmlSearching = xmlRoot.Element("searching");
|
||
|
if (xmlSearching != null)
|
||
|
{
|
||
|
var xmlBasicSearch = xmlSearching.Element("search");
|
||
|
if (xmlBasicSearch == null || xmlBasicSearch.Attribute("available").Value != "yes")
|
||
|
{
|
||
|
capabilities.SupportedSearchParameters = null;
|
||
|
}
|
||
|
else if (xmlBasicSearch.Attribute("supportedParams") != null)
|
||
|
{
|
||
|
capabilities.SupportedSearchParameters = xmlBasicSearch.Attribute("supportedParams").Value.Split(',');
|
||
|
}
|
||
|
}
|
||
|
|
||
|
var xmlCategories = xmlRoot.Element("categories");
|
||
|
if (xmlCategories != null)
|
||
|
{
|
||
|
foreach (var xmlCategory in xmlCategories.Elements("category"))
|
||
|
{
|
||
|
var cat = new NewznabCategory
|
||
|
{
|
||
|
Id = int.Parse(xmlCategory.Attribute("id").Value),
|
||
|
Name = xmlCategory.Attribute("name").Value,
|
||
|
Description = xmlCategory.Attribute("description") != null ? xmlCategory.Attribute("description").Value : string.Empty,
|
||
|
Subcategories = new List<NewznabCategory>()
|
||
|
};
|
||
|
|
||
|
foreach (var xmlSubcat in xmlCategory.Elements("subcat"))
|
||
|
{
|
||
|
cat.Subcategories.Add(new NewznabCategory
|
||
|
{
|
||
|
Id = int.Parse(xmlSubcat.Attribute("id").Value),
|
||
|
Name = xmlSubcat.Attribute("name").Value,
|
||
|
Description = xmlSubcat.Attribute("description") != null ? xmlSubcat.Attribute("description").Value : string.Empty
|
||
|
});
|
||
|
}
|
||
|
|
||
|
capabilities.Categories.Add(cat);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return capabilities;
|
||
|
}
|
||
|
}
|
||
|
}
|