From 6a332b40ac94e6e7c23217074da8e18e0ca3a319 Mon Sep 17 00:00:00 2001
From: Bogdan <mynameisbogdan@users.noreply.github.com>
Date: Thu, 5 Sep 2024 14:56:36 +0300
Subject: [PATCH] Fixed: Refresh tags after updating autotags

---
 src/NzbDrone.Core/AutoTagging/AutoTaggingService.cs   | 11 +++++++++++
 src/NzbDrone.Core/AutoTagging/AutoTagsUpdatedEvent.cs |  8 ++++++++
 src/Sonarr.Api.V3/Tags/TagController.cs               | 11 ++++++++++-
 3 files changed, 29 insertions(+), 1 deletion(-)
 create mode 100644 src/NzbDrone.Core/AutoTagging/AutoTagsUpdatedEvent.cs

diff --git a/src/NzbDrone.Core/AutoTagging/AutoTaggingService.cs b/src/NzbDrone.Core/AutoTagging/AutoTaggingService.cs
index 10c10060b..066a6e322 100644
--- a/src/NzbDrone.Core/AutoTagging/AutoTaggingService.cs
+++ b/src/NzbDrone.Core/AutoTagging/AutoTaggingService.cs
@@ -2,6 +2,7 @@ using System.Collections.Generic;
 using System.Linq;
 using NzbDrone.Common.Cache;
 using NzbDrone.Common.Extensions;
+using NzbDrone.Core.Messaging.Events;
 using NzbDrone.Core.RootFolders;
 using NzbDrone.Core.Tv;
 
@@ -22,14 +23,18 @@ namespace NzbDrone.Core.AutoTagging
     {
         private readonly IAutoTaggingRepository _repository;
         private readonly RootFolderService _rootFolderService;
+        private readonly IEventAggregator _eventAggregator;
         private readonly ICached<Dictionary<int, AutoTag>> _cache;
 
         public AutoTaggingService(IAutoTaggingRepository repository,
                                   RootFolderService rootFolderService,
+                                  IEventAggregator eventAggregator,
                                   ICacheManager cacheManager)
         {
             _repository = repository;
             _rootFolderService = rootFolderService;
+            _eventAggregator = eventAggregator;
+
             _cache = cacheManager.GetCache<Dictionary<int, AutoTag>>(typeof(AutoTag), "autoTags");
         }
 
@@ -51,13 +56,17 @@ namespace NzbDrone.Core.AutoTagging
         public void Update(AutoTag autoTag)
         {
             _repository.Update(autoTag);
+
             _cache.Clear();
+            _eventAggregator.PublishEvent(new AutoTagsUpdatedEvent());
         }
 
         public AutoTag Insert(AutoTag autoTag)
         {
             var result = _repository.Insert(autoTag);
+
             _cache.Clear();
+            _eventAggregator.PublishEvent(new AutoTagsUpdatedEvent());
 
             return result;
         }
@@ -65,7 +74,9 @@ namespace NzbDrone.Core.AutoTagging
         public void Delete(int id)
         {
             _repository.Delete(id);
+
             _cache.Clear();
+            _eventAggregator.PublishEvent(new AutoTagsUpdatedEvent());
         }
 
         public List<AutoTag> AllForTag(int tagId)
diff --git a/src/NzbDrone.Core/AutoTagging/AutoTagsUpdatedEvent.cs b/src/NzbDrone.Core/AutoTagging/AutoTagsUpdatedEvent.cs
new file mode 100644
index 000000000..730e3423d
--- /dev/null
+++ b/src/NzbDrone.Core/AutoTagging/AutoTagsUpdatedEvent.cs
@@ -0,0 +1,8 @@
+using NzbDrone.Common.Messaging;
+
+namespace NzbDrone.Core.AutoTagging
+{
+    public class AutoTagsUpdatedEvent : IEvent
+    {
+    }
+}
diff --git a/src/Sonarr.Api.V3/Tags/TagController.cs b/src/Sonarr.Api.V3/Tags/TagController.cs
index 46da11ce9..c4880657b 100644
--- a/src/Sonarr.Api.V3/Tags/TagController.cs
+++ b/src/Sonarr.Api.V3/Tags/TagController.cs
@@ -1,5 +1,6 @@
 using System.Collections.Generic;
 using Microsoft.AspNetCore.Mvc;
+using NzbDrone.Core.AutoTagging;
 using NzbDrone.Core.Datastore.Events;
 using NzbDrone.Core.Messaging.Events;
 using NzbDrone.Core.Tags;
@@ -11,7 +12,9 @@ using Sonarr.Http.REST.Attributes;
 namespace Sonarr.Api.V3.Tags
 {
     [V3ApiController]
-    public class TagController : RestControllerWithSignalR<TagResource, Tag>, IHandle<TagsUpdatedEvent>
+    public class TagController : RestControllerWithSignalR<TagResource, Tag>,
+                                 IHandle<TagsUpdatedEvent>,
+                                 IHandle<AutoTagsUpdatedEvent>
     {
         private readonly ITagService _tagService;
 
@@ -60,5 +63,11 @@ namespace Sonarr.Api.V3.Tags
         {
             BroadcastResourceChange(ModelAction.Sync);
         }
+
+        [NonAction]
+        public void Handle(AutoTagsUpdatedEvent message)
+        {
+            BroadcastResourceChange(ModelAction.Sync);
+        }
     }
 }