mirror of
https://github.com/Sonarr/Sonarr.git
synced 2024-12-16 11:37:58 +02:00
170 lines
6.1 KiB
C#
170 lines
6.1 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using NLog;
|
|
using NzbDrone.Common.EnsureThat;
|
|
using NzbDrone.Common.Eventing;
|
|
using NzbDrone.Core.Configuration;
|
|
using NzbDrone.Core.Datastore;
|
|
using NzbDrone.Core.Model;
|
|
using NzbDrone.Core.Providers;
|
|
using NzbDrone.Core.Qualities;
|
|
using NzbDrone.Core.ReferenceData;
|
|
using NzbDrone.Core.Tv.Events;
|
|
|
|
namespace NzbDrone.Core.Tv
|
|
{
|
|
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);
|
|
Series FindByTvdbId(int tvdbId);
|
|
void SetSeriesType(int seriesId, SeriesType seriesType);
|
|
}
|
|
|
|
public class SeriesService : ISeriesService
|
|
{
|
|
private readonly ISeriesRepository _seriesRepository;
|
|
private readonly IConfigService _configService;
|
|
private readonly TvDbProvider _tvDbProvider;
|
|
private readonly TvRageMappingProvider _tvRageMappingProvider;
|
|
private readonly IEventAggregator _eventAggregator;
|
|
private readonly IQualityProfileService _qualityProfileService;
|
|
|
|
private static readonly Logger logger = LogManager.GetCurrentClassLogger();
|
|
|
|
private readonly SceneMappingService _sceneNameMappingService;
|
|
|
|
public SeriesService(ISeriesRepository seriesRepository, IConfigService configServiceService,
|
|
TvDbProvider tvDbProviderProvider, SceneMappingService sceneNameMappingService,
|
|
TvRageMappingProvider tvRageMappingProvider, IEventAggregator eventAggregator, IQualityProfileService qualityProfileService)
|
|
{
|
|
_seriesRepository = seriesRepository;
|
|
_configService = configServiceService;
|
|
_tvDbProvider = tvDbProviderProvider;
|
|
_sceneNameMappingService = sceneNameMappingService;
|
|
_tvRageMappingProvider = tvRageMappingProvider;
|
|
_eventAggregator = eventAggregator;
|
|
_qualityProfileService = qualityProfileService;
|
|
}
|
|
|
|
|
|
public bool IsMonitored(int id)
|
|
{
|
|
return _seriesRepository.Get(id).Monitored;
|
|
}
|
|
|
|
|
|
public Series UpdateSeriesInfo(int seriesId)
|
|
{
|
|
var series = _seriesRepository.Get(seriesId);
|
|
var tvDbSeries = _tvDbProvider.GetSeries(series.TvDbId);
|
|
|
|
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.BannerUrl = tvDbSeries.BannerUrl;
|
|
series.Network = tvDbSeries.Network;
|
|
series.FirstAired = tvDbSeries.FirstAired;
|
|
|
|
try
|
|
{
|
|
if (series.TvRageId == 0)
|
|
series = _tvRageMappingProvider.FindMatchingTvRageSeries(series);
|
|
}
|
|
|
|
catch (Exception ex)
|
|
{
|
|
logger.ErrorException("Error getting TvRage information for series: " + series.Title, ex);
|
|
}
|
|
|
|
_seriesRepository.Update(series);
|
|
|
|
return series;
|
|
}
|
|
|
|
public Series FindSeries(string title)
|
|
{
|
|
var normalizeTitle = Parser.NormalizeTitle(title);
|
|
|
|
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)
|
|
{
|
|
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();
|
|
repoSeries.TvDbId = tvDbSeriesId;
|
|
repoSeries.Path = path;
|
|
repoSeries.Monitored = true;
|
|
repoSeries.QualityProfileId = qualityProfileId;
|
|
repoSeries.Title = title;
|
|
if (qualityProfileId == 0)
|
|
repoSeries.QualityProfileId = _configService.DefaultQualityProfile;
|
|
|
|
repoSeries.QualityProfile = _qualityProfileService.Get(repoSeries.QualityProfileId);
|
|
repoSeries.SeasonFolder = _configService.UseSeasonFolder;
|
|
repoSeries.BacklogSetting = BacklogSettingType.Inherit;
|
|
|
|
if (airedAfter.HasValue)
|
|
repoSeries.CustomStartDate = airedAfter;
|
|
|
|
_seriesRepository.Insert(repoSeries);
|
|
|
|
_eventAggregator.Publish(new SeriesAddedEvent(repoSeries));
|
|
}
|
|
|
|
public void UpdateFromSeriesEditor(IList<Series> editedSeries)
|
|
{
|
|
var allSeries = _seriesRepository.All();
|
|
|
|
foreach (var series in allSeries)
|
|
{
|
|
//Only update parameters that can be changed in MassEdit
|
|
var edited = editedSeries.Single(s => ((ModelBase)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 void SetSeriesType(int seriesId, SeriesType seriesType)
|
|
{
|
|
_seriesRepository.SetSeriesType(seriesId, seriesType);
|
|
}
|
|
|
|
|
|
}
|
|
}
|