mirror of https://github.com/Radarr/Radarr
87 lines
3.1 KiB
C#
87 lines
3.1 KiB
C#
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using FluentValidation;
|
|
using Microsoft.AspNetCore.Mvc;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Core.CustomFormats;
|
|
using NzbDrone.Core.Profiles.Qualities;
|
|
using Radarr.Http;
|
|
using Radarr.Http.REST;
|
|
using Radarr.Http.REST.Attributes;
|
|
|
|
namespace Radarr.Api.V3.Profiles.Quality
|
|
{
|
|
[V3ApiController]
|
|
public class QualityProfileController : RestController<QualityProfileResource>
|
|
{
|
|
private readonly IQualityProfileService _qualityProfileService;
|
|
private readonly ICustomFormatService _formatService;
|
|
|
|
public QualityProfileController(IQualityProfileService qualityProfileService, ICustomFormatService formatService)
|
|
{
|
|
_qualityProfileService = qualityProfileService;
|
|
_formatService = formatService;
|
|
SharedValidator.RuleFor(c => c.Name).NotEmpty();
|
|
|
|
// TODO: Need to validate the cutoff is allowed and the ID/quality ID exists
|
|
// TODO: Need to validate the Items to ensure groups have names and at no item has no name, no items and no quality
|
|
SharedValidator.RuleFor(c => c.Cutoff).ValidCutoff();
|
|
SharedValidator.RuleFor(c => c.Items).ValidItems();
|
|
|
|
SharedValidator.RuleFor(c => c.FormatItems).Must(items =>
|
|
{
|
|
var all = _formatService.All().Select(f => f.Id).ToList();
|
|
var ids = items.Select(i => i.Format);
|
|
|
|
return all.Except(ids).Empty();
|
|
}).WithMessage("All Custom Formats and no extra ones need to be present inside your Profile! Try refreshing your browser.");
|
|
|
|
SharedValidator.RuleFor(c => c).Custom((profile, context) =>
|
|
{
|
|
if (profile.FormatItems.Where(x => x.Score > 0).Sum(x => x.Score) < profile.MinFormatScore &&
|
|
profile.FormatItems.Max(x => x.Score) < profile.MinFormatScore)
|
|
{
|
|
context.AddFailure("Minimum Custom Format Score can never be satisfied");
|
|
}
|
|
});
|
|
}
|
|
|
|
[RestPostById]
|
|
[Consumes("application/json")]
|
|
public ActionResult<QualityProfileResource> Create([FromBody] QualityProfileResource resource)
|
|
{
|
|
var model = resource.ToModel();
|
|
model = _qualityProfileService.Add(model);
|
|
return Created(model.Id);
|
|
}
|
|
|
|
[RestDeleteById]
|
|
public void DeleteProfile(int id)
|
|
{
|
|
_qualityProfileService.Delete(id);
|
|
}
|
|
|
|
[RestPutById]
|
|
[Consumes("application/json")]
|
|
public ActionResult<QualityProfileResource> Update([FromBody] QualityProfileResource resource)
|
|
{
|
|
var model = resource.ToModel();
|
|
|
|
_qualityProfileService.Update(model);
|
|
|
|
return Accepted(model.Id);
|
|
}
|
|
|
|
protected override QualityProfileResource GetResourceById(int id)
|
|
{
|
|
return _qualityProfileService.Get(id).ToResource();
|
|
}
|
|
|
|
[HttpGet]
|
|
public List<QualityProfileResource> GetAll()
|
|
{
|
|
return _qualityProfileService.All().ToResource();
|
|
}
|
|
}
|
|
}
|