2018-11-23 07:03:32 +00:00
|
|
|
using System.Collections.Generic;
|
|
|
|
using System.Linq;
|
|
|
|
using FluentValidation;
|
2022-07-10 17:25:42 +00:00
|
|
|
using FluentValidation.Results;
|
2021-10-21 20:04:19 +00:00
|
|
|
using Microsoft.AspNetCore.Mvc;
|
2021-01-31 05:11:18 +00:00
|
|
|
using NzbDrone.Common.Extensions;
|
2018-11-23 07:03:32 +00:00
|
|
|
using NzbDrone.Core.CustomFormats;
|
2022-07-10 17:25:42 +00:00
|
|
|
using NzbDrone.Core.Validation;
|
2018-11-23 07:03:32 +00:00
|
|
|
using Radarr.Http;
|
2021-10-21 20:04:19 +00:00
|
|
|
using Radarr.Http.REST;
|
|
|
|
using Radarr.Http.REST.Attributes;
|
2018-11-23 07:03:32 +00:00
|
|
|
|
2020-01-22 21:47:33 +00:00
|
|
|
namespace Radarr.Api.V3.CustomFormats
|
2018-11-23 07:03:32 +00:00
|
|
|
{
|
2021-10-21 20:04:19 +00:00
|
|
|
[V3ApiController]
|
|
|
|
public class CustomFormatController : RestController<CustomFormatResource>
|
2018-11-23 07:03:32 +00:00
|
|
|
{
|
|
|
|
private readonly ICustomFormatService _formatService;
|
2020-02-18 21:03:05 +00:00
|
|
|
private readonly List<ICustomFormatSpecification> _specifications;
|
2018-11-23 07:03:32 +00:00
|
|
|
|
2021-10-21 20:04:19 +00:00
|
|
|
public CustomFormatController(ICustomFormatService formatService,
|
2020-02-18 21:03:05 +00:00
|
|
|
List<ICustomFormatSpecification> specifications)
|
2018-11-23 07:03:32 +00:00
|
|
|
{
|
|
|
|
_formatService = formatService;
|
2020-02-18 21:03:05 +00:00
|
|
|
_specifications = specifications;
|
2018-11-23 07:03:32 +00:00
|
|
|
|
|
|
|
SharedValidator.RuleFor(c => c.Name).NotEmpty();
|
|
|
|
SharedValidator.RuleFor(c => c.Name)
|
|
|
|
.Must((v, c) => !_formatService.All().Any(f => f.Name == c && f.Id != v.Id)).WithMessage("Must be unique.");
|
2021-01-31 05:11:18 +00:00
|
|
|
SharedValidator.RuleFor(c => c).Custom((customFormat, context) =>
|
|
|
|
{
|
|
|
|
if (!customFormat.Specifications.Any())
|
|
|
|
{
|
|
|
|
context.AddFailure("Must contain at least one Condition");
|
|
|
|
}
|
|
|
|
|
|
|
|
if (customFormat.Specifications.Any(s => s.Name.IsNullOrWhiteSpace()))
|
|
|
|
{
|
|
|
|
context.AddFailure("Condition name(s) cannot be empty or consist of only spaces");
|
|
|
|
}
|
|
|
|
});
|
2021-10-21 20:04:19 +00:00
|
|
|
}
|
2018-11-23 07:03:32 +00:00
|
|
|
|
2021-10-24 20:35:22 +00:00
|
|
|
protected override CustomFormatResource GetResourceById(int id)
|
2021-10-21 20:04:19 +00:00
|
|
|
{
|
2023-02-05 23:09:37 +00:00
|
|
|
return _formatService.GetById(id).ToResource(true);
|
2018-11-23 07:03:32 +00:00
|
|
|
}
|
|
|
|
|
2021-10-21 20:04:19 +00:00
|
|
|
[RestPostById]
|
2023-02-05 23:09:37 +00:00
|
|
|
[Consumes("application/json")]
|
2021-10-21 20:04:19 +00:00
|
|
|
public ActionResult<CustomFormatResource> Create(CustomFormatResource customFormatResource)
|
2018-11-23 07:03:32 +00:00
|
|
|
{
|
2020-02-18 21:03:05 +00:00
|
|
|
var model = customFormatResource.ToModel(_specifications);
|
2022-07-10 17:25:42 +00:00
|
|
|
|
|
|
|
Validate(model);
|
|
|
|
|
2021-10-21 20:04:19 +00:00
|
|
|
return Created(_formatService.Insert(model).Id);
|
2018-11-23 07:03:32 +00:00
|
|
|
}
|
|
|
|
|
2021-10-21 20:04:19 +00:00
|
|
|
[RestPutById]
|
2023-02-05 23:09:37 +00:00
|
|
|
[Consumes("application/json")]
|
2021-10-21 20:04:19 +00:00
|
|
|
public ActionResult<CustomFormatResource> Update(CustomFormatResource resource)
|
2018-11-23 07:03:32 +00:00
|
|
|
{
|
2020-02-18 21:03:05 +00:00
|
|
|
var model = resource.ToModel(_specifications);
|
2022-07-10 17:25:42 +00:00
|
|
|
|
|
|
|
Validate(model);
|
|
|
|
|
2018-11-23 07:03:32 +00:00
|
|
|
_formatService.Update(model);
|
|
|
|
|
2021-10-21 20:04:19 +00:00
|
|
|
return Accepted(model.Id);
|
2018-11-23 07:03:32 +00:00
|
|
|
}
|
|
|
|
|
2021-10-21 20:04:19 +00:00
|
|
|
[HttpGet]
|
2023-02-05 23:09:37 +00:00
|
|
|
[Produces("application/json")]
|
2021-10-21 20:04:19 +00:00
|
|
|
public List<CustomFormatResource> GetAll()
|
2018-11-23 07:03:32 +00:00
|
|
|
{
|
2023-02-05 23:09:37 +00:00
|
|
|
return _formatService.All().ToResource(true);
|
2018-11-23 07:03:32 +00:00
|
|
|
}
|
|
|
|
|
2021-10-21 20:04:19 +00:00
|
|
|
[RestDeleteById]
|
|
|
|
public void DeleteFormat(int id)
|
2018-11-23 07:03:32 +00:00
|
|
|
{
|
|
|
|
_formatService.Delete(id);
|
|
|
|
}
|
|
|
|
|
2021-10-21 20:04:19 +00:00
|
|
|
[HttpGet("schema")]
|
|
|
|
public object GetTemplates()
|
2018-11-23 07:03:32 +00:00
|
|
|
{
|
2020-02-18 21:03:05 +00:00
|
|
|
var schema = _specifications.OrderBy(x => x.Order).Select(x => x.ToSchema()).ToList();
|
|
|
|
|
|
|
|
var presets = GetPresets();
|
|
|
|
|
|
|
|
foreach (var item in schema)
|
2018-11-23 07:03:32 +00:00
|
|
|
{
|
2020-02-18 21:03:05 +00:00
|
|
|
item.Presets = presets.Where(x => x.GetType().Name == item.Implementation).Select(x => x.ToSchema()).ToList();
|
|
|
|
}
|
|
|
|
|
|
|
|
return schema;
|
2018-11-23 07:03:32 +00:00
|
|
|
}
|
|
|
|
|
2022-07-10 17:25:42 +00:00
|
|
|
private void Validate(CustomFormat definition)
|
|
|
|
{
|
|
|
|
foreach (var spec in definition.Specifications)
|
|
|
|
{
|
|
|
|
var validationResult = spec.Validate();
|
|
|
|
VerifyValidationResult(validationResult);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
protected void VerifyValidationResult(ValidationResult validationResult)
|
|
|
|
{
|
|
|
|
var result = new NzbDroneValidationResult(validationResult.Errors);
|
|
|
|
|
|
|
|
if (!result.IsValid)
|
|
|
|
{
|
|
|
|
throw new ValidationException(result.Errors);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-02-18 21:03:05 +00:00
|
|
|
private IEnumerable<ICustomFormatSpecification> GetPresets()
|
2018-11-23 07:03:32 +00:00
|
|
|
{
|
2020-02-18 21:03:05 +00:00
|
|
|
yield return new ReleaseTitleSpecification
|
2018-11-23 07:03:32 +00:00
|
|
|
{
|
2020-02-18 21:03:05 +00:00
|
|
|
Name = "x264",
|
|
|
|
Value = @"(x|h)\.?264"
|
|
|
|
};
|
2019-12-22 22:08:53 +00:00
|
|
|
|
2020-02-18 21:03:05 +00:00
|
|
|
yield return new ReleaseTitleSpecification
|
2018-11-23 07:03:32 +00:00
|
|
|
{
|
2020-02-18 21:03:05 +00:00
|
|
|
Name = "x265",
|
|
|
|
Value = @"(((x|h)\.?265)|(HEVC))"
|
2018-11-23 07:03:32 +00:00
|
|
|
};
|
|
|
|
|
2020-02-18 21:03:05 +00:00
|
|
|
yield return new ReleaseTitleSpecification
|
|
|
|
{
|
|
|
|
Name = "Simple Hardcoded Subs",
|
2020-02-25 22:10:52 +00:00
|
|
|
Value = @"subs?"
|
2020-02-18 21:03:05 +00:00
|
|
|
};
|
2018-11-23 07:03:32 +00:00
|
|
|
|
2020-02-18 21:03:05 +00:00
|
|
|
yield return new ReleaseTitleSpecification
|
|
|
|
{
|
|
|
|
Name = "Hardcoded Subs",
|
|
|
|
Value = @"\b(?<hcsub>(\w+SUBS?)\b)|(?<hc>(HC|SUBBED))\b"
|
|
|
|
};
|
2018-11-23 07:03:32 +00:00
|
|
|
|
2020-02-18 21:03:05 +00:00
|
|
|
yield return new ReleaseTitleSpecification
|
2018-11-23 07:03:32 +00:00
|
|
|
{
|
2020-02-18 21:03:05 +00:00
|
|
|
Name = "Surround Sound",
|
|
|
|
Value = @"DTS.?(HD|ES|X(?!\D))|TRUEHD|ATMOS|DD(\+|P).?([5-9])|EAC3.?([5-9])"
|
|
|
|
};
|
2019-12-22 22:08:53 +00:00
|
|
|
|
2020-02-18 21:03:05 +00:00
|
|
|
yield return new ReleaseTitleSpecification
|
2018-11-23 07:03:32 +00:00
|
|
|
{
|
2020-02-18 21:03:05 +00:00
|
|
|
Name = "Preferred Words",
|
|
|
|
Value = @"\b(SPARKS|Framestor)\b"
|
2018-11-23 07:03:32 +00:00
|
|
|
};
|
2020-02-18 21:03:05 +00:00
|
|
|
|
|
|
|
var formats = _formatService.All();
|
|
|
|
foreach (var format in formats)
|
|
|
|
{
|
|
|
|
foreach (var condition in format.Specifications)
|
|
|
|
{
|
|
|
|
var preset = condition.Clone();
|
|
|
|
preset.Name = $"{format.Name}: {preset.Name}";
|
|
|
|
yield return preset;
|
|
|
|
}
|
|
|
|
}
|
2018-11-23 07:03:32 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|