mirror of https://github.com/Radarr/Radarr
121 lines
4.2 KiB
C#
121 lines
4.2 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using NLog;
|
|
using NzbDrone.Common;
|
|
using NzbDrone.Core.DecisionEngine;
|
|
using NzbDrone.Core.Parser;
|
|
using NzbDrone.Core.Parser.Model;
|
|
using NzbDrone.Core.Tv;
|
|
|
|
|
|
namespace NzbDrone.Core.MediaFiles.EpisodeImport
|
|
{
|
|
public interface IMakeImportDecision
|
|
{
|
|
List<ImportDecision> GetImportDecisions(IEnumerable<String> videoFiles, Series series, bool sceneSource);
|
|
}
|
|
|
|
public class ImportDecisionMaker : IMakeImportDecision
|
|
{
|
|
private readonly IEnumerable<IRejectWithReason> _specifications;
|
|
private readonly IParsingService _parsingService;
|
|
private readonly IMediaFileService _mediaFileService;
|
|
private readonly IDiskProvider _diskProvider;
|
|
private readonly Logger _logger;
|
|
|
|
public ImportDecisionMaker(IEnumerable<IRejectWithReason> specifications,
|
|
IParsingService parsingService,
|
|
IMediaFileService mediaFileService,
|
|
IDiskProvider diskProvider,
|
|
|
|
Logger logger)
|
|
{
|
|
_specifications = specifications;
|
|
_parsingService = parsingService;
|
|
_mediaFileService = mediaFileService;
|
|
_diskProvider = diskProvider;
|
|
_logger = logger;
|
|
}
|
|
|
|
public List<ImportDecision> GetImportDecisions(IEnumerable<string> videoFiles, Series series, bool sceneSource)
|
|
{
|
|
var newFiles = _mediaFileService.FilterExistingFiles(videoFiles.ToList(), series.Id);
|
|
|
|
_logger.Debug("Analysing {0}/{1} files.", newFiles.Count, videoFiles.Count());
|
|
|
|
return GetDecisions(newFiles, series, sceneSource).ToList();
|
|
}
|
|
|
|
private IEnumerable<ImportDecision> GetDecisions(IEnumerable<String> videoFiles, Series series, bool sceneSource)
|
|
{
|
|
foreach (var file in videoFiles)
|
|
{
|
|
ImportDecision decision = null;
|
|
|
|
try
|
|
{
|
|
var parsedEpisode = _parsingService.GetEpisodes(file, series, sceneSource);
|
|
|
|
if (parsedEpisode != null)
|
|
{
|
|
parsedEpisode.Size = _diskProvider.GetFileSize(file);
|
|
decision = GetDecision(parsedEpisode);
|
|
}
|
|
|
|
else
|
|
{
|
|
parsedEpisode = new LocalEpisode();
|
|
parsedEpisode.Path = file;
|
|
|
|
decision = new ImportDecision(parsedEpisode, "Unable to parse file");
|
|
}
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
_logger.ErrorException("Couldn't import file." + file, e);
|
|
}
|
|
|
|
if (decision != null)
|
|
{
|
|
yield return decision;
|
|
}
|
|
}
|
|
}
|
|
|
|
private ImportDecision GetDecision(LocalEpisode localEpisode)
|
|
{
|
|
var reasons = _specifications.Select(c => EvaluateSpec(c, localEpisode))
|
|
.Where(c => !string.IsNullOrWhiteSpace(c));
|
|
|
|
return new ImportDecision(localEpisode, reasons.ToArray());
|
|
}
|
|
|
|
private string EvaluateSpec(IRejectWithReason spec, LocalEpisode localEpisode)
|
|
{
|
|
try
|
|
{
|
|
if (string.IsNullOrWhiteSpace(spec.RejectionReason))
|
|
{
|
|
throw new InvalidOperationException("[Need Rejection Text]");
|
|
}
|
|
|
|
var generalSpecification = spec as IImportDecisionEngineSpecification;
|
|
if (generalSpecification != null && !generalSpecification.IsSatisfiedBy(localEpisode))
|
|
{
|
|
return spec.RejectionReason;
|
|
}
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
//e.Data.Add("report", remoteEpisode.Report.ToJson());
|
|
//e.Data.Add("parsed", remoteEpisode.ParsedEpisodeInfo.ToJson());
|
|
_logger.ErrorException("Couldn't evaluate decision on " + localEpisode.Path, e);
|
|
return string.Format("{0}: {1}", spec.GetType().Name, e.Message);
|
|
}
|
|
|
|
return null;
|
|
}
|
|
}
|
|
}
|