Moved source code under src folder - massive change

2026-04-23 22:25:09 -04:00 · 2013-10-02 18:01:32 -07:00
parent 2fc8123d6b
commit 5bf0e197ec
1499 changed files with 1054 additions and 1444 deletions
@@ -0,0 +1,130 @@
+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.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);
+        }
+    }
+}