From 616454edb4fa8c13bdf7da443ec9344b9cc5ece6 Mon Sep 17 00:00:00 2001
From: Taloth Saldono <Taloth@users.noreply.github.com>
Date: Thu, 1 Mar 2018 19:21:37 +0100
Subject: [PATCH] Fixed: Preserve existing watched status in Kodi nfo files on
 metadata refreshes (not file upgrades).

---
 .../Metadata/Consumers/Xbmc/XbmcMetadata.cs   | 30 +++++++++++++------
 1 file changed, 21 insertions(+), 9 deletions(-)

diff --git a/src/NzbDrone.Core/Extras/Metadata/Consumers/Xbmc/XbmcMetadata.cs b/src/NzbDrone.Core/Extras/Metadata/Consumers/Xbmc/XbmcMetadata.cs
index 096d7c889..5126fbf96 100644
--- a/src/NzbDrone.Core/Extras/Metadata/Consumers/Xbmc/XbmcMetadata.cs
+++ b/src/NzbDrone.Core/Extras/Metadata/Consumers/Xbmc/XbmcMetadata.cs
@@ -7,6 +7,7 @@ using System.Text.RegularExpressions;
 using System.Xml;
 using System.Xml.Linq;
 using NLog;
+using NzbDrone.Common.Disk;
 using NzbDrone.Common.Extensions;
 using NzbDrone.Core.Extras.Metadata.Files;
 using NzbDrone.Core.MediaCover;
@@ -21,13 +22,16 @@ namespace NzbDrone.Core.Extras.Metadata.Consumers.Xbmc
         private readonly Logger _logger;
         private readonly IMapCoversToLocal _mediaCoverService;
         private readonly IDetectXbmcNfo _detectNfo;
+        private readonly IDiskProvider _diskProvider;
 
         public XbmcMetadata(IDetectXbmcNfo detectNfo,
+                            IDiskProvider diskProvider,
                             IMapCoversToLocal mediaCoverService,
                             Logger logger)
         {
             _logger = logger;
             _mediaCoverService = mediaCoverService;
+            _diskProvider = diskProvider;
             _detectNfo = detectNfo;
         }
 
@@ -203,6 +207,8 @@ namespace NzbDrone.Core.Extras.Metadata.Consumers.Xbmc
 
             _logger.Debug("Generating Episode Metadata for: {0}", Path.Combine(series.Path, episodeFile.RelativePath));
 
+            var watched = GetExistingWatchedStatus(series, episodeFile.RelativePath);
+
             var xmlResult = string.Empty;
             foreach (var episode in episodeFile.Episodes.Value)
             {
@@ -237,7 +243,7 @@ namespace NzbDrone.Core.Extras.Metadata.Consumers.Xbmc
                         details.Add(new XElement("thumb", image.Url));
                     }
 
-                    details.Add(new XElement("watched", "false"));
+                    details.Add(new XElement("watched", watched));
 
                     if (episode.Ratings != null && episode.Ratings.Votes > 0)
                     {
@@ -298,9 +304,6 @@ namespace NzbDrone.Core.Extras.Metadata.Consumers.Xbmc
                 }
             }
 
-
-            RenameExistingNfo(GetEpisodeMetadataFilename(episodeFile.RelativePath));
-
             return new MetadataFileResult(GetEpisodeMetadataFilename(episodeFile.RelativePath), xmlResult.Trim(Environment.NewLine.ToCharArray()));
         }
 
@@ -380,11 +383,6 @@ namespace NzbDrone.Core.Extras.Metadata.Consumers.Xbmc
             }
         }
 
-        private void RenameExistingNfo(string nfoFilePath)
-        {
-
-        }
-
         private string GetEpisodeMetadataFilename(string episodeFilePath)
         {
             return Path.ChangeExtension(episodeFilePath, "nfo");
@@ -394,5 +392,19 @@ namespace NzbDrone.Core.Extras.Metadata.Consumers.Xbmc
         {
             return Path.ChangeExtension(episodeFilePath, "").Trim('.') + "-thumb.jpg";
         }
+
+        private bool GetExistingWatchedStatus(Series series, string episodeFilePath)
+        {
+            var fullPath = Path.Combine(series.Path, GetEpisodeMetadataFilename(episodeFilePath));
+
+            if (!_diskProvider.FileExists(fullPath))
+            {
+                return false;
+            }
+
+            var fileContent = _diskProvider.ReadAllText(fullPath);
+
+            return Regex.IsMatch(fileContent, "<watched>true</watched>");
+        }
     }
 }