2013-11-22 16:59:03 +00:00
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
2013-11-16 09:22:31 +00:00
|
|
|
|
using System.Linq;
|
2013-08-26 05:14:55 +00:00
|
|
|
|
using FluentValidation;
|
2013-11-15 08:53:12 +00:00
|
|
|
|
using FluentValidation.Results;
|
2013-08-26 05:14:55 +00:00
|
|
|
|
using Nancy.Responses;
|
2013-11-16 05:03:42 +00:00
|
|
|
|
using NzbDrone.Api.REST;
|
2013-07-24 00:35:35 +00:00
|
|
|
|
using NzbDrone.Core.Organizer;
|
2013-08-26 05:14:55 +00:00
|
|
|
|
using Nancy.ModelBinding;
|
|
|
|
|
using NzbDrone.Api.Mapping;
|
|
|
|
|
using NzbDrone.Api.Extensions;
|
2013-11-22 07:44:43 +00:00
|
|
|
|
using Omu.ValueInjecter;
|
2013-07-24 00:35:35 +00:00
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Api.Config
|
|
|
|
|
{
|
|
|
|
|
public class NamingModule : NzbDroneRestModule<NamingConfigResource>
|
|
|
|
|
{
|
|
|
|
|
private readonly INamingConfigService _namingConfigService;
|
2013-11-16 05:03:42 +00:00
|
|
|
|
private readonly IFilenameSampleService _filenameSampleService;
|
|
|
|
|
private readonly IFilenameValidationService _filenameValidationService;
|
2013-11-22 07:44:43 +00:00
|
|
|
|
private readonly IBuildFileNames _filenameBuilder;
|
2013-07-24 00:35:35 +00:00
|
|
|
|
|
2013-11-16 05:03:42 +00:00
|
|
|
|
public NamingModule(INamingConfigService namingConfigService,
|
|
|
|
|
IFilenameSampleService filenameSampleService,
|
2013-11-22 07:44:43 +00:00
|
|
|
|
IFilenameValidationService filenameValidationService,
|
|
|
|
|
IBuildFileNames filenameBuilder)
|
2013-07-24 00:35:35 +00:00
|
|
|
|
: base("config/naming")
|
|
|
|
|
{
|
|
|
|
|
_namingConfigService = namingConfigService;
|
2013-11-16 05:03:42 +00:00
|
|
|
|
_filenameSampleService = filenameSampleService;
|
|
|
|
|
_filenameValidationService = filenameValidationService;
|
2013-11-22 07:44:43 +00:00
|
|
|
|
_filenameBuilder = filenameBuilder;
|
2013-07-24 00:35:35 +00:00
|
|
|
|
GetResourceSingle = GetNamingConfig;
|
2013-08-28 00:16:24 +00:00
|
|
|
|
GetResourceById = GetNamingConfig;
|
2013-07-24 00:35:35 +00:00
|
|
|
|
UpdateResource = UpdateNamingConfig;
|
|
|
|
|
|
2013-08-26 05:14:55 +00:00
|
|
|
|
Get["/samples"] = x => GetExamples(this.Bind<NamingConfigResource>());
|
|
|
|
|
|
2013-07-24 00:35:35 +00:00
|
|
|
|
SharedValidator.RuleFor(c => c.MultiEpisodeStyle).InclusiveBetween(0, 3);
|
2013-11-22 17:17:04 +00:00
|
|
|
|
SharedValidator.RuleFor(c => c.StandardEpisodeFormat).ValidEpisodeFormat();
|
|
|
|
|
SharedValidator.RuleFor(c => c.DailyEpisodeFormat).ValidDailyEpisodeFormat();
|
2013-07-24 00:35:35 +00:00
|
|
|
|
}
|
|
|
|
|
|
2013-08-26 05:14:55 +00:00
|
|
|
|
private void UpdateNamingConfig(NamingConfigResource resource)
|
2013-07-24 00:35:35 +00:00
|
|
|
|
{
|
2013-11-15 08:53:12 +00:00
|
|
|
|
var nameSpec = resource.InjectTo<NamingConfig>();
|
|
|
|
|
ValidateFormatResult(nameSpec);
|
|
|
|
|
|
|
|
|
|
_namingConfigService.Save(nameSpec);
|
2013-07-24 00:35:35 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private NamingConfigResource GetNamingConfig()
|
|
|
|
|
{
|
2013-11-22 07:44:43 +00:00
|
|
|
|
var nameSpec = _namingConfigService.GetConfig();
|
|
|
|
|
var resource = nameSpec.InjectTo<NamingConfigResource>();
|
2013-11-22 16:59:03 +00:00
|
|
|
|
|
|
|
|
|
if (String.IsNullOrWhiteSpace(resource.StandardEpisodeFormat))
|
|
|
|
|
{
|
|
|
|
|
return resource;
|
|
|
|
|
}
|
|
|
|
|
|
2013-11-22 07:44:43 +00:00
|
|
|
|
var basicConfig = _filenameBuilder.GetBasicNamingConfig(nameSpec);
|
|
|
|
|
resource.InjectFrom(basicConfig);
|
|
|
|
|
|
|
|
|
|
return resource;
|
2013-08-26 05:14:55 +00:00
|
|
|
|
}
|
|
|
|
|
|
2013-08-28 00:16:24 +00:00
|
|
|
|
private NamingConfigResource GetNamingConfig(int id)
|
|
|
|
|
{
|
|
|
|
|
return GetNamingConfig();
|
|
|
|
|
}
|
|
|
|
|
|
2013-08-26 05:14:55 +00:00
|
|
|
|
private JsonResponse<NamingSampleResource> GetExamples(NamingConfigResource config)
|
|
|
|
|
{
|
2013-11-16 05:03:42 +00:00
|
|
|
|
//TODO: Validate that the format is valid
|
2013-08-26 05:14:55 +00:00
|
|
|
|
var nameSpec = config.InjectTo<NamingConfig>();
|
|
|
|
|
var sampleResource = new NamingSampleResource();
|
2013-11-16 05:03:42 +00:00
|
|
|
|
|
|
|
|
|
var singleEpisodeSampleResult = _filenameSampleService.GetStandardSample(nameSpec);
|
|
|
|
|
var multiEpisodeSampleResult = _filenameSampleService.GetMultiEpisodeSample(nameSpec);
|
|
|
|
|
var dailyEpisodeSampleResult = _filenameSampleService.GetDailySample(nameSpec);
|
2013-08-26 05:14:55 +00:00
|
|
|
|
|
2013-11-16 05:03:42 +00:00
|
|
|
|
sampleResource.SingleEpisodeExample = _filenameValidationService.ValidateStandardFilename(singleEpisodeSampleResult) != null
|
|
|
|
|
? "Invalid format"
|
|
|
|
|
: singleEpisodeSampleResult.Filename;
|
2013-08-26 05:14:55 +00:00
|
|
|
|
|
2013-11-16 05:03:42 +00:00
|
|
|
|
sampleResource.MultiEpisodeExample = _filenameValidationService.ValidateStandardFilename(multiEpisodeSampleResult) != null
|
|
|
|
|
? "Invalid format"
|
|
|
|
|
: multiEpisodeSampleResult.Filename;
|
2013-11-10 04:20:45 +00:00
|
|
|
|
|
2013-11-16 05:03:42 +00:00
|
|
|
|
sampleResource.DailyEpisodeExample = _filenameValidationService.ValidateDailyFilename(dailyEpisodeSampleResult) != null
|
|
|
|
|
? "Invalid format"
|
|
|
|
|
: dailyEpisodeSampleResult.Filename;
|
2013-08-26 05:14:55 +00:00
|
|
|
|
|
|
|
|
|
return sampleResource.AsResponse();
|
2013-07-24 00:35:35 +00:00
|
|
|
|
}
|
2013-11-10 04:20:45 +00:00
|
|
|
|
|
2013-11-15 08:53:12 +00:00
|
|
|
|
private void ValidateFormatResult(NamingConfig nameSpec)
|
|
|
|
|
{
|
2013-11-16 05:03:42 +00:00
|
|
|
|
var singleEpisodeSampleResult = _filenameSampleService.GetStandardSample(nameSpec);
|
|
|
|
|
var multiEpisodeSampleResult = _filenameSampleService.GetMultiEpisodeSample(nameSpec);
|
|
|
|
|
var dailyEpisodeSampleResult = _filenameSampleService.GetDailySample(nameSpec);
|
|
|
|
|
var singleEpisodeValidationResult = _filenameValidationService.ValidateStandardFilename(singleEpisodeSampleResult);
|
|
|
|
|
var multiEpisodeValidationResult = _filenameValidationService.ValidateStandardFilename(multiEpisodeSampleResult);
|
|
|
|
|
var dailyEpisodeValidationResult = _filenameValidationService.ValidateDailyFilename(dailyEpisodeSampleResult);
|
2013-11-15 08:53:12 +00:00
|
|
|
|
|
2013-11-16 05:03:42 +00:00
|
|
|
|
var validationFailures = new List<ValidationFailure>();
|
2013-11-15 08:53:12 +00:00
|
|
|
|
|
2013-11-16 05:03:42 +00:00
|
|
|
|
if (singleEpisodeValidationResult != null)
|
2013-11-15 08:53:12 +00:00
|
|
|
|
{
|
2013-11-16 05:03:42 +00:00
|
|
|
|
validationFailures.Add(singleEpisodeValidationResult);
|
2013-11-15 08:53:12 +00:00
|
|
|
|
}
|
|
|
|
|
|
2013-11-16 05:03:42 +00:00
|
|
|
|
if (multiEpisodeValidationResult != null)
|
2013-11-15 08:53:12 +00:00
|
|
|
|
{
|
2013-11-16 05:03:42 +00:00
|
|
|
|
validationFailures.Add(multiEpisodeValidationResult);
|
2013-11-15 08:53:12 +00:00
|
|
|
|
}
|
|
|
|
|
|
2013-11-16 05:03:42 +00:00
|
|
|
|
if (dailyEpisodeValidationResult != null)
|
2013-11-15 08:53:12 +00:00
|
|
|
|
{
|
2013-11-16 05:03:42 +00:00
|
|
|
|
validationFailures.Add(dailyEpisodeValidationResult);
|
2013-11-15 08:53:12 +00:00
|
|
|
|
}
|
|
|
|
|
|
2013-11-16 09:22:31 +00:00
|
|
|
|
if (validationFailures.Any())
|
|
|
|
|
{
|
|
|
|
|
throw new ValidationException(validationFailures.ToArray());
|
|
|
|
|
}
|
2013-11-15 08:53:12 +00:00
|
|
|
|
}
|
2013-07-24 00:35:35 +00:00
|
|
|
|
}
|
2013-11-16 05:03:42 +00:00
|
|
|
|
}
|