Wino-Mail/Wino.Mail.ViewModels/MergedAccountDetailsPageViewModel.cs

using System;
using System.Collections.ObjectModel;
using System.Linq;
using System.Threading.Tasks;
using CommunityToolkit.Mvvm.ComponentModel;
using CommunityToolkit.Mvvm.Input;
using CommunityToolkit.Mvvm.Messaging;
using Wino.Core.Domain;
using Wino.Core.Domain.Interfaces;
using Wino.Core.Domain.Models.Navigation;
using Wino.Mail.ViewModels.Data;
using Wino.Messaging.Client.Navigation;
using Wino.Messaging.UI;

namespace Wino.Mail.ViewModels
{
    public partial class MergedAccountDetailsPageViewModel : BaseViewModel,
        IRecipient<MergedInboxRenamed>
    {
        [ObservableProperty]
        private MergedAccountProviderDetailViewModel editingMergedAccount;

        [ObservableProperty]
        private string mergedAccountName;

        public ObservableCollection<AccountProviderDetailViewModel> LinkedAccounts { get; set; } = [];
        public ObservableCollection<AccountProviderDetailViewModel> UnlinkedAccounts { get; set; } = [];

        // Empty Guid is passed for new created merged inboxes.
        public bool IsMergedInboxSaved => EditingMergedAccount != null && EditingMergedAccount.MergedInbox.Id != Guid.Empty;

        public bool CanUnlink => IsMergedInboxSaved;

        // There must be at least 2 accounts linked to a merged account for link to exist.
        public bool ShouldDeleteMergedAccount => LinkedAccounts.Count < 2;

        public bool CanSaveChanges
        {
            get
            {
                if (IsMergedInboxSaved)
                {
                    return ShouldDeleteMergedAccount || IsEditingAccountsDirty();
                }
                else
                {
                    return LinkedAccounts.Any();
                }
            }
        }

        private readonly IAccountService _accountService;
        private readonly IPreferencesService _preferencesService;
        private readonly IProviderService _providerService;

        public MergedAccountDetailsPageViewModel(IDialogService dialogService,
                                                 IAccountService accountService,
                                                 IPreferencesService preferencesService,
                                                 IProviderService providerService) : base(dialogService)
        {
            _accountService = accountService;
            _preferencesService = preferencesService;
            _providerService = providerService;
        }

        [RelayCommand(CanExecute = nameof(CanUnlink))]
        private async Task UnlinkAccountsAsync()
        {
            if (EditingMergedAccount == null) return;

            var isConfirmed = await DialogService.ShowConfirmationDialogAsync(Translator.DialogMessage_UnlinkAccountsConfirmationMessage, Translator.DialogMessage_UnlinkAccountsConfirmationTitle, Translator.Buttons_Yes);

            if (!isConfirmed) return;

            await _accountService.UnlinkMergedInboxAsync(EditingMergedAccount.MergedInbox.Id);

            Messenger.Send(new BackBreadcrumNavigationRequested());
        }

        [RelayCommand(CanExecute = nameof(CanSaveChanges))]
        private async Task SaveChangesAsync()
        {
            if (ShouldDeleteMergedAccount)
            {
                await UnlinkAccountsAsync();
            }
            else
            {
                if (IsMergedInboxSaved)
                {
                    await _accountService.UpdateMergedInboxAsync(EditingMergedAccount.MergedInbox.Id, LinkedAccounts.Select(a => a.Account.Id).ToList());
                }
                else
                {
                    await _accountService.CreateMergeAccountsAsync(EditingMergedAccount.MergedInbox, LinkedAccounts.Select(a => a.Account).ToList());
                }

                // Startup entity is linked now. Change the startup entity.
                if (_preferencesService.StartupEntityId != null && LinkedAccounts.Any(a => a.StartupEntityId == _preferencesService.StartupEntityId))
                {
                    _preferencesService.StartupEntityId = EditingMergedAccount.MergedInbox.Id;
                }
            }

            Messenger.Send(new BackBreadcrumNavigationRequested());
        }

        [RelayCommand]
        private async Task RenameLinkAsync()
        {
            if (EditingMergedAccount == null) return;

            var newName = await DialogService.ShowTextInputDialogAsync(EditingMergedAccount.MergedInbox.Name,
                                                                       Translator.DialogMessage_RenameLinkedAccountsTitle,
                                                                       Translator.DialogMessage_RenameLinkedAccountsMessage,
                                                                       Translator.FolderOperation_Rename);

            if (string.IsNullOrWhiteSpace(newName)) return;

            EditingMergedAccount.MergedInbox.Name = newName;

            // Update database record as well.
            if (IsMergedInboxSaved)
            {
                await _accountService.RenameMergedAccountAsync(EditingMergedAccount.MergedInbox.Id, newName);
            }
            else
            {
                // Publish the message manually since the merged inbox is not saved yet.
                // This is only for breadcrump item update.

                Messenger.Send(new MergedInboxRenamed(EditingMergedAccount.MergedInbox.Id, newName));
            }
        }

        [RelayCommand]
        private void LinkAccount(AccountProviderDetailViewModel account)
        {
            LinkedAccounts.Add(account);
            UnlinkedAccounts.Remove(account);
        }

        [RelayCommand]
        private void UnlinkAccount(AccountProviderDetailViewModel account)
        {
            UnlinkedAccounts.Add(account);
            LinkedAccounts.Remove(account);
        }

        private bool IsEditingAccountsDirty()
        {
            if (EditingMergedAccount == null) return false;

            return EditingMergedAccount.HoldingAccounts.Count != LinkedAccounts.Count ||
                   EditingMergedAccount.HoldingAccounts.Any(a => !LinkedAccounts.Any(la => la.Account.Id == a.Account.Id));
        }

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

            LinkedAccounts.CollectionChanged -= LinkedAccountsUpdated;
        }

        private void LinkedAccountsUpdated(object sender, System.Collections.Specialized.NotifyCollectionChangedEventArgs e)
        {
            OnPropertyChanged(nameof(ShouldDeleteMergedAccount));
            SaveChangesCommand.NotifyCanExecuteChanged();

            // TODO: Preview common folders for all linked accounts.
            // Basically showing a preview of how menu items will look.
        }

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

            LinkedAccounts.CollectionChanged -= LinkedAccountsUpdated;
            LinkedAccounts.CollectionChanged += LinkedAccountsUpdated;

            if (parameters is MergedAccountProviderDetailViewModel editingMergedAccount)
            {
                MergedAccountName = editingMergedAccount.MergedInbox.Name;
                EditingMergedAccount = editingMergedAccount;

                foreach (var account in editingMergedAccount.HoldingAccounts)
                {
                    LinkedAccounts.Add(account);
                }

                // Load unlinked accounts.

                var allAccounts = await _accountService.GetAccountsAsync();

                foreach (var account in allAccounts)
                {
                    if (!LinkedAccounts.Any(a => a.Account.Id == account.Id))
                    {
                        var provider = _providerService.GetProviderDetail(account.ProviderType);

                        UnlinkedAccounts.Add(new AccountProviderDetailViewModel(provider, account));
                    }
                }
            }

            UnlinkAccountsCommand.NotifyCanExecuteChanged();
        }

        public void Receive(MergedInboxRenamed message)
        {
            if (EditingMergedAccount?.MergedInbox.Id == message.MergedInboxId)
            {
                EditingMergedAccount.MergedInbox.Name = message.NewName;
            }
        }
    }
}