2018-02-01 13:18:52 +00:00
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.Linq;
|
2015-03-12 20:01:54 +00:00
|
|
|
|
using System.Text.RegularExpressions;
|
|
|
|
|
using FluentValidation;
|
2015-09-26 08:45:13 +00:00
|
|
|
|
using FluentValidation.Results;
|
2015-03-12 20:01:54 +00:00
|
|
|
|
using NzbDrone.Common.Extensions;
|
2017-06-12 12:47:05 +00:00
|
|
|
|
using NzbDrone.Core.Annotations;
|
2015-09-26 08:45:13 +00:00
|
|
|
|
using NzbDrone.Core.Indexers.Newznab;
|
2018-02-01 13:18:52 +00:00
|
|
|
|
using NzbDrone.Core.Parser.Model;
|
2015-03-12 20:01:54 +00:00
|
|
|
|
using NzbDrone.Core.Validation;
|
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Indexers.Torznab
|
|
|
|
|
{
|
|
|
|
|
public class TorznabSettingsValidator : AbstractValidator<TorznabSettings>
|
|
|
|
|
{
|
|
|
|
|
private static readonly string[] ApiKeyWhiteList =
|
|
|
|
|
{
|
2015-05-04 17:39:02 +00:00
|
|
|
|
"hd4free.xyz",
|
2015-03-12 20:01:54 +00:00
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
private static bool ShouldHaveApiKey(TorznabSettings settings)
|
|
|
|
|
{
|
2017-06-12 12:47:05 +00:00
|
|
|
|
if (settings.BaseUrl == null)
|
2015-03-12 20:01:54 +00:00
|
|
|
|
{
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
|
2017-06-12 12:47:05 +00:00
|
|
|
|
return ApiKeyWhiteList.Any(c => settings.BaseUrl.ToLowerInvariant().Contains(c));
|
2015-03-12 20:01:54 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private static readonly Regex AdditionalParametersRegex = new Regex(@"(&.+?\=.+?)+", RegexOptions.Compiled);
|
|
|
|
|
|
|
|
|
|
public TorznabSettingsValidator()
|
|
|
|
|
{
|
2015-09-26 08:45:13 +00:00
|
|
|
|
Custom(newznab =>
|
|
|
|
|
{
|
|
|
|
|
if (newznab.Categories.Empty() && newznab.AnimeCategories.Empty())
|
|
|
|
|
{
|
|
|
|
|
return new ValidationFailure("", "Either 'Categories' or 'Anime Categories' must be provided");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return null;
|
|
|
|
|
});
|
|
|
|
|
|
2017-06-12 12:47:05 +00:00
|
|
|
|
RuleFor(c => c.BaseUrl).ValidRootUrl();
|
2015-03-12 20:01:54 +00:00
|
|
|
|
RuleFor(c => c.ApiKey).NotEmpty().When(ShouldHaveApiKey);
|
2015-09-26 08:45:13 +00:00
|
|
|
|
RuleFor(c => c.AdditionalParameters).Matches(AdditionalParametersRegex)
|
|
|
|
|
.When(c => !c.AdditionalParameters.IsNullOrWhiteSpace());
|
2015-03-12 20:01:54 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2017-06-12 12:47:05 +00:00
|
|
|
|
public class TorznabSettings : NewznabSettings, ITorrentIndexerSettings
|
2015-03-12 20:01:54 +00:00
|
|
|
|
{
|
|
|
|
|
private static readonly TorznabSettingsValidator Validator = new TorznabSettingsValidator();
|
|
|
|
|
|
2017-06-12 12:47:05 +00:00
|
|
|
|
public TorznabSettings()
|
|
|
|
|
{
|
|
|
|
|
MinimumSeeders = IndexerDefaults.MINIMUM_SEEDERS;
|
|
|
|
|
}
|
|
|
|
|
|
2017-10-07 18:09:22 +00:00
|
|
|
|
[FieldDefinition(7, Type = FieldType.Textbox, Label = "Minimum Seeders", HelpText = "Minimum number of seeders required.", Advanced = true)]
|
2017-06-12 12:47:05 +00:00
|
|
|
|
public int MinimumSeeders { get; set; }
|
2018-02-01 13:18:52 +00:00
|
|
|
|
|
|
|
|
|
[FieldDefinition(8, Type = FieldType.Tag, SelectOptions = typeof(IndexerFlags), Label = "Required Flags", HelpText = "What indexer flags are required?", Advanced = true)]
|
|
|
|
|
public IEnumerable<int> RequiredFlags { get; set; }
|
2017-06-12 12:47:05 +00:00
|
|
|
|
|
2016-02-11 20:49:22 +00:00
|
|
|
|
public override NzbDroneValidationResult Validate()
|
2015-03-12 20:01:54 +00:00
|
|
|
|
{
|
2015-03-17 04:33:34 +00:00
|
|
|
|
return new NzbDroneValidationResult(Validator.Validate(this));
|
2015-03-12 20:01:54 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|