Sonarr/NzbDrone.Core/Tv/SeriesService.cs

168 lines
5.4 KiB
C#
Raw Normal View History

using System;
2011-05-27 06:03:57 +00:00
using System.Collections.Generic;
2013-04-11 04:30:51 +00:00
using System.IO;
2010-09-23 03:19:47 +00:00
using System.Linq;
2010-10-02 19:01:43 +00:00
using NLog;
2013-04-11 04:30:51 +00:00
using NzbDrone.Common;
using NzbDrone.Common.EnsureThat;
using NzbDrone.Common.Messaging;
2013-02-24 06:48:52 +00:00
using NzbDrone.Core.Configuration;
using NzbDrone.Core.DataAugmentation.Scene;
using NzbDrone.Core.Model;
2013-04-11 04:30:51 +00:00
using NzbDrone.Core.Organizer;
2013-02-23 20:34:51 +00:00
using NzbDrone.Core.Tv.Events;
2010-09-23 03:19:47 +00:00
namespace NzbDrone.Core.Tv
2010-09-23 03:19:47 +00:00
{
public interface ISeriesService
{
bool IsMonitored(int id);
2013-04-07 07:30:37 +00:00
Series GetSeries(int seriesId);
Series AddSeries(Series newSeries);
void UpdateFromSeriesEditor(IList<Series> editedSeries);
2013-03-02 18:25:39 +00:00
Series FindByTvdbId(int tvdbId);
Series FindByTitle(string title);
void SetSeriesType(int seriesId, SeriesTypes seriesTypes);
void DeleteSeries(int seriesId, bool deleteFiles);
2013-04-12 00:36:47 +00:00
List<Series> GetAllSeries();
Series UpdateSeries(Series series);
bool SeriesPathExists(string folder);
List<Series> GetSeriesInList(IEnumerable<int> seriesIds);
Series FindBySlug(string slug);
}
2013-06-02 19:29:00 +00:00
public class SeriesService : ISeriesService
2010-09-23 03:19:47 +00:00
{
2013-02-19 06:56:02 +00:00
private readonly ISeriesRepository _seriesRepository;
2013-02-24 06:48:52 +00:00
private readonly IConfigService _configService;
private readonly IMessageAggregator _messageAggregator;
private readonly ISceneMappingService _sceneMappingService;
2013-05-10 23:53:50 +00:00
private readonly IDiskProvider _diskProvider;
2013-03-28 06:49:38 +00:00
private readonly Logger _logger;
2013-06-02 19:29:00 +00:00
public SeriesService(ISeriesRepository seriesRepository, IConfigService configServiceService, IMessageAggregator messageAggregator, ISceneMappingService sceneMappingService, IDiskProvider diskProvider, Logger logger)
2010-09-23 03:19:47 +00:00
{
2013-02-19 06:56:02 +00:00
_seriesRepository = seriesRepository;
2013-02-24 06:48:52 +00:00
_configService = configServiceService;
_messageAggregator = messageAggregator;
_sceneMappingService = sceneMappingService;
2013-04-11 04:30:51 +00:00
_diskProvider = diskProvider;
2013-03-28 06:49:38 +00:00
_logger = logger;
2010-09-23 03:19:47 +00:00
}
2013-02-19 06:56:02 +00:00
public bool IsMonitored(int id)
2010-09-23 03:19:47 +00:00
{
2013-02-19 06:56:02 +00:00
return _seriesRepository.Get(id).Monitored;
}
2013-04-07 07:30:37 +00:00
public Series GetSeries(int seriesId)
{
return _seriesRepository.Get(seriesId);
}
public Series AddSeries(Series newSeries)
{
2013-03-31 21:45:16 +00:00
Ensure.That(() => newSeries).IsNotNull();
if (String.IsNullOrWhiteSpace(newSeries.Path))
{
var folderName = FileNameBuilder.CleanFilename(newSeries.Title);
newSeries.Path = Path.Combine(newSeries.RootFolderPath, folderName);
_diskProvider.CreateFolder(newSeries.Path);
}
2013-03-31 21:45:16 +00:00
_logger.Info("Adding Series [{0}] Path: [{1}]", newSeries.Title, newSeries.Path);
2013-03-31 21:45:16 +00:00
newSeries.Monitored = true;
newSeries.CleanTitle = Parser.Parser.CleanSeriesTitle(newSeries.Title);
2013-03-31 21:45:16 +00:00
newSeries.SeasonFolder = _configService.UseSeasonFolder;
newSeries.BacklogSetting = BacklogSettingType.Inherit;
2013-03-31 21:45:16 +00:00
_seriesRepository.Insert(newSeries);
_messageAggregator.PublishEvent(new SeriesAddedEvent(newSeries));
return newSeries;
}
public void UpdateFromSeriesEditor(IList<Series> editedSeries)
2012-01-20 07:50:45 +00:00
{
2013-02-19 06:56:02 +00:00
var allSeries = _seriesRepository.All();
2012-01-20 07:50:45 +00:00
2013-02-19 06:56:02 +00:00
foreach (var series in allSeries)
2012-01-20 07:50:45 +00:00
{
//Only update parameters that can be changed in MassEdit
2013-04-01 06:22:16 +00:00
var edited = editedSeries.Single(s => s.Id == series.Id);
2012-01-20 07:50:45 +00:00
series.QualityProfileId = edited.QualityProfileId;
series.Monitored = edited.Monitored;
series.SeasonFolder = edited.SeasonFolder;
2012-01-26 01:02:21 +00:00
series.BacklogSetting = edited.BacklogSetting;
//series.Path = edited.Path;
2012-09-20 15:37:40 +00:00
series.CustomStartDate = edited.CustomStartDate;
2013-02-19 06:56:02 +00:00
_seriesRepository.Update(series);
2012-01-20 07:50:45 +00:00
}
}
2013-03-02 18:25:39 +00:00
public Series FindByTvdbId(int tvdbId)
{
return _seriesRepository.FindByTvdbId(tvdbId);
}
public Series FindBySlug(string slug)
{
var series = _seriesRepository.FindBySlug(slug);
return series;
}
public Series FindByTitle(string title)
{
var tvdbId = _sceneMappingService.GetTvDbId(title);
if (tvdbId.HasValue)
{
return FindByTvdbId(tvdbId.Value);
}
return _seriesRepository.FindByTitle(Parser.Parser.CleanSeriesTitle(title));
}
public void SetSeriesType(int seriesId, SeriesTypes seriesTypes)
2013-03-02 18:25:39 +00:00
{
_seriesRepository.SetSeriesType(seriesId, seriesTypes);
2013-03-02 18:25:39 +00:00
}
public void DeleteSeries(int seriesId, bool deleteFiles)
2013-03-05 19:35:33 +00:00
{
var series = _seriesRepository.Get(seriesId);
_seriesRepository.Delete(seriesId);
_messageAggregator.PublishEvent(new SeriesDeletedEvent(series, deleteFiles));
2013-03-05 19:35:33 +00:00
}
2013-04-12 00:36:47 +00:00
public List<Series> GetAllSeries()
{
return _seriesRepository.All().ToList();
}
public Series UpdateSeries(Series series)
2013-04-12 00:36:47 +00:00
{
return _seriesRepository.Update(series);
2013-04-12 00:36:47 +00:00
}
public bool SeriesPathExists(string folder)
{
return _seriesRepository.SeriesPathExists(folder);
}
public List<Series> GetSeriesInList(IEnumerable<int> seriesIds)
{
2013-04-18 00:33:38 +00:00
return _seriesRepository.Get(seriesIds).ToList();
}
2010-09-23 03:19:47 +00:00
}
2013-03-01 07:03:41 +00:00
}