2022-12-06 06:58:53 +00:00
|
|
|
using System.Collections.Generic;
|
|
|
|
using System.Linq;
|
|
|
|
using FluentValidation;
|
2023-07-26 06:25:24 +00:00
|
|
|
using FluentValidation.Results;
|
2022-12-06 06:58:53 +00:00
|
|
|
using Microsoft.AspNetCore.Mvc;
|
|
|
|
using NzbDrone.Common.Extensions;
|
|
|
|
using NzbDrone.Core.AutoTagging;
|
|
|
|
using NzbDrone.Core.AutoTagging.Specifications;
|
2023-07-26 06:25:24 +00:00
|
|
|
using NzbDrone.Core.Validation;
|
2022-12-06 06:58:53 +00:00
|
|
|
using Radarr.Http;
|
|
|
|
using Radarr.Http.REST;
|
|
|
|
using Radarr.Http.REST.Attributes;
|
|
|
|
|
|
|
|
namespace Radarr.Api.V3.AutoTagging
|
|
|
|
{
|
|
|
|
[V3ApiController]
|
|
|
|
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");
|
|
|
|
}
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
protected override AutoTaggingResource GetResourceById(int id)
|
|
|
|
{
|
|
|
|
return _autoTaggingService.GetById(id).ToResource();
|
|
|
|
}
|
|
|
|
|
|
|
|
[RestPostById]
|
|
|
|
[Consumes("application/json")]
|
2024-04-28 22:19:25 +00:00
|
|
|
public ActionResult<AutoTaggingResource> Create([FromBody] AutoTaggingResource autoTagResource)
|
2022-12-06 06:58:53 +00:00
|
|
|
{
|
|
|
|
var model = autoTagResource.ToModel(_specifications);
|
2023-07-26 06:25:24 +00:00
|
|
|
|
|
|
|
Validate(model);
|
|
|
|
|
2022-12-06 06:58:53 +00:00
|
|
|
return Created(_autoTaggingService.Insert(model).Id);
|
|
|
|
}
|
|
|
|
|
|
|
|
[RestPutById]
|
|
|
|
[Consumes("application/json")]
|
2024-04-28 22:19:25 +00:00
|
|
|
public ActionResult<AutoTaggingResource> Update([FromBody] AutoTaggingResource resource)
|
2022-12-06 06:58:53 +00:00
|
|
|
{
|
|
|
|
var model = resource.ToModel(_specifications);
|
2023-07-26 06:25:24 +00:00
|
|
|
|
|
|
|
Validate(model);
|
|
|
|
|
2022-12-06 06:58:53 +00:00
|
|
|
_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;
|
|
|
|
}
|
2023-07-26 06:25:24 +00:00
|
|
|
|
|
|
|
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);
|
|
|
|
}
|
|
|
|
}
|
2022-12-06 06:58:53 +00:00
|
|
|
}
|
|
|
|
}
|