mirror of
https://github.com/Sonarr/Sonarr.git
synced 2024-11-01 00:12:30 +01:00
84 lines
2.9 KiB
C#
84 lines
2.9 KiB
C#
using System.Linq;
|
|
using System;
|
|
using NLog;
|
|
using NzbDrone.Core.Model.Notification;
|
|
using NzbDrone.Core.Providers;
|
|
using NzbDrone.Core.Providers.DecisionEngine;
|
|
using NzbDrone.Core.Providers.Search;
|
|
using NzbDrone.Core.Repository;
|
|
|
|
namespace NzbDrone.Core.Jobs
|
|
{
|
|
public class EpisodeSearchJob : IJob
|
|
{
|
|
private readonly EpisodeProvider _episodeProvider;
|
|
private readonly UpgradePossibleSpecification _upgradePossibleSpecification;
|
|
private readonly EpisodeSearch _episodeSearch;
|
|
private readonly DailyEpisodeSearch _dailyEpisodeSearch;
|
|
|
|
private static readonly Logger logger = LogManager.GetCurrentClassLogger();
|
|
|
|
public EpisodeSearchJob(EpisodeProvider episodeProvider, UpgradePossibleSpecification upgradePossibleSpecification,
|
|
EpisodeSearch episodeSearch, DailyEpisodeSearch dailyEpisodeSearch)
|
|
{
|
|
if(dailyEpisodeSearch == null) throw new ArgumentNullException("dailyEpisodeSearch");
|
|
_episodeProvider = episodeProvider;
|
|
_upgradePossibleSpecification = upgradePossibleSpecification;
|
|
_episodeSearch = episodeSearch;
|
|
_dailyEpisodeSearch = dailyEpisodeSearch;
|
|
}
|
|
|
|
public EpisodeSearchJob()
|
|
{
|
|
|
|
}
|
|
|
|
public string Name
|
|
{
|
|
get { return "Episode Search"; }
|
|
}
|
|
|
|
public TimeSpan DefaultInterval
|
|
{
|
|
get { return TimeSpan.FromTicks(0); }
|
|
}
|
|
|
|
public virtual void Start(ProgressNotification notification, dynamic options)
|
|
{
|
|
if (options == null || options.EpisodeId <= 0)
|
|
throw new ArgumentException("options");
|
|
|
|
Episode episode = _episodeProvider.GetEpisode(options.EpisodeId);
|
|
|
|
if (episode == null)
|
|
{
|
|
logger.Error("Unable to find an episode {0} in database", options.EpisodeId);
|
|
return;
|
|
}
|
|
|
|
if (!_upgradePossibleSpecification.IsSatisfiedBy(episode))
|
|
{
|
|
logger.Info("Search for {0} was aborted, file in disk meets or exceeds Profile's Cutoff", episode);
|
|
notification.CurrentMessage = String.Format("Skipping search for {0}, the file you have is already at cutoff", episode);
|
|
return;
|
|
}
|
|
|
|
if (episode.Series.IsDaily)
|
|
{
|
|
if (!episode.AirDate.HasValue)
|
|
{
|
|
logger.Warn("AirDate is not Valid for: {0}", episode);
|
|
notification.CurrentMessage = String.Format("Search for {0} Failed, AirDate is invalid", episode);
|
|
return;
|
|
}
|
|
|
|
_dailyEpisodeSearch.Search(episode.Series, new { Episode = episode }, notification);
|
|
}
|
|
|
|
else
|
|
{
|
|
_episodeSearch.Search(episode.Series, new { Episode = episode }, notification);
|
|
}
|
|
}
|
|
}
|
|
} |