using System; using System.Collections.Generic; using System.IO; using System.Linq; using Marr.Data; using NLog; using NzbDrone.Common; using NzbDrone.Common.EnsureThat; using NzbDrone.Common.Messaging; using NzbDrone.Core.Configuration; using NzbDrone.Core.DataAugmentation.Scene; using NzbDrone.Core.Datastore; using NzbDrone.Core.MetadataSource; using NzbDrone.Core.Model; using NzbDrone.Core.Organizer; using NzbDrone.Core.Parser; using NzbDrone.Core.RootFolders; using NzbDrone.Core.SeriesStats; using NzbDrone.Core.Tv.Events; namespace NzbDrone.Core.Tv { public interface ISeriesService { bool IsMonitored(int id); Series UpdateSeriesInfo(int seriesId); Series GetSeries(int seriesId); Series AddSeries(Series newSeries); void UpdateFromSeriesEditor(IList editedSeries); Series FindByTvdbId(int tvdbId); Series FindByTitle(string title); void SetSeriesType(int seriesId, SeriesTypes seriesTypes); void DeleteSeries(int seriesId, bool deleteFiles); List GetAllSeries(); Series UpdateSeries(Series series); bool SeriesPathExists(string folder); List GetSeriesInList(IEnumerable seriesIds); } public class SeriesService : ISeriesService, IHandleAsync { private readonly ISeriesRepository _seriesRepository; private readonly IConfigService _configService; private readonly IProvideSeriesInfo _seriesInfoProxy; private readonly IMessageAggregator _messageAggregator; private readonly ISceneMappingService _sceneMappingService; private readonly IRootFolderService _rootFolderService; private readonly DiskProvider _diskProvider; private readonly Logger _logger; public SeriesService(ISeriesRepository seriesRepository, IConfigService configServiceService, IProvideSeriesInfo seriesInfoProxy, IMessageAggregator messageAggregator, ISceneMappingService sceneMappingService, IRootFolderService rootFolderService, DiskProvider diskProvider, Logger logger) { _seriesRepository = seriesRepository; _configService = configServiceService; _seriesInfoProxy = seriesInfoProxy; _messageAggregator = messageAggregator; _sceneMappingService = sceneMappingService; _rootFolderService = rootFolderService; _diskProvider = diskProvider; _logger = logger; } public bool IsMonitored(int id) { return _seriesRepository.Get(id).Monitored; } public Series UpdateSeriesInfo(int seriesId) { var series = _seriesRepository.Get(seriesId); var seriesInfo = _seriesInfoProxy.GetSeriesInfo(series.TvdbId); series.Title = seriesInfo.Title; series.AirTime = seriesInfo.AirTime; series.Overview = seriesInfo.Overview; series.Status = seriesInfo.Status; series.CleanTitle = Parser.Parser.NormalizeTitle(seriesInfo.Title); series.LastInfoSync = DateTime.Now; series.Runtime = seriesInfo.Runtime; series.Images = seriesInfo.Images; series.Network = seriesInfo.Network; series.FirstAired = seriesInfo.FirstAired; _seriesRepository.Update(series); //Todo: We need to get the UtcOffset from TVRage, since its not available from trakt _messageAggregator.PublishEvent(new SeriesUpdatedEvent(series)); return series; } public Series GetSeries(int seriesId) { return _seriesRepository.Get(seriesId); } public Series AddSeries(Series newSeries) { Ensure.That(() => newSeries).IsNotNull(); if (String.IsNullOrWhiteSpace(newSeries.FolderName)) { newSeries.FolderName = FileNameBuilder.CleanFilename(newSeries.Title); _diskProvider.CreateFolder(Path.Combine(_rootFolderService.Get(newSeries.RootFolderId).Path, newSeries.FolderName)); } _logger.Info("Adding Series [{0}] Path: [{1}]", newSeries.Title, newSeries.Path); newSeries.Monitored = true; newSeries.CleanTitle = Parser.Parser.NormalizeTitle(newSeries.Title); if (newSeries.QualityProfileId == 0) newSeries.QualityProfileId = _configService.DefaultQualityProfile; newSeries.SeasonFolder = _configService.UseSeasonFolder; newSeries.BacklogSetting = BacklogSettingType.Inherit; _seriesRepository.Insert(newSeries); _messageAggregator.PublishEvent(new SeriesAddedEvent(newSeries)); return newSeries; } public void UpdateFromSeriesEditor(IList editedSeries) { var allSeries = _seriesRepository.All(); foreach (var series in allSeries) { //Only update parameters that can be changed in MassEdit var edited = editedSeries.Single(s => s.Id == series.Id); series.QualityProfileId = edited.QualityProfileId; series.Monitored = edited.Monitored; series.SeasonFolder = edited.SeasonFolder; series.BacklogSetting = edited.BacklogSetting; //series.Path = edited.Path; series.CustomStartDate = edited.CustomStartDate; _seriesRepository.Update(series); } } public Series FindByTvdbId(int tvdbId) { return _seriesRepository.FindByTvdbId(tvdbId); } public Series FindByTitle(string title) { var tvdbId = _sceneMappingService.GetTvDbId(title); if (tvdbId.HasValue) { return FindByTvdbId(tvdbId.Value); } return _seriesRepository.FindByTitle(Parser.Parser.NormalizeTitle(title)); } public void SetSeriesType(int seriesId, SeriesTypes seriesTypes) { _seriesRepository.SetSeriesType(seriesId, seriesTypes); } public void DeleteSeries(int seriesId, bool deleteFiles) { var series = _seriesRepository.Get(seriesId); _seriesRepository.Delete(seriesId); _messageAggregator.PublishEvent(new SeriesDeletedEvent(series, deleteFiles)); } public List GetAllSeries() { return _seriesRepository.All().ToList(); } public Series UpdateSeries(Series series) { return _seriesRepository.Update(series); } public bool SeriesPathExists(string folder) { return _seriesRepository.SeriesPathExists(folder); } public List GetSeriesInList(IEnumerable seriesIds) { return _seriesRepository.Get(seriesIds).ToList(); } public void HandleAsync(SeriesAddedEvent message) { UpdateSeriesInfo(message.Series.Id); } } }