recyclarr/src/Recyclarr.Compatibility/ServiceAgnosticCapabilityEnforcer.cs

using Recyclarr.Compatibility.Radarr;
using Recyclarr.Compatibility.Sonarr;
using Recyclarr.Config.Models;

namespace Recyclarr.Compatibility;

public class ServiceAgnosticCapabilityEnforcer
{
    private readonly SonarrCapabilityEnforcer _sonarrEnforcer;
    private readonly RadarrCapabilityEnforcer _radarrEnforcer;

    public ServiceAgnosticCapabilityEnforcer(
        SonarrCapabilityEnforcer sonarrEnforcer,
        RadarrCapabilityEnforcer radarrEnforcer)
    {
        _sonarrEnforcer = sonarrEnforcer;
        _radarrEnforcer = radarrEnforcer;
    }

    public async Task Check(IServiceConfiguration config)
    {
        switch (config)
        {
            case SonarrConfiguration c:
                await _sonarrEnforcer.Check(c);
                break;

            case RadarrConfiguration c:
                await _radarrEnforcer.Check(c);
                break;
        }
    }
}