path: root/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/UpdateViewVM.cs

using FluentFTP;
using Ionic.Zip;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Runtime.CompilerServices;
using System.ServiceModel;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using Tango.Core.Commands;
using Tango.Core.Helpers;
using Tango.Logging;
using Tango.MachineStudio.Common.Authentication;
using Tango.MachineStudio.Common.Navigation;
using Tango.MachineStudio.Common.Notifications;
using Tango.MachineStudio.Common.StudioApplication;
using Tango.MachineStudio.Common.Update;
using Tango.SharedUI;

namespace Tango.MachineStudio.UI.ViewModels
{
    public enum UpdateStatus
    {
        None,
        CheckingForUpdate,
        UpToDate,
        UpdateAvailable,
        Downloading,
        Updating,
        UpdateCompleted,
        Error,
    }

    public class UpdateViewVM : ViewModel
    {
        private String _appPath = AppDomain.CurrentDomain.BaseDirectory;

        private INotificationProvider _notification;
        private INavigationManager _navigation;
        private IStudioApplicationManager _application;
        private IAuthenticationProvider _authentication;
        private CheckForUpdatesResponse _updateInfo;

        private UpdateStatus _status;
        public UpdateStatus Status
        {
            get { return _status; }
            set { _status = value; RaisePropertyChangedAuto(); }
        }

        private String _latestVersion;
        public String LatestVersion
        {
            get { return _latestVersion; }
            set { _latestVersion = value; RaisePropertyChangedAuto(); }
        }

        private double _downloadProgress;

        public double DownloadProgress
        {
            get { return _downloadProgress; }
            set { _downloadProgress = value; RaisePropertyChangedAuto(); }
        }

        private double _updateProgress;

        public double UpdateProgress
        {
            get { return _updateProgress; }
            set { _updateProgress = value; RaisePropertyChangedAuto(); }
        }

        public RelayCommand UpdateCommand { get; set; }

        public RelayCommand BackCommand { get; set; }

        public RelayCommand RestartCommand { get; set; }

        public RelayCommand TryAgainCommand { get; set; }

        public UpdateViewVM(INotificationProvider notification, IAuthenticationProvider authentication, INavigationManager navigation, IStudioApplicationManager application)
        {
            _notification = notification;
            _navigation = navigation;
            _application = application;
            _authentication = authentication;

            LatestVersion = "1.0.0.2";
            Status = UpdateStatus.CheckingForUpdate;
            UpdateCommand = new RelayCommand(StartUpdate, () => Status == UpdateStatus.UpdateAvailable);
            BackCommand = new RelayCommand(BackToApplication, () => Status != UpdateStatus.Updating);
            RestartCommand = new RelayCommand(RestartApplication, () => Status == UpdateStatus.UpdateCompleted);
            TryAgainCommand = new RelayCommand(TryAgain, () => Status == UpdateStatus.CheckingForUpdate);

            foreach (var file in Directory.GetFiles(_appPath, "*.tmp"))
            {
                PathHelper.TryDeleteFile(file);
            }
        }

        public void OnNavigatedInto()
        {
            CheckForUpdates();
        }

        private void CheckForUpdates()
        {
            Status = UpdateStatus.CheckingForUpdate;

            ChannelFactory<IMachineStudioUpdateService> service = null;

            Task.Factory.StartNew(() =>
            {
                try
                {
                    Thread.Sleep(2000);
                
                    service = UpdateServiceHelper.GetUpdateServiceChannel();
                    var client = service.CreateChannel();

                    CheckForUpdatesResponse response = client.CheckForUpdates(new CheckForUpdatesRequest()
                    {
                        Email = _authentication.CurrentUser.Email,
                        Password = _authentication.CurrentUser.Password,
                        Version = _application.Version,
                    });

                    if (response.IsUpdateAvailable)
                    {
                        _updateInfo = response;
                        Status = UpdateStatus.UpdateAvailable;
                        LatestVersion = response.Version;
                    }
                    else
                    {
                        Status = UpdateStatus.UpToDate;
                    }
                }
                catch (Exception ex)
                {
                    LogManager.Log(ex, "Error while checking for version update!");
                    Status = UpdateStatus.Error;
                }
                finally
                {
                    if (service != null)
                    {
                        service.Close();
                    }
                }
            });
        }

        private void BackToApplication()
        {
            if (Status == UpdateStatus.Downloading)
            {
                if (!_notification.ShowQuestion("This will abort all update operations. Are you sure?"))
                {
                    return;
                }
            }

            _navigation.NavigateTo(NavigationView.MainView);
            Status = UpdateStatus.None;
        }
        
        private void StartUpdate()
        {
            DownloadProgress = 0;
            UpdateProgress = 0;

            Status = UpdateStatus.Downloading;

            Task.Factory.StartNew(() =>
            {
                try
                {
                    var tempFile = PathHelper.GetTempFilePath() + ".zip";

                    LogManager.Log("Creating temporary file " + tempFile);

                    int fileSize = 0;

                    using (FileStreamWrapper fs = new FileStreamWrapper(tempFile, FileMode.Create, (current) =>
                    {
                        InvokeUINow(() =>
                        {
                            Thread.Sleep(10);
                            DownloadProgress = ((double)current / (double)fileSize) * 100d;
                        });
                    }))
                    {
                        using (FtpClient ftp = new FtpClient(_updateInfo.FtpHost, _updateInfo.UserName, _updateInfo.Password))
                        {
                            LogManager.Log("Connecting to FTP site: " + _updateInfo.FtpHost);
                            ftp.ConnectAsync().Wait();
                            LogManager.Log("Retrieving download size...");
                            fileSize = (int)ftp.GetFileSize(_updateInfo.FilePath);
                            LogManager.Log("Download size: " + fileSize + " bytes.");
                            LogManager.Log("Starting download...");
                            ftp.DownloadAsync(fs, _updateInfo.FilePath).Wait();
                        }
                    }


                    Status = UpdateStatus.Updating;

                    int lockedCounter = 0;

                    using (ZipFile zip = ZipFile.Read(tempFile))
                    {
                        int currentEntry = 0;

                        zip.ExtractProgress += (x, args) =>
                        {
                            if (args.EventType == ZipProgressEventType.Extracting_AfterExtractEntry)
                            {
                                LogManager.Log("Extracting " + Path.GetFileName(args.CurrentEntry.FileName));
                                UpdateProgress = ((double)(currentEntry++) / (double)zip.Entries.Count) * 100d;
                            }
                        };

                        foreach (ZipEntry entry in zip)
                        {
                            Thread.Sleep(10);

                            string newPath = Path.Combine(_appPath, entry.FileName);

                            try
                            {
                                if (entry.IsDirectory)
                                {
                                    Directory.CreateDirectory(newPath);
                                }
                                else
                                {
                                    if (!IsFileLocked(newPath))
                                    {
                                        entry.Extract(_appPath, ExtractExistingFileAction.OverwriteSilently);
                                    }
                                    else
                                    {
                                        LogManager.Log("File locked: " + entry.FileName);
                                        File.Move(newPath, newPath + ".tmp");
                                        entry.Extract(_appPath, ExtractExistingFileAction.OverwriteSilently);
                                        LogManager.Log("Performed overwrite bypass... (changes will be applied on restart).");
                                        lockedCounter++;
                                    }
                                }
                            }
                            catch
                            {
                                LogManager.Log("Could not overwrite " + entry.FileName);
                            }
                        }
                    }

                    Status = UpdateStatus.UpdateCompleted;
                }
                catch (Exception ex)
                {
                    LogManager.Log(ex, "Error while extracting update package.");
                    Status = UpdateStatus.Error;
                }
            });
        }

        /// <summary>
        /// Determines whether [is file locked] [the specified file].
        /// </summary>
        /// <param name="file">The file.</param
        private bool IsFileLocked(String file)
        {
            FileStream stream = null;

            try
            {
                stream = new FileInfo(file).Open(FileMode.Open, FileAccess.ReadWrite, FileShare.None);
            }
            catch (IOException)
            {
                //the file is unavailable because it is:
                //still being written to
                //or being processed by another thread
                //or does not exist (has already been processed)
                return true;
            }
            finally
            {
                if (stream != null)
                    stream.Close();
            }

            //file is not locked
            return false;
        }

        private void TryAgain()
        {
            CheckForUpdates();
        }

        private void RestartApplication()
        {
            _application.ShutDown();
        }

        protected override void RaisePropertyChangedAuto([CallerMemberName] string caller = null)
        {
            base.RaisePropertyChangedAuto(caller);
            InvalidateRelayCommands();
        }
    }
}