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.
134 lines
4.7 KiB
134 lines
4.7 KiB
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using FluentValidation.Results;
|
|
using NLog;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Common.Http;
|
|
using NzbDrone.Core.Configuration;
|
|
using NzbDrone.Core.Indexers.Newznab;
|
|
using NzbDrone.Core.Messaging.Events;
|
|
using NzbDrone.Core.ThingiProvider;
|
|
using NzbDrone.Core.Validation;
|
|
|
|
namespace NzbDrone.Core.Indexers.Torznab
|
|
{
|
|
public class Torznab : TorrentIndexerBase<TorznabSettings>
|
|
{
|
|
private readonly INewznabCapabilitiesProvider _capabilitiesProvider;
|
|
|
|
public override string Name => "Torznab";
|
|
public override string BaseUrl => "";
|
|
|
|
public override DownloadProtocol Protocol => DownloadProtocol.Torrent;
|
|
public override IndexerPrivacy Privacy => IndexerPrivacy.Private;
|
|
|
|
public override int PageSize => _capabilitiesProvider.GetCapabilities(Settings).LimitsDefault.Value;
|
|
|
|
public override IIndexerRequestGenerator GetRequestGenerator()
|
|
{
|
|
return new NewznabRequestGenerator(_capabilitiesProvider)
|
|
{
|
|
PageSize = PageSize,
|
|
Settings = Settings
|
|
};
|
|
}
|
|
|
|
public override IParseIndexerResponse GetParser()
|
|
{
|
|
return new TorznabRssParser();
|
|
}
|
|
|
|
public override IEnumerable<ProviderDefinition> DefaultDefinitions
|
|
{
|
|
get
|
|
{
|
|
yield return GetDefinition("AnimeTosho", GetSettings("https://feed.animetosho.org"));
|
|
yield return GetDefinition("HD4Free.xyz", GetSettings("http://hd4free.xyz"));
|
|
yield return GetDefinition("Generic Torznab", GetSettings(""));
|
|
}
|
|
}
|
|
|
|
public Torznab(INewznabCapabilitiesProvider capabilitiesProvider, IHttpClient httpClient, IEventAggregator eventAggregator, IIndexerStatusService indexerStatusService, IConfigService configService, Logger logger)
|
|
: base(httpClient, eventAggregator, indexerStatusService, configService, logger)
|
|
{
|
|
_capabilitiesProvider = capabilitiesProvider;
|
|
}
|
|
|
|
private IndexerDefinition GetDefinition(string name, TorznabSettings settings)
|
|
{
|
|
return new IndexerDefinition
|
|
{
|
|
Enable = true,
|
|
Name = name,
|
|
Implementation = GetType().Name,
|
|
Settings = settings,
|
|
Protocol = DownloadProtocol.Usenet,
|
|
SupportsRss = SupportsRss,
|
|
SupportsSearch = SupportsSearch,
|
|
SupportsRedirect = SupportsRedirect,
|
|
Capabilities = new IndexerCapabilities()
|
|
};
|
|
}
|
|
|
|
private TorznabSettings GetSettings(string url, string apiPath = null)
|
|
{
|
|
var settings = new TorznabSettings { BaseUrl = url };
|
|
|
|
if (apiPath.IsNotNullOrWhiteSpace())
|
|
{
|
|
settings.ApiPath = apiPath;
|
|
}
|
|
|
|
return settings;
|
|
}
|
|
|
|
protected override async Task Test(List<ValidationFailure> failures)
|
|
{
|
|
await base.Test(failures);
|
|
if (failures.HasErrors())
|
|
{
|
|
return;
|
|
}
|
|
|
|
failures.AddIfNotNull(TestCapabilities());
|
|
}
|
|
|
|
protected static List<int> CategoryIds(IndexerCapabilitiesCategories categories)
|
|
{
|
|
var l = categories.GetTorznabCategoryTree().Select(c => c.Id).ToList();
|
|
|
|
return l;
|
|
}
|
|
|
|
protected virtual ValidationFailure TestCapabilities()
|
|
{
|
|
try
|
|
{
|
|
var capabilities = _capabilitiesProvider.GetCapabilities(Settings);
|
|
|
|
if (capabilities.SearchParams != null && capabilities.SearchParams.Contains(SearchParam.Q))
|
|
{
|
|
return null;
|
|
}
|
|
|
|
if (capabilities.MovieSearchParams != null &&
|
|
new[] { MovieSearchParam.Q, MovieSearchParam.ImdbId }.Any(v => capabilities.MovieSearchParams.Contains(v)) &&
|
|
new[] { MovieSearchParam.ImdbTitle, MovieSearchParam.ImdbYear }.All(v => capabilities.MovieSearchParams.Contains(v)))
|
|
{
|
|
return null;
|
|
}
|
|
|
|
return new ValidationFailure(string.Empty, "This indexer does not support searching for movies :(. Tell your indexer staff to enable this or force add the indexer by disabling search, adding the indexer and then enabling it again.");
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.Warn(ex, "Unable to connect to indexer: " + ex.Message);
|
|
|
|
return new ValidationFailure(string.Empty, "Unable to connect to indexer, check the log for more details");
|
|
}
|
|
}
|
|
}
|
|
}
|