path: root/Software/Visual_Studio/MachineStudio/Modules/Tango.MachineStudio.Synchronization/ViewModels/LocalSynchronizationViewVM.cs

using Microsoft.Win32;
using System;
using System.Collections.Generic;
using System.Collections.ObjectModel;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using Tango.Core.Commands;
using Tango.Logging;
using Tango.MachineStudio.Common.Notifications;
using Tango.MachineStudio.Synchronization.Navigation;
using Tango.MachineStudio.Synchronization.Properties;
using Tango.Settings;
using Tango.SharedUI;
using Tango.Synchronization;
using Tango.Synchronization.Local;

namespace Tango.MachineStudio.Synchronization.ViewModels
{
    /// <summary>
    /// Represents the 'Local Synchronization' view model.
    /// </summary>
    /// <seealso cref="Tango.SharedUI.ViewModel" />
    public class LocalSynchronizationViewVM : ViewModel
    {
        private SyncNavigationManager _navigation;
        private String _masterDBFile;
        private String _slaveDBFile;
        private LocalDBComparer _comparer;
        private INotificationProvider _notification;
        private bool _isWorking;

        #region Constructors

        /// <summary>
        /// Initializes a new instance of the <see cref="LocalSynchronizationViewVM"/> class.
        /// </summary>
        /// <param name="navigation">The navigation.</param>
        /// <param name="notification">The notification.</param>
        public LocalSynchronizationViewVM(SyncNavigationManager navigation, INotificationProvider notification)
        {
            _navigation = navigation;
            _notification = notification;

            BackCommand = new RelayCommand(() => _navigation.NavigateTo(NavigationView.MenuView));

            Differences = new ObservableCollection<Diff>();

            BrowseMasterDBCommand = new RelayCommand(BrowseMasterDB, (x) => !_isWorking);
            BrowseSlaveDBCommand = new RelayCommand(BrowseSlaveDB, (x) => !_isWorking);
            CompareCommand = new RelayCommand(Compare, (x) => MasterDBFile != null && SlaveDBFile != null && !_isWorking);
            CommitCommand = new RelayCommand(Commit, (x) => SelectedDifference != null && !_isWorking);
            CommitAllCommand = new RelayCommand(CommitAll, (x) => Differences.Count > 0 && !_isWorking);
            CleanCommand = new RelayCommand(CleanSlave, (x) => !_isWorking && SlaveDBFile != null);

            if (File.Exists(SettingsManager.Default.MachineStudio.SynchronizationModule.LocalMasterDBFile))
            {
                MasterDBFile = SettingsManager.Default.MachineStudio.SynchronizationModule.LocalMasterDBFile;
                MasterDBName = Path.GetFileName(MasterDBFile);
            }

            if (File.Exists(SettingsManager.Default.MachineStudio.SynchronizationModule.LocalSlaveDBFile))
            {
                SlaveDBFile = SettingsManager.Default.MachineStudio.SynchronizationModule.LocalSlaveDBFile;
                SlaveDBName = Path.GetFileName(SlaveDBFile);
            }
        }

        #endregion

        #region Commands

        /// <summary>
        /// Gets or sets the back command.
        /// </summary>
        public RelayCommand BackCommand { get; set; }

        /// <summary>
        /// Gets or sets the browse master database command.
        /// </summary>
        public RelayCommand BrowseMasterDBCommand { get; set; }

        /// <summary>
        /// Gets or sets the browse slave database command.
        /// </summary>
        public RelayCommand BrowseSlaveDBCommand { get; set; }

        /// <summary>
        /// Gets or sets the compare command.
        /// </summary>
        public RelayCommand CompareCommand { get; set; }

        /// <summary>
        /// Gets or sets the commit command.
        /// </summary>
        public RelayCommand CommitCommand { get; set; }

        /// <summary>
        /// Gets or sets the commit all command.
        /// </summary>
        public RelayCommand CommitAllCommand { get; set; }

        /// <summary>
        /// Gets or sets the clean command.
        /// </summary>
        public RelayCommand CleanCommand { get; set; }
        #endregion

        #region Properties

        private ObservableCollection<Diff> _differences;
        /// <summary>
        /// Gets or sets the differences.
        /// </summary>
        public ObservableCollection<Diff> Differences
        {
            get { return _differences; }
            set { _differences = value; RaisePropertyChanged(nameof(Differences)); }
        }

        private Diff _selectedDifference;
        /// <summary>
        /// Gets or sets the selected difference.
        /// </summary>
        public Diff SelectedDifference
        {
            get { return _selectedDifference; }
            set { _selectedDifference = value; RaisePropertyChanged(nameof(SelectedDifference)); InvalidateRelayCommands(); }
        }

        private String _masterDBName;
        /// <summary>
        /// Gets or sets the name of the master database.
        /// </summary>
        public String MasterDBName
        {
            get { return _masterDBName; }
            set { _masterDBName = value; RaisePropertyChanged(nameof(MasterDBName)); }
        }

        private String _slaveDBName;
        /// <summary>
        /// Gets or sets the name of the slave database.
        /// </summary>
        public String SlaveDBName
        {
            get { return _slaveDBName; }
            set { _slaveDBName = value; RaisePropertyChanged(nameof(SlaveDBName)); }
        }

        /// <summary>
        /// Gets or sets the slave database file.
        /// </summary>
        public String SlaveDBFile
        {
            get { return _slaveDBFile; }
            set { _slaveDBFile = value; RaisePropertyChangedAuto(); }
        }

        /// <summary>
        /// Gets or sets the master database file.
        /// </summary>
        public String MasterDBFile
        {
            get { return _masterDBFile; }
            set { _masterDBFile = value; RaisePropertyChangedAuto(); }
        }

        #endregion

        #region Private Methods

        /// <summary>
        /// Cleans the slave database.
        /// </summary>
        private async void CleanSlave()
        {
            if (_notification.ShowQuestion("Are you sure you want to erase all data on slave database?"))
            {
                using (_notification.PushTaskItem("Clearing database..."))
                {
                    try
                    {
                        _isWorking = true;
                        await Task.Factory.StartNew(() =>
                        {
                            SQLiteDataBase localDB = new SQLiteDataBase(SlaveDBFile);
                            localDB.LoadTables();
                            localDB.ClearDataBase();
                            try
                            {
                                localDB.Dispose();
                            }
                            catch { }
                        });

                        Differences.Clear();
                    }
                    catch (Exception ex)
                    {
                        ShowError(LogManager.Log(ex).Message);
                    }
                    finally
                    {
                        _isWorking = false;
                        InvalidateRelayCommands();
                        SelectedDifference = null;
                    }
                }
            }
        }

        /// <summary>
        /// Compares the master and slave database files.
        /// </summary>
        private void Compare()
        {
            _comparer = new LocalDBComparer(new SQLiteDataBase(MasterDBFile), new SQLiteDataBase(SlaveDBFile));

            Task.Factory.StartNew(() =>
            {
                using (_notification.PushTaskItem("Comparing Databases..."))
                {
                    try
                    {
                        _isWorking = true;
                        InvalidateRelayCommands();
                        Thread.Sleep(1500);
                        var diffs = _comparer.Compare();
                        Differences = new ObservableCollection<Diff>(diffs);

                        if (diffs.Where(x => x.Action != DiffAction.ReplaceTableDataInSlave).Count() > 0)
                        {
                            ShowInfo("Found " + Differences.Where(x => x.Action != DiffAction.ReplaceTableDataInSlave).Count() + " differences.");
                        }
                        else
                        {
                            ShowInfo("The master and slave databases are synchronized.");
                        }
                    }
                    catch (Exception ex)
                    {
                        ShowError(ex.Message);
                    }
                    finally
                    {
                        _isWorking = false;
                        SelectedDifference = null;
                        InvalidateRelayCommands();

                        SettingsManager.Default.MachineStudio.SynchronizationModule.LocalMasterDBFile = MasterDBFile;
                        SettingsManager.Default.MachineStudio.SynchronizationModule.LocalSlaveDBFile = SlaveDBFile;
                        SettingsManager.SaveDefaultSettings();
                    }
                }
            });
        }

        /// <summary>
        /// Commits the selected difference.
        /// </summary>
        private void Commit()
        {
            Task.Factory.StartNew(() =>
            {
                using (_notification.PushTaskItem("Committing difference..."))
                {
                    try
                    {
                        _isWorking = true;
                        InvalidateRelayCommands();
                        Thread.Sleep(1500);
                        SelectedDifference.Commit();

                        InvokeUINow(() => Differences.Remove(SelectedDifference));
                    }
                    catch (Exception ex)
                    {
                        ShowError(ex.Message);
                    }
                    finally
                    {
                        _isWorking = false;
                        SelectedDifference = null;
                        InvalidateRelayCommands();
                    }
                }
            });
        }

        /// <summary>
        /// Commits all the differences.
        /// </summary>
        private void CommitAll()
        {
            Task.Factory.StartNew(() =>
            {
                using (_notification.PushTaskItem("Committing all differences..."))
                {
                    try
                    {
                        _isWorking = true;
                        InvalidateRelayCommands();
                        Thread.Sleep(1500);

                        for (int i = 0; i < Differences.Count; i++)
                        {
                            var diff = Differences[i];
                            using (_notification.PushTaskItem("Committing difference " + (Differences.IndexOf(diff) + 1) + "..."))
                            {
                                diff.Commit();
                                InvokeUINow(() => Differences.Remove(diff));
                                i--;
                            }
                        }
                    }
                    catch (Exception ex)
                    {
                        ShowError(ex.Message);
                    }
                    finally
                    {
                        _isWorking = false;
                        SelectedDifference = null;
                        InvalidateRelayCommands();
                    }
                }
            });
        }

        /// <summary>
        /// Browse for slave database file.
        /// </summary>
        private void BrowseSlaveDB()
        {
            String file = BrowseForFilePath();
            if (file != null)
            {
                SlaveDBFile = file;
                SlaveDBName = Path.GetFileName(file);
                InvalidateRelayCommands();
            }
        }

        /// <summary>
        /// Browse for master database file.
        /// </summary>
        private void BrowseMasterDB()
        {
            String file = BrowseForFilePath();
            if (file != null)
            {
                MasterDBFile = file;
                MasterDBName = Path.GetFileName(file);
                InvalidateRelayCommands();
            }
        }

        /// <summary>
        /// Browse for database file.
        /// </summary>
        private String BrowseForFilePath()
        {
            OpenFileDialog dlg = new OpenFileDialog();
            dlg.Title = "Select SQLite Database File";
            dlg.Filter = "SQLite Database|*.db";
            if (dlg.ShowDialog().Value)
            {
                return dlg.FileName;
            }
            return null;
        }

        /// <summary>
        /// Displays an error message.
        /// </summary>
        /// <param name="message">The message.</param>
        private void ShowError(String message)
        {
            InvokeUINow(() => _notification.ShowError(message));
        }

        /// <summary>
        /// Displays an information message.
        /// </summary>
        /// <param name="message">The message.</param>
        private void ShowInfo(String message)
        {
            InvokeUINow(() => _notification.ShowInfo(message));
        }

        #endregion
    }
}