Cleanuparr/code/backend/Cleanuparr.Infrastructure/Features/DownloadClient/Transmission/TransmissionServiceCB.cs

using Cleanuparr.Domain.Enums;
using Cleanuparr.Infrastructure.Extensions;
using Cleanuparr.Infrastructure.Features.Context;
using Cleanuparr.Persistence.Models.Configuration.MalwareBlocker;
using Microsoft.Extensions.Logging;
using Transmission.API.RPC.Arguments;
using Transmission.API.RPC.Entity;

namespace Cleanuparr.Infrastructure.Features.DownloadClient.Transmission;

public partial class TransmissionService
{
    /// <inheritdoc/>
    public override async Task<BlockFilesResult> BlockUnwantedFilesAsync(string downloadId, IReadOnlyList<string> ignoredDownloads)
    {
        TorrentInfo? download = await GetTorrentAsync(downloadId);
        BlockFilesResult result = new();

        if (download?.Name is null)
        {
            _logger.LogDebug("Failed to find torrent {DownloadId} in the {Name} download client", downloadId, _downloadClientConfig.Name);
            return result;
        }

        if (ignoredDownloads.Count > 0 && download.ShouldIgnore(ignoredDownloads))
        {
            _logger.LogDebug("skip | download is ignored | {Name}", download.Name);
            return result;
        }

        bool isPrivate = download.IsPrivate ?? false;
        result.IsPrivate = isPrivate;
        result.Found = true;
        result.Item = new TransmissionItemWrapper(download);
        SetDownloadClientContext();

        var malwareBlockerConfig = ContextProvider.Get<ContentBlockerConfig>();
        
        if (malwareBlockerConfig.IgnorePrivate && isPrivate)
        {
            // ignore private trackers
            _logger.LogDebug("skip files check | download is private | {Name}", download.Name);
            return result;
        }

        if (download.Files is null || download.Files.Length == 0 || download.FileStats is null || download.FileStats.Length == 0)
        {
            _logger.LogDebug("skip files check | no files found | {Name}", download.Name);
            return result;
        }

        IEnumerable<(int Index, string ValidationName, string LogName, FileBlockAction Action)> BuildScanItems()
        {
            for (int i = 0; i < download.Files.Length; i++)
            {
                if (download.FileStats?[i].Wanted == null)
                {
                    _logger.LogTrace("Skipping file with no stats | {File}", download.Files[i].Name);
                    continue;
                }

                yield return (i, download.Files[i].Name, download.Files[i].Name, download.FileStats[i].Wanted!.Value
                    ? FileBlockAction.CheckBlocklist
                    : FileBlockAction.AlreadySkipped);
            }
        }

        await ApplyFileBlockingAsync(result, download.Name, BuildScanItems(), malwareBlockerConfig.DeleteIfAnyFileBlocked, async unwantedIndices =>
        {
            await _client.TorrentSetAsync(new TorrentSettings
            {
                Ids = [download.Id],
                FilesUnwanted = unwantedIndices.Select(i => (long)i).ToArray(),
            });
        });

        return result;
    }
}