131 lines
4.7 KiB
C#
131 lines
4.7 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using NLog;
|
|
using NzbDrone.Core.DataAugmentation.DailySeries;
|
|
using NzbDrone.Core.Instrumentation;
|
|
using NzbDrone.Core.Messaging;
|
|
using NzbDrone.Core.Messaging.Commands;
|
|
using NzbDrone.Core.Messaging.Events;
|
|
using NzbDrone.Core.MetadataSource;
|
|
using NzbDrone.Core.Tv.Commands;
|
|
using NzbDrone.Core.Tv.Events;
|
|
using NzbDrone.Common;
|
|
|
|
namespace NzbDrone.Core.Tv
|
|
{
|
|
public class RefreshSeriesService : IExecute<RefreshSeriesCommand>, IHandleAsync<SeriesAddedEvent>
|
|
{
|
|
private readonly IProvideSeriesInfo _seriesInfo;
|
|
private readonly ISeriesService _seriesService;
|
|
private readonly IRefreshEpisodeService _refreshEpisodeService;
|
|
private readonly IEventAggregator _eventAggregator;
|
|
private readonly IDailySeriesService _dailySeriesService;
|
|
private readonly Logger _logger;
|
|
|
|
public RefreshSeriesService(IProvideSeriesInfo seriesInfo, ISeriesService seriesService, IRefreshEpisodeService refreshEpisodeService, IEventAggregator eventAggregator, IDailySeriesService dailySeriesService, Logger logger)
|
|
{
|
|
_seriesInfo = seriesInfo;
|
|
_seriesService = seriesService;
|
|
_refreshEpisodeService = refreshEpisodeService;
|
|
_eventAggregator = eventAggregator;
|
|
_dailySeriesService = dailySeriesService;
|
|
_logger = logger;
|
|
}
|
|
|
|
private void RefreshSeriesInfo(Series series)
|
|
{
|
|
_logger.ProgressInfo("Updating Info for {0}", series.Title);
|
|
var tuple = _seriesInfo.GetSeriesInfo(series.TvdbId);
|
|
|
|
var seriesInfo = tuple.Item1;
|
|
|
|
series.Title = seriesInfo.Title;
|
|
series.AirTime = seriesInfo.AirTime;
|
|
series.Overview = seriesInfo.Overview;
|
|
series.Status = seriesInfo.Status;
|
|
series.CleanTitle = seriesInfo.CleanTitle;
|
|
series.LastInfoSync = DateTime.UtcNow;
|
|
series.Runtime = seriesInfo.Runtime;
|
|
series.Images = seriesInfo.Images;
|
|
series.Network = seriesInfo.Network;
|
|
series.FirstAired = seriesInfo.FirstAired;
|
|
|
|
if (_dailySeriesService.IsDailySeries(series.TvdbId))
|
|
{
|
|
series.SeriesType = SeriesTypes.Daily;
|
|
}
|
|
|
|
try
|
|
{
|
|
series.Path = new DirectoryInfo(series.Path).FullName;
|
|
series.Path = series.Path.GetActualCasing();
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
_logger.WarnException("Couldn't update series path for " + series.Path, e);
|
|
}
|
|
|
|
series.Seasons = UpdateSeasons(series, seriesInfo);
|
|
|
|
_seriesService.UpdateSeries(series);
|
|
_refreshEpisodeService.RefreshEpisodeInfo(series, tuple.Item2);
|
|
|
|
_logger.Debug("Finished series refresh for {0}", series.Title);
|
|
_eventAggregator.PublishEvent(new SeriesUpdatedEvent(series));
|
|
}
|
|
|
|
private List<Season> UpdateSeasons(Series series, Series seriesInfo)
|
|
{
|
|
foreach (var season in seriesInfo.Seasons)
|
|
{
|
|
var existingSeason = series.Seasons.SingleOrDefault(s => s.SeasonNumber == season.SeasonNumber);
|
|
|
|
//Todo: Should this should use the previous season's monitored state?
|
|
if (existingSeason == null)
|
|
{
|
|
_logger.Trace("New season ({0}) for series: [{1}] {2}, setting monitored to true", season.SeasonNumber, series.TvdbId, series.Title);
|
|
season.Monitored = true;
|
|
}
|
|
|
|
else
|
|
{
|
|
season.Monitored = existingSeason.Monitored;
|
|
}
|
|
}
|
|
|
|
return seriesInfo.Seasons;
|
|
}
|
|
|
|
public void Execute(RefreshSeriesCommand message)
|
|
{
|
|
if (message.SeriesId.HasValue)
|
|
{
|
|
var series = _seriesService.GetSeries(message.SeriesId.Value);
|
|
RefreshSeriesInfo(series);
|
|
}
|
|
else
|
|
{
|
|
var allSeries = _seriesService.GetAllSeries().OrderBy(c => c.LastInfoSync).ToList();
|
|
|
|
foreach (var series in allSeries)
|
|
{
|
|
try
|
|
{
|
|
RefreshSeriesInfo(series);
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
_logger.ErrorException("Couldn't refresh info for {0}".Inject(series), e);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
public void HandleAsync(SeriesAddedEvent message)
|
|
{
|
|
RefreshSeriesInfo(message.Series);
|
|
}
|
|
}
|
|
} |