using MediaBrowser.Common.Kernel; using MediaBrowser.Common.Logging; using MediaBrowser.Controller.Drawing; using MediaBrowser.Controller.Entities; using MediaBrowser.Controller.Entities.TV; using MediaBrowser.Controller.IO; using MediaBrowser.Controller.Library; using MediaBrowser.Controller.Providers; using MediaBrowser.Controller.Resolvers; using MediaBrowser.Controller.Weather; using MediaBrowser.Model.Authentication; using MediaBrowser.Model.Configuration; using MediaBrowser.Model.Progress; using MediaBrowser.Common.Extensions; using System; using System.Collections.Generic; using System.ComponentModel.Composition; using System.IO; using System.Linq; using System.Reflection; using System.Security.Cryptography; using System.Text; using System.Threading.Tasks; namespace MediaBrowser.Controller { public class Kernel : BaseKernel { #region Events /// /// Fires whenever any validation routine adds or removes items. The added and removed items are properties of the args. /// *** Will fire asynchronously. *** /// public event EventHandler LibraryChanged; public void OnLibraryChanged(ChildrenChangedEventArgs args) { if (LibraryChanged != null) { Task.Run(() => LibraryChanged(this, args)); } } #endregion public static Kernel Instance { get; private set; } public ItemController ItemController { 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 ApplicationPaths.RootFolderPath; } } public override KernelContext KernelContext { get { return KernelContext.Server; } } /// /// Gets the list of currently registered weather prvoiders /// [ImportMany(typeof(BaseWeatherProvider))] public IEnumerable WeatherProviders { get; private set; } /// /// Gets the list of currently registered metadata prvoiders /// [ImportMany(typeof(BaseMetadataProvider))] private IEnumerable MetadataProvidersEnumerable { get; set; } /// /// Once MEF has loaded the resolvers, sort them by priority and store them in this array /// Given the sheer number of times they'll be iterated over it'll be faster to loop through an array /// private BaseMetadataProvider[] MetadataProviders { get; set; } /// /// Gets the list of currently registered entity resolvers /// [ImportMany(typeof(IBaseItemResolver))] private IEnumerable EntityResolversEnumerable { get; set; } /// /// Once MEF has loaded the resolvers, sort them by priority and store them in this array /// Given the sheer number of times they'll be iterated over it'll be faster to loop through an array /// internal IBaseItemResolver[] EntityResolvers { get; private set; } /// /// Creates a kernel based on a Data path, which is akin to our current programdata path /// public Kernel() : base() { Instance = this; } /// /// Performs initializations that only occur once /// protected override void InitializeInternal(IProgress progress) { base.InitializeInternal(progress); ItemController = new ItemController(); DirectoryWatchers = new DirectoryWatchers(); ExtractFFMpeg(); } /// /// Performs initializations that can be reloaded at anytime /// protected override async Task ReloadInternal(IProgress progress) { await base.ReloadInternal(progress).ConfigureAwait(false); ReportProgress(progress, "Loading Users"); ReloadUsers(); ReportProgress(progress, "Loading Media Library"); await ReloadRoot(allowInternetProviders: false).ConfigureAwait(false); } /// /// Completely disposes the Kernel /// public override void Dispose() { base.Dispose(); DirectoryWatchers.Stop(); } protected override void OnComposablePartsLoaded() { // The base class will start up all the plugins base.OnComposablePartsLoaded(); // Sort the resolvers by priority EntityResolvers = EntityResolversEnumerable.OrderBy(e => e.Priority).ToArray(); // Sort the providers by priority MetadataProviders = MetadataProvidersEnumerable.OrderBy(e => e.Priority).ToArray(); } public BaseItem ResolveItem(ItemResolveEventArgs args) { // Try first priority resolvers for (int i = 0; i < EntityResolvers.Length; i++) { var item = EntityResolvers[i].ResolvePath(args); if (item != null) { item.ResolveArgs = args; return item; } } return null; } private void ReloadUsers() { Users = GetAllUsers(); } /// /// Reloads the root media folder /// public async Task ReloadRoot(bool allowInternetProviders = true) { if (!Directory.Exists(MediaRootFolderPath)) { Directory.CreateDirectory(MediaRootFolderPath); } DirectoryWatchers.Stop(); RootFolder = await ItemController.GetItem(MediaRootFolderPath, allowInternetProviders: allowInternetProviders).ConfigureAwait(false) as Folder; RootFolder.ChildrenChanged += RootFolder_ChildrenChanged; DirectoryWatchers.Start(); } void RootFolder_ChildrenChanged(object sender, ChildrenChangedEventArgs e) { Logger.LogDebugInfo("Root Folder Children Changed. Added: " + e.ItemsAdded.Count + " Removed: " + e.ItemsRemoved.Count()); //re-start the directory watchers DirectoryWatchers.Stop(); DirectoryWatchers.Start(); //Task.Delay(30000); //let's wait and see if more data gets filled in... var allChildren = RootFolder.RecursiveChildren; Logger.LogDebugInfo(string.Format("Loading complete. Movies: {0} Episodes: {1} Folders: {2}", allChildren.OfType().Count(), allChildren.OfType().Count(), allChildren.Where(i => i is Folder && !(i is Series || i is Season)).Count())); //foreach (var child in allChildren) //{ // Logger.LogDebugInfo("(" + child.GetType().Name + ") " + child.Name + " (" + child.Path + ")"); //} } /// /// Gets the default user to use when EnableUserProfiles is false /// public User GetDefaultUser() { User user = Users.FirstOrDefault(); return user; } /// /// Persists a User /// public void SaveUser(User user) { } /// /// Authenticates a User and returns a result indicating whether or not it succeeded /// public AuthenticationResult AuthenticateUser(User user, string password) { var result = new AuthenticationResult(); // When EnableUserProfiles is false, only the default User can login if (!Configuration.EnableUserProfiles) { result.Success = user.Id == GetDefaultUser().Id; } else if (string.IsNullOrEmpty(user.Password)) { result.Success = true; } else { password = password ?? string.Empty; result.Success = password.GetMD5().ToString().Equals(user.Password); } // Update LastActivityDate and LastLoginDate, then save if (result.Success) { user.LastActivityDate = user.LastLoginDate = DateTime.UtcNow; SaveUser(user); } return result; } /// /// Finds a library item by Id /// public BaseItem GetItemById(Guid id) { if (id == Guid.Empty) { return RootFolder; } return RootFolder.FindItemById(id); } /// /// Gets all users within the system /// private IEnumerable GetAllUsers() { var list = new List(); // Return a dummy user for now since all calls to get items requre a userId var user = new User { }; user.Name = "Default User"; user.Id = Guid.Parse("5d1cf7fce25943b790d140095457a42b"); user.PrimaryImagePath = "D:\\Video\\TV\\Archer (2009)\\backdrop.jpg"; list.Add(user); user = new User { }; user.Name = "Abobader"; user.Id = Guid.NewGuid(); user.LastLoginDate = DateTime.UtcNow.AddDays(-1); user.LastActivityDate = DateTime.UtcNow.AddHours(-3); user.Password = ("1234").GetMD5().ToString(); list.Add(user); user = new User { }; user.Name = "Scottisafool"; user.Id = Guid.NewGuid(); list.Add(user); user = new User { }; user.Name = "Redshirt"; user.Id = Guid.NewGuid(); list.Add(user); /*user = new User(); user.Name = "Test User 4"; user.Id = Guid.NewGuid(); list.Add(user); user = new User(); user.Name = "Test User 5"; user.Id = Guid.NewGuid(); list.Add(user); user = new User(); user.Name = "Test User 6"; user.Id = Guid.NewGuid(); list.Add(user);*/ return list; } /// /// Runs all metadata providers for an entity /// internal async Task ExecuteMetadataProviders(BaseEntity item, bool allowInternetProviders = true) { // Run them sequentially in order of priority for (int i = 0; i < MetadataProviders.Length; i++) { var provider = MetadataProviders[i]; // Skip if internet providers are currently disabled if (provider.RequiresInternet && (!Configuration.EnableInternetProviders || !allowInternetProviders)) { continue; } // Skip if the provider doesn't support the current item if (!provider.Supports(item)) { continue; } try { await provider.FetchIfNeededAsync(item).ConfigureAwait(false); } catch (Exception ex) { Logger.LogException(ex); } } } private void ExtractFFMpeg() { ExtractFFMpeg(ApplicationPaths.FFMpegPath); ExtractFFMpeg(ApplicationPaths.FFProbePath); } /// /// Run these during Init. /// Can't run do this on-demand because there will be multiple workers accessing them at once and we'd have to lock them /// private void ExtractFFMpeg(string exe) { if (File.Exists(exe)) { File.Delete(exe); } // Extract exe using (Stream stream = Assembly.GetExecutingAssembly().GetManifestResourceStream("MediaBrowser.Controller.FFMpeg." + Path.GetFileName(exe))) { using (var fileStream = new FileStream(exe, FileMode.Create)) { stream.CopyTo(fileStream); } } } } }