2013-07-06 21:47:49 +00:00
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
2015-01-27 05:57:07 +00:00
|
|
|
|
using System.IO;
|
2013-07-06 21:47:49 +00:00
|
|
|
|
using System.Linq;
|
|
|
|
|
using NLog;
|
2014-01-06 06:20:08 +00:00
|
|
|
|
using NzbDrone.Common.Disk;
|
2015-01-27 05:57:07 +00:00
|
|
|
|
using NzbDrone.Common.Extensions;
|
2013-07-06 21:47:49 +00:00
|
|
|
|
using NzbDrone.Core.DecisionEngine;
|
|
|
|
|
using NzbDrone.Core.Parser;
|
|
|
|
|
using NzbDrone.Core.Parser.Model;
|
2014-01-18 11:44:36 +00:00
|
|
|
|
using NzbDrone.Core.Qualities;
|
2013-07-06 21:47:49 +00:00
|
|
|
|
using NzbDrone.Core.Tv;
|
2014-04-10 17:58:50 +00:00
|
|
|
|
using NzbDrone.Core.MediaFiles.MediaInfo;
|
2013-07-06 21:47:49 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.MediaFiles.EpisodeImport
|
|
|
|
|
{
|
|
|
|
|
public interface IMakeImportDecision
|
|
|
|
|
{
|
2015-10-03 17:45:26 +00:00
|
|
|
|
List<ImportDecision> GetImportDecisions(List<string> videoFiles, Series series);
|
2015-01-27 05:57:07 +00:00
|
|
|
|
List<ImportDecision> GetImportDecisions(List<string> videoFiles, Series series, ParsedEpisodeInfo folderInfo, bool sceneSource);
|
2013-07-06 21:47:49 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public class ImportDecisionMaker : IMakeImportDecision
|
|
|
|
|
{
|
2015-01-27 05:57:07 +00:00
|
|
|
|
private readonly IEnumerable<IImportDecisionEngineSpecification> _specifications;
|
2013-07-06 21:47:49 +00:00
|
|
|
|
private readonly IParsingService _parsingService;
|
2013-07-19 05:05:07 +00:00
|
|
|
|
private readonly IMediaFileService _mediaFileService;
|
2013-07-15 23:53:06 +00:00
|
|
|
|
private readonly IDiskProvider _diskProvider;
|
2014-04-10 17:58:50 +00:00
|
|
|
|
private readonly IVideoFileInfoReader _videoFileInfoReader;
|
2015-01-27 05:57:07 +00:00
|
|
|
|
private readonly IDetectSample _detectSample;
|
2013-07-06 21:47:49 +00:00
|
|
|
|
private readonly Logger _logger;
|
|
|
|
|
|
2015-01-27 05:57:07 +00:00
|
|
|
|
public ImportDecisionMaker(IEnumerable<IImportDecisionEngineSpecification> specifications,
|
2013-07-15 23:53:06 +00:00
|
|
|
|
IParsingService parsingService,
|
2013-07-19 05:05:07 +00:00
|
|
|
|
IMediaFileService mediaFileService,
|
2013-07-15 23:53:06 +00:00
|
|
|
|
IDiskProvider diskProvider,
|
2014-04-10 17:58:50 +00:00
|
|
|
|
IVideoFileInfoReader videoFileInfoReader,
|
2015-01-27 05:57:07 +00:00
|
|
|
|
IDetectSample detectSample,
|
2013-07-15 23:53:06 +00:00
|
|
|
|
Logger logger)
|
2013-07-06 21:47:49 +00:00
|
|
|
|
{
|
|
|
|
|
_specifications = specifications;
|
|
|
|
|
_parsingService = parsingService;
|
2013-07-19 05:05:07 +00:00
|
|
|
|
_mediaFileService = mediaFileService;
|
2013-07-15 23:53:06 +00:00
|
|
|
|
_diskProvider = diskProvider;
|
2014-04-10 17:58:50 +00:00
|
|
|
|
_videoFileInfoReader = videoFileInfoReader;
|
2015-01-27 05:57:07 +00:00
|
|
|
|
_detectSample = detectSample;
|
2013-07-06 21:47:49 +00:00
|
|
|
|
_logger = logger;
|
|
|
|
|
}
|
|
|
|
|
|
2015-01-27 05:57:07 +00:00
|
|
|
|
public List<ImportDecision> GetImportDecisions(List<string> videoFiles, Series series)
|
|
|
|
|
{
|
|
|
|
|
return GetImportDecisions(videoFiles, series, null, false);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public List<ImportDecision> GetImportDecisions(List<string> videoFiles, Series series, ParsedEpisodeInfo folderInfo, bool sceneSource)
|
2013-07-06 21:47:49 +00:00
|
|
|
|
{
|
2014-07-23 23:43:54 +00:00
|
|
|
|
var newFiles = _mediaFileService.FilterExistingFiles(videoFiles.ToList(), series);
|
2013-07-19 05:05:07 +00:00
|
|
|
|
|
2013-12-01 00:33:59 +00:00
|
|
|
|
_logger.Debug("Analyzing {0}/{1} files.", newFiles.Count, videoFiles.Count());
|
2013-07-19 05:05:07 +00:00
|
|
|
|
|
2015-01-27 05:57:07 +00:00
|
|
|
|
var shouldUseFolderName = ShouldUseFolderName(videoFiles, series, folderInfo);
|
|
|
|
|
var decisions = new List<ImportDecision>();
|
|
|
|
|
|
|
|
|
|
foreach (var file in newFiles)
|
|
|
|
|
{
|
|
|
|
|
decisions.AddIfNotNull(GetDecision(file, series, folderInfo, sceneSource, shouldUseFolderName));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return decisions;
|
2013-07-06 21:47:49 +00:00
|
|
|
|
}
|
|
|
|
|
|
2015-01-27 05:57:07 +00:00
|
|
|
|
private ImportDecision GetDecision(string file, Series series, ParsedEpisodeInfo folderInfo, bool sceneSource, bool shouldUseFolderName)
|
2013-07-06 21:47:49 +00:00
|
|
|
|
{
|
2015-01-27 05:57:07 +00:00
|
|
|
|
ImportDecision decision = null;
|
|
|
|
|
|
|
|
|
|
try
|
2013-07-06 21:47:49 +00:00
|
|
|
|
{
|
2015-01-27 05:57:07 +00:00
|
|
|
|
var localEpisode = _parsingService.GetLocalEpisode(file, series, shouldUseFolderName ? folderInfo : null, sceneSource);
|
2013-07-06 21:47:49 +00:00
|
|
|
|
|
2015-01-27 05:57:07 +00:00
|
|
|
|
if (localEpisode != null)
|
2013-07-06 21:47:49 +00:00
|
|
|
|
{
|
2015-01-27 05:57:07 +00:00
|
|
|
|
localEpisode.Quality = GetQuality(folderInfo, localEpisode.Quality, series);
|
|
|
|
|
localEpisode.Size = _diskProvider.GetFileSize(file);
|
2014-09-04 00:57:27 +00:00
|
|
|
|
|
2015-01-27 05:57:07 +00:00
|
|
|
|
_logger.Debug("Size: {0}", localEpisode.Size);
|
2013-07-06 21:47:49 +00:00
|
|
|
|
|
2015-01-27 05:57:07 +00:00
|
|
|
|
//TODO: make it so media info doesn't ruin the import process of a new series
|
|
|
|
|
if (sceneSource)
|
2013-07-06 21:47:49 +00:00
|
|
|
|
{
|
2015-01-27 05:57:07 +00:00
|
|
|
|
localEpisode.MediaInfo = _videoFileInfoReader.GetMediaInfo(file);
|
2013-07-06 21:47:49 +00:00
|
|
|
|
}
|
2015-01-27 05:57:07 +00:00
|
|
|
|
|
2015-05-11 22:17:22 +00:00
|
|
|
|
if (localEpisode.Episodes.Empty())
|
|
|
|
|
{
|
2015-10-03 12:12:36 +00:00
|
|
|
|
decision = new ImportDecision(localEpisode, new Rejection("Invalid season or episode"));
|
2015-05-11 22:17:22 +00:00
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
decision = GetDecision(localEpisode);
|
|
|
|
|
}
|
2013-07-06 21:47:49 +00:00
|
|
|
|
}
|
2015-01-27 05:57:07 +00:00
|
|
|
|
|
|
|
|
|
else
|
2014-09-04 00:57:27 +00:00
|
|
|
|
{
|
2015-01-27 05:57:07 +00:00
|
|
|
|
localEpisode = new LocalEpisode();
|
2014-12-03 00:47:45 +00:00
|
|
|
|
localEpisode.Path = file;
|
2014-09-04 00:57:27 +00:00
|
|
|
|
|
2015-01-27 05:57:07 +00:00
|
|
|
|
decision = new ImportDecision(localEpisode, new Rejection("Unable to parse file"));
|
2013-07-06 21:47:49 +00:00
|
|
|
|
}
|
2015-01-27 05:57:07 +00:00
|
|
|
|
}
|
|
|
|
|
catch (Exception e)
|
|
|
|
|
{
|
2017-01-05 23:32:17 +00:00
|
|
|
|
_logger.Error(e, "Couldn't import file. {0}", file);
|
2016-04-17 21:17:46 +00:00
|
|
|
|
|
|
|
|
|
var localEpisode = new LocalEpisode { Path = file };
|
|
|
|
|
decision = new ImportDecision(localEpisode, new Rejection("Unexpected error processing file"));
|
2013-07-06 21:47:49 +00:00
|
|
|
|
}
|
2015-01-27 05:57:07 +00:00
|
|
|
|
|
|
|
|
|
return decision;
|
2013-07-06 21:47:49 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private ImportDecision GetDecision(LocalEpisode localEpisode)
|
|
|
|
|
{
|
|
|
|
|
var reasons = _specifications.Select(c => EvaluateSpec(c, localEpisode))
|
2015-01-27 05:57:07 +00:00
|
|
|
|
.Where(c => c != null);
|
2013-07-06 21:47:49 +00:00
|
|
|
|
|
|
|
|
|
return new ImportDecision(localEpisode, reasons.ToArray());
|
|
|
|
|
}
|
|
|
|
|
|
2015-01-27 05:57:07 +00:00
|
|
|
|
private Rejection EvaluateSpec(IImportDecisionEngineSpecification spec, LocalEpisode localEpisode)
|
2013-07-06 21:47:49 +00:00
|
|
|
|
{
|
|
|
|
|
try
|
|
|
|
|
{
|
2015-01-27 05:57:07 +00:00
|
|
|
|
var result = spec.IsSatisfiedBy(localEpisode);
|
2013-07-06 21:47:49 +00:00
|
|
|
|
|
2015-01-27 05:57:07 +00:00
|
|
|
|
if (!result.Accepted)
|
2013-07-06 21:47:49 +00:00
|
|
|
|
{
|
2015-01-27 05:57:07 +00:00
|
|
|
|
return new Rejection(result.Reason);
|
2013-07-06 21:47:49 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
catch (Exception e)
|
|
|
|
|
{
|
|
|
|
|
//e.Data.Add("report", remoteEpisode.Report.ToJson());
|
|
|
|
|
//e.Data.Add("parsed", remoteEpisode.ParsedEpisodeInfo.ToJson());
|
2017-01-05 23:32:17 +00:00
|
|
|
|
_logger.Error(e, "Couldn't evaluate decision on {0}", localEpisode.Path);
|
|
|
|
|
return new Rejection($"{spec.GetType().Name}: {e.Message}");
|
2013-07-06 21:47:49 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return null;
|
|
|
|
|
}
|
2015-01-27 05:57:07 +00:00
|
|
|
|
|
|
|
|
|
private bool ShouldUseFolderName(List<string> videoFiles, Series series, ParsedEpisodeInfo folderInfo)
|
|
|
|
|
{
|
|
|
|
|
if (folderInfo == null)
|
|
|
|
|
{
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (folderInfo.FullSeason)
|
|
|
|
|
{
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return videoFiles.Count(file =>
|
|
|
|
|
{
|
|
|
|
|
var size = _diskProvider.GetFileSize(file);
|
|
|
|
|
var fileQuality = QualityParser.ParseQuality(file);
|
2016-02-10 22:46:50 +00:00
|
|
|
|
var sample = _detectSample.IsSample(series, GetQuality(folderInfo, fileQuality, series), file, size, folderInfo.IsPossibleSpecialEpisode);
|
2015-01-27 05:57:07 +00:00
|
|
|
|
|
|
|
|
|
if (sample)
|
|
|
|
|
{
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (SceneChecker.IsSceneTitle(Path.GetFileName(file)))
|
|
|
|
|
{
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return true;
|
|
|
|
|
}) == 1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private QualityModel GetQuality(ParsedEpisodeInfo folderInfo, QualityModel fileQuality, Series series)
|
|
|
|
|
{
|
2016-01-05 07:11:06 +00:00
|
|
|
|
if (UseFolderQuality(folderInfo, fileQuality, series))
|
2015-01-27 05:57:07 +00:00
|
|
|
|
{
|
|
|
|
|
_logger.Debug("Using quality from folder: {0}", folderInfo.Quality);
|
|
|
|
|
return folderInfo.Quality;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return fileQuality;
|
|
|
|
|
}
|
2016-01-05 07:11:06 +00:00
|
|
|
|
|
|
|
|
|
private bool UseFolderQuality(ParsedEpisodeInfo folderInfo, QualityModel fileQuality, Series series)
|
|
|
|
|
{
|
|
|
|
|
if (folderInfo == null)
|
|
|
|
|
{
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (folderInfo.Quality.Quality == Quality.Unknown)
|
|
|
|
|
{
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (fileQuality.QualitySource == QualitySource.Extension)
|
|
|
|
|
{
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (new QualityModelComparer(series.Profile).Compare(folderInfo.Quality, fileQuality) > 0)
|
|
|
|
|
{
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return false;
|
|
|
|
|
}
|
2013-07-06 21:47:49 +00:00
|
|
|
|
}
|
|
|
|
|
}
|