1 files changed, 123 insertions, 0 deletions
diff --git a/MediaBrowser.Controller/IO/FileData.cs b/MediaBrowser.Controller/IO/FileData.cs
new file mode 100644
index 000000000..e1fabee4f
--- /dev/null
+++ b/MediaBrowser.Controller/IO/FileData.cs
@@ -0,0 +1,123 @@
+using MediaBrowser.Controller.Library;
+using MediaBrowser.Controller.Providers;
+using MediaBrowser.Model.Logging;
+using System;
+using System.Collections.Generic;
+using MediaBrowser.Model.IO;
+
+namespace MediaBrowser.Controller.IO
+{
+    /// <summary>
+    /// Provides low level File access that is much faster than the File/Directory api's
+    /// </summary>
+    public static class FileData
+    {
+        private static Dictionary<string, FileSystemMetadata> GetFileSystemDictionary(FileSystemMetadata[] list)
+        {
+            var dict = new Dictionary<string, FileSystemMetadata>(StringComparer.OrdinalIgnoreCase);
+
+            foreach (var file in list)
+            {
+                dict[file.FullName] = file;
+            }
+            return dict;
+        }
+
+        /// <summary>
+        /// Gets the filtered file system entries.
+        /// </summary>
+        /// <param name="directoryService">The directory service.</param>
+        /// <param name="path">The path.</param>
+        /// <param name="fileSystem">The file system.</param>
+        /// <param name="logger">The logger.</param>
+        /// <param name="args">The args.</param>
+        /// <param name="flattenFolderDepth">The flatten folder depth.</param>
+        /// <param name="resolveShortcuts">if set to <c>true</c> [resolve shortcuts].</param>
+        /// <returns>Dictionary{System.StringFileSystemInfo}.</returns>
+        /// <exception cref="System.ArgumentNullException">path</exception>
+        public static FileSystemMetadata[] GetFilteredFileSystemEntries(IDirectoryService directoryService,
+            string path,
+            IFileSystem fileSystem,
+            IServerApplicationHost appHost,
+            ILogger logger,
+            ItemResolveArgs args,
+            int flattenFolderDepth = 0,
+            bool resolveShortcuts = true)
+        {
+            if (string.IsNullOrEmpty(path))
+            {
+                throw new ArgumentNullException("path");
+            }
+            if (args == null)
+            {
+                throw new ArgumentNullException("args");
+            }
+
+            var entries = directoryService.GetFileSystemEntries(path);
+
+            if (!resolveShortcuts && flattenFolderDepth == 0)
+            {
+                return entries;
+            }
+
+            var dict = new Dictionary<string, FileSystemMetadata>(StringComparer.OrdinalIgnoreCase);
+
+            foreach (var entry in entries)
+            {
+                var isDirectory = entry.IsDirectory;
+
+                var fullName = entry.FullName;
+
+                if (resolveShortcuts && fileSystem.IsShortcut(fullName))
+                {
+                    try
+                    {
+                        var newPath = appHost.ExpandVirtualPath(fileSystem.ResolveShortcut(fullName));
+
+                        if (string.IsNullOrEmpty(newPath))
+                        {
+                            //invalid shortcut - could be old or target could just be unavailable
+                            logger.Warn("Encountered invalid shortcut: " + fullName);
+                            continue;
+                        }
+
+                        // Don't check if it exists here because that could return false for network shares.
+                        var data = fileSystem.GetDirectoryInfo(newPath);
+
+                        // add to our physical locations
+                        args.AddAdditionalLocation(newPath);
+
+                        dict[newPath] = data;
+                    }
+                    catch (Exception ex)
+                    {
+                        logger.ErrorException("Error resolving shortcut from {0}", ex, fullName);
+                    }
+                }
+                else if (flattenFolderDepth > 0 && isDirectory)
+                {
+                    foreach (var child in GetFilteredFileSystemEntries(directoryService, fullName, fileSystem, appHost, logger, args, flattenFolderDepth: flattenFolderDepth - 1, resolveShortcuts: resolveShortcuts))
+                    {
+                        dict[child.FullName] = child;
+                    }
+                }
+                else
+                {
+                    dict[fullName] = entry;
+                }
+            }
+
+            var returnResult = new FileSystemMetadata[dict.Count];
+            var index = 0;
+            var values = dict.Values;
+            foreach (var value in values)
+            {
+                returnResult[index] = value;
+                index++;
+            }
+            return returnResult;
+        }
+
+    }
+
+}