mirror of
https://github.com/Sonarr/Sonarr.git
synced 2024-10-31 16:02:29 +01:00
213 lines
7.3 KiB
C#
213 lines
7.3 KiB
C#
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<Series> editedSeries);
|
|
Series FindByTvdbId(int tvdbId);
|
|
Series FindByTitle(string title);
|
|
void SetSeriesType(int seriesId, SeriesTypes seriesTypes);
|
|
void DeleteSeries(int seriesId, bool deleteFiles);
|
|
List<Series> GetAllSeries();
|
|
Series UpdateSeries(Series series);
|
|
bool SeriesPathExists(string folder);
|
|
List<Series> GetSeriesInList(IEnumerable<int> seriesIds);
|
|
Series FindBySlug(string slug);
|
|
}
|
|
|
|
public class SeriesService : ISeriesService, IHandleAsync<SeriesAddedEvent>
|
|
{
|
|
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 IDiskProvider _diskProvider;
|
|
private readonly Logger _logger;
|
|
|
|
public SeriesService(ISeriesRepository seriesRepository, IConfigService configServiceService,
|
|
IProvideSeriesInfo seriesInfoProxy, IMessageAggregator messageAggregator, ISceneMappingService sceneMappingService,
|
|
IRootFolderService rootFolderService, IDiskProvider 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.Path))
|
|
{
|
|
var folderName = FileNameBuilder.CleanFilename(newSeries.Title);
|
|
newSeries.Path = Path.Combine(newSeries.RootFolderPath, folderName);
|
|
|
|
_diskProvider.CreateFolder(newSeries.Path);
|
|
}
|
|
|
|
_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<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 => 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 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.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<Series> 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<Series> GetSeriesInList(IEnumerable<int> seriesIds)
|
|
{
|
|
return _seriesRepository.Get(seriesIds).ToList();
|
|
}
|
|
|
|
public void HandleAsync(SeriesAddedEvent message)
|
|
{
|
|
UpdateSeriesInfo(message.Series.Id);
|
|
}
|
|
}
|
|
}
|