Add Lidarr support (#30)

2025-01-15 23:55:34 +02:00
parent 2bc8e445ce
commit 922f586706
63 changed files with 943 additions and 243 deletions
@@ -1,12 +1,17 @@
-using Common.Configuration.Arr;
+using System.Collections.Concurrent;
+using System.Text.RegularExpressions;
+using Common.Configuration.Arr;
+using Common.Configuration.ContentBlocker;
 using Common.Configuration.DownloadClient;
 using Domain.Enums;
+using Domain.Models.Arr;
 using Domain.Models.Arr.Queue;
 using Infrastructure.Verticals.Arr;
 using Infrastructure.Verticals.DownloadClient;
 using Infrastructure.Verticals.Jobs;
 using Microsoft.Extensions.Logging;
 using Microsoft.Extensions.Options;
+using Serilog.Context;

 namespace Infrastructure.Verticals.ContentBlocker;

@@ -19,12 +24,19 @@ public sealed class ContentBlocker : GenericHandler
        IOptions<DownloadClientConfig> downloadClientConfig,
        IOptions<SonarrConfig> sonarrConfig,
        IOptions<RadarrConfig> radarrConfig,
+        IOptions<LidarrConfig> lidarrConfig,
        SonarrClient sonarrClient,
        RadarrClient radarrClient,
+        LidarrClient lidarrClient,
        ArrQueueIterator arrArrQueueIterator,
        BlocklistProvider blocklistProvider,
        DownloadServiceFactory downloadServiceFactory
-    ) : base(logger, downloadClientConfig, sonarrConfig.Value, radarrConfig.Value, sonarrClient, radarrClient, arrArrQueueIterator, downloadServiceFactory)
+    ) : base(
+        logger, downloadClientConfig,
+        sonarrConfig, radarrConfig, lidarrConfig,
+        sonarrClient, radarrClient, lidarrClient,
+        arrArrQueueIterator, downloadServiceFactory
+    )
    {
        _blocklistProvider = blocklistProvider;
    }
@@ -37,18 +49,40 @@ public sealed class ContentBlocker : GenericHandler
            return;
        }
        
-        await _blocklistProvider.LoadBlocklistAsync();
+        bool blocklistIsConfigured = _sonarrConfig.Enabled && !string.IsNullOrEmpty(_sonarrConfig.Block.Path) ||
+                                     _radarrConfig.Enabled && !string.IsNullOrEmpty(_radarrConfig.Block.Path) ||
+                                     _lidarrConfig.Enabled && !string.IsNullOrEmpty(_lidarrConfig.Block.Path);
+
+        if (!blocklistIsConfigured)
+        {
+            _logger.LogWarning("no blocklist is configured");
+            return;
+        }
+        
+        await _blocklistProvider.LoadBlocklistsAsync();
        await base.ExecuteAsync();
    }

    protected override async Task ProcessInstanceAsync(ArrInstance instance, InstanceType instanceType)
    {
+        using var _ = LogContext.PushProperty("InstanceName", instanceType.ToString());
+
+        HashSet<SearchItem> itemsToBeRefreshed = [];
        ArrClient arrClient = GetClient(instanceType);
+        BlocklistType blocklistType = _blocklistProvider.GetBlocklistType(instanceType);
+        ConcurrentBag<string> patterns = _blocklistProvider.GetPatterns(instanceType);
+        ConcurrentBag<Regex> regexes = _blocklistProvider.GetRegexes(instanceType);

        await _arrArrQueueIterator.Iterate(arrClient, instance, async items =>
        {
-            foreach (QueueRecord record in items)
+            var groups = items
+                .GroupBy(x => x.DownloadId)
+                .ToList();
+            
+            foreach (var group in groups)
            {
+                QueueRecord record = group.First();
+                
                if (record.Protocol is not "torrent")
                {
                    continue;
@@ -61,8 +95,19 @@ public sealed class ContentBlocker : GenericHandler
                }
                
                _logger.LogDebug("searching unwanted files for {title}", record.Title);
-                await _downloadService.BlockUnwantedFilesAsync(record.DownloadId);
+
+                if (!await _downloadService.BlockUnwantedFilesAsync(record.DownloadId, blocklistType, patterns, regexes))
+                {
+                    continue;
+                }
+                
+                _logger.LogDebug("all files are marked as unwanted | {hash}", record.Title);
+                
+                itemsToBeRefreshed.Add(GetRecordSearchItem(instanceType, record, group.Count() > 1));
+                await arrClient.DeleteQueueItemAsync(instance, record);
            }
        });
+        
+        await arrClient.RefreshItemsAsync(instance, itemsToBeRefreshed);
    }
 }