using System; using FluentValidation; using NzbDrone.Common.Extensions; using NzbDrone.Core.Annotations; using NzbDrone.Core.Validation; namespace NzbDrone.Core.ImportLists.Trakt { public class TraktSettingsBaseValidator : AbstractValidator where TSettings : TraktSettingsBase { public TraktSettingsBaseValidator() { RuleFor(c => c.Link).ValidRootUrl(); RuleFor(c => c.AccessToken).NotEmpty() .OverridePropertyName("SignIn") .WithMessage("Must authenticate with Trakt"); RuleFor(c => c.RefreshToken).NotEmpty() .OverridePropertyName("SignIn") .WithMessage("Must authenticate with Trakt") .When(c => c.AccessToken.IsNotNullOrWhiteSpace()); RuleFor(c => c.Expires).NotEmpty() .OverridePropertyName("SignIn") .WithMessage("Must authenticate with Trakt") .When(c => c.AccessToken.IsNotNullOrWhiteSpace() && c.RefreshToken.IsNotNullOrWhiteSpace()); // Limit not smaller than 1 and not larger than 100 RuleFor(c => c.Limit) .GreaterThan(0) .WithMessage("Must be integer greater than 0"); } } public class TraktSettingsBase : ImportListSettingsBase where TSettings : TraktSettingsBase { private static readonly TraktSettingsBaseValidator Validator = new (); public TraktSettingsBase() { SignIn = "startOAuth"; Limit = 100; } public string Link => "https://api.trakt.tv"; public virtual string Scope => ""; [FieldDefinition(0, Label = "Access Token", Type = FieldType.Textbox, Hidden = HiddenType.Hidden)] public string AccessToken { get; set; } [FieldDefinition(0, Label = "Refresh Token", Type = FieldType.Textbox, Hidden = HiddenType.Hidden)] public string RefreshToken { get; set; } [FieldDefinition(0, Label = "Expires", Type = FieldType.Textbox, Hidden = HiddenType.Hidden)] public DateTime Expires { get; set; } [FieldDefinition(0, Label = "Auth User", Type = FieldType.Textbox, Hidden = HiddenType.Hidden)] public string AuthUser { get; set; } [FieldDefinition(5, Label = "Limit", HelpText = "Limit the number of movies to get")] public int Limit { get; set; } [FieldDefinition(6, Label = "Additional Parameters", HelpText = "Additional Trakt API parameters", Advanced = true)] public string TraktAdditionalParameters { get; set; } [FieldDefinition(99, Label = "Authenticate with Trakt", Type = FieldType.OAuth)] public string SignIn { get; set; } public override NzbDroneValidationResult Validate() { return new NzbDroneValidationResult(Validator.Validate((TSettings)this)); } } }