diff options
Diffstat (limited to 'MediaBrowser.Api/EnvironmentService.cs')
| -rw-r--r-- | MediaBrowser.Api/EnvironmentService.cs | 200 |
1 files changed, 200 insertions, 0 deletions
diff --git a/MediaBrowser.Api/EnvironmentService.cs b/MediaBrowser.Api/EnvironmentService.cs new file mode 100644 index 000000000..e552c7f1d --- /dev/null +++ b/MediaBrowser.Api/EnvironmentService.cs @@ -0,0 +1,200 @@ +using MediaBrowser.Common.IO; +using MediaBrowser.Common.Net; +using MediaBrowser.Model.IO; +using ServiceStack.ServiceHost; +using System; +using System.Collections.Generic; +using System.ComponentModel.Composition; +using System.Globalization; +using System.IO; +using System.Linq; + +namespace MediaBrowser.Api +{ + /// <summary> + /// Class GetDirectoryContents + /// </summary> + [Route("/Environment/DirectoryContents", "GET")] + public class GetDirectoryContents : IReturn<List<FileSystemEntryInfo>> + { + /// <summary> + /// Gets or sets the path. + /// </summary> + /// <value>The path.</value> + public string Path { get; set; } + /// <summary> + /// Gets or sets a value indicating whether [include files]. + /// </summary> + /// <value><c>true</c> if [include files]; otherwise, <c>false</c>.</value> + public bool IncludeFiles { get; set; } + /// <summary> + /// Gets or sets a value indicating whether [include directories]. + /// </summary> + /// <value><c>true</c> if [include directories]; otherwise, <c>false</c>.</value> + public bool IncludeDirectories { get; set; } + /// <summary> + /// Gets or sets a value indicating whether [include hidden]. + /// </summary> + /// <value><c>true</c> if [include hidden]; otherwise, <c>false</c>.</value> + public bool IncludeHidden { get; set; } + } + + /// <summary> + /// Class GetDrives + /// </summary> + [Route("/Environment/Drives", "GET")] + public class GetDrives : IReturn<List<FileSystemEntryInfo>> + { + } + + /// <summary> + /// Class GetNetworkComputers + /// </summary> + [Route("/Environment/NetworkComputers", "GET")] + public class GetNetworkComputers : IReturn<List<FileSystemEntryInfo>> + { + } + + /// <summary> + /// Class EnvironmentService + /// </summary> + [Export(typeof(IRestfulService))] + public class EnvironmentService : BaseRestService + { + /// <summary> + /// Gets the specified request. + /// </summary> + /// <param name="request">The request.</param> + /// <returns>System.Object.</returns> + /// <exception cref="System.ArgumentNullException">Path</exception> + /// <exception cref="System.ArgumentException"></exception> + public object Get(GetDirectoryContents request) + { + var path = request.Path; + + if (string.IsNullOrEmpty(path)) + { + throw new ArgumentNullException("Path"); + } + + // Reject invalid input + if (!Path.IsPathRooted(path)) + { + throw new ArgumentException(string.Format("Invalid path: {0}", path)); + } + + if (path.StartsWith(NetworkPrefix, StringComparison.OrdinalIgnoreCase) && path.LastIndexOf('\\') == 1) + { + return GetNetworkShares(path).ToList(); + } + + return GetFileSystemEntries(request).ToList(); + } + + /// <summary> + /// Gets the specified request. + /// </summary> + /// <param name="request">The request.</param> + /// <returns>System.Object.</returns> + public object Get(GetDrives request) + { + return GetDrives().ToList(); + } + + /// <summary> + /// Gets the specified request. + /// </summary> + /// <param name="request">The request.</param> + /// <returns>System.Object.</returns> + public object Get(GetNetworkComputers request) + { + return GetNetworkComputers().ToList(); + } + + /// <summary> + /// Gets the list that is returned when an empty path is supplied + /// </summary> + /// <returns>IEnumerable{FileSystemEntryInfo}.</returns> + private IEnumerable<FileSystemEntryInfo> GetDrives() + { + // Only include drives in the ready state or this method could end up being very slow, waiting for drives to timeout + return DriveInfo.GetDrives().Where(d => d.IsReady).Select(d => new FileSystemEntryInfo + { + Name = GetName(d), + Path = d.RootDirectory.FullName, + Type = FileSystemEntryType.Directory + + }); + } + + /// <summary> + /// Gets the network computers. + /// </summary> + /// <returns>IEnumerable{FileSystemEntryInfo}.</returns> + private IEnumerable<FileSystemEntryInfo> GetNetworkComputers() + { + return NetUtils.GetNetworkComputers().Select(c => new FileSystemEntryInfo + { + Name = c, + Path = NetworkPrefix + c, + Type = FileSystemEntryType.NetworkComputer + }); + } + + /// <summary> + /// Gets the name. + /// </summary> + /// <param name="drive">The drive.</param> + /// <returns>System.String.</returns> + private string GetName(DriveInfo drive) + { + return drive.Name; + } + + /// <summary> + /// Gets the network shares. + /// </summary> + /// <param name="path">The path.</param> + /// <returns>IEnumerable{FileSystemEntryInfo}.</returns> + private IEnumerable<FileSystemEntryInfo> GetNetworkShares(string path) + { + return new ShareCollection(path).OfType<Share>().Where(s => s.ShareType == ShareType.Disk).Select(c => new FileSystemEntryInfo + { + Name = c.NetName, + Path = Path.Combine(path, c.NetName), + Type = FileSystemEntryType.NetworkShare + }); + } + + /// <summary> + /// Gets the file system entries. + /// </summary> + /// <param name="request">The request.</param> + /// <returns>IEnumerable{FileSystemEntryInfo}.</returns> + private IEnumerable<FileSystemEntryInfo> GetFileSystemEntries(GetDirectoryContents request) + { + var fileSystemEntries = FileSystem.GetFileSystemEntries(request.Path, "*", request.IncludeFiles, request.IncludeDirectories).Where(f => !f.IsSystemFile); + + if (!request.IncludeHidden) + { + fileSystemEntries = fileSystemEntries.Where(f => !f.IsHidden); + } + + return fileSystemEntries.Select(f => new FileSystemEntryInfo + { + Name = f.cFileName, + Path = f.Path, + Type = f.IsDirectory ? FileSystemEntryType.Directory : FileSystemEntryType.File + }); + } + + /// <summary> + /// Gets the network prefix. + /// </summary> + /// <value>The network prefix.</value> + private string NetworkPrefix + { + get { return Path.DirectorySeparatorChar.ToString(CultureInfo.InvariantCulture) + Path.DirectorySeparatorChar.ToString(CultureInfo.InvariantCulture); } + } + } +} |