jellyfin-server/MediaBrowser.Controller/Providers/DirectoryService.cs

120 lines
3.6 KiB
C#
Raw Normal View History

using MediaBrowser.Model.Logging;
2014-02-11 21:41:01 +00:00
using System;
using System.Collections.Concurrent;
2014-02-08 22:38:02 +00:00
using System.Collections.Generic;
using System.IO;
using System.Linq;
2015-10-04 04:23:11 +00:00
using CommonIO;
2015-09-13 21:32:02 +00:00
using MediaBrowser.Common.IO;
2014-02-08 22:38:02 +00:00
namespace MediaBrowser.Controller.Providers
{
2014-02-10 18:39:41 +00:00
public class DirectoryService : IDirectoryService
2014-02-08 22:38:02 +00:00
{
private readonly ILogger _logger;
2015-09-13 21:32:02 +00:00
private readonly IFileSystem _fileSystem;
2014-02-08 22:38:02 +00:00
2015-10-04 03:38:46 +00:00
private readonly ConcurrentDictionary<string, Dictionary<string, FileSystemMetadata>> _cache =
new ConcurrentDictionary<string, Dictionary<string, FileSystemMetadata>>(StringComparer.OrdinalIgnoreCase);
2014-02-08 22:38:02 +00:00
2015-09-13 21:32:02 +00:00
public DirectoryService(ILogger logger, IFileSystem fileSystem)
2014-02-08 22:38:02 +00:00
{
_logger = logger;
2015-09-13 21:32:02 +00:00
_fileSystem = fileSystem;
2014-02-08 22:38:02 +00:00
}
2015-09-13 21:32:02 +00:00
public DirectoryService(IFileSystem fileSystem)
: this(new NullLogger(), fileSystem)
2014-10-25 18:32:58 +00:00
{
}
2015-10-04 03:38:46 +00:00
public IEnumerable<FileSystemMetadata> GetFileSystemEntries(string path)
2014-05-08 05:04:39 +00:00
{
return GetFileSystemEntries(path, false);
}
2015-10-04 03:38:46 +00:00
public Dictionary<string, FileSystemMetadata> GetFileSystemDictionary(string path)
2014-10-25 18:32:58 +00:00
{
return GetFileSystemDictionary(path, false);
}
2015-10-04 03:38:46 +00:00
private Dictionary<string, FileSystemMetadata> GetFileSystemDictionary(string path, bool clearCache)
2014-02-08 22:38:02 +00:00
{
2014-12-28 17:59:40 +00:00
if (string.IsNullOrWhiteSpace(path))
{
throw new ArgumentNullException("path");
}
2015-10-04 03:38:46 +00:00
Dictionary<string, FileSystemMetadata> entries;
2014-02-08 22:38:02 +00:00
2014-05-08 05:04:39 +00:00
if (clearCache)
{
2015-10-04 03:38:46 +00:00
Dictionary<string, FileSystemMetadata> removed;
2014-05-08 05:04:39 +00:00
_cache.TryRemove(path, out removed);
}
2014-02-08 22:38:02 +00:00
if (!_cache.TryGetValue(path, out entries))
{
//_logger.Debug("Getting files for " + path);
2015-10-04 03:38:46 +00:00
entries = new Dictionary<string, FileSystemMetadata>(StringComparer.OrdinalIgnoreCase);
2014-10-25 18:32:58 +00:00
2014-02-11 21:41:01 +00:00
try
{
2015-08-15 20:01:40 +00:00
// using EnumerateFileSystemInfos doesn't handle reparse points (symlinks)
2016-02-29 16:23:30 +00:00
var list = _fileSystem.GetFileSystemEntries(path)
.ToList();
2014-10-25 18:32:58 +00:00
// Seeing dupes on some users file system for some reason
foreach (var item in list)
{
entries[item.FullName] = item;
}
2014-02-11 21:41:01 +00:00
}
catch (DirectoryNotFoundException)
{
}
2014-10-25 18:32:58 +00:00
//var group = entries.ToLookup(i => Path.GetDirectoryName(i.FullName)).ToList();
2014-02-15 16:36:09 +00:00
_cache.TryAdd(path, entries);
2014-02-08 22:38:02 +00:00
}
return entries;
}
2015-10-04 03:38:46 +00:00
private IEnumerable<FileSystemMetadata> GetFileSystemEntries(string path, bool clearCache)
2014-10-25 18:32:58 +00:00
{
return GetFileSystemDictionary(path, clearCache).Values;
}
2015-10-04 03:38:46 +00:00
public IEnumerable<FileSystemMetadata> GetFiles(string path)
2014-02-08 22:38:02 +00:00
{
2014-05-08 05:04:39 +00:00
return GetFiles(path, false);
}
2015-10-04 03:38:46 +00:00
public IEnumerable<FileSystemMetadata> GetFiles(string path, bool clearCache)
2014-05-08 05:04:39 +00:00
{
2015-11-12 20:51:39 +00:00
return GetFileSystemEntries(path, clearCache).Where(i => !i.IsDirectory);
2014-02-08 22:38:02 +00:00
}
2015-10-04 03:38:46 +00:00
public FileSystemMetadata GetFile(string path)
2014-02-08 22:38:02 +00:00
{
var directory = Path.GetDirectoryName(path);
2014-10-25 18:32:58 +00:00
var dict = GetFileSystemDictionary(directory, false);
2015-10-04 03:38:46 +00:00
FileSystemMetadata entry;
2014-10-25 18:32:58 +00:00
dict.TryGetValue(path, out entry);
return entry;
2014-02-08 22:38:02 +00:00
}
2014-11-18 02:48:22 +00:00
2015-10-04 03:38:46 +00:00
public IEnumerable<FileSystemMetadata> GetDirectories(string path)
2014-11-18 02:48:22 +00:00
{
2015-11-12 20:51:39 +00:00
return GetFileSystemEntries(path, false).Where(i => i.IsDirectory);
2014-11-18 02:48:22 +00:00
}
2014-02-08 22:38:02 +00:00
}
}