From 743c977e5b242bb68e658c2ba9dba609cd00e86b Mon Sep 17 00:00:00 2001
From: Bogdan <mynameisbogdan@users.noreply.github.com>
Date: Fri, 31 Jan 2025 21:55:45 +0200
Subject: [PATCH] New: Refresh cache for tracked queue on movies update

---
 .../TrackedDownloadService.cs                 | 34 +++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/src/NzbDrone.Core/Download/TrackedDownloads/TrackedDownloadService.cs b/src/NzbDrone.Core/Download/TrackedDownloads/TrackedDownloadService.cs
index 341536f5bf..fee706f7b1 100644
--- a/src/NzbDrone.Core/Download/TrackedDownloads/TrackedDownloadService.cs
+++ b/src/NzbDrone.Core/Download/TrackedDownloads/TrackedDownloadService.cs
@@ -29,6 +29,8 @@ public interface ITrackedDownloadService
 
     public class TrackedDownloadService : ITrackedDownloadService,
                                           IHandle<MovieAddedEvent>,
+                                          IHandle<MovieEditedEvent>,
+                                          IHandle<MoviesBulkEditedEvent>,
                                           IHandle<MoviesDeletedEvent>
     {
         private readonly IParsingService _parsingService;
@@ -272,6 +274,38 @@ public void Handle(MovieAddedEvent message)
             }
         }
 
+        public void Handle(MovieEditedEvent message)
+        {
+            var cachedItems = _cache.Values
+                .Where(t =>
+                    t.RemoteMovie?.Movie != null &&
+                    (t.RemoteMovie.Movie.Id == message.Movie?.Id || t.RemoteMovie.Movie.TmdbId == message.Movie?.TmdbId))
+                .ToList();
+
+            if (cachedItems.Any())
+            {
+                cachedItems.ForEach(UpdateCachedItem);
+
+                _eventAggregator.PublishEvent(new TrackedDownloadRefreshedEvent(GetTrackedDownloads()));
+            }
+        }
+
+        public void Handle(MoviesBulkEditedEvent message)
+        {
+            var cachedItems = _cache.Values
+                .Where(t =>
+                    t.RemoteMovie?.Movie != null &&
+                    message.Movies.Any(m => m.Id == t.RemoteMovie.Movie.Id || m.TmdbId == t.RemoteMovie.Movie.TmdbId))
+                .ToList();
+
+            if (cachedItems.Any())
+            {
+                cachedItems.ForEach(UpdateCachedItem);
+
+                _eventAggregator.PublishEvent(new TrackedDownloadRefreshedEvent(GetTrackedDownloads()));
+            }
+        }
+
         public void Handle(MoviesDeletedEvent message)
         {
             var cachedItems = _cache.Values