mirror of https://github.com/Radarr/Radarr
91 lines
3.4 KiB
C#
91 lines
3.4 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Text.RegularExpressions;
|
|
using Equ;
|
|
using FluentValidation;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Core.Annotations;
|
|
using NzbDrone.Core.Indexers.Newznab;
|
|
using NzbDrone.Core.Parser.Model;
|
|
using NzbDrone.Core.Validation;
|
|
|
|
namespace NzbDrone.Core.Indexers.Torznab
|
|
{
|
|
public class TorznabSettingsValidator : AbstractValidator<TorznabSettings>
|
|
{
|
|
private static readonly string[] ApiKeyWhiteList = Array.Empty<string>();
|
|
|
|
private static bool ShouldHaveApiKey(TorznabSettings settings)
|
|
{
|
|
return settings.BaseUrl != null && ApiKeyWhiteList.Any(c => settings.BaseUrl.ToLowerInvariant().Contains(c));
|
|
}
|
|
|
|
private static readonly Regex AdditionalParametersRegex = new (@"(&.+?\=.+?)+", RegexOptions.Compiled);
|
|
|
|
public TorznabSettingsValidator()
|
|
{
|
|
RuleFor(c => c).Custom((c, context) =>
|
|
{
|
|
if (c.Categories.Empty())
|
|
{
|
|
context.AddFailure("'Categories' must be provided");
|
|
}
|
|
});
|
|
|
|
RuleFor(c => c.BaseUrl).ValidRootUrl();
|
|
RuleFor(c => c.ApiPath).ValidUrlBase("/api");
|
|
RuleFor(c => c.ApiKey).NotEmpty().When(ShouldHaveApiKey);
|
|
RuleFor(c => c.AdditionalParameters).Matches(AdditionalParametersRegex)
|
|
.When(c => !c.AdditionalParameters.IsNullOrWhiteSpace());
|
|
|
|
RuleFor(c => c.SeedCriteria).SetValidator(_ => new SeedCriteriaSettingsValidator());
|
|
}
|
|
}
|
|
|
|
public class TorznabSettings : NewznabSettings, ITorrentIndexerSettings, IEquatable<TorznabSettings>
|
|
{
|
|
private static readonly TorznabSettingsValidator Validator = new ();
|
|
|
|
private static readonly MemberwiseEqualityComparer<TorznabSettings> Comparer = MemberwiseEqualityComparer<TorznabSettings>.ByProperties;
|
|
|
|
public TorznabSettings()
|
|
{
|
|
MinimumSeeders = IndexerDefaults.MINIMUM_SEEDERS;
|
|
RequiredFlags = Array.Empty<int>();
|
|
}
|
|
|
|
[FieldDefinition(8, Type = FieldType.Number, Label = "Minimum Seeders", HelpText = "Minimum number of seeders required.", Advanced = true)]
|
|
public int MinimumSeeders { get; set; }
|
|
|
|
[FieldDefinition(9)]
|
|
public SeedCriteriaSettings SeedCriteria { get; set; } = new ();
|
|
|
|
[FieldDefinition(10, Type = FieldType.Checkbox, Label = "IndexerSettingsRejectBlocklistedTorrentHashes", HelpText = "IndexerSettingsRejectBlocklistedTorrentHashesHelpText", Advanced = true)]
|
|
public bool RejectBlocklistedTorrentHashesWhileGrabbing { get; set; }
|
|
|
|
[FieldDefinition(11, Type = FieldType.Select, SelectOptions = typeof(IndexerFlags), Label = "Required Flags", HelpText = "What indexer flags are required?", HelpLink = "https://wiki.servarr.com/radarr/settings#indexer-flags", Advanced = true)]
|
|
public IEnumerable<int> RequiredFlags { get; set; }
|
|
|
|
public override NzbDroneValidationResult Validate()
|
|
{
|
|
return new NzbDroneValidationResult(Validator.Validate(this));
|
|
}
|
|
|
|
public bool Equals(TorznabSettings other)
|
|
{
|
|
return Comparer.Equals(this, other);
|
|
}
|
|
|
|
public override bool Equals(object obj)
|
|
{
|
|
return Equals(obj as TorznabSettings);
|
|
}
|
|
|
|
public override int GetHashCode()
|
|
{
|
|
return Comparer.GetHashCode(this);
|
|
}
|
|
}
|
|
}
|