mirror of https://github.com/Sonarr/Sonarr
67 lines
2.4 KiB
C#
67 lines
2.4 KiB
C#
using System.Collections.Generic;
|
|
using FluentValidation;
|
|
using NzbDrone.Core.RootFolders;
|
|
using NzbDrone.Core.Validation.Paths;
|
|
using NzbDrone.SignalR;
|
|
using Sonarr.Http;
|
|
using Sonarr.Http.Mapping;
|
|
|
|
namespace NzbDrone.Api.RootFolders
|
|
{
|
|
public class RootFolderModule : SonarrRestModuleWithSignalR<RootFolderResource, RootFolder>
|
|
{
|
|
private readonly IRootFolderService _rootFolderService;
|
|
|
|
public RootFolderModule(IRootFolderService rootFolderService,
|
|
IBroadcastSignalRMessage signalRBroadcaster,
|
|
RootFolderValidator rootFolderValidator,
|
|
PathExistsValidator pathExistsValidator,
|
|
MappedNetworkDriveValidator mappedNetworkDriveValidator,
|
|
StartupFolderValidator startupFolderValidator,
|
|
SystemFolderValidator systemFolderValidator,
|
|
FolderWritableValidator folderWritableValidator
|
|
)
|
|
: base(signalRBroadcaster)
|
|
{
|
|
_rootFolderService = rootFolderService;
|
|
|
|
GetResourceAll = GetRootFolders;
|
|
GetResourceById = GetRootFolder;
|
|
CreateResource = CreateRootFolder;
|
|
DeleteResource = DeleteFolder;
|
|
|
|
SharedValidator.RuleFor(c => c.Path)
|
|
.Cascade(CascadeMode.StopOnFirstFailure)
|
|
.IsValidPath()
|
|
.SetValidator(rootFolderValidator)
|
|
.SetValidator(mappedNetworkDriveValidator)
|
|
.SetValidator(startupFolderValidator)
|
|
.SetValidator(pathExistsValidator)
|
|
.SetValidator(systemFolderValidator)
|
|
.SetValidator(folderWritableValidator);
|
|
}
|
|
|
|
private RootFolderResource GetRootFolder(int id)
|
|
{
|
|
return _rootFolderService.Get(id).ToResource();
|
|
}
|
|
|
|
private int CreateRootFolder(RootFolderResource rootFolderResource)
|
|
{
|
|
var model = rootFolderResource.ToModel();
|
|
|
|
return _rootFolderService.Add(model).Id;
|
|
}
|
|
|
|
private List<RootFolderResource> GetRootFolders()
|
|
{
|
|
return _rootFolderService.AllWithUnmappedFolders().ToResource();
|
|
}
|
|
|
|
private void DeleteFolder(int id)
|
|
{
|
|
_rootFolderService.Remove(id);
|
|
}
|
|
}
|
|
}
|