diff options
Diffstat (limited to 'MediaBrowser.Common/Kernel/BaseKernel.cs')
| -rw-r--r-- | MediaBrowser.Common/Kernel/BaseKernel.cs | 1116 |
1 files changed, 771 insertions, 345 deletions
diff --git a/MediaBrowser.Common/Kernel/BaseKernel.cs b/MediaBrowser.Common/Kernel/BaseKernel.cs index a6081a688..a4cd81665 100644 --- a/MediaBrowser.Common/Kernel/BaseKernel.cs +++ b/MediaBrowser.Common/Kernel/BaseKernel.cs @@ -1,345 +1,771 @@ -using MediaBrowser.Common.Events;
-using MediaBrowser.Common.Logging;
-using MediaBrowser.Common.Mef;
-using MediaBrowser.Common.Net;
-using MediaBrowser.Common.Net.Handlers;
-using MediaBrowser.Common.Plugins;
-using MediaBrowser.Common.Serialization;
-using MediaBrowser.Model.Configuration;
-using MediaBrowser.Model.Progress;
-using System;
-using System.Collections.Generic;
-using System.ComponentModel.Composition;
-using System.ComponentModel.Composition.Hosting;
-using System.ComponentModel.Composition.Primitives;
-using System.IO;
-using System.Linq;
-using System.Reflection;
-using System.Threading.Tasks;
-
-namespace MediaBrowser.Common.Kernel
-{
- /// <summary>
- /// Represents a shared base kernel for both the Ui and server apps
- /// </summary>
- public abstract class BaseKernel<TConfigurationType, TApplicationPathsType> : IDisposable, IKernel
- where TConfigurationType : BaseApplicationConfiguration, new()
- where TApplicationPathsType : BaseApplicationPaths, new()
- {
- #region ReloadBeginning Event
- /// <summary>
- /// Fires whenever the kernel begins reloading
- /// </summary>
- public event EventHandler<GenericEventArgs<IProgress<TaskProgress>>> ReloadBeginning;
- private void OnReloadBeginning(IProgress<TaskProgress> progress)
- {
- if (ReloadBeginning != null)
- {
- ReloadBeginning(this, new GenericEventArgs<IProgress<TaskProgress>> { Argument = progress });
- }
- }
- #endregion
-
- #region ReloadCompleted Event
- /// <summary>
- /// Fires whenever the kernel completes reloading
- /// </summary>
- public event EventHandler<GenericEventArgs<IProgress<TaskProgress>>> ReloadCompleted;
- private void OnReloadCompleted(IProgress<TaskProgress> progress)
- {
- if (ReloadCompleted != null)
- {
- ReloadCompleted(this, new GenericEventArgs<IProgress<TaskProgress>> { Argument = progress });
- }
- }
- #endregion
-
- /// <summary>
- /// Gets the current configuration
- /// </summary>
- public TConfigurationType Configuration { get; private set; }
-
- public TApplicationPathsType ApplicationPaths { get; private set; }
-
- /// <summary>
- /// Gets the list of currently loaded plugins
- /// </summary>
- [ImportMany(typeof(BasePlugin))]
- public IEnumerable<BasePlugin> Plugins { get; private set; }
-
- /// <summary>
- /// Gets the list of currently registered http handlers
- /// </summary>
- [ImportMany(typeof(BaseHandler))]
- private IEnumerable<BaseHandler> HttpHandlers { get; set; }
-
- /// <summary>
- /// Gets the list of currently registered Loggers
- /// </summary>
- [ImportMany(typeof(BaseLogger))]
- public IEnumerable<BaseLogger> Loggers { get; set; }
-
- /// <summary>
- /// 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.
- /// </summary>
- public HttpServer HttpServer { get; private set; }
-
- /// <summary>
- /// This subscribes to HttpListener requests and finds the appropate BaseHandler to process it
- /// </summary>
- private IDisposable HttpListener { get; set; }
-
- /// <summary>
- /// Gets the MEF CompositionContainer
- /// </summary>
- private CompositionContainer CompositionContainer { get; set; }
-
- protected virtual string HttpServerUrlPrefix
- {
- get
- {
- return "http://+:" + Configuration.HttpServerPortNumber + "/mediabrowser/";
- }
- }
-
- /// <summary>
- /// Gets the kernel context. Subclasses will have to override.
- /// </summary>
- public abstract KernelContext KernelContext { get; }
-
- /// <summary>
- /// Initializes the Kernel
- /// </summary>
- public async Task Init(IProgress<TaskProgress> progress)
- {
- Logger.Kernel = this;
-
- // Performs initializations that only occur once
- InitializeInternal(progress);
-
- // Performs initializations that can be reloaded at anytime
- await Reload(progress).ConfigureAwait(false);
- }
-
- /// <summary>
- /// Performs initializations that only occur once
- /// </summary>
- protected virtual void InitializeInternal(IProgress<TaskProgress> progress)
- {
- ApplicationPaths = new TApplicationPathsType();
-
- ReportProgress(progress, "Loading Configuration");
- ReloadConfiguration();
-
- ReportProgress(progress, "Loading Http Server");
- ReloadHttpServer();
- }
-
- /// <summary>
- /// Performs initializations that can be reloaded at anytime
- /// </summary>
- public async Task Reload(IProgress<TaskProgress> progress)
- {
- OnReloadBeginning(progress);
-
- await ReloadInternal(progress).ConfigureAwait(false);
-
- OnReloadCompleted(progress);
-
- ReportProgress(progress, "Kernel.Reload Complete");
- }
-
- /// <summary>
- /// Performs initializations that can be reloaded at anytime
- /// </summary>
- protected virtual async Task ReloadInternal(IProgress<TaskProgress> progress)
- {
- await Task.Run(() =>
- {
- ReportProgress(progress, "Loading Plugins");
- ReloadComposableParts();
-
- }).ConfigureAwait(false);
- }
-
- /// <summary>
- /// Uses MEF to locate plugins
- /// Subclasses can use this to locate types within plugins
- /// </summary>
- private void ReloadComposableParts()
- {
- DisposeComposableParts();
-
- CompositionContainer = GetCompositionContainer(includeCurrentAssembly: true);
-
- CompositionContainer.ComposeParts(this);
-
- OnComposablePartsLoaded();
-
- CompositionContainer.Catalog.Dispose();
- }
-
- /// <summary>
- /// Constructs an MEF CompositionContainer based on the current running assembly and all plugin assemblies
- /// </summary>
- public CompositionContainer GetCompositionContainer(bool includeCurrentAssembly = false)
- {
- // 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<Assembly> pluginAssemblies = Directory.GetFiles(ApplicationPaths.PluginsPath, "*.dll", SearchOption.TopDirectoryOnly).Select(f => Assembly.Load(File.ReadAllBytes((f))));
-
- var catalogs = new List<ComposablePartCatalog>();
-
- catalogs.AddRange(pluginAssemblies.Select(a => new AssemblyCatalog(a)));
-
- // Include composable parts in the Common assembly
- catalogs.Add(new AssemblyCatalog(Assembly.GetExecutingAssembly()));
-
- if (includeCurrentAssembly)
- {
- // Include composable parts in the subclass assembly
- catalogs.Add(new AssemblyCatalog(GetType().Assembly));
- }
-
- return MefUtils.GetSafeCompositionContainer(catalogs);
- }
-
- /// <summary>
- /// Fires after MEF finishes finding composable parts within plugin assemblies
- /// </summary>
- protected virtual void OnComposablePartsLoaded()
- {
- foreach (var logger in Loggers)
- {
- logger.Initialize(this);
- }
-
- // Start-up each plugin
- foreach (var plugin in Plugins)
- {
- plugin.Initialize(this);
- }
- }
-
- /// <summary>
- /// Reloads application configuration from the config file
- /// </summary>
- private void ReloadConfiguration()
- {
- //Configuration information for anything other than server-specific configuration will have to come via the API... -ebr
-
- // Deserialize config
- // Use try/catch to avoid the extra file system lookup using File.Exists
- try
- {
- Configuration = XmlSerializer.DeserializeFromFile<TConfigurationType>(ApplicationPaths.SystemConfigurationFilePath);
- }
- catch (FileNotFoundException)
- {
- Configuration = new TConfigurationType();
- XmlSerializer.SerializeToFile(Configuration, ApplicationPaths.SystemConfigurationFilePath);
- }
- }
-
- /// <summary>
- /// Restarts the Http Server, or starts it if not currently running
- /// </summary>
- private void ReloadHttpServer()
- {
- DisposeHttpServer();
-
- HttpServer = new HttpServer(HttpServerUrlPrefix);
-
- HttpListener = HttpServer.Subscribe(ctx =>
- {
- BaseHandler handler = HttpHandlers.FirstOrDefault(h => h.HandlesRequest(ctx.Request));
-
- // Find the appropiate http handler
- if (handler != null)
- {
- // Need to create a new instance because handlers are currently stateful
- handler = Activator.CreateInstance(handler.GetType()) as BaseHandler;
-
- // No need to await this, despite the compiler warning
- handler.ProcessRequest(ctx);
- }
- });
- }
-
- /// <summary>
- /// Disposes all resources currently in use.
- /// </summary>
- public virtual void Dispose()
- {
- Logger.LogInfo("Beginning Kernel.Dispose");
-
- DisposeHttpServer();
-
- DisposeComposableParts();
- }
-
- /// <summary>
- /// Disposes all objects gathered through MEF composable parts
- /// </summary>
- protected virtual void DisposeComposableParts()
- {
- if (CompositionContainer != null)
- {
- CompositionContainer.Dispose();
- }
- }
-
- /// <summary>
- /// Disposes the current HttpServer
- /// </summary>
- private void DisposeHttpServer()
- {
- if (HttpServer != null)
- {
- Logger.LogInfo("Disposing Http Server");
-
- HttpServer.Dispose();
- }
-
- if (HttpListener != null)
- {
- HttpListener.Dispose();
- }
- }
-
- /// <summary>
- /// Gets the current application version
- /// </summary>
- public Version ApplicationVersion
- {
- get
- {
- return GetType().Assembly.GetName().Version;
- }
- }
-
- protected void ReportProgress(IProgress<TaskProgress> progress, string message)
- {
- progress.Report(new TaskProgress { Description = message });
-
- Logger.LogInfo(message);
- }
-
- BaseApplicationPaths IKernel.ApplicationPaths
- {
- get { return ApplicationPaths; }
- }
- }
-
- public interface IKernel
- {
- BaseApplicationPaths ApplicationPaths { get; }
- KernelContext KernelContext { get; }
-
- Task Init(IProgress<TaskProgress> progress);
- Task Reload(IProgress<TaskProgress> progress);
- IEnumerable<BaseLogger> Loggers { get; }
- void Dispose();
- }
-}
+using MediaBrowser.Common.Events; +using MediaBrowser.Common.IO; +using MediaBrowser.Common.Localization; +using MediaBrowser.Common.Mef; +using MediaBrowser.Common.Net; +using MediaBrowser.Common.Net.Handlers; +using MediaBrowser.Common.Plugins; +using MediaBrowser.Common.ScheduledTasks; +using MediaBrowser.Common.Serialization; +using MediaBrowser.Model.Configuration; +using MediaBrowser.Model.Logging; +using MediaBrowser.Model.System; +using NLog; +using NLog.Config; +using NLog.Targets; +using System; +using System.Collections.Generic; +using System.ComponentModel.Composition; +using System.ComponentModel.Composition.Hosting; +using System.ComponentModel.Composition.Primitives; +using System.Deployment.Application; +using System.Diagnostics; +using System.IO; +using System.Linq; +using System.Reflection; +using System.Threading; +using System.Threading.Tasks; + +namespace MediaBrowser.Common.Kernel +{ + /// <summary> + /// Represents a shared base kernel for both the Ui and server apps + /// </summary> + /// <typeparam name="TConfigurationType">The type of the T configuration type.</typeparam> + /// <typeparam name="TApplicationPathsType">The type of the T application paths type.</typeparam> + public abstract class BaseKernel<TConfigurationType, TApplicationPathsType> : IDisposable, IKernel + where TConfigurationType : BaseApplicationConfiguration, new() + where TApplicationPathsType : BaseApplicationPaths, new() + { + /// <summary> + /// Occurs when [has pending restart changed]. + /// </summary> + public event EventHandler HasPendingRestartChanged; + + /// <summary> + /// Notifiies the containing application that a restart has been requested + /// </summary> + public event EventHandler ApplicationRestartRequested; + + #region ConfigurationUpdated Event + /// <summary> + /// Occurs when [configuration updated]. + /// </summary> + public event EventHandler<EventArgs> ConfigurationUpdated; + + /// <summary> + /// Called when [configuration updated]. + /// </summary> + internal void OnConfigurationUpdated() + { + EventHelper.QueueEventIfNotNull(ConfigurationUpdated, this, EventArgs.Empty); + + // Notify connected clients + TcpManager.SendWebSocketMessage("ConfigurationUpdated", Configuration); + } + #endregion + + #region LoggerLoaded Event + /// <summary> + /// Fires whenever the logger is loaded + /// </summary> + public event EventHandler LoggerLoaded; + /// <summary> + /// Called when [logger loaded]. + /// </summary> + private void OnLoggerLoaded() + { + EventHelper.QueueEventIfNotNull(LoggerLoaded, this, EventArgs.Empty); + } + #endregion + + #region ReloadBeginning Event + /// <summary> + /// Fires whenever the kernel begins reloading + /// </summary> + public event EventHandler<EventArgs> ReloadBeginning; + /// <summary> + /// Called when [reload beginning]. + /// </summary> + private void OnReloadBeginning() + { + EventHelper.QueueEventIfNotNull(ReloadBeginning, this, EventArgs.Empty); + } + #endregion + + #region ReloadCompleted Event + /// <summary> + /// Fires whenever the kernel completes reloading + /// </summary> + public event EventHandler<EventArgs> ReloadCompleted; + /// <summary> + /// Called when [reload completed]. + /// </summary> + private void OnReloadCompleted() + { + EventHelper.QueueEventIfNotNull(ReloadCompleted, this, EventArgs.Empty); + } + #endregion + + #region ApplicationUpdated Event + /// <summary> + /// Occurs when [application updated]. + /// </summary> + public event EventHandler<GenericEventArgs<Version>> ApplicationUpdated; + /// <summary> + /// Called when [application updated]. + /// </summary> + /// <param name="newVersion">The new version.</param> + public void OnApplicationUpdated(Version newVersion) + { + EventHelper.QueueEventIfNotNull(ApplicationUpdated, this, new GenericEventArgs<Version> {Argument = newVersion}); + + NotifyPendingRestart(); + } + #endregion + + /// <summary> + /// The _configuration loaded + /// </summary> + private bool _configurationLoaded; + /// <summary> + /// The _configuration sync lock + /// </summary> + private object _configurationSyncLock = new object(); + /// <summary> + /// The _configuration + /// </summary> + private TConfigurationType _configuration; + /// <summary> + /// Gets the system configuration + /// </summary> + /// <value>The configuration.</value> + public TConfigurationType Configuration + { + get + { + // Lazy load + LazyInitializer.EnsureInitialized(ref _configuration, ref _configurationLoaded, ref _configurationSyncLock, () => XmlSerializer.GetXmlConfiguration<TConfigurationType>(ApplicationPaths.SystemConfigurationFilePath)); + return _configuration; + } + protected set + { + _configuration = value; + + if (value == null) + { + _configurationLoaded = false; + } + } + } + + /// <summary> + /// Gets a value indicating whether this instance is first run. + /// </summary> + /// <value><c>true</c> if this instance is first run; otherwise, <c>false</c>.</value> + public bool IsFirstRun { get; private set; } + + /// <summary> + /// The version of the application to display + /// </summary> + /// <value>The display version.</value> + public string DisplayVersion { get { return ApplicationVersion.ToString(); } } + + /// <summary> + /// Gets or sets a value indicating whether this instance has changes that require the entire application to restart. + /// </summary> + /// <value><c>true</c> if this instance has pending application restart; otherwise, <c>false</c>.</value> + public bool HasPendingRestart { get; private set; } + + /// <summary> + /// Gets the application paths. + /// </summary> + /// <value>The application paths.</value> + public TApplicationPathsType ApplicationPaths { get; private set; } + + /// <summary> + /// The _failed assembly loads + /// </summary> + private readonly List<string> _failedPluginAssemblies = new List<string>(); + /// <summary> + /// Gets the plugin assemblies that failed to load. + /// </summary> + /// <value>The failed assembly loads.</value> + public IEnumerable<string> FailedPluginAssemblies + { + get { return _failedPluginAssemblies; } + } + + /// <summary> + /// Gets the list of currently loaded plugins + /// </summary> + /// <value>The plugins.</value> + [ImportMany(typeof(IPlugin))] + public IEnumerable<IPlugin> Plugins { get; protected set; } + + /// <summary> + /// Gets the list of Scheduled Tasks + /// </summary> + /// <value>The scheduled tasks.</value> + [ImportMany(typeof(IScheduledTask))] + public IEnumerable<IScheduledTask> ScheduledTasks { get; private set; } + + /// <summary> + /// Gets the web socket listeners. + /// </summary> + /// <value>The web socket listeners.</value> + [ImportMany(typeof(IWebSocketListener))] + public IEnumerable<IWebSocketListener> WebSocketListeners { get; private set; } + + /// <summary> + /// Gets the list of Localized string files + /// </summary> + /// <value>The string files.</value> + [ImportMany(typeof(LocalizedStringData))] + public IEnumerable<LocalizedStringData> StringFiles { get; private set; } + + /// <summary> + /// Gets the MEF CompositionContainer + /// </summary> + /// <value>The composition container.</value> + private CompositionContainer CompositionContainer { get; set; } + + /// <summary> + /// The _HTTP manager + /// </summary> + /// <value>The HTTP manager.</value> + public HttpManager HttpManager { get; private set; } + + /// <summary> + /// Gets or sets the TCP manager. + /// </summary> + /// <value>The TCP manager.</value> + public TcpManager TcpManager { get; private set; } + + /// <summary> + /// Gets the task manager. + /// </summary> + /// <value>The task manager.</value> + public TaskManager TaskManager { get; private set; } + + /// <summary> + /// Gets the iso manager. + /// </summary> + /// <value>The iso manager.</value> + public IIsoManager IsoManager { get; private set; } + + /// <summary> + /// Gets the rest services. + /// </summary> + /// <value>The rest services.</value> + [ImportMany(typeof(IRestfulService))] + public IEnumerable<IRestfulService> RestServices { get; private set; } + + /// <summary> + /// The _protobuf serializer initialized + /// </summary> + private bool _protobufSerializerInitialized; + /// <summary> + /// The _protobuf serializer sync lock + /// </summary> + private object _protobufSerializerSyncLock = new object(); + /// <summary> + /// Gets a dynamically compiled generated serializer that can serialize protocontracts without reflection + /// </summary> + private DynamicProtobufSerializer _protobufSerializer; + /// <summary> + /// Gets the protobuf serializer. + /// </summary> + /// <value>The protobuf serializer.</value> + public DynamicProtobufSerializer ProtobufSerializer + { + get + { + // Lazy load + LazyInitializer.EnsureInitialized(ref _protobufSerializer, ref _protobufSerializerInitialized, ref _protobufSerializerSyncLock, () => DynamicProtobufSerializer.Create(Assemblies)); + return _protobufSerializer; + } + private set + { + _protobufSerializer = value; + + if (value == null) + { + _protobufSerializerInitialized = false; + } + } + } + + /// <summary> + /// Gets the UDP server port number. + /// This can't be configurable because then the user would have to configure their client to discover the server. + /// </summary> + /// <value>The UDP server port number.</value> + public abstract int UdpServerPortNumber { get; } + + /// <summary> + /// Gets the name of the web application that can be used for url building. + /// All api urls will be of the form {protocol}://{host}:{port}/{appname}/... + /// </summary> + /// <value>The name of the web application.</value> + public string WebApplicationName + { + get { return "mediabrowser"; } + } + + /// <summary> + /// Gets the HTTP server URL prefix. + /// </summary> + /// <value>The HTTP server URL prefix.</value> + public virtual string HttpServerUrlPrefix + { + get + { + return "http://+:" + Configuration.HttpServerPortNumber + "/" + WebApplicationName + "/"; + } + } + + /// <summary> + /// Gets the kernel context. Subclasses will have to override. + /// </summary> + /// <value>The kernel context.</value> + public abstract KernelContext KernelContext { get; } + + /// <summary> + /// Gets the log file path. + /// </summary> + /// <value>The log file path.</value> + public string LogFilePath { get; private set; } + + /// <summary> + /// Gets the logger. + /// </summary> + /// <value>The logger.</value> + protected ILogger Logger { get; private set; } + + /// <summary> + /// Gets the assemblies. + /// </summary> + /// <value>The assemblies.</value> + public Assembly[] Assemblies { get; private set; } + + /// <summary> + /// Initializes the Kernel + /// </summary> + /// <param name="isoManager">The iso manager.</param> + /// <returns>Task.</returns> + public async Task Init(IIsoManager isoManager) + { + IsoManager = isoManager; + + Logger = Logging.LogManager.GetLogger(GetType().Name); + + ApplicationPaths = new TApplicationPathsType(); + + IsFirstRun = !File.Exists(ApplicationPaths.SystemConfigurationFilePath); + + // Performs initializations that can be reloaded at anytime + await Reload().ConfigureAwait(false); + } + + /// <summary> + /// Performs initializations that can be reloaded at anytime + /// </summary> + /// <returns>Task.</returns> + public async Task Reload() + { + OnReloadBeginning(); + + await ReloadInternal().ConfigureAwait(false); + + OnReloadCompleted(); + + Logger.Info("Kernel.Reload Complete"); + } + + /// <summary> + /// Performs initializations that can be reloaded at anytime + /// </summary> + /// <returns>Task.</returns> + protected virtual async Task ReloadInternal() + { + // Set these to null so that they can be lazy loaded again + Configuration = null; + ProtobufSerializer = null; + + ReloadLogger(); + + Logger.Info("Version {0} initializing", ApplicationVersion); + + DisposeHttpManager(); + HttpManager = new HttpManager(this); + + await OnConfigurationLoaded().ConfigureAwait(false); + + DisposeTaskManager(); + TaskManager = new TaskManager(this); + + Logger.Info("Loading Plugins"); + await ReloadComposableParts().ConfigureAwait(false); + + DisposeTcpManager(); + TcpManager = new TcpManager(this); + } + + /// <summary> + /// Called when [configuration loaded]. + /// </summary> + /// <returns>Task.</returns> + protected virtual Task OnConfigurationLoaded() + { + return Task.FromResult<object>(null); + } + + /// <summary> + /// Disposes and reloads all loggers + /// </summary> + public void ReloadLogger() + { + DisposeLogger(); + + LogFilePath = Path.Combine(ApplicationPaths.LogDirectoryPath, KernelContext + "-" + DateTime.Now.Ticks + ".log"); + + var logFile = new FileTarget(); + + logFile.FileName = LogFilePath; + logFile.Layout = "${longdate}, ${level}, ${logger}, ${message}"; + + AddLogTarget(logFile, "ApplicationLogFile"); + + Logging.Logger.LoggerInstance = Logging.LogManager.GetLogger("Global"); + + OnLoggerLoaded(); + } + + /// <summary> + /// Adds the log target. + /// </summary> + /// <param name="target">The target.</param> + /// <param name="name">The name.</param> + private void AddLogTarget(Target target, string name) + { + var config = LogManager.Configuration; + + config.RemoveTarget(name); + + target.Name = name; + config.AddTarget(name, target); + + var level = Configuration.EnableDebugLevelLogging ? LogLevel.Debug : LogLevel.Info; + + var rule = new LoggingRule("*", level, target); + config.LoggingRules.Add(rule); + + LogManager.Configuration = config; + } + + /// <summary> + /// Uses MEF to locate plugins + /// Subclasses can use this to locate types within plugins + /// </summary> + /// <returns>Task.</returns> + private async Task ReloadComposableParts() + { + _failedPluginAssemblies.Clear(); + + DisposeComposableParts(); + + Assemblies = GetComposablePartAssemblies().ToArray(); + + CompositionContainer = MefUtils.GetSafeCompositionContainer(Assemblies.Select(i => new AssemblyCatalog(i))); + + CompositionContainer.ComposeExportedValue("kernel", this); + + CompositionContainer.ComposeParts(this); + + await OnComposablePartsLoaded().ConfigureAwait(false); + + CompositionContainer.Catalog.Dispose(); + } + + /// <summary> + /// Gets the composable part assemblies. + /// </summary> + /// <returns>IEnumerable{Assembly}.</returns> + protected virtual IEnumerable<Assembly> GetComposablePartAssemblies() + { + // 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 + var pluginAssemblies = Directory.EnumerateFiles(ApplicationPaths.PluginsPath, "*.dll", SearchOption.TopDirectoryOnly) + .Select(file => + { + try + { + return Assembly.Load(File.ReadAllBytes((file))); + } + catch (Exception ex) + { + _failedPluginAssemblies.Add(file); + Logger.ErrorException("Error loading {0}", ex, file); + return null; + } + + }).Where(a => a != null); + + foreach (var pluginAssembly in pluginAssemblies) + { + yield return pluginAssembly; + } + + // Include composable parts in the Model assembly + yield return typeof (SystemInfo).Assembly; + + // Include composable parts in the Common assembly + yield return Assembly.GetExecutingAssembly(); + + // Include composable parts in the subclass assembly + yield return GetType().Assembly; + } + + /// <summary> + /// Fires after MEF finishes finding composable parts within plugin assemblies + /// </summary> + /// <returns>Task.</returns> + protected virtual Task OnComposablePartsLoaded() + { + return Task.Run(() => + { + foreach (var listener in WebSocketListeners) + { + listener.Initialize(this); + } + + foreach (var task in ScheduledTasks) + { + task.Initialize(this); + } + + // Start-up each plugin + Parallel.ForEach(Plugins, plugin => + { + Logger.Info("Initializing {0} {1}", plugin.Name, plugin.Version); + + try + { + plugin.Initialize(this); + + Logger.Info("{0} {1} initialized.", plugin.Name, plugin.Version); + } + catch (Exception ex) + { + Logger.ErrorException("Error initializing {0}", ex, plugin.Name); + } + }); + }); + } + + /// <summary> + /// Notifies that the kernel that a change has been made that requires a restart + /// </summary> + public void NotifyPendingRestart() + { + HasPendingRestart = true; + + TcpManager.SendWebSocketMessage("HasPendingRestartChanged", GetSystemInfo()); + + EventHelper.QueueEventIfNotNull(HasPendingRestartChanged, this, EventArgs.Empty); + } + + /// <summary> + /// Performs application-defined tasks associated with freeing, releasing, or resetting unmanaged resources. + /// </summary> + public void Dispose() + { + Dispose(true); + GC.SuppressFinalize(this); + } + + /// <summary> + /// Releases unmanaged and - optionally - managed resources. + /// </summary> + /// <param name="dispose"><c>true</c> to release both managed and unmanaged resources; <c>false</c> to release only unmanaged resources.</param> + protected virtual void Dispose(bool dispose) + { + if (dispose) + { + DisposeTcpManager(); + DisposeTaskManager(); + DisposeIsoManager(); + DisposeHttpManager(); + + DisposeComposableParts(); + } + } + + /// <summary> + /// Disposes the iso manager. + /// </summary> + private void DisposeIsoManager() + { + if (IsoManager != null) + { + IsoManager.Dispose(); + IsoManager = null; + } + } + + /// <summary> + /// Disposes the TCP manager. + /// </summary> + private void DisposeTcpManager() + { + if (TcpManager != null) + { + TcpManager.Dispose(); + TcpManager = null; + } + } + + /// <summary> + /// Disposes the task manager. + /// </summary> + private void DisposeTaskManager() + { + if (TaskManager != null) + { + TaskManager.Dispose(); + TaskManager = null; + } + } + + /// <summary> + /// Disposes the HTTP manager. + /// </summary> + private void DisposeHttpManager() + { + if (HttpManager != null) + { + HttpManager.Dispose(); + HttpManager = null; + } + } + + /// <summary> + /// Disposes all objects gathered through MEF composable parts + /// </summary> + protected virtual void DisposeComposableParts() + { + if (CompositionContainer != null) + { + CompositionContainer.Dispose(); + } + } + + /// <summary> + /// Disposes all logger resources + /// </summary> + private void DisposeLogger() + { + // Dispose all current loggers + var listeners = Trace.Listeners.OfType<TraceListener>().ToList(); + + Trace.Listeners.Clear(); + + foreach (var listener in listeners) + { + listener.Dispose(); + } + + } + + /// <summary> + /// Gets the current application version + /// </summary> + /// <value>The application version.</value> + public Version ApplicationVersion + { + get + { + return GetType().Assembly.GetName().Version; + } + } + + /// <summary> + /// Performs the pending restart. + /// </summary> + /// <returns>Task.</returns> + public void PerformPendingRestart() + { + if (HasPendingRestart) + { + RestartApplication(); + } + else + { + Logger.Info("PerformPendingRestart - not needed"); + } + } + + /// <summary> + /// Restarts the application. + /// </summary> + protected void RestartApplication() + { + Logger.Info("Restarting the application"); + + EventHelper.QueueEventIfNotNull(ApplicationRestartRequested, this, EventArgs.Empty); + } + + /// <summary> + /// Gets the system status. + /// </summary> + /// <returns>SystemInfo.</returns> + public virtual SystemInfo GetSystemInfo() + { + return new SystemInfo + { + HasPendingRestart = HasPendingRestart, + Version = DisplayVersion, + IsNetworkDeployed = ApplicationDeployment.IsNetworkDeployed, + WebSocketPortNumber = TcpManager.WebSocketPortNumber, + SupportsNativeWebSocket = TcpManager.SupportsNativeWebSocket, + FailedPluginAssemblies = FailedPluginAssemblies.ToArray() + }; + } + + /// <summary> + /// The _save lock + /// </summary> + private readonly object _configurationSaveLock = new object(); + + /// <summary> + /// Saves the current configuration + /// </summary> + public void SaveConfiguration() + { + lock (_configurationSaveLock) + { + XmlSerializer.SerializeToFile(Configuration, ApplicationPaths.SystemConfigurationFilePath); + } + + OnConfigurationUpdated(); + } + + /// <summary> + /// Gets the application paths. + /// </summary> + /// <value>The application paths.</value> + BaseApplicationPaths IKernel.ApplicationPaths + { + get { return ApplicationPaths; } + } + /// <summary> + /// Gets the configuration. + /// </summary> + /// <value>The configuration.</value> + BaseApplicationConfiguration IKernel.Configuration + { + get { return Configuration; } + } + } +} |
