Add option to set a tag instead of changing the category for unlinked downloads (#133)

code/Infrastructure/Verticals/DownloadCleaner/DownloadCleaner.cs

@@ -86,9 +86,12 @@ public sealed class DownloadCleaner : GenericHandler
         {
             if (!_hardLinkCategoryCreated)
             {
-                _logger.LogDebug("creating category {cat}", _config.UnlinkedTargetCategory);
-
-                await _downloadService.CreateCategoryAsync(_config.UnlinkedTargetCategory);
+                if (_downloadClientConfig.DownloadClient is Common.Enums.DownloadClient.QBittorrent && !_config.UnlinkedUseTag)
+                {
+                    _logger.LogDebug("creating category {cat}", _config.UnlinkedTargetCategory);
+                    await _downloadService.CreateCategoryAsync(_config.UnlinkedTargetCategory);
+                }
+                
                 _hardLinkCategoryCreated = true;
             }
             

code/Infrastructure/Verticals/DownloadClient/QBittorrent/QBitService.cs

@@ -251,6 +251,15 @@ public class QBitService : DownloadService, IQBitService
             ?.Cast<TorrentInfo>()
             .Where(x => !string.IsNullOrEmpty(x.Hash))
             .Where(x => categories.Any(cat => cat.Equals(x.Category, StringComparison.InvariantCultureIgnoreCase)))
+            .Where(x =>
+            {
+                if (_downloadCleanerConfig.UnlinkedUseTag)
+                {
+                    return !x.Tags.Any(tag => tag.Equals(_downloadCleanerConfig.UnlinkedTargetCategory, StringComparison.InvariantCultureIgnoreCase));
+                }
+
+                return true;
+            })
             .Cast<object>()
             .ToList();
 
@@ -436,12 +445,18 @@ public class QBitService : DownloadService, IQBitService
             }
             
             await _dryRunInterceptor.InterceptAsync(ChangeCategory, download.Hash, _downloadCleanerConfig.UnlinkedTargetCategory);
+
+            if (_downloadCleanerConfig.UnlinkedUseTag)
+            {
+                _logger.LogInformation("category changed for {name}", download.Name);
+                download.Category = _downloadCleanerConfig.UnlinkedTargetCategory;
+            }
+            else
+            {
+                _logger.LogInformation("tag added for {name}", download.Name);
+            }
             
-            _logger.LogInformation("category changed for {name}", download.Name);
-            
-            await _notifier.NotifyCategoryChanged(download.Category, _downloadCleanerConfig.UnlinkedTargetCategory);
-            
-            download.Category = _downloadCleanerConfig.UnlinkedTargetCategory;
+            await _notifier.NotifyCategoryChanged(download.Category, _downloadCleanerConfig.UnlinkedTargetCategory, _downloadCleanerConfig.UnlinkedUseTag);
         }
     }
     
@@ -467,6 +482,12 @@ public class QBitService : DownloadService, IQBitService
     [DryRunSafeguard]
     protected virtual async Task ChangeCategory(string hash, string newCategory)
     {
+        if (_downloadCleanerConfig.UnlinkedUseTag)
+        {
+            await _client.AddTorrentTagAsync([hash], newCategory);
+            return;
+        }
+        
         await _client.SetTorrentCategoryAsync([hash], newCategory);
     }
 

code/Infrastructure/Verticals/Notifications/INotificationPublisher.cs

@@ -10,5 +10,5 @@ public interface INotificationPublisher
     
     Task NotifyDownloadCleaned(double ratio, TimeSpan seedingTime, string categoryName, CleanReason reason);
 
-    Task NotifyCategoryChanged(string oldCategory, string newCategory);
+    Task NotifyCategoryChanged(string oldCategory, string newCategory, bool isTag = false);
 }

code/Infrastructure/Verticals/Notifications/NotificationPublisher.cs

@@ -123,21 +123,29 @@ public class NotificationPublisher : INotificationPublisher
         }
     }
     
-    public virtual async Task NotifyCategoryChanged(string oldCategory, string newCategory)
+    public virtual async Task NotifyCategoryChanged(string oldCategory, string newCategory, bool isTag = false)
     {
         CategoryChangedNotification notification = new()
         {
-            Title = "Category changed",
+            Title = isTag? "Tag added" : "Category changed",
             Description = ContextProvider.Get<string>("downloadName"),
             Fields =
             [
-                new() { Title = "Hash", Text = ContextProvider.Get<string>("hash").ToLowerInvariant() },
-                new() { Title = "Old category", Text = oldCategory },
-                new() { Title = "New category", Text = newCategory }
+                new() { Title = "Hash", Text = ContextProvider.Get<string>("hash").ToLowerInvariant() }
             ],
             Level = NotificationLevel.Important
         };
 
+        if (isTag)
+        {
+            notification.Fields.Add(new() { Title = "Tag", Text = newCategory });
+        }
+        else
+        {
+            notification.Fields.Add(new() { Title = "Old category", Text = oldCategory });
+            notification.Fields.Add(new() { Title = "New category", Text = newCategory });
+        }
+
         await NotifyInternal(notification);
     }