using System;
using System.Collections.Generic;
using System.ComponentModel.Composition;
using System.ComponentModel.Composition.Hosting;
using System.Configuration;
using System.IO;
using System.Linq;
using System.Reflection;
using MediaBrowser.Common.Json;
using MediaBrowser.Common.Net;
using MediaBrowser.Common.Plugins;
using MediaBrowser.Logging;
using MediaBrowser.Model.Configuration;
using MediaBrowser.Model.Progress;
namespace MediaBrowser.Common.Kernel
{
///
/// Represents a shared base kernel for both the UI and server apps
///
public abstract class BaseKernel : IDisposable
where TConfigurationType : BaseApplicationConfiguration, new()
{
///
/// Gets the path to the program data folder
///
public string ProgramDataPath { get; private set; }
///
/// Gets the path to the plugin directory
///
protected string PluginsPath
{
get
{
return Path.Combine(ProgramDataPath, "plugins");
}
}
///
/// Gets the path to the application configuration file
///
protected string ConfigurationPath
{
get
{
return Path.Combine(ProgramDataPath, "config.js");
}
}
///
/// Gets the path to the log directory
///
private string LogDirectoryPath
{
get
{
return Path.Combine(ProgramDataPath, "logs");
}
}
///
/// Gets or sets the path to the current log file
///
private string LogFilePath { get; set; }
///
/// Gets the current configuration
///
public TConfigurationType Configuration { get; private set; }
///
/// Gets the list of currently loaded plugins
///
[ImportMany(typeof(BasePlugin))]
public IEnumerable Plugins { get; private set; }
///
/// Both the UI and server will have a built-in HttpServer.
/// People will inevitably want remote control apps so it's needed in the UI too.
///
public HttpServer HttpServer { get; private set; }
///
/// Gets the kernel context. The UI kernel will have to override this.
///
protected KernelContext KernelContext { get { return KernelContext.Server; } }
public BaseKernel()
{
ProgramDataPath = GetProgramDataPath();
}
public virtual void Init(IProgress progress)
{
ReloadLogger();
ReloadConfiguration();
ReloadHttpServer();
ReloadComposableParts();
}
private void ReloadLogger()
{
DisposeLogger();
if (!Directory.Exists(LogDirectoryPath))
{
Directory.CreateDirectory(LogDirectoryPath);
}
DateTime now = DateTime.Now;
LogFilePath = Path.Combine(LogDirectoryPath, now.ToString("dMyyyy") + "-" + now.Ticks + ".log");
FileStream fs = new FileStream(LogFilePath, FileMode.Append, FileAccess.Write, FileShare.Read);
Logger.LoggerInstance = new StreamLogger(fs);
}
///
/// Uses MEF to locate plugins
/// Subclasses can use this to locate types within plugins
///
protected void ReloadComposableParts()
{
if (!Directory.Exists(PluginsPath))
{
Directory.CreateDirectory(PluginsPath);
}
// Gets all plugin assemblies by first reading all bytes of the .dll and calling Assembly.Load against that
// This will prevent the .dll file from getting locked, and allow us to replace it when needed
IEnumerable pluginAssemblies = Directory.GetFiles(PluginsPath, "*.dll", SearchOption.AllDirectories).Select(f => Assembly.Load(File.ReadAllBytes((f))));
var catalog = new AggregateCatalog(pluginAssemblies.Select(a => new AssemblyCatalog(a)));
//catalog.Catalogs.Add(new AssemblyCatalog(Assembly.GetExecutingAssembly()));
//catalog.Catalogs.Add(new AssemblyCatalog(GetType().Assembly));
var container = new CompositionContainer(catalog);
container.ComposeParts(this);
OnComposablePartsLoaded();
catalog.Dispose();
container.Dispose();
}
///
/// Fires after MEF finishes finding composable parts within plugin assemblies
///
protected virtual void OnComposablePartsLoaded()
{
// This event handler will allow any plugin to reference another
AppDomain.CurrentDomain.AssemblyResolve -= new ResolveEventHandler(CurrentDomain_AssemblyResolve);
AppDomain.CurrentDomain.AssemblyResolve += new ResolveEventHandler(CurrentDomain_AssemblyResolve);
StartPlugins();
}
///
/// Initializes all plugins
///
private void StartPlugins()
{
foreach (BasePlugin plugin in Plugins)
{
Assembly assembly = plugin.GetType().Assembly;
AssemblyName assemblyName = assembly.GetName();
plugin.Version = assemblyName.Version;
plugin.Path = Path.Combine(PluginsPath, assemblyName.Name);
plugin.Context = KernelContext;
plugin.ReloadConfiguration();
if (plugin.Enabled)
{
plugin.Init();
}
}
}
///
/// Gets the path to the application's ProgramDataFolder
///
private string GetProgramDataPath()
{
string programDataPath = ConfigurationManager.AppSettings["ProgramDataPath"];
// If it's a relative path, e.g. "..\"
if (!Path.IsPathRooted(programDataPath))
{
string path = Assembly.GetExecutingAssembly().Location;
path = Path.GetDirectoryName(path);
programDataPath = Path.Combine(path, programDataPath);
programDataPath = Path.GetFullPath(programDataPath);
}
if (!Directory.Exists(programDataPath))
{
Directory.CreateDirectory(programDataPath);
}
return programDataPath;
}
///
/// Reloads application configuration from the config file
///
private void ReloadConfiguration()
{
// Deserialize config
if (!File.Exists(ConfigurationPath))
{
Configuration = new TConfigurationType();
}
else
{
Configuration = JsonSerializer.DeserializeFromFile(ConfigurationPath);
}
Logger.LoggerInstance.LogSeverity = Configuration.LogSeverity;
}
///
/// Saves the current application configuration to the config file
///
public void SaveConfiguration()
{
JsonSerializer.SerializeToFile(Configuration, ConfigurationPath);
}
///
/// Restarts the Http Server, or starts it if not currently running
///
private void ReloadHttpServer()
{
DisposeHttpServer();
HttpServer = new HttpServer("http://+:" + Configuration.HttpServerPortNumber + "/mediabrowser/");
}
///
/// This snippet will allow any plugin to reference another
///
Assembly CurrentDomain_AssemblyResolve(object sender, ResolveEventArgs args)
{
AssemblyName assemblyName = new AssemblyName(args.Name);
// Look for the .dll recursively within the plugins directory
string dll = Directory.GetFiles(PluginsPath, "*.dll", SearchOption.AllDirectories)
.FirstOrDefault(f => Path.GetFileNameWithoutExtension(f) == assemblyName.Name);
// If we found a matching assembly, load it now
if (!string.IsNullOrEmpty(dll))
{
return Assembly.Load(File.ReadAllBytes(dll));
}
return null;
}
///
/// Disposes all resources currently in use.
///
public void Dispose()
{
DisposeHttpServer();
DisposeLogger();
}
///
/// Disposes the current HttpServer
///
private void DisposeHttpServer()
{
if (HttpServer != null)
{
HttpServer.Dispose();
}
}
///
/// Disposes the current Logger instance
///
private void DisposeLogger()
{
if (Logger.LoggerInstance != null)
{
Logger.LoggerInstance.Dispose();
}
}
}
}