using System; using System.Collections.Generic; using System.IO; using System.Linq; using NLog; using NzbDrone.Common.EnsureThat; using NzbDrone.Common.Extensions; using NzbDrone.Core.DataAugmentation.Scene; using NzbDrone.Core.Messaging.Events; using NzbDrone.Core.Organizer; using NzbDrone.Core.Parser; using NzbDrone.Core.Tv.Events; namespace NzbDrone.Core.Tv { public interface ISeriesService { Series GetSeries(int seriesId); List GetSeries(IEnumerable seriesIds); Series AddSeries(Series newSeries); Series FindByTvdbId(int tvdbId); Series FindByTvRageId(int tvRageId); Series FindByTitle(string title); Series FindByTitle(string title, int year); Series FindByTitleInexact(string title); void DeleteSeries(int seriesId, bool deleteFiles); List GetAllSeries(); Series UpdateSeries(Series series); List UpdateSeries(List series); bool SeriesPathExists(string folder); void RemoveAddOptions(Series series); } public class SeriesService : ISeriesService { private readonly ISeriesRepository _seriesRepository; private readonly IEventAggregator _eventAggregator; // private readonly ISceneMappingService _sceneMappingService; private readonly IEpisodeService _episodeService; private readonly IBuildFileNames _fileNameBuilder; private readonly Logger _logger; public SeriesService(ISeriesRepository seriesRepository, IEventAggregator eventAggregator, // ISceneMappingService sceneMappingService, IEpisodeService episodeService, IBuildFileNames fileNameBuilder, Logger logger) { _seriesRepository = seriesRepository; _eventAggregator = eventAggregator; //_sceneMappingService = sceneMappingService; _episodeService = episodeService; _fileNameBuilder = fileNameBuilder; _logger = logger; } public Series GetSeries(int seriesId) { return _seriesRepository.Get(seriesId); } public List GetSeries(IEnumerable seriesIds) { return _seriesRepository.Get(seriesIds).ToList(); } public Series AddSeries(Series newSeries) { _seriesRepository.Insert(newSeries); _eventAggregator.PublishEvent(new SeriesAddedEvent(GetSeries(newSeries.Id))); return newSeries; } public Series FindByTvdbId(int tvRageId) { return _seriesRepository.FindByTvdbId(tvRageId); } public Series FindByTvRageId(int tvRageId) { return _seriesRepository.FindByTvRageId(tvRageId); } public Series FindByTitle(string title) { //var tvdbId = _sceneMappingService.FindTvdbId(title); //if (tvdbId.HasValue) //{ // return _seriesRepository.FindByTvdbId(tvdbId.Value); //} return _seriesRepository.FindByTitle(title.CleanSeriesTitle()); } public Series FindByTitleInexact(string title) { // find any series clean title within the provided release title string cleanTitle = title.CleanSeriesTitle(); var list = _seriesRepository.All().Where(s => cleanTitle.Contains(s.CleanTitle)).ToList(); if (!list.Any()) { // no series matched return null; } if (list.Count == 1) { // return the first series if there is only one return list.Single(); } // build ordered list of series by position in the search string var query = list.Select(series => new { position = cleanTitle.IndexOf(series.CleanTitle), length = series.CleanTitle.Length, series = series }) .Where(s => (s.position>=0)) .ToList() .OrderBy(s => s.position) .ThenByDescending(s => s.length) .ToList(); // get the leftmost series that is the longest // series are usually the first thing in release title, so we select the leftmost and longest match var match = query.First().series; _logger.Debug("Multiple series matched {0} from title {1}", match.Title, title); foreach (var entry in list) { _logger.Debug("Multiple series match candidate: {0} cleantitle: {1}", entry.Title, entry.CleanTitle); } return match; } public Series FindByTitle(string title, int year) { return _seriesRepository.FindByTitle(title.CleanSeriesTitle(), year); } public void DeleteSeries(int seriesId, bool deleteFiles) { var series = _seriesRepository.Get(seriesId); _seriesRepository.Delete(seriesId); _eventAggregator.PublishEvent(new SeriesDeletedEvent(series, deleteFiles)); } public List GetAllSeries() { return _seriesRepository.All().ToList(); } public Series UpdateSeries(Series series) { var storedSeries = GetSeries(series.Id); foreach (var season in series.Seasons) { var storedSeason = storedSeries.Seasons.SingleOrDefault(s => s.SeasonNumber == season.SeasonNumber); if (storedSeason != null && season.Monitored != storedSeason.Monitored) { _episodeService.SetEpisodeMonitoredBySeason(series.Id, season.SeasonNumber, season.Monitored); } } var updatedSeries = _seriesRepository.Update(series); _eventAggregator.PublishEvent(new SeriesEditedEvent(updatedSeries, storedSeries)); return updatedSeries; } public List UpdateSeries(List series) { _logger.Debug("Updating {0} series", series.Count); foreach (var s in series) { _logger.Trace("Updating: {0}", s.Title); if (!s.RootFolderPath.IsNullOrWhiteSpace()) { var folderName = new DirectoryInfo(s.Path).Name; s.Path = Path.Combine(s.RootFolderPath, folderName); _logger.Trace("Changing path for {0} to {1}", s.Title, s.Path); } else { _logger.Trace("Not changing path for: {0}", s.Title); } } _seriesRepository.UpdateMany(series); _logger.Debug("{0} series updated", series.Count); return series; } public bool SeriesPathExists(string folder) { return _seriesRepository.SeriesPathExists(folder); } public void RemoveAddOptions(Series series) { _seriesRepository.SetFields(series, s => s.AddOptions); } } }