Wino-Mail/Wino.Mail.ViewModels/AliasManagementPageViewModel.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Security.Cryptography.X509Certificates;
using System.Threading.Tasks;
using CommunityToolkit.Mvvm.ComponentModel;
using CommunityToolkit.Mvvm.Input;
using EmailValidation;
using Wino.Core.Domain;
using Wino.Core.Domain.Entities.Mail;
using Wino.Core.Domain.Entities.Shared;
using Wino.Core.Domain.Enums;
using Wino.Core.Domain.Interfaces;
using Wino.Core.Domain.Models.Navigation;
using Wino.Core.Domain.Models.Synchronization;
using Wino.Core.Services;

namespace Wino.Mail.ViewModels;

public partial class AliasManagementPageViewModel : MailBaseViewModel
{
    private readonly IMailDialogService _dialogService;
    private readonly IAccountService _accountService;
    private readonly ISmimeCertificateService _smimeCertificateService;

    [ObservableProperty]
    [NotifyPropertyChangedFor(nameof(CanSynchronizeAliases))]
    private MailAccount account;

    [ObservableProperty]
    private List<MailAccountAlias> accountAliases = [];

    public bool CanSynchronizeAliases => Account?.IsAliasSyncSupported ?? false;

    public AliasManagementPageViewModel(IMailDialogService dialogService,
                                        IAccountService accountService,
                                        ISmimeCertificateService smimeCertificateService)
    {
        _dialogService = dialogService;
        _accountService = accountService;
        _smimeCertificateService = smimeCertificateService;
    }

    public override async void OnNavigatedTo(NavigationMode mode, object parameters)
    {
        base.OnNavigatedTo(mode, parameters);

        if (parameters is Guid accountId)
            Account = await _accountService.GetAccountAsync(accountId);

        if (Account == null) return;

        await LoadAliasesAsync();
    }

    private async Task LoadAliasesAsync()
    {
        var aliases = await _accountService.GetAccountAliasesAsync(Account.Id);
        foreach (var alias in aliases)
        {
            alias.Certificates.Clear();
            alias.Certificates.Add(null); // First blank optioon
            var certs = _smimeCertificateService.GetCertificates()
                .Where(cert => cert.Subject.Contains(alias.AliasAddress, StringComparison.OrdinalIgnoreCase))
                .ToList();
            foreach (var cert in certs)
                alias.Certificates.Add(cert);

            alias.SelectedSigningCertificate = !string.IsNullOrEmpty(alias.SelectedSigningCertificateThumbprint)
                ? alias.Certificates.FirstOrDefault(c => c?.Thumbprint == alias.SelectedSigningCertificateThumbprint)
                : null;
        }
        AccountAliases = aliases;
    }

    [RelayCommand]
    private async Task SetAliasPrimaryAsync(MailAccountAlias alias)
    {
        if (alias.IsPrimary) return;

        AccountAliases.ForEach(a =>
        {
            a.IsPrimary = a == alias;
        });

        await _accountService.UpdateAccountAliasesAsync(Account.Id, AccountAliases);
        await LoadAliasesAsync();
    }

    [RelayCommand]
    private async Task SyncAliasesAsync()
    {
        if (!CanSynchronizeAliases) return;

        var aliasSyncOptions = new MailSynchronizationOptions()
        {
            AccountId = Account.Id,
            Type = MailSynchronizationType.Alias
        };

        var aliasSyncResult = await SynchronizationManager.Instance.SynchronizeAliasesAsync(Account.Id);

        if (aliasSyncResult.CompletedState == SynchronizationCompletedState.Success)
            await LoadAliasesAsync();
        else
            _dialogService.InfoBarMessage(Translator.GeneralTitle_Error, "Failed to synchronize aliases", InfoBarMessageType.Error);
    }

    [RelayCommand]
    private async Task AddNewAliasAsync()
    {
        var createdAliasDialog = await _dialogService.ShowCreateAccountAliasDialogAsync();

        if (createdAliasDialog.CreatedAccountAlias == null) return;

        var newAlias = createdAliasDialog.CreatedAccountAlias;

        // Check existence.
        if (AccountAliases.Any(a => a.AliasAddress == newAlias.AliasAddress))
        {
            await _dialogService.ShowMessageAsync(Translator.DialogMessage_AliasExistsTitle,
                                                 Translator.DialogMessage_AliasExistsMessage,
                                                 WinoCustomMessageDialogIcon.Warning);
            return;
        }

        // Validate all addresses.
        if (!EmailValidator.Validate(newAlias.AliasAddress) || (!string.IsNullOrEmpty(newAlias.ReplyToAddress) && !EmailValidator.Validate(newAlias.ReplyToAddress)))
        {
            await _dialogService.ShowMessageAsync(Translator.DialogMessage_InvalidAliasMessage,
                                                 Translator.DialogMessage_InvalidAliasTitle,
                                                 WinoCustomMessageDialogIcon.Warning);
            return;
        }

        newAlias.AccountId = Account.Id;

        AccountAliases.Add(newAlias);

        await _accountService.UpdateAccountAliasesAsync(Account.Id, AccountAliases);
        _dialogService.InfoBarMessage(Translator.DialogMessage_AliasCreatedTitle, Translator.DialogMessage_AliasCreatedMessage, InfoBarMessageType.Success);

        await LoadAliasesAsync();
    }

    [RelayCommand]
    private async Task DeleteAliasAsync(MailAccountAlias alias)
    {
        // Primary aliases can't be deleted.
        if (alias.IsPrimary)
        {
            await _dialogService.ShowMessageAsync(Translator.Info_CantDeletePrimaryAliasMessage,
                                                 Translator.GeneralTitle_Warning,
                                                 WinoCustomMessageDialogIcon.Warning);
            return;
        }

        // Root aliases can't be deleted.
        if (alias.IsRootAlias)
        {
            await _dialogService.ShowMessageAsync(Translator.DialogMessage_CantDeleteRootAliasTitle,
                                                 Translator.DialogMessage_CantDeleteRootAliasMessage,
                                                 WinoCustomMessageDialogIcon.Warning);
            return;
        }

        await _accountService.DeleteAccountAliasAsync(alias.Id);
        await LoadAliasesAsync();
    }

    public async Task SetAliasSmimeEncryption(MailAccountAlias alias, bool value)
    {
        alias.IsSmimeEncryptionEnabled = value;
        await _accountService.UpdateAccountAliasesAsync(Account.Id, AccountAliases);
        await LoadAliasesAsync();
    }

    public async Task SetSelectedSigningCertificate(MailAccountAlias alias, X509Certificate2 cert)
    {
        alias.SelectedSigningCertificate = cert;
        alias.SelectedSigningCertificateThumbprint = cert?.Thumbprint;

        await _accountService.UpdateAccountAliasesAsync(Account.Id, AccountAliases);
        await LoadAliasesAsync();
    }
}