New: Add Download Client validation for indexers

This commit is contained in:
Bogdan 2023-10-25 15:11:41 +03:00 committed by Mark McDowall
parent 653aede0b7
commit e53b7f8c94
2 changed files with 30 additions and 1 deletions

View File

@ -0,0 +1,27 @@
using FluentValidation.Validators;
using NzbDrone.Core.Download;
namespace NzbDrone.Core.Validation
{
public class DownloadClientExistsValidator : PropertyValidator
{
private readonly IDownloadClientFactory _downloadClientFactory;
public DownloadClientExistsValidator(IDownloadClientFactory downloadClientFactory)
{
_downloadClientFactory = downloadClientFactory;
}
protected override string GetDefaultMessageTemplate() => "Download Client does not exist";
protected override bool IsValid(PropertyValidatorContext context)
{
if (context?.PropertyValue == null || (int)context.PropertyValue == 0)
{
return true;
}
return _downloadClientFactory.Exists((int)context.PropertyValue);
}
}
}

View File

@ -1,4 +1,5 @@
using NzbDrone.Core.Indexers; using NzbDrone.Core.Indexers;
using NzbDrone.Core.Validation;
using Sonarr.Http; using Sonarr.Http;
namespace Sonarr.Api.V3.Indexers namespace Sonarr.Api.V3.Indexers
@ -9,9 +10,10 @@ namespace Sonarr.Api.V3.Indexers
public static readonly IndexerResourceMapper ResourceMapper = new (); public static readonly IndexerResourceMapper ResourceMapper = new ();
public static readonly IndexerBulkResourceMapper BulkResourceMapper = new (); public static readonly IndexerBulkResourceMapper BulkResourceMapper = new ();
public IndexerController(IndexerFactory indexerFactory) public IndexerController(IndexerFactory indexerFactory, DownloadClientExistsValidator downloadClientExistsValidator)
: base(indexerFactory, "indexer", ResourceMapper, BulkResourceMapper) : base(indexerFactory, "indexer", ResourceMapper, BulkResourceMapper)
{ {
SharedValidator.RuleFor(c => c.DownloadClientId).SetValidator(downloadClientExistsValidator);
} }
} }
} }