95 lines
3.6 KiB
C#
95 lines
3.6 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using NLog;
|
|
using NzbDrone.Common;
|
|
using NzbDrone.Common.Messaging;
|
|
using NzbDrone.Core.MediaFiles.Events;
|
|
|
|
namespace NzbDrone.Core.MediaFiles.EpisodeImport
|
|
{
|
|
public interface IImportApprovedEpisodes
|
|
{
|
|
List<ImportDecision> Import(List<ImportDecision> decisions, bool newDownloads = false);
|
|
}
|
|
|
|
public class ImportApprovedEpisodes : IImportApprovedEpisodes
|
|
{
|
|
private readonly IMoveEpisodeFiles _episodeFileMover;
|
|
private readonly MediaFileService _mediaFileService;
|
|
private readonly DiskProvider _diskProvider;
|
|
private readonly IMessageAggregator _messageAggregator;
|
|
private readonly Logger _logger;
|
|
|
|
public ImportApprovedEpisodes(IMoveEpisodeFiles episodeFileMover,
|
|
MediaFileService mediaFileService,
|
|
DiskProvider diskProvider,
|
|
IMessageAggregator messageAggregator,
|
|
Logger logger)
|
|
{
|
|
_episodeFileMover = episodeFileMover;
|
|
_mediaFileService = mediaFileService;
|
|
_diskProvider = diskProvider;
|
|
_messageAggregator = messageAggregator;
|
|
_logger = logger;
|
|
}
|
|
|
|
public List<ImportDecision> Import(List<ImportDecision> decisions, bool newDownload = false)
|
|
{
|
|
var qualifiedReports = GetQualifiedReports(decisions);
|
|
var imported = new List<ImportDecision>();
|
|
|
|
foreach (var report in qualifiedReports)
|
|
{
|
|
var localEpisode = report.LocalEpisode;
|
|
|
|
try
|
|
{
|
|
if (imported.SelectMany(r => r.LocalEpisode.Episodes)
|
|
.Select(e => e.Id)
|
|
.ToList()
|
|
.Intersect(localEpisode.Episodes.Select(e => e.Id))
|
|
.Any())
|
|
{
|
|
continue;
|
|
}
|
|
|
|
var episodeFile = new EpisodeFile();
|
|
episodeFile.DateAdded = DateTime.UtcNow;
|
|
episodeFile.SeriesId = localEpisode.Series.Id;
|
|
episodeFile.Path = localEpisode.Path.CleanPath();
|
|
episodeFile.Size = _diskProvider.GetFileSize(localEpisode.Path);
|
|
episodeFile.Quality = localEpisode.Quality;
|
|
episodeFile.SeasonNumber = localEpisode.SeasonNumber;
|
|
episodeFile.SceneName = Path.GetFileNameWithoutExtension(localEpisode.Path.CleanPath());
|
|
episodeFile.Episodes = localEpisode.Episodes;
|
|
|
|
if (newDownload)
|
|
{
|
|
episodeFile = _episodeFileMover.MoveEpisodeFile(episodeFile, localEpisode);
|
|
}
|
|
|
|
_mediaFileService.Add(episodeFile);
|
|
_messageAggregator.PublishEvent(new EpisodeImportedEvent(episodeFile));
|
|
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
_logger.WarnException("Couldn't add report to download queue. " + localEpisode, e);
|
|
}
|
|
}
|
|
|
|
return imported;
|
|
}
|
|
|
|
private List<ImportDecision> GetQualifiedReports(List<ImportDecision> decisions)
|
|
{
|
|
return decisions.Where(c => c.Approved)
|
|
.OrderByDescending(c => c.LocalEpisode.Quality)
|
|
.ToList();
|
|
}
|
|
}
|
|
}
|