2013-05-02 01:59:09 +00:00
|
|
|
using System;
|
2013-07-31 03:41:46 +00:00
|
|
|
using System.Collections.Generic;
|
2013-12-01 21:33:53 +00:00
|
|
|
using System.Linq;
|
2014-05-19 19:14:41 +00:00
|
|
|
using System.Text.RegularExpressions;
|
2013-08-22 08:08:43 +00:00
|
|
|
using FluentValidation;
|
2015-05-31 07:35:25 +00:00
|
|
|
using FluentValidation.Results;
|
2014-12-02 06:26:25 +00:00
|
|
|
using NzbDrone.Common.Extensions;
|
2013-05-03 05:24:52 +00:00
|
|
|
using NzbDrone.Core.Annotations;
|
2013-09-21 07:09:26 +00:00
|
|
|
using NzbDrone.Core.ThingiProvider;
|
2013-08-22 08:08:43 +00:00
|
|
|
using NzbDrone.Core.Validation;
|
2013-05-02 01:59:09 +00:00
|
|
|
|
|
|
|
namespace NzbDrone.Core.Indexers.Newznab
|
|
|
|
{
|
2013-08-26 04:01:03 +00:00
|
|
|
public class NewznabSettingsValidator : AbstractValidator<NewznabSettings>
|
|
|
|
{
|
2013-12-01 21:33:53 +00:00
|
|
|
private static readonly string[] ApiKeyWhiteList =
|
|
|
|
{
|
|
|
|
"nzbs.org",
|
|
|
|
"nzb.su",
|
|
|
|
"dognzb.cr",
|
|
|
|
"nzbplanet.net",
|
|
|
|
"nzbid.org",
|
|
|
|
"nzbndx.com",
|
2013-12-02 22:41:19 +00:00
|
|
|
"nzbindex.in"
|
2013-12-01 21:33:53 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
private static bool ShouldHaveApiKey(NewznabSettings settings)
|
|
|
|
{
|
2013-12-02 22:11:17 +00:00
|
|
|
if (settings.Url == null)
|
|
|
|
{
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2013-12-01 21:33:53 +00:00
|
|
|
return ApiKeyWhiteList.Any(c => settings.Url.ToLowerInvariant().Contains(c));
|
|
|
|
}
|
|
|
|
|
2014-05-19 19:14:41 +00:00
|
|
|
private static readonly Regex AdditionalParametersRegex = new Regex(@"(&.+?\=.+?)+", RegexOptions.Compiled);
|
|
|
|
|
2013-08-26 04:01:03 +00:00
|
|
|
public NewznabSettingsValidator()
|
|
|
|
{
|
2015-05-31 07:35:25 +00:00
|
|
|
Custom(newznab =>
|
|
|
|
{
|
|
|
|
if (newznab.Categories.Empty() && newznab.AnimeCategories.Empty())
|
|
|
|
{
|
|
|
|
return new ValidationFailure("", "Either 'Categories' or 'Anime Categories' must be provided");
|
|
|
|
}
|
|
|
|
|
|
|
|
return null;
|
|
|
|
});
|
|
|
|
|
2013-08-26 04:01:03 +00:00
|
|
|
RuleFor(c => c.Url).ValidRootUrl();
|
2013-12-01 21:33:53 +00:00
|
|
|
RuleFor(c => c.ApiKey).NotEmpty().When(ShouldHaveApiKey);
|
2015-05-31 07:35:25 +00:00
|
|
|
RuleFor(c => c.AdditionalParameters).Matches(AdditionalParametersRegex)
|
|
|
|
.When(c => !c.AdditionalParameters.IsNullOrWhiteSpace());
|
2013-08-26 04:01:03 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-09-21 07:09:26 +00:00
|
|
|
public class NewznabSettings : IProviderConfig
|
2013-05-02 01:59:09 +00:00
|
|
|
{
|
2013-08-26 04:01:03 +00:00
|
|
|
private static readonly NewznabSettingsValidator Validator = new NewznabSettingsValidator();
|
|
|
|
|
2013-07-31 03:41:46 +00:00
|
|
|
public NewznabSettings()
|
|
|
|
{
|
2015-03-17 04:33:34 +00:00
|
|
|
Categories = new[] {5030, 5040};
|
2014-05-19 19:14:41 +00:00
|
|
|
AnimeCategories = Enumerable.Empty<Int32>();
|
2013-07-31 03:41:46 +00:00
|
|
|
}
|
|
|
|
|
2013-05-29 07:06:25 +00:00
|
|
|
[FieldDefinition(0, Label = "URL")]
|
2013-05-02 01:59:09 +00:00
|
|
|
public String Url { get; set; }
|
2013-05-03 05:24:52 +00:00
|
|
|
|
2013-05-29 07:06:25 +00:00
|
|
|
[FieldDefinition(1, Label = "API Key")]
|
2013-05-02 01:59:09 +00:00
|
|
|
public String ApiKey { get; set; }
|
|
|
|
|
2015-05-31 07:35:25 +00:00
|
|
|
[FieldDefinition(2, Label = "Categories", HelpText = "Comma Separated list, leave blank to disable standard/daily shows", Advanced = true)]
|
2013-07-31 03:41:46 +00:00
|
|
|
public IEnumerable<Int32> Categories { get; set; }
|
|
|
|
|
2015-05-31 07:35:25 +00:00
|
|
|
[FieldDefinition(3, Label = "Anime Categories", HelpText = "Comma Separated list, leave blank to disable anime", Advanced = true)]
|
2014-05-19 19:14:41 +00:00
|
|
|
public IEnumerable<Int32> AnimeCategories { get; set; }
|
|
|
|
|
2015-05-31 07:35:25 +00:00
|
|
|
[FieldDefinition(4, Label = "Additional Parameters", HelpText = "Additional newznab parameters", Advanced = true)]
|
2014-05-19 19:14:41 +00:00
|
|
|
public String AdditionalParameters { get; set; }
|
|
|
|
|
2015-03-17 04:33:34 +00:00
|
|
|
public NzbDroneValidationResult Validate()
|
2013-08-22 08:08:43 +00:00
|
|
|
{
|
2015-03-17 04:33:34 +00:00
|
|
|
return new NzbDroneValidationResult(Validator.Validate(this));
|
2013-08-22 08:08:43 +00:00
|
|
|
}
|
2013-05-02 01:59:09 +00:00
|
|
|
}
|
|
|
|
}
|