mirror of https://github.com/Radarr/Radarr
67 lines
2.3 KiB
C#
67 lines
2.3 KiB
C#
![]() |
using System;
|
|||
|
using System.Collections.Generic;
|
|||
|
using System.Linq;
|
|||
|
using NLog;
|
|||
|
using NzbDrone.Core.DecisionEngine;
|
|||
|
|
|||
|
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(List<DownloadDecision> decisions)
|
|||
|
{
|
|||
|
return decisions.Where(c => c.Approved)
|
|||
|
.OrderByDescending(c => c.RemoteEpisode.ParsedEpisodeInfo.Quality)
|
|||
|
.ThenBy(c => c.RemoteEpisode.Episodes.Select(e => e.EpisodeNumber).MinOrDefault())
|
|||
|
.ThenBy(c => c.RemoteEpisode.Report.Age)
|
|||
|
.ToList();
|
|||
|
}
|
|||
|
}
|
|||
|
}
|