58 lines
1.5 KiB
C#
58 lines
1.5 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using FluentValidation;
|
|
using FluentValidation.Results;
|
|
using NzbDrone.Core.Annotations;
|
|
using NzbDrone.Core.ThingiProvider;
|
|
using NzbDrone.Core.Validation;
|
|
|
|
namespace NzbDrone.Core.Indexers.Newznab
|
|
{
|
|
public class NewznabSettingsValidator : AbstractValidator<NewznabSettings>
|
|
{
|
|
private static readonly string[] ApiKeyWhiteList =
|
|
{
|
|
"nzbs.org",
|
|
"nzb.su",
|
|
"dognzb.cr",
|
|
"nzbplanet.net",
|
|
"nzbid.org",
|
|
"nzbndx.com",
|
|
};
|
|
|
|
private static bool ShouldHaveApiKey(NewznabSettings settings)
|
|
{
|
|
return ApiKeyWhiteList.Any(c => settings.Url.ToLowerInvariant().Contains(c));
|
|
}
|
|
|
|
public NewznabSettingsValidator()
|
|
{
|
|
RuleFor(c => c.Url).ValidRootUrl();
|
|
RuleFor(c => c.ApiKey).NotEmpty().When(ShouldHaveApiKey);
|
|
}
|
|
}
|
|
|
|
public class NewznabSettings : IProviderConfig
|
|
{
|
|
private static readonly NewznabSettingsValidator Validator = new NewznabSettingsValidator();
|
|
|
|
public NewznabSettings()
|
|
{
|
|
Categories = new[] { 5030, 5040 };
|
|
}
|
|
|
|
[FieldDefinition(0, Label = "URL")]
|
|
public String Url { get; set; }
|
|
|
|
[FieldDefinition(1, Label = "API Key")]
|
|
public String ApiKey { get; set; }
|
|
|
|
public IEnumerable<Int32> Categories { get; set; }
|
|
|
|
public ValidationResult Validate()
|
|
{
|
|
return Validator.Validate(this);
|
|
}
|
|
}
|
|
} |