2012-07-12 08:55:27 +02:00
|
|
|
|
using System;
|
2012-08-19 22:38:31 +02:00
|
|
|
|
using System.Collections.Concurrent;
|
2012-07-12 08:55:27 +02:00
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.IO;
|
|
|
|
|
using System.Linq;
|
|
|
|
|
using System.Threading.Tasks;
|
|
|
|
|
using MediaBrowser.Controller.Events;
|
|
|
|
|
using MediaBrowser.Controller.IO;
|
|
|
|
|
using MediaBrowser.Model.Entities;
|
|
|
|
|
|
|
|
|
|
namespace MediaBrowser.Controller.Library
|
|
|
|
|
{
|
|
|
|
|
public class ItemController
|
|
|
|
|
{
|
|
|
|
|
#region PreBeginResolvePath Event
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// Fires when a path is about to be resolved, but before child folders and files
|
|
|
|
|
/// have been collected from the file system.
|
|
|
|
|
/// This gives listeners a chance to cancel the operation and cause the path to be ignored.
|
|
|
|
|
/// </summary>
|
|
|
|
|
public event EventHandler<PreBeginResolveEventArgs> PreBeginResolvePath;
|
2012-08-23 07:45:26 +02:00
|
|
|
|
private bool OnPreBeginResolvePath(PreBeginResolveEventArgs args)
|
2012-07-12 08:55:27 +02:00
|
|
|
|
{
|
|
|
|
|
if (PreBeginResolvePath != null)
|
|
|
|
|
{
|
|
|
|
|
PreBeginResolvePath(this, args);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return !args.Cancel;
|
|
|
|
|
}
|
|
|
|
|
#endregion
|
|
|
|
|
|
|
|
|
|
#region BeginResolvePath Event
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// Fires when a path is about to be resolved, but after child folders and files
|
|
|
|
|
/// have been collected from the file system.
|
|
|
|
|
/// This gives listeners a chance to cancel the operation and cause the path to be ignored.
|
|
|
|
|
/// </summary>
|
|
|
|
|
public event EventHandler<ItemResolveEventArgs> BeginResolvePath;
|
|
|
|
|
private bool OnBeginResolvePath(ItemResolveEventArgs args)
|
|
|
|
|
{
|
|
|
|
|
if (BeginResolvePath != null)
|
|
|
|
|
{
|
|
|
|
|
BeginResolvePath(this, args);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return !args.Cancel;
|
|
|
|
|
}
|
|
|
|
|
#endregion
|
|
|
|
|
|
2012-08-20 23:48:11 +02:00
|
|
|
|
private BaseItem ResolveItem(ItemResolveEventArgs args)
|
2012-07-12 08:55:27 +02:00
|
|
|
|
{
|
2012-08-20 17:55:05 +02:00
|
|
|
|
// Try first priority resolvers
|
2012-08-21 03:21:03 +02:00
|
|
|
|
for (int i = 0; i < Kernel.Instance.EntityResolvers.Length; i++)
|
2012-07-12 08:55:27 +02:00
|
|
|
|
{
|
2012-08-21 03:21:03 +02:00
|
|
|
|
var item = Kernel.Instance.EntityResolvers[i].ResolvePath(args);
|
2012-08-20 17:55:05 +02:00
|
|
|
|
|
|
|
|
|
if (item != null)
|
|
|
|
|
{
|
|
|
|
|
return item;
|
|
|
|
|
}
|
|
|
|
|
}
|
2012-08-20 23:48:11 +02:00
|
|
|
|
|
2012-07-12 08:55:27 +02:00
|
|
|
|
return null;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// Resolves a path into a BaseItem
|
|
|
|
|
/// </summary>
|
2012-08-23 14:26:08 +02:00
|
|
|
|
public async Task<BaseItem> GetItem(string path, Folder parent = null, WIN32_FIND_DATA? fileInfo = null, bool allowInternetProviders = true)
|
2012-07-12 08:55:27 +02:00
|
|
|
|
{
|
2012-08-23 07:45:26 +02:00
|
|
|
|
ItemResolveEventArgs args = new ItemResolveEventArgs()
|
|
|
|
|
{
|
|
|
|
|
FileInfo = fileInfo ?? FileData.GetFileData(path),
|
|
|
|
|
Parent = parent,
|
|
|
|
|
Cancel = false,
|
|
|
|
|
Path = path
|
|
|
|
|
};
|
2012-08-21 01:53:32 +02:00
|
|
|
|
|
2012-08-23 07:45:26 +02:00
|
|
|
|
if (!OnPreBeginResolvePath(args))
|
2012-07-12 08:55:27 +02:00
|
|
|
|
{
|
|
|
|
|
return null;
|
|
|
|
|
}
|
|
|
|
|
|
2012-08-23 07:45:26 +02:00
|
|
|
|
WIN32_FIND_DATA[] fileSystemChildren;
|
2012-07-12 08:55:27 +02:00
|
|
|
|
|
|
|
|
|
// Gather child folder and files
|
2012-08-23 07:45:26 +02:00
|
|
|
|
if (args.IsDirectory)
|
2012-07-12 08:55:27 +02:00
|
|
|
|
{
|
2012-08-23 07:45:26 +02:00
|
|
|
|
fileSystemChildren = FileData.GetFileSystemEntries(path, "*").ToArray();
|
2012-07-12 08:55:27 +02:00
|
|
|
|
|
|
|
|
|
bool isVirtualFolder = parent != null && parent.IsRoot;
|
|
|
|
|
fileSystemChildren = FilterChildFileSystemEntries(fileSystemChildren, isVirtualFolder);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
2012-08-23 07:45:26 +02:00
|
|
|
|
fileSystemChildren = new WIN32_FIND_DATA[] { };
|
2012-07-12 08:55:27 +02:00
|
|
|
|
}
|
|
|
|
|
|
2012-08-23 07:45:26 +02:00
|
|
|
|
args.FileSystemChildren = fileSystemChildren;
|
2012-07-12 08:55:27 +02:00
|
|
|
|
|
|
|
|
|
// Fire BeginResolvePath to see if anyone wants to cancel this operation
|
|
|
|
|
if (!OnBeginResolvePath(args))
|
|
|
|
|
{
|
|
|
|
|
return null;
|
|
|
|
|
}
|
|
|
|
|
|
2012-08-20 23:48:11 +02:00
|
|
|
|
BaseItem item = ResolveItem(args);
|
2012-07-12 08:55:27 +02:00
|
|
|
|
|
2012-08-20 23:48:11 +02:00
|
|
|
|
if (item != null)
|
2012-07-12 08:55:27 +02:00
|
|
|
|
{
|
2012-08-23 14:26:08 +02:00
|
|
|
|
await Kernel.Instance.ExecuteMetadataProviders(item, args, allowInternetProviders: allowInternetProviders).ConfigureAwait(false);
|
2012-08-20 23:48:11 +02:00
|
|
|
|
|
2012-08-22 04:50:59 +02:00
|
|
|
|
if (item.IsFolder)
|
2012-08-20 23:48:11 +02:00
|
|
|
|
{
|
|
|
|
|
// If it's a folder look for child entities
|
2012-08-23 14:26:08 +02:00
|
|
|
|
(item as Folder).Children = (await Task.WhenAll<BaseItem>(GetChildren(item as Folder, fileSystemChildren, allowInternetProviders)).ConfigureAwait(false))
|
2012-08-22 04:50:59 +02:00
|
|
|
|
.Where(i => i != null).OrderBy(f =>
|
|
|
|
|
{
|
|
|
|
|
return string.IsNullOrEmpty(f.SortName) ? f.Name : f.SortName;
|
|
|
|
|
|
|
|
|
|
});
|
2012-08-20 23:48:11 +02:00
|
|
|
|
}
|
2012-07-12 08:55:27 +02:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return item;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// Finds child BaseItems for a given Folder
|
|
|
|
|
/// </summary>
|
2012-08-23 14:26:08 +02:00
|
|
|
|
private Task<BaseItem>[] GetChildren(Folder folder, WIN32_FIND_DATA[] fileSystemChildren, bool allowInternetProviders)
|
2012-07-12 08:55:27 +02:00
|
|
|
|
{
|
2012-08-22 04:50:59 +02:00
|
|
|
|
Task<BaseItem>[] tasks = new Task<BaseItem>[fileSystemChildren.Length];
|
2012-07-12 08:55:27 +02:00
|
|
|
|
|
2012-08-22 04:50:59 +02:00
|
|
|
|
for (int i = 0; i < fileSystemChildren.Length; i++)
|
2012-08-19 17:58:35 +02:00
|
|
|
|
{
|
2012-08-21 03:21:03 +02:00
|
|
|
|
var child = fileSystemChildren[i];
|
2012-07-12 08:55:27 +02:00
|
|
|
|
|
2012-08-23 14:26:08 +02:00
|
|
|
|
tasks[i] = GetItem(child.Path, folder, child, allowInternetProviders: allowInternetProviders);
|
2012-08-19 17:58:35 +02:00
|
|
|
|
}
|
2012-07-12 08:55:27 +02:00
|
|
|
|
|
2012-08-22 04:50:59 +02:00
|
|
|
|
return tasks;
|
2012-07-12 08:55:27 +02:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// Transforms shortcuts into their actual paths
|
|
|
|
|
/// </summary>
|
2012-08-23 07:45:26 +02:00
|
|
|
|
private WIN32_FIND_DATA[] FilterChildFileSystemEntries(WIN32_FIND_DATA[] fileSystemChildren, bool flattenShortcuts)
|
2012-07-12 08:55:27 +02:00
|
|
|
|
{
|
2012-08-23 07:45:26 +02:00
|
|
|
|
WIN32_FIND_DATA[] returnArray = new WIN32_FIND_DATA[fileSystemChildren.Length];
|
|
|
|
|
List<WIN32_FIND_DATA> resolvedShortcuts = new List<WIN32_FIND_DATA>();
|
2012-07-12 08:55:27 +02:00
|
|
|
|
|
2012-08-21 03:21:03 +02:00
|
|
|
|
for (int i = 0; i < fileSystemChildren.Length; i++)
|
2012-07-12 08:55:27 +02:00
|
|
|
|
{
|
2012-08-23 07:45:26 +02:00
|
|
|
|
WIN32_FIND_DATA file = fileSystemChildren[i];
|
2012-08-21 03:21:03 +02:00
|
|
|
|
|
2012-07-12 08:55:27 +02:00
|
|
|
|
// If it's a shortcut, resolve it
|
2012-08-21 05:56:28 +02:00
|
|
|
|
if (Shortcut.IsShortcut(file.Path))
|
2012-07-12 08:55:27 +02:00
|
|
|
|
{
|
2012-08-21 05:56:28 +02:00
|
|
|
|
string newPath = Shortcut.ResolveShortcut(file.Path);
|
2012-08-21 01:53:32 +02:00
|
|
|
|
WIN32_FIND_DATA newPathData = FileData.GetFileData(newPath);
|
2012-07-12 08:55:27 +02:00
|
|
|
|
|
|
|
|
|
// Find out if the shortcut is pointing to a directory or file
|
2012-08-21 01:53:32 +02:00
|
|
|
|
if (newPathData.IsDirectory)
|
2012-07-12 08:55:27 +02:00
|
|
|
|
{
|
|
|
|
|
// If we're flattening then get the shortcut's children
|
|
|
|
|
|
|
|
|
|
if (flattenShortcuts)
|
|
|
|
|
{
|
2012-08-21 03:21:03 +02:00
|
|
|
|
returnArray[i] = file;
|
2012-08-23 07:45:26 +02:00
|
|
|
|
WIN32_FIND_DATA[] newChildren = FileData.GetFileSystemEntries(newPath, "*").ToArray();
|
2012-07-12 08:55:27 +02:00
|
|
|
|
|
2012-08-21 03:21:03 +02:00
|
|
|
|
resolvedShortcuts.AddRange(FilterChildFileSystemEntries(newChildren, false));
|
2012-07-12 08:55:27 +02:00
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
2012-08-23 07:45:26 +02:00
|
|
|
|
returnArray[i] = newPathData;
|
2012-07-12 08:55:27 +02:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
2012-08-23 07:45:26 +02:00
|
|
|
|
returnArray[i] = newPathData;
|
2012-07-12 08:55:27 +02:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
2012-08-21 03:21:03 +02:00
|
|
|
|
returnArray[i] = file;
|
2012-07-12 08:55:27 +02:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2012-08-21 03:21:03 +02:00
|
|
|
|
if (resolvedShortcuts.Count > 0)
|
|
|
|
|
{
|
|
|
|
|
resolvedShortcuts.InsertRange(0, returnArray);
|
|
|
|
|
return resolvedShortcuts.ToArray();
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
return returnArray;
|
|
|
|
|
}
|
2012-07-12 08:55:27 +02:00
|
|
|
|
}
|
2012-07-16 18:50:44 +02:00
|
|
|
|
|
2012-08-14 18:06:46 +02:00
|
|
|
|
/// <summary>
|
|
|
|
|
/// Gets a Person
|
|
|
|
|
/// </summary>
|
2012-08-22 04:50:59 +02:00
|
|
|
|
public Task<Person> GetPerson(string name)
|
2012-07-16 18:50:44 +02:00
|
|
|
|
{
|
2012-08-18 22:38:02 +02:00
|
|
|
|
string path = Path.Combine(Kernel.Instance.ApplicationPaths.PeoplePath, name);
|
2012-08-14 18:06:46 +02:00
|
|
|
|
|
2012-08-22 04:50:59 +02:00
|
|
|
|
return GetImagesByNameItem<Person>(path, name);
|
2012-07-16 18:50:44 +02:00
|
|
|
|
}
|
|
|
|
|
|
2012-08-14 18:06:46 +02:00
|
|
|
|
/// <summary>
|
|
|
|
|
/// Gets a Studio
|
|
|
|
|
/// </summary>
|
2012-08-22 04:50:59 +02:00
|
|
|
|
public Task<Studio> GetStudio(string name)
|
2012-07-16 18:50:44 +02:00
|
|
|
|
{
|
2012-08-18 22:38:02 +02:00
|
|
|
|
string path = Path.Combine(Kernel.Instance.ApplicationPaths.StudioPath, name);
|
2012-08-14 18:06:46 +02:00
|
|
|
|
|
2012-08-22 04:50:59 +02:00
|
|
|
|
return GetImagesByNameItem<Studio>(path, name);
|
2012-07-16 18:50:44 +02:00
|
|
|
|
}
|
|
|
|
|
|
2012-08-14 18:06:46 +02:00
|
|
|
|
/// <summary>
|
|
|
|
|
/// Gets a Genre
|
|
|
|
|
/// </summary>
|
2012-08-22 04:50:59 +02:00
|
|
|
|
public Task<Genre> GetGenre(string name)
|
2012-07-20 04:22:44 +02:00
|
|
|
|
{
|
2012-08-18 22:38:02 +02:00
|
|
|
|
string path = Path.Combine(Kernel.Instance.ApplicationPaths.GenrePath, name);
|
2012-08-14 18:06:46 +02:00
|
|
|
|
|
2012-08-22 04:50:59 +02:00
|
|
|
|
return GetImagesByNameItem<Genre>(path, name);
|
2012-07-20 04:22:44 +02:00
|
|
|
|
}
|
|
|
|
|
|
2012-08-14 18:06:46 +02:00
|
|
|
|
/// <summary>
|
|
|
|
|
/// Gets a Year
|
|
|
|
|
/// </summary>
|
2012-08-22 04:50:59 +02:00
|
|
|
|
public Task<Year> GetYear(int value)
|
2012-07-16 18:50:44 +02:00
|
|
|
|
{
|
2012-08-18 22:38:02 +02:00
|
|
|
|
string path = Path.Combine(Kernel.Instance.ApplicationPaths.YearPath, value.ToString());
|
2012-08-14 18:06:46 +02:00
|
|
|
|
|
2012-08-22 04:50:59 +02:00
|
|
|
|
return GetImagesByNameItem<Year>(path, value.ToString());
|
2012-08-14 18:06:46 +02:00
|
|
|
|
}
|
|
|
|
|
|
2012-08-19 22:38:31 +02:00
|
|
|
|
private ConcurrentDictionary<string, object> ImagesByNameItemCache = new ConcurrentDictionary<string, object>();
|
2012-08-14 18:06:46 +02:00
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// Generically retrieves an IBN item
|
|
|
|
|
/// </summary>
|
2012-08-22 04:50:59 +02:00
|
|
|
|
private Task<T> GetImagesByNameItem<T>(string path, string name)
|
2012-08-14 18:06:46 +02:00
|
|
|
|
where T : BaseEntity, new()
|
|
|
|
|
{
|
|
|
|
|
string key = path.ToLower();
|
|
|
|
|
|
|
|
|
|
// Look for it in the cache, if it's not there, create it
|
|
|
|
|
if (!ImagesByNameItemCache.ContainsKey(key))
|
|
|
|
|
{
|
2012-08-22 04:50:59 +02:00
|
|
|
|
ImagesByNameItemCache[key] = CreateImagesByNameItem<T>(path, name);
|
2012-08-14 18:06:46 +02:00
|
|
|
|
}
|
|
|
|
|
|
2012-08-22 04:50:59 +02:00
|
|
|
|
return ImagesByNameItemCache[key] as Task<T>;
|
2012-08-14 18:06:46 +02:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// Creates an IBN item based on a given path
|
|
|
|
|
/// </summary>
|
2012-08-19 22:38:31 +02:00
|
|
|
|
private async Task<T> CreateImagesByNameItem<T>(string path, string name)
|
2012-08-19 17:58:35 +02:00
|
|
|
|
where T : BaseEntity, new()
|
2012-08-14 18:06:46 +02:00
|
|
|
|
{
|
|
|
|
|
T item = new T();
|
|
|
|
|
|
|
|
|
|
item.Name = name;
|
|
|
|
|
item.Id = Kernel.GetMD5(path);
|
|
|
|
|
|
2012-08-19 22:38:31 +02:00
|
|
|
|
if (!Directory.Exists(path))
|
2012-08-14 18:06:46 +02:00
|
|
|
|
{
|
2012-08-19 22:38:31 +02:00
|
|
|
|
Directory.CreateDirectory(path);
|
2012-08-14 18:06:46 +02:00
|
|
|
|
}
|
2012-08-19 22:38:31 +02:00
|
|
|
|
|
|
|
|
|
item.DateCreated = Directory.GetCreationTime(path);
|
|
|
|
|
item.DateModified = Directory.GetLastAccessTime(path);
|
|
|
|
|
|
2012-08-20 14:59:21 +02:00
|
|
|
|
ItemResolveEventArgs args = new ItemResolveEventArgs();
|
2012-08-23 07:45:26 +02:00
|
|
|
|
args.FileInfo = FileData.GetFileData(path);
|
|
|
|
|
args.FileSystemChildren = FileData.GetFileSystemEntries(path, "*").ToArray();
|
2012-08-19 22:38:31 +02:00
|
|
|
|
|
2012-08-20 23:48:11 +02:00
|
|
|
|
await Kernel.Instance.ExecuteMetadataProviders(item, args).ConfigureAwait(false);
|
2012-08-14 18:06:46 +02:00
|
|
|
|
|
|
|
|
|
return item;
|
2012-07-16 18:50:44 +02:00
|
|
|
|
}
|
2012-07-12 08:55:27 +02:00
|
|
|
|
}
|
|
|
|
|
}
|