2018-11-23 07:03:32 +00:00
|
|
|
using System;
|
|
|
|
using System.Linq;
|
|
|
|
using Nancy.Responses;
|
|
|
|
using NzbDrone.Common.TPL;
|
|
|
|
using NzbDrone.Core.Datastore.Events;
|
|
|
|
using NzbDrone.Core.Download.Pending;
|
|
|
|
using NzbDrone.Core.Messaging.Events;
|
|
|
|
using NzbDrone.Core.Queue;
|
|
|
|
using NzbDrone.SignalR;
|
|
|
|
using Radarr.Http;
|
|
|
|
using Radarr.Http.Extensions;
|
|
|
|
|
|
|
|
namespace Radarr.Api.V2.Queue
|
|
|
|
{
|
|
|
|
public class QueueStatusModule : RadarrRestModuleWithSignalR<QueueStatusResource, NzbDrone.Core.Queue.Queue>,
|
|
|
|
IHandle<QueueUpdatedEvent>, IHandle<PendingReleasesUpdatedEvent>
|
|
|
|
{
|
|
|
|
private readonly IQueueService _queueService;
|
|
|
|
private readonly IPendingReleaseService _pendingReleaseService;
|
|
|
|
private readonly Debouncer _broadcastDebounce;
|
|
|
|
|
|
|
|
|
|
|
|
public QueueStatusModule(IBroadcastSignalRMessage broadcastSignalRMessage, IQueueService queueService, IPendingReleaseService pendingReleaseService)
|
|
|
|
: base(broadcastSignalRMessage, "queue/status")
|
|
|
|
{
|
|
|
|
_queueService = queueService;
|
|
|
|
_pendingReleaseService = pendingReleaseService;
|
|
|
|
|
|
|
|
_broadcastDebounce = new Debouncer(BroadcastChange, TimeSpan.FromSeconds(5));
|
|
|
|
|
|
|
|
|
|
|
|
Get["/"] = x => GetQueueStatusResponse();
|
|
|
|
}
|
|
|
|
|
|
|
|
private JsonResponse<QueueStatusResource> GetQueueStatusResponse()
|
|
|
|
{
|
|
|
|
return GetQueueStatus().AsResponse();
|
|
|
|
}
|
|
|
|
|
|
|
|
private QueueStatusResource GetQueueStatus()
|
|
|
|
{
|
|
|
|
_broadcastDebounce.Pause();
|
|
|
|
|
|
|
|
var queue = _queueService.GetQueue();
|
|
|
|
var pending = _pendingReleaseService.GetPendingQueue();
|
|
|
|
|
|
|
|
var resource = new QueueStatusResource
|
|
|
|
{
|
2019-06-14 03:54:25 +00:00
|
|
|
TotalCount = queue.Count + pending.Count,
|
|
|
|
Count = queue.Count(q => q.Movie != null) + pending.Count,
|
|
|
|
UnknownCount = queue.Count(q => q.Movie == null),
|
|
|
|
Errors = queue.Any(q => q.Movie != null && q.TrackedDownloadStatus.Equals("Error", StringComparison.InvariantCultureIgnoreCase)),
|
|
|
|
Warnings = queue.Any(q => q.Movie != null && q.TrackedDownloadStatus.Equals("Warning", StringComparison.InvariantCultureIgnoreCase)),
|
|
|
|
UnknownErrors = queue.Any(q => q.Movie == null && q.TrackedDownloadStatus.Equals("Error", StringComparison.InvariantCultureIgnoreCase)),
|
|
|
|
UnknownWarnings = queue.Any(q => q.Movie == null && q.TrackedDownloadStatus.Equals("Warning", StringComparison.InvariantCultureIgnoreCase))
|
2018-11-23 07:03:32 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
_broadcastDebounce.Resume();
|
|
|
|
|
|
|
|
return resource;
|
|
|
|
}
|
|
|
|
|
|
|
|
private void BroadcastChange()
|
|
|
|
{
|
|
|
|
BroadcastResourceChange(ModelAction.Updated, GetQueueStatus());
|
|
|
|
}
|
|
|
|
|
|
|
|
public void Handle(QueueUpdatedEvent message)
|
|
|
|
{
|
|
|
|
_broadcastDebounce.Execute();
|
|
|
|
}
|
|
|
|
|
|
|
|
public void Handle(PendingReleasesUpdatedEvent message)
|
|
|
|
{
|
|
|
|
_broadcastDebounce.Execute();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|