using System; using System.Collections.Generic; using System.IO; using System.Linq; using NLog; using NzbDrone.Core.Model; using NzbDrone.Core.Repository; using SubSonic.Repository; namespace NzbDrone.Core.Providers { public class EpisodeProvider { private static readonly Logger Logger = LogManager.GetCurrentClassLogger(); private readonly IRepository _repository; private readonly TvDbProvider _tvDbProvider; public EpisodeProvider(IRepository repository, TvDbProvider tvDbProviderProvider) { _repository = repository; _tvDbProvider = tvDbProviderProvider; } public EpisodeProvider() { } public virtual void AddEpisode(Episode episode) { _repository.Add(episode); } public virtual Episode GetEpisode(long id) { return _repository.Single(id); } public virtual Episode GetEpisode(int seriesId, int seasonNumber, int episodeNumber) { return _repository.Single( c => c.SeriesId == seriesId && c.SeasonNumber == seasonNumber && c.EpisodeNumber == episodeNumber); } public virtual Episode GetEpisode(int seriesId, DateTime date) { return _repository.Single( c => c.SeriesId == seriesId && c.AirDate == date.Date); } public virtual IQueryable GetEpisodeBySeries(long seriesId) { return _repository.All().Where(e => e.SeriesId == seriesId); } public virtual IList GetEpisodesBySeason(long seriesId, int seasonNumber) { return _repository.Find(e => e.SeriesId == seriesId && e.SeasonNumber == seasonNumber); } public virtual List GetEpisodes(EpisodeParseResult parseResult) { if (parseResult.Series == null) { Logger.Debug("Episode Parse Result is Invalid, skipping"); return new List(); } var episodes = new List(); foreach (var ep in parseResult.EpisodeNumbers) { var episode = GetEpisode(parseResult.Series.SeriesId, parseResult.SeasonNumber, ep); if (episode == null) return new List(); episodes.Add(episode); } return episodes; } public virtual IList EpisodesWithoutFiles(bool includeSpecials) { if (includeSpecials) return _repository.All().Where(e => e.EpisodeFileId == 0 && e.AirDate <= DateTime.Today && e.AirDate > new DateTime(1899, 12, 31)).ToList(); return _repository.All().Where(e => e.EpisodeFileId == 0 && e.AirDate <= DateTime.Today && e.AirDate > new DateTime(1899, 12, 31) && e.SeasonNumber > 0).ToList(); } public virtual IList EpisodesByFileId(int episodeFileId) { return _repository.All().Where(e => e.EpisodeFileId == episodeFileId).ToList(); } public virtual void RefreshEpisodeInfo(Series series) { Logger.Info("Starting episode info refresh for series:{0}", series.SeriesId); int successCount = 0; int failCount = 0; var tvDbSeriesInfo = _tvDbProvider.GetSeries(series.SeriesId, true); var updateList = new List(); var newList = new List(); foreach (var episode in tvDbSeriesInfo.Episodes) { try { //DateTime throws an error in SQLServer per message below: //SqlDateTime overflow. Must be between 1/1/1753 12:00:00 AM and 12/31/9999 11:59:59 PM. //So lets hack it so it works for SQLServer (as well as SQLite), perhaps we can find a better solution //Todo: Fix this hack if (episode.FirstAired < new DateTime(1753, 1, 1)) episode.FirstAired = new DateTime(1753, 1, 1); Logger.Trace("Updating info for [{0}] - S{1}E{2}", tvDbSeriesInfo.SeriesName, episode.SeasonNumber, episode.EpisodeNumber); //first check using tvdbId, this should cover cases when and episode number in a season is changed var episodeToUpdate = _repository.Single(c => c.TvDbEpisodeId == episode.Id); //not found, try using season/episode number if (episodeToUpdate == null) { episodeToUpdate = GetEpisode(series.SeriesId, episode.SeasonNumber, episode.EpisodeNumber); } //Episode doesn't exist locally if (episodeToUpdate == null) { episodeToUpdate = new Episode(); newList.Add(episodeToUpdate); } else { updateList.Add(episodeToUpdate); } episodeToUpdate.SeriesId = series.SeriesId; episodeToUpdate.TvDbEpisodeId = episode.Id; episodeToUpdate.AirDate = episode.FirstAired.Date; episodeToUpdate.EpisodeNumber = episode.EpisodeNumber; episodeToUpdate.SeasonNumber = episode.SeasonNumber; episodeToUpdate.Title = episode.EpisodeName; episodeToUpdate.Overview = episode.Overview; successCount++; } catch (Exception e) { Logger.FatalException( String.Format("An error has occurred while updating episode info for series {0}", series.SeriesId), e); failCount++; } } _repository.AddMany(newList); _repository.UpdateMany(updateList); Logger.Debug("Finished episode refresh for series:{0}. Successful:{1} - Failed:{2} ", tvDbSeriesInfo.SeriesName, successCount, failCount); } public virtual void DeleteEpisode(int episodeId) { _repository.Delete(episodeId); } public virtual void UpdateEpisode(Episode episode) { _repository.Update(episode); } public virtual bool IsIgnored(int seriesId, int seasonNumber) { return !_repository.Exists(e => e.SeriesId == seriesId && e.SeasonNumber == seasonNumber && !e.Ignored); } public virtual IList GetSeasons(int seriesId) { return _repository.All().Where(e => e.SeriesId == seriesId).Select(s => s.SeasonNumber).Distinct().OrderBy(c => c).ToList(); } public virtual void SetSeasonIgnore(long seriesId, int seasonNumber, bool isIgnored) { Logger.Info("Setting ignore flag on Series:{0} Season:{1} to {2}", seriesId, seasonNumber, isIgnored); var episodes = GetEpisodesBySeason(seriesId, seasonNumber); foreach (var episode in episodes) { episode.Ignored = isIgnored; } _repository.UpdateMany(episodes); Logger.Info("Ignore flag for Series:{0} Season:{1} successfully set to {2}", seriesId, seasonNumber, isIgnored); } } }