1
0
mirror of https://github.com/Sonarr/Sonarr.git synced 2024-12-16 11:37:58 +02:00
Sonarr/NzbDrone.Core/Tv/SeriesService.cs

190 lines
6.9 KiB
C#
Raw Normal View History

using System;
2011-05-27 09:03:57 +03:00
using System.Collections.Generic;
2010-09-23 06:19:47 +03:00
using System.Linq;
2010-10-02 22:01:43 +03:00
using NLog;
using NzbDrone.Common.EnsureThat;
2013-02-23 23:34:51 +03:00
using NzbDrone.Common.Eventing;
2013-02-24 09:48:52 +03:00
using NzbDrone.Core.Configuration;
2013-02-24 22:57:33 +03:00
using NzbDrone.Core.Datastore;
using NzbDrone.Core.MetadataSource;
using NzbDrone.Core.Model;
using NzbDrone.Core.Qualities;
2013-03-02 21:25:39 +03:00
using NzbDrone.Core.ReferenceData;
2013-02-23 23:34:51 +03:00
using NzbDrone.Core.Tv.Events;
2010-09-23 06:19:47 +03:00
namespace NzbDrone.Core.Tv
2010-09-23 06:19:47 +03:00
{
public interface ISeriesService
{
bool IsMonitored(int id);
Series UpdateSeriesInfo(int seriesId);
Series FindSeries(string title);
void AddSeries(string title, string path, int tvDbSeriesId, int qualityProfileId, DateTime? airedAfter);
void UpdateFromSeriesEditor(IList<Series> editedSeries);
2013-03-02 21:25:39 +03:00
Series FindByTvdbId(int tvdbId);
void SetSeriesType(int seriesId, SeriesTypes seriesTypes);
void DeleteSeries(int seriesId, bool deleteFiles);
}
public class SeriesService : ISeriesService, IHandleAsync<SeriesAddedEvent>
2010-09-23 06:19:47 +03:00
{
2013-02-19 09:56:02 +03:00
private readonly ISeriesRepository _seriesRepository;
2013-02-24 09:48:52 +03:00
private readonly IConfigService _configService;
private readonly TvDbProxy _tvDbProxy;
private readonly TvRageMappingProvider _tvRageMappingProvider;
2013-02-23 23:34:51 +03:00
private readonly IEventAggregator _eventAggregator;
private readonly IQualityProfileService _qualityProfileService;
2013-03-28 09:49:38 +03:00
private readonly Logger _logger;
2013-03-28 09:49:38 +03:00
private readonly ISceneMappingService _sceneNameMappingService;
2010-09-23 06:19:47 +03:00
2013-02-24 23:24:31 +03:00
public SeriesService(ISeriesRepository seriesRepository, IConfigService configServiceService,
2013-03-28 09:49:38 +03:00
TvDbProxy tvDbProxyProxy, ISceneMappingService sceneNameMappingService,
TvRageMappingProvider tvRageMappingProvider, IEventAggregator eventAggregator, IQualityProfileService qualityProfileService, Logger logger)
2010-09-23 06:19:47 +03:00
{
2013-02-19 09:56:02 +03:00
_seriesRepository = seriesRepository;
2013-02-24 09:48:52 +03:00
_configService = configServiceService;
_tvDbProxy = tvDbProxyProxy;
2013-03-02 21:25:39 +03:00
_sceneNameMappingService = sceneNameMappingService;
_tvRageMappingProvider = tvRageMappingProvider;
2013-02-23 23:34:51 +03:00
_eventAggregator = eventAggregator;
_qualityProfileService = qualityProfileService;
2013-03-28 09:49:38 +03:00
_logger = logger;
2010-09-23 06:19:47 +03:00
}
2013-02-19 09:56:02 +03:00
public bool IsMonitored(int id)
2010-09-23 06:19:47 +03:00
{
2013-02-19 09:56:02 +03:00
return _seriesRepository.Get(id).Monitored;
}
public Series UpdateSeriesInfo(int seriesId)
{
2013-02-19 09:56:02 +03:00
var series = _seriesRepository.Get(seriesId);
var tvDbSeries = _tvDbProxy.GetSeries(series.TvDbId);
2013-02-24 22:57:33 +03:00
series.Title = tvDbSeries.Title;
series.AirTime = tvDbSeries.AirTime;
series.Overview = tvDbSeries.Overview;
series.Status = tvDbSeries.Status;
series.Language = tvDbSeries.Language;
series.CleanTitle = tvDbSeries.CleanTitle;
series.LastInfoSync = DateTime.Now;
series.Runtime = tvDbSeries.Runtime;
series.Covers = tvDbSeries.Covers;
series.Network = tvDbSeries.Network;
series.FirstAired = tvDbSeries.FirstAired;
try
{
2013-02-19 09:56:02 +03:00
if (series.TvRageId == 0)
{
series = _tvRageMappingProvider.FindMatchingTvRageSeries(series);
}
}
2013-02-19 09:56:02 +03:00
catch (Exception ex)
{
2013-03-28 09:49:38 +03:00
_logger.ErrorException("Error getting TvRage information for series: " + series.Title, ex);
}
2013-02-19 09:56:02 +03:00
_seriesRepository.Update(series);
_eventAggregator.Publish(new SeriesUpdatedEvent(series));
return series;
}
public Series FindSeries(string title)
2010-09-23 06:19:47 +03:00
{
var normalizeTitle = Parser.NormalizeTitle(title);
2013-03-02 21:25:39 +03:00
var mapping = _sceneNameMappingService.GetTvDbId(normalizeTitle);
if (mapping.HasValue)
{
var sceneSeries = _seriesRepository.Get(mapping.Value);
return sceneSeries;
}
return _seriesRepository.GetByTitle(normalizeTitle);
}
public void AddSeries(string title, string path, int tvDbSeriesId, int qualityProfileId, DateTime? airedAfter)
{
2013-03-28 09:49:38 +03:00
_logger.Info("Adding Series [{0}] Path: [{1}]", tvDbSeriesId, path);
Ensure.That(() => tvDbSeriesId).IsGreaterThan(0);
Ensure.That(() => title).IsNotNullOrWhiteSpace();
Ensure.That(() => path).IsNotNullOrWhiteSpace();
var repoSeries = new Series();
2013-02-24 22:57:33 +03:00
repoSeries.TvDbId = tvDbSeriesId;
repoSeries.Path = path;
repoSeries.Monitored = true;
repoSeries.QualityProfileId = qualityProfileId;
2012-02-29 11:25:41 +03:00
repoSeries.Title = title;
repoSeries.CleanTitle = Parser.NormalizeTitle(title);
if (qualityProfileId == 0)
2013-02-24 09:48:52 +03:00
repoSeries.QualityProfileId = _configService.DefaultQualityProfile;
repoSeries.QualityProfile = _qualityProfileService.Get(repoSeries.QualityProfileId);
2013-02-24 09:48:52 +03:00
repoSeries.SeasonFolder = _configService.UseSeasonFolder;
2012-01-26 04:02:21 +03:00
repoSeries.BacklogSetting = BacklogSettingType.Inherit;
if (airedAfter.HasValue)
2012-09-20 18:37:40 +03:00
repoSeries.CustomStartDate = airedAfter;
//Todo: Allow the user to set this as part of the addition process.
repoSeries.Language = "en";
2013-02-19 09:56:02 +03:00
_seriesRepository.Insert(repoSeries);
2013-02-23 23:34:51 +03:00
_eventAggregator.Publish(new SeriesAddedEvent(repoSeries));
}
public void UpdateFromSeriesEditor(IList<Series> editedSeries)
2012-01-20 10:50:45 +03:00
{
2013-02-19 09:56:02 +03:00
var allSeries = _seriesRepository.All();
2012-01-20 10:50:45 +03:00
2013-02-19 09:56:02 +03:00
foreach (var series in allSeries)
2012-01-20 10:50:45 +03:00
{
//Only update parameters that can be changed in MassEdit
var edited = editedSeries.Single(s => ((ModelBase)s).Id == series.Id);
2012-01-20 10:50:45 +03:00
series.QualityProfileId = edited.QualityProfileId;
series.Monitored = edited.Monitored;
series.SeasonFolder = edited.SeasonFolder;
2012-01-26 04:02:21 +03:00
series.BacklogSetting = edited.BacklogSetting;
2012-01-20 10:50:45 +03:00
series.Path = edited.Path;
2012-09-20 18:37:40 +03:00
series.CustomStartDate = edited.CustomStartDate;
2013-02-19 09:56:02 +03:00
_seriesRepository.Update(series);
2012-01-20 10:50:45 +03:00
}
}
2013-03-02 21:25:39 +03:00
public Series FindByTvdbId(int tvdbId)
{
return _seriesRepository.FindByTvdbId(tvdbId);
}
public void SetSeriesType(int seriesId, SeriesTypes seriesTypes)
2013-03-02 21:25:39 +03:00
{
_seriesRepository.SetSeriesType(seriesId, seriesTypes);
2013-03-02 21:25:39 +03:00
}
public void DeleteSeries(int seriesId, bool deleteFiles)
2013-03-05 22:35:33 +03:00
{
var series = _seriesRepository.Get(seriesId);
_seriesRepository.Delete(seriesId);
_eventAggregator.Publish(new SeriesDeletedEvent(series, deleteFiles));
2013-03-05 22:35:33 +03:00
}
public void HandleAsync(SeriesAddedEvent message)
{
UpdateSeriesInfo(message.Series.Id);
}
2010-09-23 06:19:47 +03:00
}
2013-03-01 10:03:41 +03:00
}