add content blocker (#5)

* refactored code added deluge support added transmission support added content blocker added blacklist and whitelist * increased level on some logs; updated test docker compose; updated dev appsettings * updated docker compose and readme * moved some logs * fixed env var typo; fixed sonarr and radarr default download client
2024-11-18 20:08:01 +02:00
parent b323cb40ae
commit e0a6c7842b
154 changed files with 4752 additions and 789 deletions
@@ -0,0 +1,95 @@
+using Common.Configuration;
+using Infrastructure.Verticals.ContentBlocker;
+using Microsoft.Extensions.Logging;
+using Microsoft.Extensions.Options;
+using QBittorrent.Client;
+
+namespace Infrastructure.Verticals.DownloadClient.QBittorrent;
+
+public sealed class QBitService : IDownloadService
+{
+    private readonly ILogger<QBitService> _logger;
+    private readonly QBitConfig _config;
+    private readonly QBittorrentClient _client;
+    private readonly FilenameEvaluator _filenameEvaluator;
+
+    public QBitService(
+        ILogger<QBitService> logger,
+        IOptions<QBitConfig> config,
+        FilenameEvaluator filenameEvaluator
+    )
+    {
+        _logger = logger;
+        _config = config.Value;
+        _client = new(_config.Url);
+        _filenameEvaluator = filenameEvaluator;
+    }
+
+    public async Task LoginAsync()
+    {
+        await _client.LoginAsync(_config.Username, _config.Password);
+    }
+
+    public async Task<bool> ShouldRemoveFromArrQueueAsync(string hash)
+    {
+        TorrentInfo? torrent = (await _client.GetTorrentListAsync(new TorrentListQuery { Hashes = [hash] }))
+            .FirstOrDefault();
+
+        if (torrent is null)
+        {
+            return false;
+        }
+
+        // if all files were blocked by qBittorrent
+        if (torrent is { CompletionOn: not null, Downloaded: null or 0 })
+        {
+            return true;
+        }
+
+        IReadOnlyList<TorrentContent>? files = await _client.GetTorrentContentsAsync(hash);
+
+        if (files is null)
+        {
+            return false;
+        }
+        
+        // if all files are marked as skip
+        if (files.All(x => x.Priority is TorrentContentPriority.Skip))
+        {
+            return true;
+        }
+
+        return false;
+    }
+
+    public async Task BlockUnwantedFilesAsync(string hash)
+    {
+        IReadOnlyList<TorrentContent>? files = await _client.GetTorrentContentsAsync(hash);
+
+        if (files is null)
+        {
+            return;
+        }
+        
+        foreach (TorrentContent file in files)
+        {
+            if (!file.Index.HasValue)
+            {
+                continue;
+            }
+
+            if (file.Priority is TorrentContentPriority.Skip || _filenameEvaluator.IsValid(file.Name))
+            {
+                continue;
+            }
+            
+            _logger.LogInformation("unwanted file found | {file}", file.Name);
+            await _client.SetFilePriorityAsync(hash, file.Index.Value, TorrentContentPriority.Skip);
+        }
+    }
+
+    public void Dispose()
+    {
+        _client.Dispose();
+    }
+}