2018-03-14 20:41:36 +00:00
|
|
|
using System;
|
2014-02-21 02:30:30 +00:00
|
|
|
using System.Collections.Generic;
|
2014-12-26 13:55:35 +00:00
|
|
|
using System.Linq;
|
|
|
|
using NzbDrone.Common.Crypto;
|
2014-12-19 00:26:42 +00:00
|
|
|
using NzbDrone.Core.Download.TrackedDownloads;
|
2019-06-14 03:54:25 +00:00
|
|
|
using NzbDrone.Core.Languages;
|
2014-12-19 00:26:42 +00:00
|
|
|
using NzbDrone.Core.Messaging.Events;
|
2018-03-14 20:41:36 +00:00
|
|
|
using NzbDrone.Core.Movies;
|
2019-11-01 22:31:48 +00:00
|
|
|
using NzbDrone.Core.Qualities;
|
2013-10-02 05:20:30 +00:00
|
|
|
|
|
|
|
namespace NzbDrone.Core.Queue
|
|
|
|
{
|
|
|
|
public interface IQueueService
|
|
|
|
{
|
|
|
|
List<Queue> GetQueue();
|
2014-12-03 00:47:45 +00:00
|
|
|
Queue Find(int id);
|
2019-06-14 03:54:25 +00:00
|
|
|
void Remove(int id);
|
2013-10-02 05:20:30 +00:00
|
|
|
}
|
|
|
|
|
2014-12-19 00:26:42 +00:00
|
|
|
public class QueueService : IQueueService, IHandle<TrackedDownloadRefreshedEvent>
|
2013-10-02 05:20:30 +00:00
|
|
|
{
|
2014-12-19 00:26:42 +00:00
|
|
|
private readonly IEventAggregator _eventAggregator;
|
|
|
|
private static List<Queue> _queue = new List<Queue>();
|
2013-10-02 05:20:30 +00:00
|
|
|
|
2014-12-19 00:26:42 +00:00
|
|
|
public QueueService(IEventAggregator eventAggregator)
|
2013-10-02 05:20:30 +00:00
|
|
|
{
|
2014-12-19 00:26:42 +00:00
|
|
|
_eventAggregator = eventAggregator;
|
2013-10-02 05:20:30 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
public List<Queue> GetQueue()
|
|
|
|
{
|
2014-12-19 00:26:42 +00:00
|
|
|
return _queue;
|
2013-10-02 05:20:30 +00:00
|
|
|
}
|
|
|
|
|
2014-12-03 00:47:45 +00:00
|
|
|
public Queue Find(int id)
|
|
|
|
{
|
2014-12-19 00:26:42 +00:00
|
|
|
return _queue.SingleOrDefault(q => q.Id == id);
|
2014-12-03 00:47:45 +00:00
|
|
|
}
|
|
|
|
|
2019-06-14 03:54:25 +00:00
|
|
|
public void Remove(int id)
|
2013-10-02 05:20:30 +00:00
|
|
|
{
|
2019-06-14 03:54:25 +00:00
|
|
|
_queue.Remove(Find(id));
|
2014-12-19 00:26:42 +00:00
|
|
|
}
|
2013-10-02 05:20:30 +00:00
|
|
|
|
2014-12-26 13:55:35 +00:00
|
|
|
private IEnumerable<Queue> MapQueue(TrackedDownload trackedDownload)
|
2014-12-19 00:26:42 +00:00
|
|
|
{
|
2018-03-14 20:41:36 +00:00
|
|
|
if (trackedDownload.RemoteMovie != null && trackedDownload.RemoteMovie.Movie != null)
|
2015-07-11 20:49:38 +00:00
|
|
|
{
|
2017-01-02 21:01:11 +00:00
|
|
|
yield return MapMovie(trackedDownload, trackedDownload.RemoteMovie.Movie);
|
2015-07-11 20:49:38 +00:00
|
|
|
}
|
2019-06-14 03:54:25 +00:00
|
|
|
else
|
|
|
|
{
|
|
|
|
yield return MapMovie(trackedDownload, null);
|
|
|
|
}
|
2015-07-11 20:49:38 +00:00
|
|
|
}
|
2014-12-19 00:26:42 +00:00
|
|
|
|
2017-01-02 21:01:11 +00:00
|
|
|
private Queue MapMovie(TrackedDownload trackedDownload, Movie movie)
|
|
|
|
{
|
|
|
|
var queue = new Queue
|
|
|
|
{
|
2019-12-22 21:24:11 +00:00
|
|
|
Languages = trackedDownload.RemoteMovie?.ParsedMovieInfo.Languages ?? new List<Language> { Language.Unknown },
|
2019-11-01 22:31:48 +00:00
|
|
|
Quality = trackedDownload.RemoteMovie?.ParsedMovieInfo.Quality ?? new QualityModel(Quality.Unknown),
|
2017-01-02 21:01:11 +00:00
|
|
|
Title = trackedDownload.DownloadItem.Title,
|
|
|
|
Size = trackedDownload.DownloadItem.TotalSize,
|
|
|
|
Sizeleft = trackedDownload.DownloadItem.RemainingSize,
|
|
|
|
Timeleft = trackedDownload.DownloadItem.RemainingTime,
|
|
|
|
Status = trackedDownload.DownloadItem.Status.ToString(),
|
2020-03-29 17:48:34 +00:00
|
|
|
TrackedDownloadStatus = trackedDownload.Status,
|
|
|
|
TrackedDownloadState = trackedDownload.State,
|
2017-01-02 21:01:11 +00:00
|
|
|
StatusMessages = trackedDownload.StatusMessages.ToList(),
|
2019-06-14 03:54:25 +00:00
|
|
|
ErrorMessage = trackedDownload.DownloadItem.Message,
|
2017-01-02 21:01:11 +00:00
|
|
|
RemoteMovie = trackedDownload.RemoteMovie,
|
|
|
|
DownloadId = trackedDownload.DownloadItem.DownloadId,
|
|
|
|
Protocol = trackedDownload.Protocol,
|
2018-11-23 07:03:32 +00:00
|
|
|
Movie = movie,
|
|
|
|
DownloadClient = trackedDownload.DownloadItem.DownloadClient,
|
2019-06-14 03:54:25 +00:00
|
|
|
Indexer = trackedDownload.Indexer,
|
|
|
|
OutputPath = trackedDownload.DownloadItem.OutputPath.ToString()
|
2017-01-02 21:01:11 +00:00
|
|
|
};
|
|
|
|
|
2019-06-14 03:54:25 +00:00
|
|
|
queue.Id = HashConverter.GetHashInt31(string.Format("trackedDownload-{0}", trackedDownload.DownloadItem.DownloadId));
|
|
|
|
|
2017-01-02 21:01:11 +00:00
|
|
|
if (queue.Timeleft.HasValue)
|
|
|
|
{
|
|
|
|
queue.EstimatedCompletionTime = DateTime.UtcNow.Add(queue.Timeleft.Value);
|
|
|
|
}
|
|
|
|
|
|
|
|
return queue;
|
|
|
|
}
|
2019-06-14 03:54:25 +00:00
|
|
|
|
|
|
|
public void Handle(TrackedDownloadRefreshedEvent message)
|
|
|
|
{
|
|
|
|
_queue = message.TrackedDownloads.OrderBy(c => c.DownloadItem.RemainingTime).SelectMany(MapQueue)
|
|
|
|
.ToList();
|
|
|
|
|
|
|
|
_eventAggregator.PublishEvent(new QueueUpdatedEvent());
|
|
|
|
}
|
2013-10-02 05:20:30 +00:00
|
|
|
}
|
|
|
|
}
|