using MediaBrowser.Common.IO; using MediaBrowser.Controller; using MediaBrowser.Controller.IO; using MediaBrowser.Controller.Library; using MediaBrowser.Model.Entities; using MediaBrowser.Model.Logging; using ServiceStack.ServiceHost; using System; using System.Collections.Generic; using System.IO; using System.Linq; using System.Threading; using System.Threading.Tasks; namespace MediaBrowser.Api.Library { /// /// Class GetDefaultVirtualFolders /// [Route("/Library/VirtualFolders", "GET")] [Route("/Users/{UserId}/VirtualFolders", "GET")] public class GetVirtualFolders : IReturn> { /// /// Gets or sets the user id. /// /// The user id. public string UserId { get; set; } } [Route("/Library/VirtualFolders/{Name}", "POST")] [Route("/Users/{UserId}/VirtualFolders/{Name}", "POST")] public class AddVirtualFolder : IReturnVoid { /// /// Gets or sets the user id. /// /// The user id. public string UserId { get; set; } /// /// Gets or sets the name. /// /// The name. public string Name { get; set; } /// /// Gets or sets the type of the collection. /// /// The type of the collection. public string CollectionType { get; set; } /// /// Gets or sets a value indicating whether [refresh library]. /// /// true if [refresh library]; otherwise, false. public bool RefreshLibrary { get; set; } } [Route("/Library/VirtualFolders/{Name}", "DELETE")] [Route("/Users/{UserId}/VirtualFolders/{Name}", "DELETE")] public class RemoveVirtualFolder : IReturnVoid { /// /// Gets or sets the user id. /// /// The user id. public string UserId { get; set; } /// /// Gets or sets the name. /// /// The name. public string Name { get; set; } /// /// Gets or sets a value indicating whether [refresh library]. /// /// true if [refresh library]; otherwise, false. public bool RefreshLibrary { get; set; } } [Route("/Library/VirtualFolders/{Name}/Name", "POST")] [Route("/Users/{UserId}/VirtualFolders/{Name}/Name", "POST")] public class RenameVirtualFolder : IReturnVoid { /// /// Gets or sets the user id. /// /// The user id. public string UserId { get; set; } /// /// Gets or sets the name. /// /// The name. public string Name { get; set; } /// /// Gets or sets the name. /// /// The name. public string NewName { get; set; } /// /// Gets or sets a value indicating whether [refresh library]. /// /// true if [refresh library]; otherwise, false. public bool RefreshLibrary { get; set; } } [Route("/Library/VirtualFolders/{Name}/Paths", "POST")] [Route("/Users/{UserId}/VirtualFolders/{Name}/Paths", "POST")] public class AddMediaPath : IReturnVoid { /// /// Gets or sets the user id. /// /// The user id. public string UserId { get; set; } /// /// Gets or sets the name. /// /// The name. public string Name { get; set; } /// /// Gets or sets the name. /// /// The name. public string Path { get; set; } /// /// Gets or sets a value indicating whether [refresh library]. /// /// true if [refresh library]; otherwise, false. public bool RefreshLibrary { get; set; } } [Route("/Library/VirtualFolders/{Name}/Paths", "DELETE")] [Route("/Users/{UserId}/VirtualFolders/{Name}/Paths", "DELETE")] public class RemoveMediaPath : IReturnVoid { /// /// Gets or sets the user id. /// /// The user id. public string UserId { get; set; } /// /// Gets or sets the name. /// /// The name. public string Name { get; set; } /// /// Gets or sets the name. /// /// The name. public string Path { get; set; } /// /// Gets or sets a value indicating whether [refresh library]. /// /// true if [refresh library]; otherwise, false. public bool RefreshLibrary { get; set; } } /// /// Class LibraryStructureService /// public class LibraryStructureService : BaseApiService { /// /// The _app paths /// private readonly IServerApplicationPaths _appPaths; /// /// The _user manager /// private readonly IUserManager _userManager; /// /// The _library manager /// private readonly ILibraryManager _libraryManager; private readonly IDirectoryWatchers _directoryWatchers; private readonly IFileSystem _fileSystem; private readonly ILogger _logger; /// /// Initializes a new instance of the class. /// /// The app paths. /// The user manager. /// The library manager. /// appPaths public LibraryStructureService(IServerApplicationPaths appPaths, IUserManager userManager, ILibraryManager libraryManager, IDirectoryWatchers directoryWatchers, IFileSystem fileSystem, ILogger logger) { if (appPaths == null) { throw new ArgumentNullException("appPaths"); } _userManager = userManager; _appPaths = appPaths; _libraryManager = libraryManager; _directoryWatchers = directoryWatchers; _fileSystem = fileSystem; _logger = logger; } /// /// Gets the specified request. /// /// The request. /// System.Object. public object Get(GetVirtualFolders request) { if (string.IsNullOrEmpty(request.UserId)) { var result = _libraryManager.GetDefaultVirtualFolders().OrderBy(i => i.Name).ToList(); return ToOptimizedResult(result); } else { var user = _userManager.GetUserById(new Guid(request.UserId)); var result = _libraryManager.GetVirtualFolders(user).OrderBy(i => i.Name).ToList(); return ToOptimizedResult(result); } } /// /// Posts the specified request. /// /// The request. public void Post(AddVirtualFolder request) { var name = _fileSystem.GetValidFilename(request.Name); string rootFolderPath; if (string.IsNullOrEmpty(request.UserId)) { rootFolderPath = _appPaths.DefaultUserViewsPath; } else { var user = _userManager.GetUserById(new Guid(request.UserId)); rootFolderPath = user.RootFolderPath; } var virtualFolderPath = Path.Combine(rootFolderPath, name); if (Directory.Exists(virtualFolderPath)) { throw new ArgumentException("There is already a media collection with the name " + name + "."); } _directoryWatchers.Stop(); _directoryWatchers.TemporarilyIgnore(virtualFolderPath); try { Directory.CreateDirectory(virtualFolderPath); if (!string.IsNullOrEmpty(request.CollectionType)) { var path = Path.Combine(virtualFolderPath, request.CollectionType + ".collection"); File.Create(path); } // Need to add a delay here or directory watchers may still pick up the changes var task = Task.Delay(1000); // Have to block here to allow exceptions to bubble Task.WaitAll(task); } finally { // No need to start if scanning the library because it will handle it if (!request.RefreshLibrary) { _directoryWatchers.Start(); } _directoryWatchers.RemoveTempIgnore(virtualFolderPath); } if (request.RefreshLibrary) { _libraryManager.ValidateMediaLibrary(new Progress(), CancellationToken.None); } } /// /// Posts the specified request. /// /// The request. public void Post(RenameVirtualFolder request) { string rootFolderPath; if (string.IsNullOrEmpty(request.UserId)) { rootFolderPath = _appPaths.DefaultUserViewsPath; } else { var user = _userManager.GetUserById(new Guid(request.UserId)); rootFolderPath = user.RootFolderPath; } var currentPath = Path.Combine(rootFolderPath, request.Name); var newPath = Path.Combine(rootFolderPath, request.NewName); if (!Directory.Exists(currentPath)) { throw new DirectoryNotFoundException("The media collection does not exist"); } if (!string.Equals(currentPath, newPath, StringComparison.OrdinalIgnoreCase) && Directory.Exists(newPath)) { throw new ArgumentException("There is already a media collection with the name " + newPath + "."); } _directoryWatchers.Stop(); _directoryWatchers.TemporarilyIgnore(currentPath); _directoryWatchers.TemporarilyIgnore(newPath); try { // Only make a two-phase move when changing capitalization if (string.Equals(currentPath, newPath, StringComparison.OrdinalIgnoreCase)) { //Create an unique name var temporaryName = Guid.NewGuid().ToString(); var temporaryPath = Path.Combine(rootFolderPath, temporaryName); Directory.Move(currentPath, temporaryPath); currentPath = temporaryPath; } Directory.Move(currentPath, newPath); // Need to add a delay here or directory watchers may still pick up the changes var task = Task.Delay(1000); // Have to block here to allow exceptions to bubble Task.WaitAll(task); } finally { // No need to start if scanning the library because it will handle it if (!request.RefreshLibrary) { _directoryWatchers.Start(); } _directoryWatchers.RemoveTempIgnore(currentPath); _directoryWatchers.RemoveTempIgnore(newPath); } if (request.RefreshLibrary) { _libraryManager.ValidateMediaLibrary(new Progress(), CancellationToken.None); } } /// /// Deletes the specified request. /// /// The request. public void Delete(RemoveVirtualFolder request) { string rootFolderPath; if (string.IsNullOrEmpty(request.UserId)) { rootFolderPath = _appPaths.DefaultUserViewsPath; } else { var user = _userManager.GetUserById(new Guid(request.UserId)); rootFolderPath = user.RootFolderPath; } var path = Path.Combine(rootFolderPath, request.Name); if (!Directory.Exists(path)) { throw new DirectoryNotFoundException("The media folder does not exist"); } _directoryWatchers.Stop(); _directoryWatchers.TemporarilyIgnore(path); try { Directory.Delete(path, true); // Need to add a delay here or directory watchers may still pick up the changes var delayTask = Task.Delay(1000); // Have to block here to allow exceptions to bubble Task.WaitAll(delayTask); } finally { // No need to start if scanning the library because it will handle it if (!request.RefreshLibrary) { _directoryWatchers.Start(); } _directoryWatchers.RemoveTempIgnore(path); } if (request.RefreshLibrary) { _libraryManager.ValidateMediaLibrary(new Progress(), CancellationToken.None); } } /// /// Posts the specified request. /// /// The request. public void Post(AddMediaPath request) { _directoryWatchers.Stop(); try { if (string.IsNullOrEmpty(request.UserId)) { LibraryHelpers.AddMediaPath(_fileSystem, request.Name, request.Path, null, _appPaths); } else { var user = _userManager.GetUserById(new Guid(request.UserId)); LibraryHelpers.AddMediaPath(_fileSystem, request.Name, request.Path, user, _appPaths); } // Need to add a delay here or directory watchers may still pick up the changes var task = Task.Delay(1000); // Have to block here to allow exceptions to bubble Task.WaitAll(task); } finally { // No need to start if scanning the library because it will handle it if (!request.RefreshLibrary) { _directoryWatchers.Start(); } } if (request.RefreshLibrary) { _libraryManager.ValidateMediaLibrary(new Progress(), CancellationToken.None); } } /// /// Deletes the specified request. /// /// The request. public void Delete(RemoveMediaPath request) { _directoryWatchers.Stop(); try { if (string.IsNullOrEmpty(request.UserId)) { LibraryHelpers.RemoveMediaPath(_fileSystem, request.Name, request.Path, null, _appPaths); } else { var user = _userManager.GetUserById(new Guid(request.UserId)); LibraryHelpers.RemoveMediaPath(_fileSystem, request.Name, request.Path, user, _appPaths); } // Need to add a delay here or directory watchers may still pick up the changes var task = Task.Delay(1000); // Have to block here to allow exceptions to bubble Task.WaitAll(task); } finally { // No need to start if scanning the library because it will handle it if (!request.RefreshLibrary) { _directoryWatchers.Start(); } } if (request.RefreshLibrary) { _libraryManager.ValidateMediaLibrary(new Progress(), CancellationToken.None); } } } }