using System.Linq; using System.Threading; using MediaBrowser.Common.Events; using MediaBrowser.Common.Net; using MediaBrowser.Common.Plugins; using MediaBrowser.Common.ScheduledTasks; using MediaBrowser.Controller; using MediaBrowser.Controller.Dto; using MediaBrowser.Controller.Entities; using MediaBrowser.Controller.Library; using MediaBrowser.Controller.Plugins; using MediaBrowser.Controller.Updates; using MediaBrowser.Model.Entities; using MediaBrowser.Model.Logging; using MediaBrowser.Model.Tasks; using MediaBrowser.Model.Updates; using System; namespace MediaBrowser.ServerApplication.EntryPoints { /// /// Class WebSocketEvents /// public class WebSocketEvents : IServerEntryPoint { /// /// The _server manager /// private readonly IServerManager _serverManager; /// /// The _logger /// private readonly ILogger _logger; /// /// The _user manager /// private readonly IUserManager _userManager; /// /// The _library manager /// private readonly ILibraryManager _libraryManager; /// /// The _installation manager /// private readonly IInstallationManager _installationManager; /// /// The _kernel /// private readonly IServerApplicationHost _appHost; /// /// The _task manager /// private readonly ITaskManager _taskManager; /// /// The _library changed sync lock /// private readonly object _libraryChangedSyncLock = new object(); /// /// Gets or sets the library update info. /// /// The library update info. private LibraryUpdateInfo LibraryUpdateInfo { get; set; } /// /// Gets or sets the library update timer. /// /// The library update timer. private Timer LibraryUpdateTimer { get; set; } /// /// The library update duration /// private const int LibraryUpdateDuration = 60000; /// /// Initializes a new instance of the class. /// /// The server manager. /// The logger. /// The user manager. public WebSocketEvents(IServerManager serverManager, IServerApplicationHost appHost, ILogger logger, IUserManager userManager, ILibraryManager libraryManager, IInstallationManager installationManager, ITaskManager taskManager) { _serverManager = serverManager; _logger = logger; _userManager = userManager; _libraryManager = libraryManager; _installationManager = installationManager; _appHost = appHost; _taskManager = taskManager; } public void Run() { _userManager.UserDeleted += userManager_UserDeleted; _userManager.UserUpdated += userManager_UserUpdated; _libraryManager.LibraryChanged += libraryManager_LibraryChanged; _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 e) { _serverManager.SendWebSocketMessage("ScheduledTaskEnded", e.Argument); } void _taskManager_TaskExecuting(object sender, EventArgs e) { var task = (IScheduledTask) sender; _serverManager.SendWebSocketMessage("ScheduledTaskStarted", task.Name); } /// /// Installations the manager_ package installation failed. /// /// The sender. /// The e. void installationManager_PackageInstallationFailed(object sender, GenericEventArgs e) { _serverManager.SendWebSocketMessage("PackageInstallationFailed", e.Argument); } /// /// Installations the manager_ package installation completed. /// /// The sender. /// The e. void installationManager_PackageInstallationCompleted(object sender, GenericEventArgs e) { _serverManager.SendWebSocketMessage("PackageInstallationCompleted", e.Argument); } /// /// Installations the manager_ package installation cancelled. /// /// The sender. /// The e. void installationManager_PackageInstallationCancelled(object sender, GenericEventArgs e) { _serverManager.SendWebSocketMessage("PackageInstallationCancelled", e.Argument); } /// /// Installations the manager_ package installing. /// /// The sender. /// The e. void installationManager_PackageInstalling(object sender, GenericEventArgs e) { _serverManager.SendWebSocketMessage("PackageInstalling", e.Argument); } /// /// Handles the LibraryChanged event of the libraryManager control. /// /// The source of the event. /// The instance containing the event data. void libraryManager_LibraryChanged(object sender, ChildrenChangedEventArgs e) { lock (_libraryChangedSyncLock) { if (LibraryUpdateInfo == null) { LibraryUpdateInfo = new LibraryUpdateInfo(); } if (LibraryUpdateTimer == null) { LibraryUpdateTimer = new Timer(LibraryUpdateTimerCallback, null, LibraryUpdateDuration, Timeout.Infinite); } else { LibraryUpdateTimer.Change(LibraryUpdateDuration, Timeout.Infinite); } LibraryUpdateInfo.Folders.Add(e.Folder.Id); LibraryUpdateInfo.ItemsAdded.AddRange(e.ItemsAdded.Select(i => i.Id)); LibraryUpdateInfo.ItemsUpdated.AddRange(e.ItemsUpdated.Select(i => i.Id)); LibraryUpdateInfo.ItemsRemoved.AddRange(e.ItemsRemoved.Select(i => i.Id)); } } /// /// Libraries the update timer callback. /// /// The state. private void LibraryUpdateTimerCallback(object state) { lock (_libraryChangedSyncLock) { _serverManager.SendWebSocketMessage("LibraryChanged", LibraryUpdateInfo); if (LibraryUpdateTimer != null) { LibraryUpdateTimer.Dispose(); LibraryUpdateTimer = null; } } } /// /// Installations the manager_ plugin uninstalled. /// /// The sender. /// The e. void InstallationManager_PluginUninstalled(object sender, GenericEventArgs e) { _serverManager.SendWebSocketMessage("PluginUninstalled", e.Argument.GetPluginInfo()); } /// /// Handles the HasPendingRestartChanged event of the kernel control. /// /// The source of the event. /// The instance containing the event data. void kernel_HasPendingRestartChanged(object sender, EventArgs e) { _serverManager.SendWebSocketMessage("RestartRequired", _appHost.GetSystemInfo()); } /// /// Users the manager_ user updated. /// /// The sender. /// The e. void userManager_UserUpdated(object sender, GenericEventArgs e) { var dto = new UserDtoBuilder(_logger).GetUserDto(e.Argument); _serverManager.SendWebSocketMessage("UserUpdated", dto); } /// /// Users the manager_ user deleted. /// /// The sender. /// The e. void userManager_UserDeleted(object sender, GenericEventArgs e) { _serverManager.SendWebSocketMessage("UserDeleted", e.Argument.Id.ToString()); } /// /// Performs application-defined tasks associated with freeing, releasing, or resetting unmanaged resources. /// public void Dispose() { Dispose(true); } /// /// Releases unmanaged and - optionally - managed resources. /// /// true to release both managed and unmanaged resources; false to release only unmanaged resources. protected virtual void Dispose(bool dispose) { if (dispose) { if (LibraryUpdateTimer != null) { LibraryUpdateTimer.Dispose(); LibraryUpdateTimer = null; } _userManager.UserDeleted -= userManager_UserDeleted; _userManager.UserUpdated -= userManager_UserUpdated; _libraryManager.LibraryChanged -= libraryManager_LibraryChanged; _installationManager.PluginUninstalled -= InstallationManager_PluginUninstalled; _installationManager.PackageInstalling -= installationManager_PackageInstalling; _installationManager.PackageInstallationCancelled -= installationManager_PackageInstallationCancelled; _installationManager.PackageInstallationCompleted -= installationManager_PackageInstallationCompleted; _installationManager.PackageInstallationFailed -= installationManager_PackageInstallationFailed; _appHost.HasPendingRestartChanged -= kernel_HasPendingRestartChanged; } } } }