Sonarr/NzbDrone.Core/Providers/InventoryProvider.cs

136 lines
4.8 KiB
C#
Raw Normal View History

using System;
using System.Collections.Generic;
2011-06-22 01:12:20 +00:00
using System.Linq;
using Ninject;
using NLog;
using NzbDrone.Core.Model;
using NzbDrone.Core.Repository;
2011-05-28 19:23:35 +00:00
using NzbDrone.Core.Repository.Quality;
namespace NzbDrone.Core.Providers
{
public class InventoryProvider
{
private readonly SeriesProvider _seriesProvider;
private readonly EpisodeProvider _episodeProvider;
private readonly HistoryProvider _historyProvider;
private static readonly Logger Logger = LogManager.GetCurrentClassLogger();
[Inject]
public InventoryProvider(SeriesProvider seriesProvider, EpisodeProvider episodeProvider, HistoryProvider historyProvider)
{
_seriesProvider = seriesProvider;
_episodeProvider = episodeProvider;
_historyProvider = historyProvider;
}
2011-05-27 03:07:32 +00:00
public InventoryProvider()
{
2011-05-28 19:23:35 +00:00
2011-05-27 03:07:32 +00:00
}
2011-05-28 19:23:35 +00:00
public virtual bool IsMonitored(EpisodeParseResult parseResult)
{
var series = _seriesProvider.FindSeries(parseResult.CleanTitle);
if (series == null)
{
Logger.Trace("{0} is not mapped to any series in DB. skipping", parseResult.CleanTitle);
return false;
}
2011-05-20 05:52:05 +00:00
parseResult.Series = series;
if (!series.Monitored)
{
Logger.Debug("{0} is present in the DB but not tracked. skipping.", parseResult.CleanTitle);
return false;
}
2011-06-22 01:12:20 +00:00
var episodes = _episodeProvider.GetEpisodesByParseResult(parseResult, true);
2011-06-22 01:12:20 +00:00
//return monitored if any of the episodes are monitored
if (episodes.Any(episode => !episode.Ignored))
{
return true;
}
2011-06-22 01:12:20 +00:00
Logger.Debug("All episodes are ignored. skipping.");
return false;
2011-05-28 19:23:35 +00:00
}
2011-05-28 19:23:35 +00:00
/// <summary>
/// Comprehensive check on whether or not this episode is needed.
/// </summary>
/// <param name = "parsedReport">Episode that needs to be checked</param>
/// <returns>Whether or not the file quality meets the requirements </returns>
/// <remarks>for multi episode files, all episodes need to meet the requirement
/// before the report is downloaded</remarks>
public virtual bool IsQualityNeeded(EpisodeParseResult parsedReport)
{
Logger.Trace("Checking if report meets quality requirements. {0}", parsedReport.Quality);
if (!parsedReport.Series.QualityProfile.Allowed.Contains(parsedReport.Quality.QualityType))
{
Logger.Trace("Quality {0} rejected by Series' quality profile", parsedReport.Quality);
return false;
}
2011-05-28 19:23:35 +00:00
var cutoff = parsedReport.Series.QualityProfile.Cutoff;
2011-06-22 01:12:20 +00:00
foreach (var episode in _episodeProvider.GetEpisodesByParseResult(parsedReport, true))
2011-05-28 19:23:35 +00:00
{
//Checking File
var file = episode.EpisodeFile;
if (file != null)
{
2011-05-28 19:23:35 +00:00
Logger.Trace("Comparing file quality with report. Existing file is {0} proper:{1}", file.Quality, file.Proper);
if (!IsUpgrade(new Quality { QualityType = file.Quality, Proper = file.Proper }, parsedReport.Quality, cutoff))
return false;
}
2011-05-28 19:23:35 +00:00
//Checking History
var bestQualityInHistory = _historyProvider.GetBestQualityInHistory(episode.EpisodeId);
if (bestQualityInHistory != null)
{
2011-05-28 19:23:35 +00:00
Logger.Trace("Comparing history quality with report. History is {0}", bestQualityInHistory);
if (!IsUpgrade(bestQualityInHistory, parsedReport.Quality, cutoff))
return false;
}
2011-05-28 19:23:35 +00:00
}
2011-05-28 19:23:35 +00:00
Logger.Debug("Episode {0} is needed", parsedReport);
return true; //If we get to this point and the file has not yet been rejected then accept it
}
public static bool IsUpgrade(Quality currentQuality, Quality newQuality, QualityTypes cutOff)
{
if (currentQuality.QualityType >= cutOff)
{
Logger.Trace("Existing item meets cut-off. skipping.");
2011-05-28 19:23:35 +00:00
return false;
}
if (newQuality > currentQuality)
return true;
2011-05-28 19:23:35 +00:00
if (currentQuality > newQuality)
{
Logger.Trace("existing item has better quality. skipping");
return false;
}
if (currentQuality == newQuality && !newQuality.Proper)
{
Logger.Trace("same quality. not proper skipping");
return false;
}
2011-05-28 19:23:35 +00:00
Logger.Debug("New item has better quality than existing item");
return true;
}
}
}