86 lines
2.7 KiB
C#
86 lines
2.7 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using Nancy;
|
|
using NzbDrone.Api.Extensions;
|
|
using NzbDrone.Api.REST;
|
|
using NzbDrone.Core.Configuration;
|
|
using NzbDrone.Core.Organizer;
|
|
using FluentValidation;
|
|
|
|
namespace NzbDrone.Api.Config
|
|
{
|
|
public class NamingModule : NzbDroneRestModule<NamingConfigResource>
|
|
{
|
|
private readonly INamingConfigService _namingConfigService;
|
|
|
|
public NamingModule(INamingConfigService namingConfigService)
|
|
: base("config/naming")
|
|
{
|
|
_namingConfigService = namingConfigService;
|
|
GetResourceSingle = GetNamingConfig;
|
|
|
|
UpdateResource = UpdateNamingConfig;
|
|
|
|
SharedValidator.RuleFor(c => c.MultiEpisodeStyle).InclusiveBetween(0, 3);
|
|
SharedValidator.RuleFor(c => c.NumberStyle).InclusiveBetween(0, 3);
|
|
SharedValidator.RuleFor(c => c.SeasonFolderFormat).NotEmpty();
|
|
SharedValidator.RuleFor(c => c.Separator).NotEmpty();
|
|
}
|
|
|
|
private NamingConfigResource UpdateNamingConfig(NamingConfigResource resource)
|
|
{
|
|
return Apply<NamingConfig>(_namingConfigService.Save, resource);
|
|
}
|
|
|
|
private NamingConfigResource GetNamingConfig()
|
|
{
|
|
return Apply(_namingConfigService.GetConfig);
|
|
}
|
|
}
|
|
|
|
public class NamingConfigResource : RestResource
|
|
{
|
|
public Boolean IncludeEpisodeTitle { get; set; }
|
|
public Boolean ReplaceSpaces { get; set; }
|
|
public Boolean UseSceneName { get; set; }
|
|
public Int32 MultiEpisodeStyle { get; set; }
|
|
public Int32 NumberStyle { get; set; }
|
|
public String SeasonFolderFormat { get; set; }
|
|
public String Separator { get; set; }
|
|
public Boolean IncludeQuality { get; set; }
|
|
public Boolean IncludeSeriesTitle { get; set; }
|
|
}
|
|
|
|
|
|
public class SettingsModule : NzbDroneApiModule
|
|
{
|
|
private readonly IConfigService _configService;
|
|
|
|
public SettingsModule(IConfigService configService)
|
|
: base("/settings")
|
|
{
|
|
_configService = configService;
|
|
Get["/"] = x => GetAllSettings();
|
|
Post["/"] = x => SaveSettings();
|
|
}
|
|
|
|
private Response GetAllSettings()
|
|
{
|
|
var collection = Request.Query.Collection;
|
|
|
|
if (collection.HasValue && Boolean.Parse(collection.Value))
|
|
return _configService.All().AsResponse();
|
|
|
|
return _configService.AllWithDefaults().AsResponse();
|
|
}
|
|
|
|
private Response SaveSettings()
|
|
{
|
|
var request = Request.Body.FromJson<Dictionary<string, object>>();
|
|
_configService.SaveValues(request);
|
|
|
|
|
|
return request.AsResponse();
|
|
}
|
|
}
|
|
} |