recyclarr/src/Recyclarr/Command/SonarrCommand.cs

using CliFx.Attributes;
using CliFx.Exceptions;
using JetBrains.Annotations;
using Recyclarr.Config;
using Serilog;
using TrashLib.Extensions;
using TrashLib.Services.CustomFormat;
using TrashLib.Services.Sonarr;
using TrashLib.Services.Sonarr.Config;
using TrashLib.Services.Sonarr.QualityDefinition;
using TrashLib.Services.Sonarr.ReleaseProfile;
using TrashLib.Services.Sonarr.ReleaseProfile.Guide;

namespace Recyclarr.Command;

[Command("sonarr", Description = "Perform operations on a Sonarr instance")]
[UsedImplicitly]
public class SonarrCommand : ServiceCommand
{
    [CommandOption("list-release-profiles", Description =
        "List available release profiles from the guide in YAML format.")]
    public bool ListReleaseProfiles { get; [UsedImplicitly] set; }

    // The default value is "empty" because I need to know when the user specifies the option but no value with it.
    // Discussed here: https://github.com/Tyrrrz/CliFx/discussions/128#discussioncomment-2647015
    [CommandOption("list-terms", Description =
        "For the given Release Profile Trash ID, list terms in it that can be filtered in YAML format. " +
        "Note that not every release profile has terms that may be filtered.")]
    public string? ListTerms { get; [UsedImplicitly] set; } = "empty";

    [CommandOption("list-qualities", Description =
        "List available quality definition types from the guide.")]
    public bool ListQualities { get; [UsedImplicitly] set; }

    [CommandOption("list-custom-formats", Description =
        "List available custom formats from the guide in YAML format.")]
    public bool ListCustomFormats { get; [UsedImplicitly] set; }

    public override string Name => "Sonarr";

    public override async Task Process(IServiceLocatorProxy container)
    {
        await base.Process(container);

        var lister = container.Resolve<ISonarrGuideDataLister>();
        var profileUpdaterFactory = container.Resolve<Func<IReleaseProfileUpdater>>();
        var qualityUpdaterFactory = container.Resolve<Func<ISonarrQualityDefinitionUpdater>>();
        var configLoader = container.Resolve<IConfigurationLoader<SonarrConfiguration>>();
        var log = container.Resolve<ILogger>();
        var customFormatUpdaterFactory = container.Resolve<Func<ICustomFormatUpdater>>();
        var guideService = container.Resolve<ISonarrGuideService>();
        var versionEnforcement = container.Resolve<ISonarrVersionEnforcement>();

        if (ListReleaseProfiles)
        {
            lister.ListReleaseProfiles();
            return;
        }

        if (ListQualities)
        {
            lister.ListQualities();
            return;
        }

        if (ListCustomFormats)
        {
            lister.ListCustomFormats();
            return;
        }

        if (ListTerms != "empty")
        {
            if (!string.IsNullOrEmpty(ListTerms))
            {
                lister.ListTerms(ListTerms);
            }
            else
            {
                throw new CommandException(
                    "The --list-terms option was specified without a Release Profile Trash ID specified");
            }

            return;
        }

        foreach (var config in configLoader.LoadMany(Config, "sonarr"))
        {
            log.Information("Processing server {Url}", FlurlLogging.SanitizeUrl(config.BaseUrl));

            await versionEnforcement.DoVersionEnforcement(config);

            if (config.ReleaseProfiles.Count > 0)
            {
                await profileUpdaterFactory().Process(Preview, config);
            }

            if (!string.IsNullOrEmpty(config.QualityDefinition))
            {
                await qualityUpdaterFactory().Process(Preview, config);
            }

            if (config.CustomFormats.Count > 0)
            {
                await customFormatUpdaterFactory().Process(Preview, config.CustomFormats, guideService);
            }
        }
    }
}