using System; using System.Collections.Generic; using System.IO; using System.Linq; using System.Security.Cryptography; using System.Text; using System.Threading.Tasks; using MediaBrowser.Common.Json; using MediaBrowser.Common.Logging; using MediaBrowser.Common.Plugins; using MediaBrowser.Controller.Events; using MediaBrowser.Controller.IO; using MediaBrowser.Controller.Library; using MediaBrowser.Controller.Resolvers; using MediaBrowser.Model.Configuration; using MediaBrowser.Model.Entities; using MediaBrowser.Model.Users; using MediaBrowser.Net; namespace MediaBrowser.Controller { public class Kernel { public static Kernel Instance { get; private set; } public string DataPath { get; private set; } public HttpServer HttpServer { get; private set; } public ItemController ItemController { get; private set; } public UserController UserController { get; private set; } public PluginController PluginController { get; private set; } public Configuration Configuration { get; private set; } public IEnumerable Plugins { get; private set; } public IEnumerable Users { get; private set; } public Folder RootFolder { get; private set; } private DirectoryWatchers DirectoryWatchers { get; set; } private string MediaRootFolderPath { get { return Path.Combine(DataPath, "Root"); } } /// /// Creates a kernal based on a Data path, which is akin to our current programdata path /// public Kernel(string dataPath) { Instance = this; DataPath = dataPath; Logger.LoggerInstance = new FileLogger(Path.Combine(DataPath, "Logs")); ItemController = new ItemController(); UserController = new UserController(Path.Combine(DataPath, "Users")); PluginController = new PluginController(Path.Combine(DataPath, "Plugins")); DirectoryWatchers = new DirectoryWatchers(); ItemController.PreBeginResolvePath += ItemController_PreBeginResolvePath; ItemController.BeginResolvePath += ItemController_BeginResolvePath; // Add support for core media types - audio, video, etc AddBaseItemType(); AddBaseItemType(); AddBaseItemType(); } /// /// Tells the kernel to start spinning up /// public void Init() { ReloadConfiguration(); ReloadHttpServer(); LoadPlugins(); // Get users from users folder // Load root media folder Parallel.Invoke(ReloadUsers, ReloadRoot); } private void ReloadConfiguration() { // Deserialize config Configuration = GetConfiguration(DataPath); Logger.LoggerInstance.LogSeverity = Configuration.LogSeverity; } private void LoadPlugins() { // Find plugins Plugins = PluginController.GetAllPlugins(); Parallel.For(0, Plugins.Count(), i => { Plugins.ElementAt(i).Init(); }); } private void ReloadHttpServer() { if (HttpServer != null) { HttpServer.Dispose(); } HttpServer = new HttpServer("http://+:" + Configuration.HttpServerPortNumber + "/mediabrowser/"); } /// /// Registers a new BaseItem subclass /// public void AddBaseItemType() where TBaseItemType : BaseItem, new() where TResolverType : BaseItemResolver, new() { ItemController.AddResovler(); } /// /// Fires when a path is about to be resolved, but before child folders and files /// have been collected from the file system. /// This gives us a chance to cancel it if needed, resulting in the path being ignored /// void ItemController_PreBeginResolvePath(object sender, PreBeginResolveEventArgs e) { if (e.IsHidden || e.IsSystemFile) { // Ignore hidden files and folders e.Cancel = true; } else if (Path.GetFileName(e.Path).Equals("trailers", StringComparison.OrdinalIgnoreCase)) { // Ignore any folders named "trailers" e.Cancel = true; } } /// /// Fires when a path is about to be resolved, but after child folders and files /// This gives us a chance to cancel it if needed, resulting in the path being ignored /// void ItemController_BeginResolvePath(object sender, ItemResolveEventArgs e) { if (e.IsFolder) { if (e.ContainsFile(".ignore")) { // Ignore any folders containing a file called .ignore e.Cancel = true; } } } private void ReloadUsers() { Users = UserController.GetAllUsers(); } /// /// Reloads the root media folder /// public void ReloadRoot() { if (!Directory.Exists(MediaRootFolderPath)) { Directory.CreateDirectory(MediaRootFolderPath); } DirectoryWatchers.Stop(); RootFolder = ItemController.GetItem(MediaRootFolderPath) as Folder; DirectoryWatchers.Start(); } private static MD5CryptoServiceProvider md5Provider = new MD5CryptoServiceProvider(); public static Guid GetMD5(string str) { lock (md5Provider) { return new Guid(md5Provider.ComputeHash(Encoding.Unicode.GetBytes(str))); } } private static Configuration GetConfiguration(string directory) { string file = Path.Combine(directory, "config.js"); if (!File.Exists(file)) { return new Configuration(); } return JsonSerializer.DeserializeFromFile(file); } public void ReloadItem(BaseItem item) { Folder folder = item as Folder; if (folder != null && folder.IsRoot) { ReloadRoot(); } else { if (!Directory.Exists(item.Path) && !File.Exists(item.Path)) { ReloadItem(item.Parent); return; } BaseItem newItem = ItemController.GetItem(item.Parent, item.Path); List children = item.Parent.Children.ToList(); int index = children.IndexOf(item); children.RemoveAt(index); children.Insert(index, newItem); item.Parent.Children = children.ToArray(); } } /// /// Finds a library item by Id /// public BaseItem GetItemById(Guid id) { if (id == Guid.Empty) { return RootFolder; } return RootFolder.FindById(id); } /// /// Determines if an item is allowed for a given user /// public bool IsParentalAllowed(BaseItem item, Guid userId) { // not yet implemented return true; } /// /// Gets allowed children of an item /// public IEnumerable GetParentalAllowedChildren(Folder folder, Guid userId) { return folder.Children.Where(i => IsParentalAllowed(i, userId)); } /// /// Gets allowed recursive children of an item /// public IEnumerable GetParentalAllowedRecursiveChildren(Folder folder, Guid userId) { foreach (var item in GetParentalAllowedChildren(folder, userId)) { yield return item; var subFolder = item as Folder; if (subFolder != null) { foreach (var subitem in GetParentalAllowedRecursiveChildren(subFolder, userId)) { yield return subitem; } } } } /// /// Gets user data for an item, if there is any /// public UserItemData GetUserItemData(Guid userId, Guid itemId) { User user = Users.First(u => u.Id == userId); if (user.ItemData.ContainsKey(itemId)) { return user.ItemData[itemId]; } return null; } /// /// Gets all recently added items (recursive) within a folder, based on configuration and parental settings /// public IEnumerable GetRecentlyAddedItems(Folder parent, Guid userId) { DateTime now = DateTime.Now; return GetParentalAllowedRecursiveChildren(parent, userId).Where(i => !(i is Folder) && (now - i.DateCreated).TotalDays < Configuration.RecentItemDays); } /// /// Gets all recently added unplayed items (recursive) within a folder, based on configuration and parental settings /// public IEnumerable GetRecentlyAddedUnplayedItems(Folder parent, Guid userId) { return GetRecentlyAddedItems(parent, userId).Where(i => { var userdata = GetUserItemData(userId, i.Id); return userdata == null || userdata.PlayCount == 0; }); } /// /// Gets all in-progress items (recursive) within a folder /// public IEnumerable GetInProgressItems(Folder parent, Guid userId) { return GetParentalAllowedRecursiveChildren(parent, userId).Where(i => { if (i is Folder) { return false; } var userdata = GetUserItemData(userId, i.Id); return userdata != null && userdata.PlaybackPosition.Ticks > 0; }); } /// /// Finds all recursive items within a top-level parent that contain the given studio and are allowed for the current user /// public IEnumerable GetItemsWithStudio(Folder parent, string studio, Guid userId) { return GetParentalAllowedRecursiveChildren(parent, userId).Where(f => f.Studios != null && f.Studios.Any(s => s.Equals(studio, StringComparison.OrdinalIgnoreCase))); } /// /// Finds all recursive items within a top-level parent that contain the given genre and are allowed for the current user /// public IEnumerable GetItemsWithGenre(Folder parent, string genre, Guid userId) { return GetParentalAllowedRecursiveChildren(parent, userId).Where(f => f.Genres != null && f.Genres.Any(s => s.Equals(genre, StringComparison.OrdinalIgnoreCase))); } /// /// Finds all recursive items within a top-level parent that contain the given person and are allowed for the current user /// public IEnumerable GetItemsWithPerson(Folder parent, string personName, Guid userId) { return GetParentalAllowedRecursiveChildren(parent, userId).Where(f => f.People != null && f.People.Any(s => s.Name.Equals(personName, StringComparison.OrdinalIgnoreCase))); } /// /// Gets all studios from all recursive children of a folder /// The CategoryInfo class is used to keep track of the number of times each studio appears /// public IEnumerable GetAllStudios(Folder parent, Guid userId) { Dictionary data = new Dictionary(); // Get all the allowed recursive children IEnumerable allItems = Kernel.Instance.GetParentalAllowedRecursiveChildren(parent, userId); foreach (var item in allItems) { // Add each studio from the item to the data dictionary // If the studio already exists, increment the count if (item.Studios == null) { continue; } foreach (string val in item.Studios) { if (!data.ContainsKey(val)) { data.Add(val, 1); } else { data[val]++; } } } // Now go through the dictionary and create a Category for each studio List list = new List(); foreach (string key in data.Keys) { // Get the original entity so that we can also supply the PrimaryImagePath Studio entity = Kernel.Instance.ItemController.GetStudio(key); if (entity != null) { list.Add(new CategoryInfo() { Name = entity.Name, ItemCount = data[key], PrimaryImagePath = entity.PrimaryImagePath }); } } return list; } /// /// Gets all genres from all recursive children of a folder /// The CategoryInfo class is used to keep track of the number of times each genres appears /// public IEnumerable GetAllGenres(Folder parent, Guid userId) { Dictionary data = new Dictionary(); // Get all the allowed recursive children IEnumerable allItems = Kernel.Instance.GetParentalAllowedRecursiveChildren(parent, userId); foreach (var item in allItems) { // Add each genre from the item to the data dictionary // If the genre already exists, increment the count if (item.Genres == null) { continue; } foreach (string val in item.Genres) { if (!data.ContainsKey(val)) { data.Add(val, 1); } else { data[val]++; } } } // Now go through the dictionary and create a Category for each genre List list = new List(); foreach (string key in data.Keys) { // Get the original entity so that we can also supply the PrimaryImagePath Genre entity = Kernel.Instance.ItemController.GetGenre(key); if (entity != null) { list.Add(new CategoryInfo() { Name = entity.Name, ItemCount = data[key], PrimaryImagePath = entity.PrimaryImagePath }); } } return list; } } }