recyclarr/src/Recyclarr.TrashLib/Services/CustomFormat/Processors/PersistenceProcessor.cs

using Recyclarr.TrashLib.Config.Services;
using Recyclarr.TrashLib.Services.CustomFormat.Api;
using Recyclarr.TrashLib.Services.CustomFormat.Models;
using Recyclarr.TrashLib.Services.CustomFormat.Models.Cache;
using Recyclarr.TrashLib.Services.CustomFormat.Processors.PersistenceSteps;

namespace Recyclarr.TrashLib.Services.CustomFormat.Processors;

public interface IPersistenceProcessorSteps
{
    public IJsonTransactionStep JsonTransactionStep { get; }
    public ICustomFormatApiPersistenceStep CustomFormatCustomFormatApiPersister { get; }
    public IQualityProfileApiPersistenceStep ProfileQualityProfileApiPersister { get; }
}

internal class PersistenceProcessor : IPersistenceProcessor
{
    private readonly ICustomFormatService _customFormatService;
    private readonly IPersistenceProcessorSteps _steps;

    public PersistenceProcessor(
        ICustomFormatService customFormatService,
        IPersistenceProcessorSteps steps)
    {
        _customFormatService = customFormatService;
        _steps = steps;
    }

    public CustomFormatTransactionData Transactions
        => _steps.JsonTransactionStep.Transactions;

    public IDictionary<string, List<UpdatedFormatScore>> UpdatedScores
        => _steps.ProfileQualityProfileApiPersister.UpdatedScores;

    public IReadOnlyCollection<string> InvalidProfileNames
        => _steps.ProfileQualityProfileApiPersister.InvalidProfileNames;

    public async Task PersistCustomFormats(
        IServiceConfiguration config,
        IReadOnlyCollection<ProcessedCustomFormatData> guideCfs,
        IEnumerable<TrashIdMapping> deletedCfsInCache,
        IDictionary<string, QualityProfileCustomFormatScoreMapping> profileScores)
    {
        var serviceCfs = await _customFormatService.GetCustomFormats(config);

        // Step 1: Match CFs between the guide & Radarr and merge the data. The goal is to retain as much of the
        // original data from Radarr as possible. There are many properties in the response JSON that we don't
        // directly care about. We keep those and just update the ones we do care about.
        _steps.JsonTransactionStep.Process(guideCfs, serviceCfs);

        // Step 1.1: Optionally record deletions of custom formats in cache but not in the guide
        if (config.DeleteOldCustomFormats)
        {
            _steps.JsonTransactionStep.RecordDeletions(deletedCfsInCache, serviceCfs);
        }

        // Step 2: For each merged CF, persist it to Radarr via its API. This will involve a combination of updates
        // to existing CFs and creation of brand new ones, depending on what's already available in Radarr.
        await _steps.CustomFormatCustomFormatApiPersister.Process(config, _steps.JsonTransactionStep.Transactions);

        // Step 3: Update all quality profiles with the scores from the guide for the uploaded custom formats
        await _steps.ProfileQualityProfileApiPersister.Process(config, profileScores);
    }
}