Cleanuparr/code/Infrastructure/Verticals/Notifications/Notifiarr/NotifiarrProvider.cs

using Infrastructure.Verticals.Notifications.Models;
using Mapster;
using Microsoft.Extensions.Options;

namespace Infrastructure.Verticals.Notifications.Notifiarr;

public class NotifiarrProvider : NotificationProvider
{
    private readonly NotifiarrConfig _config;
    private readonly INotifiarrProxy _proxy;

    private const string WarningColor = "f0ad4e";
    private const string ImportantColor = "bb2124";
    private const string Logo = "https://github.com/flmorg/cleanuperr/blob/main/Logo/48.png?raw=true";
    
    public override string Name => "Notifiarr";

    public NotifiarrProvider(IOptions<NotifiarrConfig> config, INotifiarrProxy proxy)
        : base(config)
    {
        _config = config.Value;
        _proxy = proxy;
    }

    public override async Task OnFailedImportStrike(FailedImportStrikeNotification notification)
    {
        await _proxy.SendNotification(BuildPayload(notification, WarningColor), _config);
    }
    
    public override async Task OnStalledStrike(StalledStrikeNotification notification)
    {
        await _proxy.SendNotification(BuildPayload(notification, WarningColor), _config);
    }
    
    public override async Task OnSlowStrike(SlowStrikeNotification notification)
    {
        await _proxy.SendNotification(BuildPayload(notification, WarningColor), _config);
    }
    
    public override async Task OnQueueItemDeleted(QueueItemDeletedNotification notification)
    {
        await _proxy.SendNotification(BuildPayload(notification, ImportantColor), _config);
    }

    public override async Task OnDownloadCleaned(DownloadCleanedNotification notification)
    {
        await _proxy.SendNotification(BuildPayload(notification), _config);
    }
    
    public override async Task OnCategoryChanged(CategoryChangedNotification notification)
    {
        await _proxy.SendNotification(BuildPayload(notification), _config);
    }

    private NotifiarrPayload BuildPayload(ArrNotification notification, string color)
    {
        NotifiarrPayload payload = new()
        {
            Discord = new()
            {
                Color = color,
                Text = new()
                {
                    Title = notification.Title,
                    Icon = Logo,
                    Description = notification.Description,
                    Fields = new()
                    {
                        new() { Title = "Instance type", Text = notification.InstanceType.ToString() },
                        new() { Title = "Url", Text = notification.InstanceUrl.ToString() },
                        new() { Title = "Download hash", Text = notification.Hash }
                    }
                },
                Ids = new Ids
                {
                    Channel = _config.ChannelId
                },
                Images = new()
                {
                    Thumbnail = new Uri(Logo),
                    Image = notification.Image
                }
            }
        };
        
        payload.Discord.Text.Fields.AddRange(notification.Fields?.Adapt<List<Field>>() ?? []);

        return payload;
    }

    private NotifiarrPayload BuildPayload(DownloadCleanedNotification notification)
    {
        NotifiarrPayload payload = new()
        {
            Discord = new()
            {
                Color = ImportantColor,
                Text = new()
                {
                    Title = notification.Title,
                    Icon = Logo,
                    Description = notification.Description,
                    Fields = notification.Fields?.Adapt<List<Field>>() ?? []
                },
                Ids = new Ids
                {
                    Channel = _config.ChannelId
                },
                Images = new()
                {
                    Thumbnail = new Uri(Logo)
                }
            }
        };
        
        return payload;
    }

    private NotifiarrPayload BuildPayload(CategoryChangedNotification notification)
    {
        NotifiarrPayload payload = new()
        {
            Discord = new()
            {
                Color = WarningColor,
                Text = new()
                {
                    Title = notification.Title,
                    Icon = Logo,
                    Description = notification.Description,
                    Fields = notification.Fields?.Adapt<List<Field>>() ?? []
                },
                Ids = new Ids
                {
                    Channel = _config.ChannelId
                },
                Images = new()
                {
                    Thumbnail = new Uri(Logo)
                }
            }
        };
        
        return payload;
    }
}