You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
116 lines
3.6 KiB
116 lines
3.6 KiB
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using FluentValidation;
|
|
using FluentValidation.Results;
|
|
using Lidarr.Http;
|
|
using Lidarr.Http.REST;
|
|
using Lidarr.Http.REST.Attributes;
|
|
using Microsoft.AspNetCore.Mvc;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Core.AutoTagging;
|
|
using NzbDrone.Core.AutoTagging.Specifications;
|
|
using NzbDrone.Core.Validation;
|
|
|
|
namespace Lidarr.Api.V1.AutoTagging
|
|
{
|
|
[V1ApiController]
|
|
public class AutoTaggingController : RestController<AutoTaggingResource>
|
|
{
|
|
private readonly IAutoTaggingService _autoTaggingService;
|
|
private readonly List<IAutoTaggingSpecification> _specifications;
|
|
|
|
public AutoTaggingController(IAutoTaggingService autoTaggingService,
|
|
List<IAutoTaggingSpecification> specifications)
|
|
{
|
|
_autoTaggingService = autoTaggingService;
|
|
_specifications = specifications;
|
|
|
|
SharedValidator.RuleFor(c => c.Name).NotEmpty();
|
|
SharedValidator.RuleFor(c => c.Name)
|
|
.Must((v, c) => !_autoTaggingService.All().Any(f => f.Name == c && f.Id != v.Id)).WithMessage("Must be unique.");
|
|
SharedValidator.RuleFor(c => c.Tags).NotEmpty();
|
|
SharedValidator.RuleFor(c => c.Specifications).NotEmpty();
|
|
SharedValidator.RuleFor(c => c).Custom((autoTag, context) =>
|
|
{
|
|
if (!autoTag.Specifications.Any())
|
|
{
|
|
context.AddFailure("Must contain at least one Condition");
|
|
}
|
|
|
|
if (autoTag.Specifications.Any(s => s.Name.IsNullOrWhiteSpace()))
|
|
{
|
|
context.AddFailure("Condition name(s) cannot be empty or consist of only spaces");
|
|
}
|
|
});
|
|
}
|
|
|
|
public override AutoTaggingResource GetResourceById(int id)
|
|
{
|
|
return _autoTaggingService.GetById(id).ToResource();
|
|
}
|
|
|
|
[RestPostById]
|
|
[Consumes("application/json")]
|
|
public ActionResult<AutoTaggingResource> Create(AutoTaggingResource autoTagResource)
|
|
{
|
|
var model = autoTagResource.ToModel(_specifications);
|
|
|
|
Validate(model);
|
|
|
|
return Created(_autoTaggingService.Insert(model).Id);
|
|
}
|
|
|
|
[RestPutById]
|
|
[Consumes("application/json")]
|
|
public ActionResult<AutoTaggingResource> Update(AutoTaggingResource resource)
|
|
{
|
|
var model = resource.ToModel(_specifications);
|
|
|
|
Validate(model);
|
|
|
|
_autoTaggingService.Update(model);
|
|
|
|
return Accepted(model.Id);
|
|
}
|
|
|
|
[HttpGet]
|
|
[Produces("application/json")]
|
|
public List<AutoTaggingResource> GetAll()
|
|
{
|
|
return _autoTaggingService.All().ToResource();
|
|
}
|
|
|
|
[RestDeleteById]
|
|
public void DeleteFormat(int id)
|
|
{
|
|
_autoTaggingService.Delete(id);
|
|
}
|
|
|
|
[HttpGet("schema")]
|
|
public object GetTemplates()
|
|
{
|
|
var schema = _specifications.OrderBy(x => x.Order).Select(x => x.ToSchema()).ToList();
|
|
|
|
return schema;
|
|
}
|
|
|
|
private void Validate(AutoTag definition)
|
|
{
|
|
foreach (var validationResult in definition.Specifications.Select(spec => spec.Validate()))
|
|
{
|
|
VerifyValidationResult(validationResult);
|
|
}
|
|
}
|
|
|
|
private void VerifyValidationResult(ValidationResult validationResult)
|
|
{
|
|
var result = new NzbDroneValidationResult(validationResult.Errors);
|
|
|
|
if (!result.IsValid)
|
|
{
|
|
throw new ValidationException(result.Errors);
|
|
}
|
|
}
|
|
}
|
|
}
|