You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Readarr/NzbDrone.Core/Tv/SeriesService.cs

162 lines
5.7 KiB

using System;
14 years ago
using System.Collections.Generic;
using System.IO;
14 years ago
using System.Linq;
using Marr.Data;
14 years ago
using NLog;
using NzbDrone.Common;
using NzbDrone.Common.EnsureThat;
using NzbDrone.Common.Eventing;
using NzbDrone.Core.Configuration;
using NzbDrone.Core.Datastore;
using NzbDrone.Core.MetadataSource;
using NzbDrone.Core.Model;
using NzbDrone.Core.Organizer;
using NzbDrone.Core.RootFolders;
using NzbDrone.Core.Tv.Events;
14 years ago
namespace NzbDrone.Core.Tv
14 years ago
{
public interface ISeriesService
{
bool IsMonitored(int id);
Series UpdateSeriesInfo(int seriesId);
Series GetSeries(int seriesId);
void AddSeries(Series newSeries);
void UpdateFromSeriesEditor(IList<Series> editedSeries);
12 years ago
Series FindByTvdbId(int tvdbId);
void SetSeriesType(int seriesId, SeriesTypes seriesTypes);
void DeleteSeries(int seriesId, bool deleteFiles);
}
public class SeriesService : ISeriesService, IHandleAsync<SeriesAddedEvent>
14 years ago
{
12 years ago
private readonly ISeriesRepository _seriesRepository;
private readonly IConfigService _configService;
private readonly IProvideSeriesInfo _seriesInfoProxy;
private readonly IEventAggregator _eventAggregator;
private readonly IBasicRepository<RootFolder> _rootFolderRepository;
private readonly DiskProvider _diskProvider;
private readonly Logger _logger;
public SeriesService(ISeriesRepository seriesRepository, IConfigService configServiceService,
IProvideSeriesInfo seriesInfoProxy, IEventAggregator eventAggregator,
IBasicRepository<RootFolder> rootFolderRepository, DiskProvider diskProvider, Logger logger)
14 years ago
{
12 years ago
_seriesRepository = seriesRepository;
_configService = configServiceService;
_seriesInfoProxy = seriesInfoProxy;
_eventAggregator = eventAggregator;
_rootFolderRepository = rootFolderRepository;
_diskProvider = diskProvider;
_logger = logger;
14 years ago
}
12 years ago
public bool IsMonitored(int id)
14 years ago
{
12 years ago
return _seriesRepository.Get(id).Monitored;
}
public Series UpdateSeriesInfo(int seriesId)
{
12 years ago
var series = _seriesRepository.Get(seriesId);
var tvDbSeries = _seriesInfoProxy.GetSeriesInfo(series.TvDbId);
12 years ago
series.Title = tvDbSeries.Title;
series.AirTime = tvDbSeries.AirTime;
series.Overview = tvDbSeries.Overview;
series.Status = tvDbSeries.Status;
series.CleanTitle = Parser.NormalizeTitle(tvDbSeries.Title);
series.LastInfoSync = DateTime.Now;
series.Runtime = tvDbSeries.Runtime;
series.Images = tvDbSeries.Images;
series.Network = tvDbSeries.Network;
series.FirstAired = tvDbSeries.FirstAired;
12 years ago
_seriesRepository.Update(series);
_eventAggregator.Publish(new SeriesUpdatedEvent(series));
return series;
}
public Series GetSeries(int seriesId)
{
return _seriesRepository.Get(seriesId);
}
public void AddSeries(Series newSeries)
{
Ensure.That(() => newSeries).IsNotNull();
if(String.IsNullOrWhiteSpace(newSeries.Folder))
{
newSeries.Folder = FileNameBuilder.CleanFilename(newSeries.Title);
_diskProvider.CreateDirectory(Path.Combine(_rootFolderRepository.Get(newSeries.RootFolderId).Path, newSeries.Folder));
}
_logger.Info("Adding Series [{0}] Path: [{1}]", newSeries.Title, newSeries.Path);
newSeries.Monitored = true;
newSeries.CleanTitle = Parser.NormalizeTitle(newSeries.Title);
if (newSeries.QualityProfileId == 0)
newSeries.QualityProfileId = _configService.DefaultQualityProfile;
newSeries.SeasonFolder = _configService.UseSeasonFolder;
newSeries.BacklogSetting = BacklogSettingType.Inherit;
_seriesRepository.Insert(newSeries);
_eventAggregator.Publish(new SeriesAddedEvent(newSeries));
}
public void UpdateFromSeriesEditor(IList<Series> editedSeries)
{
12 years ago
var allSeries = _seriesRepository.All();
12 years ago
foreach (var series in allSeries)
{
//Only update parameters that can be changed in MassEdit
12 years ago
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;
12 years ago
_seriesRepository.Update(series);
}
}
public void SetTvRageId(int seriesId, int tvRageId)
{
_seriesRepository.SetTvRageId(seriesId, tvRageId);
}
12 years ago
public Series FindByTvdbId(int tvdbId)
{
return _seriesRepository.FindByTvdbId(tvdbId);
}
public void SetSeriesType(int seriesId, SeriesTypes seriesTypes)
12 years ago
{
_seriesRepository.SetSeriesType(seriesId, seriesTypes);
12 years ago
}
public void DeleteSeries(int seriesId, bool deleteFiles)
{
var series = _seriesRepository.Get(seriesId);
_seriesRepository.Delete(seriesId);
_eventAggregator.Publish(new SeriesDeletedEvent(series, deleteFiles));
}
public void HandleAsync(SeriesAddedEvent message)
{
UpdateSeriesInfo(message.Series.Id);
}
14 years ago
}
}