romrepomgr/RomRepoMgr/ViewModels/MainWindowViewModel.cs

/******************************************************************************
// RomRepoMgr - ROM repository manager
// ----------------------------------------------------------------------------
//
// Author(s)      : Natalia Portillo <claunia@claunia.com>
//
// --[ License ] --------------------------------------------------------------
//
//     This program is free software: you can redistribute it and/or modify
//     it under the terms of the GNU General Public License as
//     published by the Free Software Foundation, either version 3 of the
//     License, or (at your option) any later version.
//
//     This program is distributed in the hope that it will be useful,
//     but WITHOUT ANY WARRANTY; without even the implied warranty of
//     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//     GNU General Public License for more details.
//
//     You should have received a copy of the GNU General Public License
//     along with this program.  If not, see <http://www.gnu.org/licenses/>.
//
// ----------------------------------------------------------------------------
// Copyright © 2020 Natalia Portillo
*******************************************************************************/

using System.Collections.Generic;
using System.Collections.ObjectModel;
using System.Linq;
using System.Reactive;
using Avalonia;
using Avalonia.Controls;
using Avalonia.Controls.ApplicationLifetimes;
using Avalonia.Threading;
using MessageBox.Avalonia;
using MessageBox.Avalonia.Enums;
using ReactiveUI;
using RomRepoMgr.Core.EventArgs;
using RomRepoMgr.Core.Models;
using RomRepoMgr.Views;

namespace RomRepoMgr.ViewModels
{
    public class MainWindowViewModel : ViewModelBase
    {
        readonly MainWindow _view;

        RomSetModel _selectedRomSet;

        public MainWindowViewModel(MainWindow view, List<RomSetModel> romSets)
        {
            _view                  = view;
            ExitCommand            = ReactiveCommand.Create(ExecuteExitCommand);
            SettingsCommand        = ReactiveCommand.Create(ExecuteSettingsCommand);
            AboutCommand           = ReactiveCommand.Create(ExecuteAboutCommand);
            ImportDatCommand       = ReactiveCommand.Create(ExecuteImportDatCommand);
            ImportDatFolderCommand = ReactiveCommand.Create(ExecuteImportDatFolderCommand);
            ImportRomFolderCommand = ReactiveCommand.Create(ExecuteImportRomFolderCommand);
            DeleteRomSetCommand    = ReactiveCommand.Create(ExecuteDeleteRomSetCommand);
            EditRomSetCommand      = ReactiveCommand.Create(ExecuteEditRomSetCommand);
            ExportDatCommand       = ReactiveCommand.Create(ExecuteExportDatCommand);
            RomSets                = new ObservableCollection<RomSetModel>(romSets);
        }

        public ObservableCollection<RomSetModel> RomSets                       { get; }
        public string                            RomSetLabel                   => "ROM sets";
        public string                            RomSetNameLabel               => "Name";
        public string                            RomSetVersionLabel            => "Version";
        public string                            RomSetAuthorLabel             => "Author";
        public string                            RomSetDateLabel               => "Date";
        public string                            RomSetDescriptionLabel        => "Description";
        public string                            RomSetCommentLabel            => "Comment";
        public string                            RomSetTotalMachinesLabel      => "Games";
        public string                            RomSetCompleteMachinesLabel   => "Complete";
        public string                            RomSetIncompleteMachinesLabel => "Incomplete";
        public string                            RomSetTotalRomsLabel          => "ROMs";
        public string                            RomSetHaveRomsLabel           => "Have";
        public string                            RomSetMissRomsLabel           => "Miss";

        public bool NativeMenuSupported =>
            NativeMenu.GetIsNativeMenuExported((Application.Current.ApplicationLifetime as
                                                    IClassicDesktopStyleApplicationLifetime)?.MainWindow);

        public ReactiveCommand<Unit, Unit> AboutCommand           { get; }
        public ReactiveCommand<Unit, Unit> ExitCommand            { get; }
        public ReactiveCommand<Unit, Unit> SettingsCommand        { get; }
        public ReactiveCommand<Unit, Unit> ImportDatCommand       { get; }
        public ReactiveCommand<Unit, Unit> ImportDatFolderCommand { get; }
        public ReactiveCommand<Unit, Unit> ImportRomFolderCommand { get; }
        public ReactiveCommand<Unit, Unit> DeleteRomSetCommand    { get; }
        public ReactiveCommand<Unit, Unit> EditRomSetCommand      { get; }
        public ReactiveCommand<Unit, Unit> ExportDatCommand       { get; }

        public RomSetModel SelectedRomSet
        {
            get => _selectedRomSet;
            set => this.RaiseAndSetIfChanged(ref _selectedRomSet, value);
        }

        internal async void ExecuteSettingsCommand()
        {
            var dialog = new SettingsDialog();
            dialog.DataContext = new SettingsViewModel(dialog);
            await dialog.ShowDialog(_view);
        }

        internal void ExecuteExitCommand() =>
            (Application.Current.ApplicationLifetime as ClassicDesktopStyleApplicationLifetime)?.Shutdown();

        internal void ExecuteAboutCommand()
        {
            var dialog = new About();
            dialog.DataContext = new AboutViewModel(dialog);
            dialog.ShowDialog(_view);
        }

        internal async void ExecuteImportDatCommand()
        {
            var dlgOpen = new OpenFileDialog
            {
                AllowMultiple = false,
                Title         = "Import DAT file..."
            };

            dlgOpen.Filters.Add(new FileDialogFilter
            {
                Extensions = new List<string>
                {
                    "*.dat",
                    "*.xml"
                },
                Name = "DAT files"
            });

            dlgOpen.Filters.Add(new FileDialogFilter
            {
                Extensions = new List<string>
                {
                    "*.*"
                },
                Name = "All files"
            });

            string[] result = await dlgOpen.ShowAsync(_view);

            if(result?.Length != 1)
                return;

            var dialog             = new ImportDat();
            var importDatViewModel = new ImportDatViewModel(dialog, result[0]);
            importDatViewModel.RomSetAdded += ImportDatViewModelOnRomSetAdded;
            dialog.DataContext             =  importDatViewModel;
            await dialog.ShowDialog(_view);
        }

        void ImportDatViewModelOnRomSetAdded(object sender, RomSetEventArgs e) => Dispatcher.UIThread.Post(() =>
        {
            RomSets.Add(e.RomSet);
        });

        internal async void ExecuteImportDatFolderCommand()
        {
            var dlgOpen = new OpenFolderDialog
            {
                Title = "Import DATs from folder..."
            };

            string result = await dlgOpen.ShowAsync(_view);

            if(result == null)
                return;

            var dialog                   = new ImportDatFolder();
            var importDatFolderViewModel = new ImportDatFolderViewModel(dialog, result);
            importDatFolderViewModel.RomSetAdded += ImportDatViewModelOnRomSetAdded;
            dialog.DataContext                   =  importDatFolderViewModel;
            await dialog.ShowDialog(_view);
        }

        internal async void ExecuteImportRomFolderCommand()
        {
            var dlgOpen = new OpenFolderDialog
            {
                Title = "Import ROMs from folder..."
            };

            string result = await dlgOpen.ShowAsync(_view);

            if(result == null)
                return;

            var dialog                   = new ImportRomFolder();
            var importRomFolderViewModel = new ImportRomFolderViewModel(dialog, result);
            dialog.DataContext = importRomFolderViewModel;
            await dialog.ShowDialog(_view);
        }

        async void ExecuteDeleteRomSetCommand()
        {
            if(SelectedRomSet == null)
                return;

            ButtonResult result = await MessageBoxManager.GetMessageBoxStandardWindow("Delete ROM set",
                                                              string.
                                                                  Format("Are you sure you want to delete the ROM set {0}?",
                                                                         SelectedRomSet.Name), ButtonEnum.YesNo,
                                                              Icon.Database).
                                                          ShowDialog(_view);

            if(result == ButtonResult.No)
                return;

            var dialog             = new RemoveDat();
            var removeDatViewModel = new RemoveDatViewModel(dialog, SelectedRomSet.Id);
            dialog.DataContext = removeDatViewModel;
            await dialog.ShowDialog(_view);

            RomSets.Remove(SelectedRomSet);
            SelectedRomSet = null;
        }

        void ExecuteEditRomSetCommand()
        {
            if(SelectedRomSet == null)
                return;

            var window    = new EditDat();
            var viewModel = new EditDatViewModel(window, SelectedRomSet);

            viewModel.RomSetModified += (sender, args) =>
            {
                RomSetModel old = RomSets.FirstOrDefault(r => r.Id == args.RomSet.Id);

                if(old == null)
                    return;

                RomSets.Remove(old);
                RomSets.Add(args.RomSet);
            };

            window.DataContext = viewModel;

            window.Show();
        }

        async void ExecuteExportDatCommand()
        {
            if(SelectedRomSet == null)
                return;

            var dlgSave = new SaveFileDialog
            {
                InitialFileName = SelectedRomSet.Filename
            };

            string result = await dlgSave.ShowAsync(_view);

            if(result == null)
                return;

            var dialog    = new ExportDat();
            var viewModel = new ExportDatViewModel(dialog, SelectedRomSet.Sha384, result);
            dialog.DataContext = viewModel;
            await dialog.ShowDialog(_view);
        }
    }
}