From 5a2fa41af7bec16a8ca279966c2113566310e705 Mon Sep 17 00:00:00 2001
From: Mark McDowall <markus.mcd5@gmail.com>
Date: Sun, 7 Jun 2015 01:15:41 -0700
Subject: [PATCH] New: Search for newly added past episodes after series is
 refreshed

Closes #503
---
 src/NzbDrone.Core/NzbDrone.Core.csproj       |  1 +
 src/NzbDrone.Core/Tv/EpisodeAddedService.cs  | 88 ++++++++++++++++++++
 src/NzbDrone.Core/Tv/SeriesScannedHandler.cs |  6 +-
 3 files changed, 94 insertions(+), 1 deletion(-)
 create mode 100644 src/NzbDrone.Core/Tv/EpisodeAddedService.cs

diff --git a/src/NzbDrone.Core/NzbDrone.Core.csproj b/src/NzbDrone.Core/NzbDrone.Core.csproj
index 6c7875871..56c632442 100644
--- a/src/NzbDrone.Core/NzbDrone.Core.csproj
+++ b/src/NzbDrone.Core/NzbDrone.Core.csproj
@@ -892,6 +892,7 @@
     <Compile Include="Tv\Commands\MoveSeriesCommand.cs" />
     <Compile Include="Tv\Commands\RefreshSeriesCommand.cs" />
     <Compile Include="Tv\Episode.cs" />
+    <Compile Include="Tv\EpisodeAddedService.cs" />
     <Compile Include="Tv\EpisodeCutoffService.cs" />
     <Compile Include="Tv\EpisodeMonitoredService.cs" />
     <Compile Include="Tv\EpisodeRepository.cs">
diff --git a/src/NzbDrone.Core/Tv/EpisodeAddedService.cs b/src/NzbDrone.Core/Tv/EpisodeAddedService.cs
new file mode 100644
index 000000000..54e3d2991
--- /dev/null
+++ b/src/NzbDrone.Core/Tv/EpisodeAddedService.cs
@@ -0,0 +1,88 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using NLog;
+using NzbDrone.Common.Cache;
+using NzbDrone.Common.Extensions;
+using NzbDrone.Core.IndexerSearch;
+using NzbDrone.Core.Messaging.Commands;
+using NzbDrone.Core.Messaging.Events;
+using NzbDrone.Core.Tv.Events;
+
+namespace NzbDrone.Core.Tv
+{
+    public interface IEpisodeAddedService
+    {
+        void SearchForRecentlyAdded(int seriesId);
+    }
+
+    public class EpisodeAddedService : IHandle<EpisodeInfoRefreshedEvent>, IEpisodeAddedService
+    {
+        private readonly IManageCommandQueue _commandQueueManager;
+        private readonly IEpisodeService _episodeService;
+        private readonly Logger _logger;
+        private readonly ICached<List<int>> _addedEpisodesCache;
+
+        public EpisodeAddedService(ICacheManager cacheManager,
+                                   IManageCommandQueue commandQueueManager,
+                                   IEpisodeService episodeService,
+                                   Logger logger)
+        {
+            _commandQueueManager = commandQueueManager;
+            _episodeService = episodeService;
+            _logger = logger;
+            _addedEpisodesCache = cacheManager.GetCache<List<int>>(GetType());
+        }
+
+        public void SearchForRecentlyAdded(int seriesId)
+        {
+            var previouslyAired = _addedEpisodesCache.Find(seriesId.ToString());
+
+            if (previouslyAired != null && previouslyAired.Any())
+            {
+                var missing = previouslyAired.Select(e => _episodeService.GetEpisode(e)).Where(e => !e.HasFile).ToList();
+
+                if (missing.Any())
+                {
+                    _commandQueueManager.Push(new EpisodeSearchCommand(missing.Select(e => e.Id).ToList()));
+                }
+            }
+            
+            _addedEpisodesCache.Remove(seriesId.ToString());
+        }
+
+        public void Handle(EpisodeInfoRefreshedEvent message)
+        {
+            if (message.Series.AddOptions == null)
+            {
+                if (!message.Series.Monitored)
+                {
+                    _logger.Debug("Series is not monitored");
+                    return;
+                }
+
+                if (message.Added.Empty())
+                {
+                    _logger.Debug("No new episodes, skipping search");
+                    return;
+                }
+
+                if (message.Added.None(a => a.AirDateUtc.HasValue))
+                {
+                    _logger.Debug("No new episodes have an air date");
+                    return;
+                }
+
+                var previouslyAired = message.Added.Where(a => a.AirDateUtc.HasValue && a.AirDateUtc.Value.Before(DateTime.UtcNow.AddDays(1)) && a.Monitored).ToList();
+
+                if (previouslyAired.Empty())
+                {
+                    _logger.Debug("Newly added episodes all air in the future");
+                    return;
+                }
+
+                _addedEpisodesCache.Set(message.Series.Id.ToString(), previouslyAired.Select(e => e.Id).ToList());
+            }
+        }
+    }
+}
diff --git a/src/NzbDrone.Core/Tv/SeriesScannedHandler.cs b/src/NzbDrone.Core/Tv/SeriesScannedHandler.cs
index aa8d90d5a..9d208c764 100644
--- a/src/NzbDrone.Core/Tv/SeriesScannedHandler.cs
+++ b/src/NzbDrone.Core/Tv/SeriesScannedHandler.cs
@@ -12,17 +12,20 @@ namespace NzbDrone.Core.Tv
         private readonly IEpisodeMonitoredService _episodeMonitoredService;
         private readonly ISeriesService _seriesService;
         private readonly IManageCommandQueue _commandQueueManager;
-       
+        private readonly IEpisodeAddedService _episodeAddedService;
+
         private readonly Logger _logger;
 
         public SeriesScannedHandler(IEpisodeMonitoredService episodeMonitoredService,
                                     ISeriesService seriesService,
                                     IManageCommandQueue commandQueueManager,
+                                    IEpisodeAddedService episodeAddedService,
                                     Logger logger)
         {
             _episodeMonitoredService = episodeMonitoredService;
             _seriesService = seriesService;
             _commandQueueManager = commandQueueManager;
+            _episodeAddedService = episodeAddedService;
             _logger = logger;
         }
 
@@ -30,6 +33,7 @@ namespace NzbDrone.Core.Tv
         {
             if (series.AddOptions == null)
             {
+                _episodeAddedService.SearchForRecentlyAdded(series.Id);
                 return;
             }