Wino-Mail/Wino.Mail/Activation/ToastNotificationActivationHandler.cs

using System;
using System.Threading.Tasks;
using CommunityToolkit.Mvvm.Messaging;
using CommunityToolkit.WinUI.Notifications;
using Microsoft.Extensions.DependencyInjection;
using Serilog;
using Windows.ApplicationModel.Activation;
using Wino.Core.Domain;
using Wino.Core.Domain.Interfaces;
using Wino.Messaging.Client.Accounts;

namespace Wino.Activation
{
    /// <summary>
    /// This handler will only handle the toasts that runs on foreground.
    /// Background executions are not handled here like mark as read or delete.
    /// </summary>
    internal class ToastNotificationActivationHandler : ActivationHandler<ToastNotificationActivatedEventArgs>
    {
        private readonly IMailService _mailService;
        private readonly IFolderService _folderService;

        private ToastArguments _toastArguments;

        public ToastNotificationActivationHandler(IMailService mailService,
                                                  IFolderService folderService)
        {
            _mailService = mailService;
            _folderService = folderService;
        }

        protected override async Task HandleInternalAsync(ToastNotificationActivatedEventArgs args)
        {
            // Create the mail item navigation event.
            // If the app is running, it'll be picked up by the Messenger.
            // Otherwise we'll save it and handle it when the shell loads all accounts.

            // Parse the mail unique id and perform above actions.
            if (Guid.TryParse(_toastArguments[Constants.ToastMailUniqueIdKey], out Guid mailItemUniqueId))
            {
                var account = await _mailService.GetMailAccountByUniqueIdAsync(mailItemUniqueId).ConfigureAwait(false);
                if (account == null) return;

                var mailItem = await _mailService.GetSingleMailItemAsync(mailItemUniqueId).ConfigureAwait(false);
                if (mailItem == null) return;

                var message = new AccountMenuItemExtended(mailItem.AssignedFolder.Id, mailItem);

                // Delegate this event to LaunchProtocolService so app shell can pick it up on launch if app doesn't work.
                var launchProtocolService = App.Current.Services.GetService<ILaunchProtocolService>();
                launchProtocolService.LaunchParameter = message;

                // Send the messsage anyways. Launch protocol service will be ignored if the message is picked up by subscriber shell.
                WeakReferenceMessenger.Default.Send(message);
            }
        }

        protected override bool CanHandleInternal(ToastNotificationActivatedEventArgs args)
        {
            try
            {
                _toastArguments = ToastArguments.Parse(args.Argument);

                return
                    _toastArguments.Contains(Constants.ToastMailUniqueIdKey) &&
                    _toastArguments.Contains(Constants.ToastActionKey);
            }
            catch (Exception ex)
            {
                Log.Error(ex, "Couldn't handle parsing toast notification arguments for foreground navigate.");
            }

            return false;
        }
    }
}