Now importing downloaded episodes

NzbDrone.Core/MediaFiles/Commands/DiskScanCommand.cs

@@ -13,5 +13,9 @@ namespace NzbDrone.Core.MediaFiles.Commands
                 SeriesId = seriesId;
             }
         }
+
+        public DiskScanCommand()
+        {
+        }
     }
 }

NzbDrone.Core/MediaFiles/Commands/DownloadedEpisodesScanCommand.cs

@@ -0,0 +1,11 @@
+using NzbDrone.Common.Messaging;
+
+namespace NzbDrone.Core.MediaFiles.Commands
+{
+    public class DownloadedEpisodesScanCommand : ICommand
+    {
+        public DownloadedEpisodesScanCommand()
+        {
+        }
+    }
+}

NzbDrone.Core/MediaFiles/DownloadedEpisodesImportService.cs

@@ -0,0 +1,128 @@
+using System;
+using System.IO;
+using NLog;
+using NzbDrone.Common;
+using NzbDrone.Common.Messaging;
+using NzbDrone.Core.Configuration;
+using NzbDrone.Core.MediaFiles.Commands;
+using NzbDrone.Core.Parser;
+using NzbDrone.Core.Tv;
+
+namespace NzbDrone.Core.MediaFiles
+{
+    public class DownloadedEpisodesImportService : IExecute<DownloadedEpisodesScanCommand>
+    {
+        private readonly IDiskProvider _diskProvider;
+        private readonly IDiskScanService _diskScanService;
+        private readonly ISeriesService _seriesService;
+        private readonly IMoveEpisodeFiles _episodeFileMover;
+        private readonly IParsingService _parsingService;
+        private readonly IConfigService _configService;
+        private readonly Logger _logger;
+
+        public DownloadedEpisodesImportService(IDiskProvider diskProvider,
+            IDiskScanService diskScanService,
+            ISeriesService seriesService,
+            IMoveEpisodeFiles episodeFileMover,
+            IParsingService parsingService,
+            IConfigService configService,
+            Logger logger)
+        {
+            _diskProvider = diskProvider;
+            _diskScanService = diskScanService;
+            _seriesService = seriesService;
+            _episodeFileMover = episodeFileMover;
+            _parsingService = parsingService;
+            _configService = configService;
+            _logger = logger;
+        }
+
+        public void ProcessSubFolder(DirectoryInfo subfolderInfo)
+        {
+            if (_diskProvider.GetLastFolderWrite(subfolderInfo.FullName).AddMinutes(2) > DateTime.UtcNow)
+            {
+                _logger.Trace("[{0}] is too fresh. skipping", subfolderInfo.FullName);
+                return;
+            }
+
+            var series = _parsingService.GetSeries(subfolderInfo.Name);
+
+            if (series == null)
+            {
+                _logger.Trace("Unknown Series {0}", subfolderInfo.Name);
+                return;
+            }
+
+            var files = _diskScanService.GetVideoFiles(subfolderInfo.FullName);
+
+            foreach (var file in files)
+            {
+                ProcessVideoFile(file, series);
+            }
+        }
+
+        public void ProcessVideoFile(string videoFile, Series series)
+        {
+            if (_diskProvider.GetLastFileWrite(videoFile).AddMinutes(2) > DateTime.UtcNow)
+            {
+                _logger.Trace("[{0}] is too fresh. skipping", videoFile);
+                return;
+            }
+
+            if (_diskProvider.IsFileLocked(new FileInfo(videoFile)))
+            {
+                _logger.Trace("[{0}] is currently locked by another process, skipping", videoFile);
+                return;
+            }
+
+            var episodeFile = _diskScanService.ImportFile(series, videoFile);
+
+            if (episodeFile != null)
+            {
+                _episodeFileMover.MoveEpisodeFile(episodeFile, true);
+            }
+        }
+
+        public void Execute(DownloadedEpisodesScanCommand message)
+        {
+            //TODO: We should also process the download client's category folder
+            var downloadedEpisodesFolder = _configService.DownloadedEpisodesFolder;
+
+            if (String.IsNullOrEmpty(downloadedEpisodesFolder))
+            {
+                _logger.Warn("Downloaded Episodes Folder is not configured");
+                return;
+            }
+
+            foreach (var subfolder in _diskProvider.GetDirectories(downloadedEpisodesFolder))
+            {
+                try
+                {
+                    if (!_seriesService.SeriesPathExists(subfolder))
+                    {
+                        ProcessSubFolder(new DirectoryInfo(subfolder));
+                    }
+                }
+                catch (Exception e)
+                {
+                    _logger.ErrorException("An error has occurred while importing folder: " + subfolder, e);
+                }
+            }
+
+            foreach (var videoFile in _diskScanService.GetVideoFiles(downloadedEpisodesFolder, false))
+            {
+                try
+                {
+                    var series = _parsingService.GetSeries(videoFile);
+                    ProcessVideoFile(videoFile, series);
+                }
+                catch (Exception ex)
+                {
+                    _logger.ErrorException("An error has occurred while importing video file" + videoFile, ex);
+                }
+            }
+
+            //TODO: cleanup empty folders
+        }
+    }
+}