diff options
Diffstat (limited to 'MediaBrowser.Api/HttpHandlers/UpdateMediaLibraryHandler.cs')
| -rw-r--r-- | MediaBrowser.Api/HttpHandlers/UpdateMediaLibraryHandler.cs | 263 |
1 files changed, 263 insertions, 0 deletions
diff --git a/MediaBrowser.Api/HttpHandlers/UpdateMediaLibraryHandler.cs b/MediaBrowser.Api/HttpHandlers/UpdateMediaLibraryHandler.cs new file mode 100644 index 000000000..e5c42008e --- /dev/null +++ b/MediaBrowser.Api/HttpHandlers/UpdateMediaLibraryHandler.cs @@ -0,0 +1,263 @@ +using MediaBrowser.Common.IO; +using MediaBrowser.Common.Net.Handlers; +using MediaBrowser.Controller; +using MediaBrowser.Controller.Entities; +using System; +using System.ComponentModel.Composition; +using System.IO; +using System.Linq; +using System.Threading.Tasks; + +namespace MediaBrowser.Api.HttpHandlers +{ + /// <summary> + /// Makes changes to the user's media library + /// </summary> + [Export(typeof(IHttpServerHandler))] + public class UpdateMediaLibraryHandler : BaseActionHandler<Kernel> + { + /// <summary> + /// Executes the action. + /// </summary> + /// <returns>Task.</returns> + /// <exception cref="System.NotImplementedException"></exception> + protected override Task ExecuteAction() + { + return Task.Run(() => + { + var action = QueryString["action"]; + + if (string.IsNullOrEmpty(action)) + { + throw new ArgumentNullException(); + } + + User user = null; + + if (!string.IsNullOrEmpty(QueryString["userId"])) + { + user = ApiService.GetUserById(QueryString["userId"]); + } + + if (action.Equals("AddVirtualFolder", StringComparison.OrdinalIgnoreCase)) + { + AddVirtualFolder(Uri.UnescapeDataString(QueryString["name"]), user); + } + + if (action.Equals("RemoveVirtualFolder", StringComparison.OrdinalIgnoreCase)) + { + RemoveVirtualFolder(QueryString["name"], user); + } + + if (action.Equals("RenameVirtualFolder", StringComparison.OrdinalIgnoreCase)) + { + RenameVirtualFolder(QueryString["name"], QueryString["newName"], user); + } + + if (action.Equals("RemoveMediaPath", StringComparison.OrdinalIgnoreCase)) + { + RemoveMediaPath(QueryString["virtualFolderName"], QueryString["mediaPath"], user); + } + + if (action.Equals("AddMediaPath", StringComparison.OrdinalIgnoreCase)) + { + AddMediaPath(QueryString["virtualFolderName"], QueryString["mediaPath"], user); + } + + throw new ArgumentOutOfRangeException(); + }); + } + + /// <summary> + /// Adds a virtual folder to either the default view or a user view + /// </summary> + /// <param name="name">The name.</param> + /// <param name="user">The user.</param> + private void AddVirtualFolder(string name, User user) + { + name = FileSystem.GetValidFilename(name); + + var rootFolderPath = user != null ? user.RootFolderPath : Kernel.ApplicationPaths.DefaultUserViewsPath; + var virtualFolderPath = Path.Combine(rootFolderPath, name); + + if (Directory.Exists(virtualFolderPath)) + { + throw new ArgumentException("There is already a media collection with the name " + name + "."); + } + + Directory.CreateDirectory(virtualFolderPath); + } + + /// <summary> + /// Adds an additional mediaPath to an existing virtual folder, within either the default view or a user view + /// </summary> + /// <param name="virtualFolderName">Name of the virtual folder.</param> + /// <param name="path">The path.</param> + /// <param name="user">The user.</param> + private void AddMediaPath(string virtualFolderName, string path, User user) + { + if (!Path.IsPathRooted(path)) + { + throw new ArgumentException("The path is not valid."); + } + + if (!Directory.Exists(path)) + { + throw new DirectoryNotFoundException("The path does not exist."); + } + + // Strip off trailing slash, but not on drives + path = path.TrimEnd(Path.DirectorySeparatorChar); + if (path.EndsWith(":", StringComparison.OrdinalIgnoreCase)) + { + path += Path.DirectorySeparatorChar; + } + + var rootFolderPath = user != null ? user.RootFolderPath : Kernel.ApplicationPaths.DefaultUserViewsPath; + var virtualFolderPath = Path.Combine(rootFolderPath, virtualFolderName); + + ValidateNewMediaPath(rootFolderPath, path); + + var shortcutFilename = Path.GetFileNameWithoutExtension(path); + + var lnk = Path.Combine(virtualFolderPath, shortcutFilename + ".lnk"); + + while (File.Exists(lnk)) + { + shortcutFilename += "1"; + lnk = Path.Combine(virtualFolderPath, shortcutFilename + ".lnk"); + } + + FileSystem.CreateShortcut(lnk, path); + } + + /// <summary> + /// Validates that a new media path can be added + /// </summary> + /// <param name="currentViewRootFolderPath">The current view root folder path.</param> + /// <param name="mediaPath">The media path.</param> + private void ValidateNewMediaPath(string currentViewRootFolderPath, string mediaPath) + { + var duplicate = Directory.EnumerateFiles(Kernel.ApplicationPaths.RootFolderPath, "*.lnk", SearchOption.AllDirectories) + .Select(FileSystem.ResolveShortcut) + .FirstOrDefault(p => !IsNewPathValid(mediaPath, p)); + + if (!string.IsNullOrEmpty(duplicate)) + { + throw new ArgumentException(string.Format("The path cannot be added to the library because {0} already exists.", duplicate)); + } + + // Make sure the current root folder doesn't already have a shortcut to the same path + duplicate = Directory.EnumerateFiles(currentViewRootFolderPath, "*.lnk", SearchOption.AllDirectories) + .Select(FileSystem.ResolveShortcut) + .FirstOrDefault(p => mediaPath.Equals(p, StringComparison.OrdinalIgnoreCase)); + + if (!string.IsNullOrEmpty(duplicate)) + { + throw new ArgumentException(string.Format("The path {0} already exists in the library", mediaPath)); + } + } + + /// <summary> + /// Validates that a new path can be added based on an existing path + /// </summary> + /// <param name="newPath">The new path.</param> + /// <param name="existingPath">The existing path.</param> + /// <returns><c>true</c> if [is new path valid] [the specified new path]; otherwise, <c>false</c>.</returns> + private bool IsNewPathValid(string newPath, string existingPath) + { + // Example: D:\Movies is the existing path + // D:\ cannot be added + // Neither can D:\Movies\Kids + // A D:\Movies duplicate is ok here since that will be caught later + + if (newPath.Equals(existingPath, StringComparison.OrdinalIgnoreCase)) + { + return true; + } + + // Validate the D:\Movies\Kids scenario + if (newPath.StartsWith(existingPath.TrimEnd(Path.DirectorySeparatorChar) + Path.DirectorySeparatorChar, StringComparison.OrdinalIgnoreCase)) + { + return false; + } + + // Validate the D:\ scenario + if (existingPath.StartsWith(newPath.TrimEnd(Path.DirectorySeparatorChar) + Path.DirectorySeparatorChar, StringComparison.OrdinalIgnoreCase)) + { + return false; + } + + return true; + } + + /// <summary> + /// Renames a virtual folder within either the default view or a user view + /// </summary> + /// <param name="name">The name.</param> + /// <param name="newName">The new name.</param> + /// <param name="user">The user.</param> + private void RenameVirtualFolder(string name, string newName, User user) + { + var rootFolderPath = user != null ? user.RootFolderPath : Kernel.ApplicationPaths.DefaultUserViewsPath; + + var currentPath = Path.Combine(rootFolderPath, name); + var newPath = Path.Combine(rootFolderPath, newName); + + if (!Directory.Exists(currentPath)) + { + throw new DirectoryNotFoundException("The media collection does not exist"); + } + + if (Directory.Exists(newPath)) + { + throw new ArgumentException("There is already a media collection with the name " + newPath + "."); + } + + Directory.Move(currentPath, newPath); + } + + /// <summary> + /// Deletes a virtual folder from either the default view or a user view + /// </summary> + /// <param name="name">The name.</param> + /// <param name="user">The user.</param> + private void RemoveVirtualFolder(string name, User user) + { + var rootFolderPath = user != null ? user.RootFolderPath : Kernel.ApplicationPaths.DefaultUserViewsPath; + var path = Path.Combine(rootFolderPath, name); + + if (!Directory.Exists(path)) + { + throw new DirectoryNotFoundException("The media folder does not exist"); + } + + Directory.Delete(path, true); + } + + /// <summary> + /// Deletes a shortcut from within a virtual folder, within either the default view or a user view + /// </summary> + /// <param name="virtualFolderName">Name of the virtual folder.</param> + /// <param name="mediaPath">The media path.</param> + /// <param name="user">The user.</param> + private void RemoveMediaPath(string virtualFolderName, string mediaPath, User user) + { + var rootFolderPath = user != null ? user.RootFolderPath : Kernel.ApplicationPaths.DefaultUserViewsPath; + var path = Path.Combine(rootFolderPath, virtualFolderName); + + if (!Directory.Exists(path)) + { + throw new DirectoryNotFoundException("The media folder does not exist"); + } + + var shortcut = Directory.EnumerateFiles(path, "*.lnk", SearchOption.AllDirectories).FirstOrDefault(f => FileSystem.ResolveShortcut(f).Equals(mediaPath, StringComparison.OrdinalIgnoreCase)); + + if (string.IsNullOrEmpty(shortcut)) + { + throw new DirectoryNotFoundException("The media folder does not exist"); + } + File.Delete(shortcut); + } + } +} |