69 lines
2.5 KiB
C#
69 lines
2.5 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using NLog;
|
|
using NzbDrone.Core.DecisionEngine;
|
|
using NzbDrone.Core.Parser.Model;
|
|
|
|
namespace NzbDrone.Core.Download
|
|
{
|
|
public interface IDownloadApprovedReports
|
|
{
|
|
List<DownloadDecision> DownloadApproved(List<DownloadDecision> decisions);
|
|
}
|
|
|
|
public class DownloadApprovedReports : IDownloadApprovedReports
|
|
{
|
|
private readonly IDownloadService _downloadService;
|
|
private readonly Logger _logger;
|
|
|
|
public DownloadApprovedReports(IDownloadService downloadService, Logger logger)
|
|
{
|
|
_downloadService = downloadService;
|
|
_logger = logger;
|
|
}
|
|
|
|
public List<DownloadDecision> DownloadApproved(List<DownloadDecision> decisions)
|
|
{
|
|
var qualifiedReports = GetQualifiedReports(decisions);
|
|
var downloadedReports = new List<DownloadDecision>();
|
|
|
|
foreach (var report in qualifiedReports)
|
|
{
|
|
var remoteEpisode = report.RemoteEpisode;
|
|
|
|
try
|
|
{
|
|
if (downloadedReports.SelectMany(r => r.RemoteEpisode.Episodes)
|
|
.Select(e => e.Id)
|
|
.ToList()
|
|
.Intersect(remoteEpisode.Episodes.Select(e => e.Id))
|
|
.Any())
|
|
{
|
|
continue;
|
|
}
|
|
|
|
_downloadService.DownloadReport(remoteEpisode);
|
|
downloadedReports.Add(report);
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
_logger.WarnException("Couldn't add report to download queue. " + remoteEpisode, e);
|
|
}
|
|
}
|
|
|
|
return downloadedReports;
|
|
}
|
|
|
|
public List<DownloadDecision> GetQualifiedReports(IEnumerable<DownloadDecision> decisions)
|
|
{
|
|
return decisions.Where(c => c.Approved && c.RemoteEpisode.Episodes.Any())
|
|
.OrderByDescending(c => c.RemoteEpisode.ParsedEpisodeInfo.Quality)
|
|
.ThenBy(c => c.RemoteEpisode.Episodes.Select(e => e.EpisodeNumber).MinOrDefault())
|
|
.ThenBy(c => c.RemoteEpisode.Report.Size.Round(200.Megabytes()) / c.RemoteEpisode.Episodes.Count)
|
|
.ThenBy(c => c.RemoteEpisode.Report.Age)
|
|
.ToList();
|
|
}
|
|
}
|
|
}
|