using System.Collections.Generic; using System.Linq; using FluentValidation; using FluentValidation.Results; using Lidarr.Http.REST; using Lidarr.Http.REST.Attributes; using Microsoft.AspNetCore.Mvc; using NzbDrone.Common.Serializer; using NzbDrone.Core.ThingiProvider; using NzbDrone.Core.Validation; namespace Lidarr.Api.V1 { public abstract class ProviderControllerBase : RestController where TProviderDefinition : ProviderDefinition, new() where TProvider : IProvider where TProviderResource : ProviderResource, new() { private readonly IProviderFactory _providerFactory; private readonly ProviderResourceMapper _resourceMapper; protected ProviderControllerBase(IProviderFactory providerFactory, string resource, ProviderResourceMapper resourceMapper) { _providerFactory = providerFactory; _resourceMapper = resourceMapper; SharedValidator.RuleFor(c => c.Name).NotEmpty(); SharedValidator.RuleFor(c => c.Name).Must((v, c) => !_providerFactory.All().Any(p => p.Name == c && p.Id != v.Id)).WithMessage("Should be unique"); SharedValidator.RuleFor(c => c.Implementation).NotEmpty(); SharedValidator.RuleFor(c => c.ConfigContract).NotEmpty(); PostValidator.RuleFor(c => c.Fields).NotNull(); } public override TProviderResource GetResourceById(int id) { var definition = _providerFactory.Get(id); _providerFactory.SetProviderCharacteristics(definition); return _resourceMapper.ToResource(definition); } [HttpGet] public List GetAll() { var providerDefinitions = _providerFactory.All().OrderBy(p => p.ImplementationName); var result = new List(providerDefinitions.Count()); foreach (var definition in providerDefinitions) { _providerFactory.SetProviderCharacteristics(definition); result.Add(_resourceMapper.ToResource(definition)); } return result.OrderBy(p => p.Name).ToList(); } [RestPostById] public ActionResult CreateProvider(TProviderResource providerResource) { var providerDefinition = GetDefinition(providerResource, false); if (providerDefinition.Enable) { Test(providerDefinition, false); } providerDefinition = _providerFactory.Create(providerDefinition); return Created(providerDefinition.Id); } [RestPutById] public ActionResult UpdateProvider(TProviderResource providerResource) { var providerDefinition = GetDefinition(providerResource, false); var existingDefinition = _providerFactory.Get(providerDefinition.Id); // Only test existing definitions if it was previously disabled if (providerDefinition.Enable && !existingDefinition.Enable) { Test(providerDefinition, false); } _providerFactory.Update(providerDefinition); return Accepted(providerResource.Id); } private TProviderDefinition GetDefinition(TProviderResource providerResource, bool includeWarnings = false, bool validate = true) { var definition = _resourceMapper.ToModel(providerResource); if (validate) { Validate(definition, includeWarnings); } return definition; } [RestDeleteById] public object DeleteProvider(int id) { _providerFactory.Delete(id); return new { }; } [HttpGet("schema")] public List GetTemplates() { var defaultDefinitions = _providerFactory.GetDefaultDefinitions().OrderBy(p => p.ImplementationName).ToList(); var result = new List(defaultDefinitions.Count()); foreach (var providerDefinition in defaultDefinitions) { var providerResource = _resourceMapper.ToResource(providerDefinition); var presetDefinitions = _providerFactory.GetPresetDefinitions(providerDefinition); providerResource.Presets = presetDefinitions .Select(v => _resourceMapper.ToResource(v)) .ToList(); result.Add(providerResource); } return result; } [SkipValidation(true, false)] [HttpPost("test")] public object Test([FromBody] TProviderResource providerResource) { var providerDefinition = GetDefinition(providerResource, true); Test(providerDefinition, true); return "{}"; } [HttpPost("testall")] public IActionResult TestAll() { var providerDefinitions = _providerFactory.All() .Where(c => c.Settings.Validate().IsValid && c.Enable) .ToList(); var result = new List(); foreach (var definition in providerDefinitions) { var validationResult = _providerFactory.Test(definition); result.Add(new ProviderTestAllResult { Id = definition.Id, ValidationFailures = validationResult.Errors.ToList() }); } return result.Any(c => !c.IsValid) ? BadRequest(result) : Ok(result); } [SkipValidation] [HttpPost("action/{name}")] public IActionResult RequestAction(string name, [FromBody] TProviderResource resource) { var providerDefinition = GetDefinition(resource, true, false); var query = Request.Query.ToDictionary(x => x.Key, x => x.Value.ToString()); var data = _providerFactory.RequestAction(providerDefinition, name, query); return Content(data.ToJson(), "application/json"); } protected virtual void Validate(TProviderDefinition definition, bool includeWarnings) { var validationResult = definition.Settings.Validate(); VerifyValidationResult(validationResult, includeWarnings); } protected virtual void Test(TProviderDefinition definition, bool includeWarnings) { var validationResult = _providerFactory.Test(definition); VerifyValidationResult(validationResult, includeWarnings); } protected void VerifyValidationResult(ValidationResult validationResult, bool includeWarnings) { var result = validationResult as NzbDroneValidationResult; if (result == null) { result = new NzbDroneValidationResult(validationResult.Errors); } if (includeWarnings && (!result.IsValid || result.HasWarnings)) { throw new ValidationException(result.Failures); } if (!result.IsValid) { throw new ValidationException(result.Errors); } } } }