aboutsummaryrefslogtreecommitdiff
path: root/Jellyfin.Api/WebSocketListeners/ScheduledTasksWebSocketListener.cs
diff options
context:
space:
mode:
Diffstat (limited to 'Jellyfin.Api/WebSocketListeners/ScheduledTasksWebSocketListener.cs')
-rw-r--r--Jellyfin.Api/WebSocketListeners/ScheduledTasksWebSocketListener.cs80
1 files changed, 80 insertions, 0 deletions
diff --git a/Jellyfin.Api/WebSocketListeners/ScheduledTasksWebSocketListener.cs b/Jellyfin.Api/WebSocketListeners/ScheduledTasksWebSocketListener.cs
new file mode 100644
index 000000000..12f815ff7
--- /dev/null
+++ b/Jellyfin.Api/WebSocketListeners/ScheduledTasksWebSocketListener.cs
@@ -0,0 +1,80 @@
+using System.Collections.Generic;
+using System.Linq;
+using System.Threading.Tasks;
+using MediaBrowser.Controller.Net;
+using MediaBrowser.Model.Events;
+using MediaBrowser.Model.Tasks;
+using Microsoft.Extensions.Logging;
+
+namespace Jellyfin.Api.WebSocketListeners
+{
+ /// <summary>
+ /// Class ScheduledTasksWebSocketListener.
+ /// </summary>
+ public class ScheduledTasksWebSocketListener : BasePeriodicWebSocketListener<IEnumerable<TaskInfo>, WebSocketListenerState>
+ {
+ /// <summary>
+ /// Gets or sets the task manager.
+ /// </summary>
+ /// <value>The task manager.</value>
+ private readonly ITaskManager _taskManager;
+
+ /// <summary>
+ /// Initializes a new instance of the <see cref="ScheduledTasksWebSocketListener"/> class.
+ /// </summary>
+ /// <param name="logger">Instance of the <see cref="ILogger{ScheduledTasksWebSocketListener}"/> interface.</param>
+ /// <param name="taskManager">Instance of the <see cref="ITaskManager"/> interface.</param>
+ public ScheduledTasksWebSocketListener(ILogger<ScheduledTasksWebSocketListener> logger, ITaskManager taskManager)
+ : base(logger)
+ {
+ _taskManager = taskManager;
+
+ _taskManager.TaskExecuting += OnTaskExecuting;
+ _taskManager.TaskCompleted += OnTaskCompleted;
+ }
+
+ /// <summary>
+ /// Gets the name.
+ /// </summary>
+ /// <value>The name.</value>
+ protected override string Name => "ScheduledTasksInfo";
+
+ /// <summary>
+ /// Gets the data to send.
+ /// </summary>
+ /// <returns>Task{IEnumerable{TaskInfo}}.</returns>
+ protected override Task<IEnumerable<TaskInfo>> GetDataToSend()
+ {
+ return Task.FromResult(_taskManager.ScheduledTasks
+ .OrderBy(i => i.Name)
+ .Select(ScheduledTaskHelpers.GetTaskInfo)
+ .Where(i => !i.IsHidden));
+ }
+
+ /// <inheritdoc />
+ protected override void Dispose(bool dispose)
+ {
+ _taskManager.TaskExecuting -= OnTaskExecuting;
+ _taskManager.TaskCompleted -= OnTaskCompleted;
+
+ base.Dispose(dispose);
+ }
+
+ private void OnTaskCompleted(object sender, TaskCompletionEventArgs e)
+ {
+ SendData(true);
+ e.Task.TaskProgress -= OnTaskProgress;
+ }
+
+ private void OnTaskExecuting(object sender, GenericEventArgs<IScheduledTaskWorker> e)
+ {
+ SendData(true);
+ e.Argument.TaskProgress += OnTaskProgress;
+ }
+
+ private void OnTaskProgress(object sender, GenericEventArgs<double> e)
+ {
+ SendData(false);
+ }
+ }
+}