mirror of https://github.com/lidarr/Lidarr
92 lines
3.6 KiB
C#
92 lines
3.6 KiB
C#
using System.IO;
|
|
using System.Linq;
|
|
using System.Reflection;
|
|
using FluentValidation;
|
|
using Lidarr.Http;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Core.Authentication;
|
|
using NzbDrone.Core.Configuration;
|
|
using NzbDrone.Core.Update;
|
|
using NzbDrone.Core.Validation;
|
|
using NzbDrone.Core.Validation.Paths;
|
|
|
|
namespace Lidarr.Api.V1.Config
|
|
{
|
|
public class HostConfigModule : LidarrRestModule<HostConfigResource>
|
|
{
|
|
private readonly IConfigFileProvider _configFileProvider;
|
|
private readonly IConfigService _configService;
|
|
private readonly IUserService _userService;
|
|
|
|
public HostConfigModule(IConfigFileProvider configFileProvider, IConfigService configService, IUserService userService)
|
|
: base("/config/host")
|
|
{
|
|
_configFileProvider = configFileProvider;
|
|
_configService = configService;
|
|
_userService = userService;
|
|
|
|
GetResourceSingle = GetHostConfig;
|
|
GetResourceById = GetHostConfig;
|
|
UpdateResource = SaveHostConfig;
|
|
|
|
SharedValidator.RuleFor(c => c.BindAddress)
|
|
.ValidIp4Address()
|
|
.NotListenAllIp4Address()
|
|
.When(c => c.BindAddress != "*");
|
|
|
|
SharedValidator.RuleFor(c => c.Port).ValidPort();
|
|
|
|
SharedValidator.RuleFor(c => c.UrlBase).ValidUrlBase();
|
|
|
|
SharedValidator.RuleFor(c => c.Username).NotEmpty().When(c => c.AuthenticationMethod != AuthenticationType.None);
|
|
SharedValidator.RuleFor(c => c.Password).NotEmpty().When(c => c.AuthenticationMethod != AuthenticationType.None);
|
|
|
|
SharedValidator.RuleFor(c => c.SslPort).ValidPort().When(c => c.EnableSsl);
|
|
SharedValidator.RuleFor(c => c.SslPort).NotEqual(c => c.Port).When(c => c.EnableSsl);
|
|
SharedValidator.RuleFor(c => c.SslCertPath).NotEmpty().When(c => c.EnableSsl);
|
|
|
|
SharedValidator.RuleFor(c => c.Branch).NotEmpty().WithMessage("Branch name is required, 'master' is the default");
|
|
SharedValidator.RuleFor(c => c.UpdateScriptPath).IsValidPath().When(c => c.UpdateMechanism == UpdateMechanism.Script);
|
|
|
|
SharedValidator.RuleFor(c => c.BackupFolder).IsValidPath().When(c => Path.IsPathRooted(c.BackupFolder));
|
|
SharedValidator.RuleFor(c => c.BackupInterval).InclusiveBetween(1, 7);
|
|
SharedValidator.RuleFor(c => c.BackupRetention).InclusiveBetween(1, 90);
|
|
}
|
|
|
|
private HostConfigResource GetHostConfig()
|
|
{
|
|
var resource = _configFileProvider.ToResource(_configService);
|
|
resource.Id = 1;
|
|
|
|
var user = _userService.FindUser();
|
|
if (user != null)
|
|
{
|
|
resource.Username = user.Username;
|
|
resource.Password = user.Password;
|
|
}
|
|
|
|
return resource;
|
|
}
|
|
|
|
private HostConfigResource GetHostConfig(int id)
|
|
{
|
|
return GetHostConfig();
|
|
}
|
|
|
|
private void SaveHostConfig(HostConfigResource resource)
|
|
{
|
|
var dictionary = resource.GetType()
|
|
.GetProperties(BindingFlags.Instance | BindingFlags.Public)
|
|
.ToDictionary(prop => prop.Name, prop => prop.GetValue(resource, null));
|
|
|
|
_configFileProvider.SaveConfigDictionary(dictionary);
|
|
_configService.SaveConfigDictionary(dictionary);
|
|
|
|
if (resource.Username.IsNotNullOrWhiteSpace() && resource.Password.IsNotNullOrWhiteSpace())
|
|
{
|
|
_userService.Upsert(resource.Username, resource.Password);
|
|
}
|
|
}
|
|
}
|
|
}
|