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