using DigitalData.Core.Abstraction.Application.Repository;
using EnvelopeGenerator.Domain.Entities;
using MediatR;

namespace EnvelopeGenerator.Application.Common.Notifications.RemoveSignature.Handlers;

/// <summary>
/// 
/// </summary>
public class RemoveAnnotationHandler : INotificationHandler<RemoveSignatureNotification>
{
    private readonly IRepository<Annotation> _repo;

    /// <summary>
    /// 
    /// </summary>
    /// <param name="repository"></param>
    public RemoveAnnotationHandler(IRepository<Annotation> repository)
    {
        _repo = repository;
    }

    /// <summary>
    /// 
    /// </summary>
    /// <param name="notification"></param>
    /// <param name="cancel"></param>
    /// <returns></returns>
    public Task Handle(RemoveSignatureNotification notification, CancellationToken cancel)
    {
        return _repo.DeleteAsync(annots =>
        {
            annots = annots.Where(annot => annot.Element!.Document.Envelope!.Uuid == notification.EnvelopeUuid);

            if (notification.ReceiverSignature is string signature)
                annots = annots.Where(annot => annot.Element!.Receiver!.Signature == signature);

            return annots;
        }, cancel);
    }
}