1 files changed, 207 insertions, 0 deletions
diff --git a/MediaBrowser.Server.Implementations/EntryPoints/WebSocketEvents.cs b/MediaBrowser.Server.Implementations/EntryPoints/WebSocketEvents.cs
new file mode 100644
index 000000000..b4bffc077
--- /dev/null
+++ b/MediaBrowser.Server.Implementations/EntryPoints/WebSocketEvents.cs
@@ -0,0 +1,207 @@
+using MediaBrowser.Common.Events;
+using MediaBrowser.Common.Net;
+using MediaBrowser.Common.Plugins;
+using MediaBrowser.Common.ScheduledTasks;
+using MediaBrowser.Common.Updates;
+using MediaBrowser.Controller;
+using MediaBrowser.Controller.Dto;
+using MediaBrowser.Controller.Entities;
+using MediaBrowser.Controller.Library;
+using MediaBrowser.Controller.Plugins;
+using MediaBrowser.Model.Logging;
+using MediaBrowser.Model.Tasks;
+using MediaBrowser.Model.Updates;
+using System;
+
+namespace MediaBrowser.Server.Implementations.EntryPoints
+{
+    /// <summary>
+    /// Class WebSocketEvents
+    /// </summary>
+    public class WebSocketEvents : IServerEntryPoint
+    {
+        /// <summary>
+        /// The _server manager
+        /// </summary>
+        private readonly IServerManager _serverManager;
+        /// <summary>
+        /// The _logger
+        /// </summary>
+        private readonly ILogger _logger;
+
+        /// <summary>
+        /// The _user manager
+        /// </summary>
+        private readonly IUserManager _userManager;
+
+        /// <summary>
+        /// The _installation manager
+        /// </summary>
+        private readonly IInstallationManager _installationManager;
+
+        /// <summary>
+        /// The _kernel
+        /// </summary>
+        private readonly IServerApplicationHost _appHost;
+
+        /// <summary>
+        /// The _task manager
+        /// </summary>
+        private readonly ITaskManager _taskManager;
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="WebSocketEvents" /> class.
+        /// </summary>
+        /// <param name="serverManager">The server manager.</param>
+        /// <param name="logger">The logger.</param>
+        /// <param name="userManager">The user manager.</param>
+        public WebSocketEvents(IServerManager serverManager, IServerApplicationHost appHost, ILogger logger, IUserManager userManager, IInstallationManager installationManager, ITaskManager taskManager)
+        {
+            _serverManager = serverManager;
+            _logger = logger;
+            _userManager = userManager;
+            _installationManager = installationManager;
+            _appHost = appHost;
+            _taskManager = taskManager;
+        }
+
+        public void Run()
+        {
+            _userManager.UserDeleted += userManager_UserDeleted;
+            _userManager.UserUpdated += userManager_UserUpdated;
+
+            _appHost.HasPendingRestartChanged += kernel_HasPendingRestartChanged;
+
+            _installationManager.PluginUninstalled += InstallationManager_PluginUninstalled;
+            _installationManager.PackageInstalling += installationManager_PackageInstalling;
+            _installationManager.PackageInstallationCancelled += installationManager_PackageInstallationCancelled;
+            _installationManager.PackageInstallationCompleted += installationManager_PackageInstallationCompleted;
+            _installationManager.PackageInstallationFailed += installationManager_PackageInstallationFailed;
+
+            _taskManager.TaskExecuting += _taskManager_TaskExecuting;
+            _taskManager.TaskCompleted += _taskManager_TaskCompleted;
+        }
+
+        void _taskManager_TaskCompleted(object sender, GenericEventArgs<TaskResult> e)
+        {
+            _serverManager.SendWebSocketMessage("ScheduledTaskEnded", e.Argument);
+        }
+
+        void _taskManager_TaskExecuting(object sender, EventArgs e)
+        {
+            var task = (IScheduledTask)sender;
+            _serverManager.SendWebSocketMessage("ScheduledTaskStarted", task.Name);
+        }
+
+        /// <summary>
+        /// Installations the manager_ package installation failed.
+        /// </summary>
+        /// <param name="sender">The sender.</param>
+        /// <param name="e">The e.</param>
+        void installationManager_PackageInstallationFailed(object sender, GenericEventArgs<InstallationInfo> e)
+        {
+            _serverManager.SendWebSocketMessage("PackageInstallationFailed", e.Argument);
+        }
+
+        /// <summary>
+        /// Installations the manager_ package installation completed.
+        /// </summary>
+        /// <param name="sender">The sender.</param>
+        /// <param name="e">The e.</param>
+        void installationManager_PackageInstallationCompleted(object sender, GenericEventArgs<InstallationInfo> e)
+        {
+            _serverManager.SendWebSocketMessage("PackageInstallationCompleted", e.Argument);
+        }
+
+        /// <summary>
+        /// Installations the manager_ package installation cancelled.
+        /// </summary>
+        /// <param name="sender">The sender.</param>
+        /// <param name="e">The e.</param>
+        void installationManager_PackageInstallationCancelled(object sender, GenericEventArgs<InstallationInfo> e)
+        {
+            _serverManager.SendWebSocketMessage("PackageInstallationCancelled", e.Argument);
+        }
+
+        /// <summary>
+        /// Installations the manager_ package installing.
+        /// </summary>
+        /// <param name="sender">The sender.</param>
+        /// <param name="e">The e.</param>
+        void installationManager_PackageInstalling(object sender, GenericEventArgs<InstallationInfo> e)
+        {
+            _serverManager.SendWebSocketMessage("PackageInstalling", e.Argument);
+        }
+
+        /// <summary>
+        /// Installations the manager_ plugin uninstalled.
+        /// </summary>
+        /// <param name="sender">The sender.</param>
+        /// <param name="e">The e.</param>
+        void InstallationManager_PluginUninstalled(object sender, GenericEventArgs<IPlugin> e)
+        {
+            _serverManager.SendWebSocketMessage("PluginUninstalled", e.Argument.GetPluginInfo());
+        }
+
+        /// <summary>
+        /// Handles the HasPendingRestartChanged event of the kernel control.
+        /// </summary>
+        /// <param name="sender">The source of the event.</param>
+        /// <param name="e">The <see cref="EventArgs" /> instance containing the event data.</param>
+        void kernel_HasPendingRestartChanged(object sender, EventArgs e)
+        {
+            _serverManager.SendWebSocketMessage("RestartRequired", _appHost.GetSystemInfo());
+        }
+
+        /// <summary>
+        /// Users the manager_ user updated.
+        /// </summary>
+        /// <param name="sender">The sender.</param>
+        /// <param name="e">The e.</param>
+        async void userManager_UserUpdated(object sender, GenericEventArgs<User> e)
+        {
+            var dto = await new UserDtoBuilder(_logger).GetUserDto(e.Argument).ConfigureAwait(false);
+
+            _serverManager.SendWebSocketMessage("UserUpdated", dto);
+        }
+
+        /// <summary>
+        /// Users the manager_ user deleted.
+        /// </summary>
+        /// <param name="sender">The sender.</param>
+        /// <param name="e">The e.</param>
+        void userManager_UserDeleted(object sender, GenericEventArgs<User> e)
+        {
+            _serverManager.SendWebSocketMessage("UserDeleted", e.Argument.Id.ToString("N"));
+        }
+
+        /// <summary>
+        /// Performs application-defined tasks associated with freeing, releasing, or resetting unmanaged resources.
+        /// </summary>
+        public void Dispose()
+        {
+            Dispose(true);
+        }
+
+        /// <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)
+            {
+                _userManager.UserDeleted -= userManager_UserDeleted;
+                _userManager.UserUpdated -= userManager_UserUpdated;
+
+                _installationManager.PluginUninstalled -= InstallationManager_PluginUninstalled;
+                _installationManager.PackageInstalling -= installationManager_PackageInstalling;
+                _installationManager.PackageInstallationCancelled -= installationManager_PackageInstallationCancelled;
+                _installationManager.PackageInstallationCompleted -= installationManager_PackageInstallationCompleted;
+                _installationManager.PackageInstallationFailed -= installationManager_PackageInstallationFailed;
+
+                _appHost.HasPendingRestartChanged -= kernel_HasPendingRestartChanged;
+            }
+        }
+    }
+}