recyclarr/src/Trash/Sonarr/ReleaseProfile/ReleaseProfileUpdater.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using CliFx.Exceptions;
using Serilog;
using Trash.Extensions;
using Trash.Sonarr.Api;
using Trash.Sonarr.Api.Objects;

namespace Trash.Sonarr.ReleaseProfile
{
    public class ReleaseProfileUpdater
    {
        private readonly ISonarrApi _api;
        private readonly IReleaseProfileGuideParser _parser;

        public ReleaseProfileUpdater(IReleaseProfileGuideParser parser, ISonarrApi api)
        {
            _parser = parser;
            _api = api;
        }

        private async Task DoVersionEnforcement()
        {
            // Since this script requires a specific version of v3 Sonarr that implements name support for
            // release profiles, we perform that version check here and bail out if it does not meet a minimum
            // required version.
            var minimumVersion = new Version("3.0.4.1098");
            var version = await _api.GetVersion();
            if (version < minimumVersion)
            {
                Log.Error("Your Sonarr version {CurrentVersion} does not meet the minimum " +
                          "required version of {MinimumVersion} to use this program", version, minimumVersion);
                throw new CommandException("Exiting due to version incompatibility");
            }
        }

        private async Task CreateMissingTags(ICollection<SonarrTag> sonarrTags, IEnumerable<string> configTags)
        {
            var missingTags = configTags.Where(t => !sonarrTags.Any(t2 => t2.Label.EqualsIgnoreCase(t)));
            foreach (var tag in missingTags)
            {
                Log.Debug("Creating Tag: {Tag}", tag);
                var newTag = await _api.CreateTag(tag);
                sonarrTags.Add(newTag);
            }
        }

        private string BuildProfileTitle(ReleaseProfileType profileType, string profileName)
        {
            var titleType = profileType.ToString();
            return $"[Trash] {titleType} - {profileName}";
        }

        private static SonarrReleaseProfile? GetProfileToUpdate(List<SonarrReleaseProfile> profiles, string profileName)
        {
            return profiles.FirstOrDefault(p => p.Name == profileName);
        }

        private static void SetupProfileRequestObject(SonarrReleaseProfile profileToUpdate, ProfileData profile,
            List<int> tagIds)
        {
            profileToUpdate.Preferred = profile.Preferred
                .SelectMany(kvp => kvp.Value.Select(term => new SonarrPreferredTerm(kvp.Key, term)))
                .ToList();

            profileToUpdate.Ignored = string.Join(',', profile.Ignored);
            profileToUpdate.Required = string.Join(',', profile.Required);

            // Null means the guide didn't specify a value for this, so we leave the existing setting intact.
            if (profile.IncludePreferredWhenRenaming != null)
            {
                profileToUpdate.IncludePreferredWhenRenaming = profile.IncludePreferredWhenRenaming.Value;
            }

            profileToUpdate.Tags = tagIds;
        }

        private async Task UpdateExistingProfile(SonarrReleaseProfile profileToUpdate, ProfileData profile,
            List<int> tagIds)
        {
            Log.Debug("Update existing profile with id {ProfileId}", profileToUpdate.Id);
            SetupProfileRequestObject(profileToUpdate, profile, tagIds);
            await _api.UpdateReleaseProfile(profileToUpdate);
        }

        private async Task CreateNewProfile(string title, ProfileData profile, List<int> tagIds)
        {
            var newProfile = new SonarrReleaseProfile
            {
                Name = title,
                Enabled = true
            };

            SetupProfileRequestObject(newProfile, profile, tagIds);
            await _api.CreateReleaseProfile(newProfile);
        }

        private async Task ProcessReleaseProfiles(IDictionary<string, ProfileData> profiles,
            ReleaseProfileConfig profile)
        {
            await DoVersionEnforcement();

            List<int> tagIds = new();

            // If tags were provided, ensure they exist. Tags that do not exist are added first, so that we
            // may specify them with the release profile request payload.
            if (profile.Tags.Count > 0)
            {
                var sonarrTags = await _api.GetTags();
                await CreateMissingTags(sonarrTags, profile.Tags);
                tagIds = sonarrTags.Where(t => profile.Tags.Any(ct => ct.EqualsIgnoreCase(t.Label)))
                    .Select(t => t.Id)
                    .ToList();
            }

            // Obtain all of the existing release profiles first. If any were previously created by our program
            // here, we favor replacing those instead of creating new ones, which would just be mostly duplicates
            // (but with some differences, since there have likely been updates since the last run).
            var existingProfiles = await _api.GetReleaseProfiles();

            foreach (var (name, profileData) in profiles)
            {
                var title = BuildProfileTitle(profile.Type, name);
                var profileToUpdate = GetProfileToUpdate(existingProfiles, title);
                if (profileToUpdate != null)
                {
                    Log.Information("Update existing profile: {ProfileName}", title);
                    await UpdateExistingProfile(profileToUpdate, profileData, tagIds);
                }
                else
                {
                    Log.Information("Create new profile: {ProfileName}", title);
                    await CreateNewProfile(title, profileData, tagIds);
                }
            }
        }

        public async Task Process(ISonarrCommand args, SonarrConfiguration config)
        {
            foreach (var profile in config.ReleaseProfiles)
            {
                Log.Information("Processing Release Profile: {ProfileName}", profile.Type);
                var markdown = await _parser.GetMarkdownData(profile.Type);

                var profiles = Utils.FilterProfiles(_parser.ParseMarkdown(profile, markdown));

                if (args.Preview)
                {
                    Utils.PrintTermsAndScores(profiles);
                    continue;
                }

                await ProcessReleaseProfiles(profiles, profile);
            }
        }
    }
}