2014-08-02 01:08:32 +00:00
|
|
|
|
using System;
|
|
|
|
|
using System.Linq;
|
2014-02-21 02:30:30 +00:00
|
|
|
|
using System.Collections.Generic;
|
2013-10-02 05:20:30 +00:00
|
|
|
|
using NzbDrone.Core.Download;
|
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Queue
|
|
|
|
|
{
|
|
|
|
|
public interface IQueueService
|
|
|
|
|
{
|
|
|
|
|
List<Queue> GetQueue();
|
2014-12-03 00:47:45 +00:00
|
|
|
|
Queue Find(int id);
|
2013-10-02 05:20:30 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public class QueueService : IQueueService
|
|
|
|
|
{
|
2014-04-19 15:09:22 +00:00
|
|
|
|
private readonly IDownloadTrackingService _downloadTrackingService;
|
2013-10-02 05:20:30 +00:00
|
|
|
|
|
2014-08-07 15:24:23 +00:00
|
|
|
|
public QueueService(IDownloadTrackingService downloadTrackingService)
|
2013-10-02 05:20:30 +00:00
|
|
|
|
{
|
2014-04-19 15:09:22 +00:00
|
|
|
|
_downloadTrackingService = downloadTrackingService;
|
2013-10-02 05:20:30 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public List<Queue> GetQueue()
|
|
|
|
|
{
|
2014-05-24 10:50:34 +00:00
|
|
|
|
var queueItems = _downloadTrackingService.GetQueuedDownloads()
|
2014-07-17 19:02:27 +00:00
|
|
|
|
.OrderBy(v => v.DownloadItem.RemainingTime)
|
2014-05-24 10:50:34 +00:00
|
|
|
|
.ToList();
|
2014-02-14 05:31:49 +00:00
|
|
|
|
|
2014-04-19 15:09:22 +00:00
|
|
|
|
return MapQueue(queueItems);
|
2013-10-02 05:20:30 +00:00
|
|
|
|
}
|
|
|
|
|
|
2014-12-03 00:47:45 +00:00
|
|
|
|
public Queue Find(int id)
|
|
|
|
|
{
|
|
|
|
|
return GetQueue().SingleOrDefault(q => q.Id == id);
|
|
|
|
|
}
|
|
|
|
|
|
2014-09-13 23:13:00 +00:00
|
|
|
|
private List<Queue> MapQueue(IEnumerable<TrackedDownload> trackedDownloads)
|
2013-10-02 05:20:30 +00:00
|
|
|
|
{
|
|
|
|
|
var queued = new List<Queue>();
|
|
|
|
|
|
2014-09-13 23:13:00 +00:00
|
|
|
|
foreach (var trackedDownload in trackedDownloads)
|
2013-10-02 05:20:30 +00:00
|
|
|
|
{
|
2014-09-13 23:13:00 +00:00
|
|
|
|
foreach (var episode in trackedDownload.RemoteEpisode.Episodes)
|
2013-10-02 05:20:30 +00:00
|
|
|
|
{
|
2014-06-08 08:22:55 +00:00
|
|
|
|
var queue = new Queue
|
|
|
|
|
{
|
2014-10-22 23:57:37 +00:00
|
|
|
|
Id = episode.Id ^ (trackedDownload.DownloadItem.DownloadClientId.GetHashCode() << 16),
|
2014-09-13 23:13:00 +00:00
|
|
|
|
Series = trackedDownload.RemoteEpisode.Series,
|
2014-06-08 08:22:55 +00:00
|
|
|
|
Episode = episode,
|
2014-09-13 23:13:00 +00:00
|
|
|
|
Quality = trackedDownload.RemoteEpisode.ParsedEpisodeInfo.Quality,
|
|
|
|
|
Title = trackedDownload.DownloadItem.Title,
|
|
|
|
|
Size = trackedDownload.DownloadItem.TotalSize,
|
|
|
|
|
Sizeleft = trackedDownload.DownloadItem.RemainingSize,
|
|
|
|
|
Timeleft = trackedDownload.DownloadItem.RemainingTime,
|
|
|
|
|
Status = trackedDownload.DownloadItem.Status.ToString(),
|
|
|
|
|
RemoteEpisode = trackedDownload.RemoteEpisode,
|
|
|
|
|
TrackedDownloadStatus = trackedDownload.Status.ToString(),
|
2014-10-22 23:57:37 +00:00
|
|
|
|
StatusMessages = trackedDownload.StatusMessages,
|
|
|
|
|
TrackingId = trackedDownload.TrackingId
|
2014-06-08 08:22:55 +00:00
|
|
|
|
};
|
2014-07-17 19:02:27 +00:00
|
|
|
|
|
2014-08-02 01:08:32 +00:00
|
|
|
|
if (queue.Timeleft.HasValue)
|
|
|
|
|
{
|
|
|
|
|
queue.EstimatedCompletionTime = DateTime.UtcNow.Add(queue.Timeleft.Value);
|
|
|
|
|
}
|
|
|
|
|
|
2013-10-09 06:58:29 +00:00
|
|
|
|
queued.Add(queue);
|
2013-10-02 05:20:30 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return queued;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|