Fixed: Setting file dates

Fixes #940

src/NzbDrone.Core/MediaFiles/UpdateBookFileService.cs

@@ -0,0 +1,127 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using NLog;
+using NzbDrone.Common.Disk;
+using NzbDrone.Common.EnvironmentInfo;
+using NzbDrone.Common.Instrumentation.Extensions;
+using NzbDrone.Core.Books;
+using NzbDrone.Core.Configuration;
+using NzbDrone.Core.MediaFiles.Events;
+using NzbDrone.Core.Messaging.Events;
+
+namespace NzbDrone.Core.MediaFiles
+{
+    public interface IUpdateBookFileService
+    {
+        void ChangeFileDateForFile(BookFile bookFile, Author author, Book book);
+    }
+
+    public class UpdateBookFileService : IUpdateBookFileService,
+                                            IHandle<AuthorScannedEvent>
+    {
+        private readonly IDiskProvider _diskProvider;
+        private readonly IBookService _bookService;
+        private readonly IConfigService _configService;
+        private readonly Logger _logger;
+        private static readonly DateTime EpochTime = new DateTime(1970, 1, 1, 0, 0, 0, DateTimeKind.Utc);
+
+        public UpdateBookFileService(IDiskProvider diskProvider,
+                                      IConfigService configService,
+                                      IBookService bookService,
+                                      Logger logger)
+        {
+            _diskProvider = diskProvider;
+            _configService = configService;
+            _bookService = bookService;
+            _logger = logger;
+        }
+
+        public void ChangeFileDateForFile(BookFile bookFile, Author author, Book book)
+        {
+            ChangeFileDate(bookFile, book);
+        }
+
+        private bool ChangeFileDate(BookFile bookFile, Book book)
+        {
+            var bookFilePath = bookFile.Path;
+
+            switch (_configService.FileDate)
+            {
+                case FileDateType.BookReleaseDate:
+                    {
+                        if (!book.ReleaseDate.HasValue)
+                        {
+                            _logger.Debug("Could not create valid date to change file [{0}]", bookFilePath);
+                            return false;
+                        }
+
+                        var relDate = book.ReleaseDate.Value;
+
+                        // avoiding false +ve checks and set date skewing by not using UTC (Windows)
+                        var oldDateTime = _diskProvider.FileGetLastWrite(bookFilePath);
+
+                        if (OsInfo.IsNotWindows && relDate < EpochTime)
+                        {
+                            _logger.Debug("Setting date of file to 1970-01-01 as actual airdate is before that time and will not be set properly");
+                            relDate = EpochTime;
+                        }
+
+                        if (!DateTime.Equals(relDate, oldDateTime))
+                        {
+                            try
+                            {
+                                _diskProvider.FileSetLastWriteTime(bookFilePath, relDate);
+                                _logger.Debug("Date of file [{0}] changed from '{1}' to '{2}'", bookFilePath, oldDateTime, relDate);
+
+                                return true;
+                            }
+                            catch (Exception ex)
+                            {
+                                _logger.Warn(ex, "Unable to set date of file [" + bookFilePath + "]");
+                            }
+                        }
+
+                        return false;
+                    }
+            }
+
+            return false;
+        }
+
+        public void Handle(AuthorScannedEvent message)
+        {
+            if (_configService.FileDate == FileDateType.None)
+            {
+                return;
+            }
+
+            var books = _bookService.GetAuthorBooksWithFiles(message.Author);
+
+            var bookFiles = new List<BookFile>();
+            var updated = new List<BookFile>();
+
+            foreach (var book in books)
+            {
+                var files = book.BookFiles.Value;
+                foreach (var file in files)
+                {
+                    bookFiles.Add(file);
+                    if (ChangeFileDate(file, book))
+                    {
+                        updated.Add(file);
+                    }
+                }
+            }
+
+            if (updated.Any())
+            {
+                _logger.ProgressDebug("Changed file date for {0} files of {1} in {2}", updated.Count, bookFiles.Count, message.Author.Name);
+            }
+            else
+            {
+                _logger.ProgressDebug("No file dates changed for {0}", message.Author.Name);
+            }
+        }
+    }
+}