using System.Collections.Generic; using FluentValidation; using NzbDrone.Core.RootFolders; using NzbDrone.Core.Validation.Paths; using NzbDrone.SignalR; using Lidarr.Http; namespace Lidarr.Api.V1.RootFolders { public class RootFolderModule : LidarrRestModuleWithSignalR { private readonly IRootFolderService _rootFolderService; public RootFolderModule(IRootFolderService rootFolderService, IBroadcastSignalRMessage signalRBroadcaster, RootFolderValidator rootFolderValidator, PathExistsValidator pathExistsValidator, MappedNetworkDriveValidator mappedNetworkDriveValidator) : 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(pathExistsValidator); } 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 GetRootFolders() { return _rootFolderService.AllWithUnmappedFolders().ToResource(); } private void DeleteFolder(int id) { _rootFolderService.Remove(id); } } }