2013-02-20 02:05:15 +00:00
|
|
|
using System;
|
|
|
|
using System.Collections.Generic;
|
|
|
|
using System.Linq;
|
|
|
|
using NLog;
|
2013-08-31 01:42:30 +00:00
|
|
|
using NzbDrone.Common.Instrumentation;
|
2013-03-07 04:34:56 +00:00
|
|
|
using NzbDrone.Core.Configuration;
|
2013-05-02 05:50:34 +00:00
|
|
|
using NzbDrone.Core.Datastore;
|
2013-03-07 04:34:56 +00:00
|
|
|
using NzbDrone.Core.MediaFiles.Events;
|
2013-09-14 06:36:07 +00:00
|
|
|
using NzbDrone.Core.Messaging.Events;
|
2013-03-03 22:26:41 +00:00
|
|
|
using NzbDrone.Core.Tv.Events;
|
2013-02-20 02:05:15 +00:00
|
|
|
|
|
|
|
namespace NzbDrone.Core.Tv
|
|
|
|
{
|
|
|
|
public interface IEpisodeService
|
|
|
|
{
|
|
|
|
Episode GetEpisode(int id);
|
2013-05-29 01:28:47 +00:00
|
|
|
Episode FindEpisode(int seriesId, int seasonNumber, int episodeNumber, bool useScene = false);
|
2013-02-20 02:05:15 +00:00
|
|
|
Episode GetEpisode(int seriesId, DateTime date);
|
2013-05-29 01:28:47 +00:00
|
|
|
Episode FindEpisode(int seriesId, DateTime date);
|
2013-03-02 19:13:23 +00:00
|
|
|
List<Episode> GetEpisodeBySeries(int seriesId);
|
2013-03-03 23:18:43 +00:00
|
|
|
List<Episode> GetEpisodesBySeason(int seriesId, int seasonNumber);
|
2013-05-13 06:12:19 +00:00
|
|
|
PagingSpec<Episode> EpisodesWithoutFiles(PagingSpec<Episode> pagingSpec);
|
2013-03-03 23:18:43 +00:00
|
|
|
List<Episode> GetEpisodesByFileId(int episodeFileId);
|
2013-02-20 02:05:15 +00:00
|
|
|
void UpdateEpisode(Episode episode);
|
2013-07-09 01:22:02 +00:00
|
|
|
void SetEpisodeMonitored(int episodeId, bool monitored);
|
2013-04-15 01:41:39 +00:00
|
|
|
bool IsFirstOrLastEpisodeOfSeason(int episodeId);
|
2013-02-20 02:05:15 +00:00
|
|
|
void UpdateEpisodes(List<Episode> episodes);
|
2013-02-26 03:30:24 +00:00
|
|
|
List<Episode> EpisodesBetweenDates(DateTime start, DateTime end);
|
2013-06-02 19:29:00 +00:00
|
|
|
void InsertMany(List<Episode> episodes);
|
|
|
|
void UpdateMany(List<Episode> episodes);
|
2013-08-01 02:02:36 +00:00
|
|
|
void DeleteMany(List<Episode> episodes);
|
2013-08-07 01:25:06 +00:00
|
|
|
void SetEpisodeMonitoredBySeason(int seriesId, int seasonNumber, bool monitored);
|
2013-02-20 02:05:15 +00:00
|
|
|
}
|
|
|
|
|
2013-03-05 19:49:34 +00:00
|
|
|
public class EpisodeService : IEpisodeService,
|
2013-03-07 04:34:56 +00:00
|
|
|
IHandle<EpisodeFileDeletedEvent>,
|
2013-04-15 01:41:39 +00:00
|
|
|
IHandle<EpisodeFileAddedEvent>,
|
2013-06-02 19:29:00 +00:00
|
|
|
IHandleAsync<SeriesDeletedEvent>
|
2013-02-20 02:05:15 +00:00
|
|
|
{
|
|
|
|
|
2013-02-24 20:24:31 +00:00
|
|
|
private readonly IEpisodeRepository _episodeRepository;
|
2013-03-07 04:34:56 +00:00
|
|
|
private readonly IConfigService _configService;
|
|
|
|
private readonly Logger _logger;
|
2013-02-20 02:05:15 +00:00
|
|
|
|
2013-06-02 19:29:00 +00:00
|
|
|
public EpisodeService(IEpisodeRepository episodeRepository, IConfigService configService, Logger logger)
|
2013-02-20 02:05:15 +00:00
|
|
|
{
|
|
|
|
_episodeRepository = episodeRepository;
|
2013-03-07 04:34:56 +00:00
|
|
|
_configService = configService;
|
|
|
|
_logger = logger;
|
2013-02-20 02:05:15 +00:00
|
|
|
}
|
|
|
|
|
2013-03-02 19:13:23 +00:00
|
|
|
public Episode GetEpisode(int id)
|
2013-02-20 02:05:15 +00:00
|
|
|
{
|
|
|
|
return _episodeRepository.Get(id);
|
|
|
|
}
|
|
|
|
|
2013-05-29 01:28:47 +00:00
|
|
|
public Episode FindEpisode(int seriesId, int seasonNumber, int episodeNumber, bool useSceneNumbering = false)
|
|
|
|
{
|
|
|
|
if (useSceneNumbering)
|
|
|
|
{
|
|
|
|
return _episodeRepository.FindEpisodeBySceneNumbering(seriesId, seasonNumber, episodeNumber);
|
|
|
|
}
|
|
|
|
return _episodeRepository.Find(seriesId, seasonNumber, episodeNumber);
|
2013-02-20 02:05:15 +00:00
|
|
|
}
|
|
|
|
|
2013-03-02 19:13:23 +00:00
|
|
|
public Episode GetEpisode(int seriesId, DateTime date)
|
2013-02-20 02:05:15 +00:00
|
|
|
{
|
|
|
|
return _episodeRepository.Get(seriesId, date);
|
|
|
|
}
|
|
|
|
|
2013-05-29 01:28:47 +00:00
|
|
|
public Episode FindEpisode(int seriesId, DateTime date)
|
|
|
|
{
|
|
|
|
return _episodeRepository.Find(seriesId, date);
|
|
|
|
}
|
|
|
|
|
2013-03-02 19:13:23 +00:00
|
|
|
public List<Episode> GetEpisodeBySeries(int seriesId)
|
2013-02-20 02:05:15 +00:00
|
|
|
{
|
2013-03-02 19:13:23 +00:00
|
|
|
return _episodeRepository.GetEpisodes(seriesId).ToList();
|
2013-02-20 02:05:15 +00:00
|
|
|
}
|
|
|
|
|
2013-03-03 23:18:43 +00:00
|
|
|
public List<Episode> GetEpisodesBySeason(int seriesId, int seasonNumber)
|
2013-02-20 02:05:15 +00:00
|
|
|
{
|
|
|
|
return _episodeRepository.GetEpisodes(seriesId, seasonNumber);
|
|
|
|
}
|
|
|
|
|
2013-05-13 06:12:19 +00:00
|
|
|
public PagingSpec<Episode> EpisodesWithoutFiles(PagingSpec<Episode> pagingSpec)
|
2013-02-20 02:05:15 +00:00
|
|
|
{
|
2013-05-13 06:12:19 +00:00
|
|
|
var episodeResult = _episodeRepository.EpisodesWithoutFiles(pagingSpec, false);
|
2013-05-01 03:04:06 +00:00
|
|
|
|
2013-05-02 05:50:34 +00:00
|
|
|
return episodeResult;
|
2013-02-20 02:05:15 +00:00
|
|
|
}
|
|
|
|
|
2013-03-03 23:18:43 +00:00
|
|
|
public List<Episode> GetEpisodesByFileId(int episodeFileId)
|
2013-02-20 02:05:15 +00:00
|
|
|
{
|
|
|
|
return _episodeRepository.GetEpisodeByFileId(episodeFileId);
|
|
|
|
}
|
|
|
|
|
2013-03-02 19:13:23 +00:00
|
|
|
public void UpdateEpisode(Episode episode)
|
2013-02-20 02:05:15 +00:00
|
|
|
{
|
|
|
|
_episodeRepository.Update(episode);
|
|
|
|
}
|
|
|
|
|
2013-07-09 01:22:02 +00:00
|
|
|
public void SetEpisodeMonitored(int episodeId, bool monitored)
|
2013-02-20 02:05:15 +00:00
|
|
|
{
|
|
|
|
var episode = _episodeRepository.Get(episodeId);
|
2013-07-09 01:22:02 +00:00
|
|
|
_episodeRepository.SetMonitoredFlat(episode, monitored);
|
2013-02-20 02:05:15 +00:00
|
|
|
|
2013-09-17 05:16:16 +00:00
|
|
|
_logger.Debug("Monitored flag for Episode:{0} was set to {1}", episodeId, monitored);
|
2013-02-20 02:05:15 +00:00
|
|
|
}
|
|
|
|
|
2013-07-10 02:11:00 +00:00
|
|
|
public void SetEpisodeMonitoredBySeason(int seriesId, int seasonNumber, bool monitored)
|
|
|
|
{
|
|
|
|
_episodeRepository.SetMonitoredBySeason(seriesId, seasonNumber, monitored);
|
|
|
|
}
|
|
|
|
|
2013-04-15 01:41:39 +00:00
|
|
|
public bool IsFirstOrLastEpisodeOfSeason(int episodeId)
|
2013-02-20 02:05:15 +00:00
|
|
|
{
|
2013-04-15 01:41:39 +00:00
|
|
|
var episode = GetEpisode(episodeId);
|
|
|
|
var seasonEpisodes = GetEpisodesBySeason(episode.SeriesId, episode.SeasonNumber);
|
2013-02-20 02:05:15 +00:00
|
|
|
|
|
|
|
|
|
|
|
//Ensure that this is either the first episode
|
|
|
|
//or is the last episode in a season that has 10 or more episodes
|
2013-04-15 01:41:39 +00:00
|
|
|
if (seasonEpisodes.First().EpisodeNumber == episode.EpisodeNumber || (seasonEpisodes.Count() >= 10 && seasonEpisodes.Last().EpisodeNumber == episode.EpisodeNumber))
|
2013-02-20 02:05:15 +00:00
|
|
|
return true;
|
|
|
|
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2013-03-02 19:13:23 +00:00
|
|
|
public void UpdateEpisodes(List<Episode> episodes)
|
2013-02-20 02:05:15 +00:00
|
|
|
{
|
|
|
|
_episodeRepository.UpdateMany(episodes);
|
|
|
|
}
|
|
|
|
|
2013-02-26 03:30:24 +00:00
|
|
|
public List<Episode> EpisodesBetweenDates(DateTime start, DateTime end)
|
2013-02-23 23:08:22 +00:00
|
|
|
{
|
2013-04-17 06:55:39 +00:00
|
|
|
var episodes = _episodeRepository.EpisodesBetweenDates(start.ToUniversalTime(), end.ToUniversalTime());
|
|
|
|
|
2013-05-30 04:06:24 +00:00
|
|
|
return episodes;
|
2013-02-23 23:08:22 +00:00
|
|
|
}
|
2013-02-24 19:18:48 +00:00
|
|
|
|
2013-06-02 19:29:00 +00:00
|
|
|
public void InsertMany(List<Episode> episodes)
|
|
|
|
{
|
|
|
|
_episodeRepository.InsertMany(episodes);
|
|
|
|
}
|
|
|
|
|
|
|
|
public void UpdateMany(List<Episode> episodes)
|
|
|
|
{
|
2013-06-03 03:56:17 +00:00
|
|
|
_episodeRepository.UpdateMany(episodes);
|
2013-06-02 19:29:00 +00:00
|
|
|
}
|
|
|
|
|
2013-08-01 02:02:36 +00:00
|
|
|
public void DeleteMany(List<Episode> episodes)
|
|
|
|
{
|
|
|
|
_episodeRepository.DeleteMany(episodes);
|
|
|
|
}
|
|
|
|
|
2013-03-05 19:49:34 +00:00
|
|
|
public void HandleAsync(SeriesDeletedEvent message)
|
|
|
|
{
|
|
|
|
var episodes = GetEpisodeBySeries(message.Series.Id);
|
|
|
|
_episodeRepository.DeleteMany(episodes);
|
|
|
|
}
|
2013-03-07 04:34:56 +00:00
|
|
|
|
|
|
|
public void Handle(EpisodeFileDeletedEvent message)
|
|
|
|
{
|
|
|
|
foreach (var episode in GetEpisodesByFileId(message.EpisodeFile.Id))
|
|
|
|
{
|
|
|
|
_logger.Trace("Detaching episode {0} from file.", episode.Id);
|
2013-05-13 00:36:23 +00:00
|
|
|
episode.EpisodeFileId = 0;
|
2013-07-17 03:01:38 +00:00
|
|
|
|
2013-07-23 00:50:37 +00:00
|
|
|
if (!message.ForUpgrade && _configService.AutoUnmonitorPreviouslyDownloadedEpisodes)
|
2013-07-17 03:01:38 +00:00
|
|
|
{
|
|
|
|
episode.Monitored = false;
|
|
|
|
}
|
|
|
|
|
2013-03-07 04:34:56 +00:00
|
|
|
UpdateEpisode(episode);
|
|
|
|
}
|
|
|
|
}
|
2013-03-30 22:43:19 +00:00
|
|
|
|
2013-04-15 01:41:39 +00:00
|
|
|
public void Handle(EpisodeFileAddedEvent message)
|
|
|
|
{
|
|
|
|
foreach (var episode in message.EpisodeFile.Episodes.Value)
|
|
|
|
{
|
|
|
|
_episodeRepository.SetFileId(episode.Id, message.EpisodeFile.Id);
|
|
|
|
_logger.Debug("Linking [{0}] > [{1}]", message.EpisodeFile.Path, episode);
|
|
|
|
}
|
|
|
|
}
|
2013-02-20 02:05:15 +00:00
|
|
|
}
|
|
|
|
}
|