Implemented machine studio periodical update checking.

2 files changed, 106 insertions, 8 deletions

diff --git a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/MainViewVM.cs b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/MainViewVM.cs
index acd9e7925..fb2cd5c82 100644
--- a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/MainViewVM.cs
+++ b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/MainViewVM.cs
@@ -6,6 +6,7 @@ using System.IO;
 using System.Linq;
 using System.Reflection;
 using System.Text;
+using System.Threading;
 using System.Threading.Tasks;
 using System.Windows;
 using System.Windows.Controls;
@@ -19,6 +20,7 @@ using Tango.MachineStudio.Common.Modules;
 using Tango.MachineStudio.Common.Navigation;
 using Tango.MachineStudio.Common.Notifications;
 using Tango.MachineStudio.Common.StudioApplication;
+using Tango.MachineStudio.Common.Update;
 using Tango.MachineStudio.UI.StudioApplication;
 using Tango.MachineStudio.UI.SupervisingController;
 using Tango.MachineStudio.UI.Views;
@@ -39,6 +41,7 @@ namespace Tango.MachineStudio.UI.ViewModels
         private IStudioModule _currentModule;
         private INavigationManager _navigation;
         private bool _isDisconnecting;
+        private Thread _updateCheckThread;
 
         /// <summary>
         /// Gets or sets the current loaded module.
@@ -149,6 +152,38 @@ namespace Tango.MachineStudio.UI.ViewModels
             set { _applicationManager = value; RaisePropertyChangedAuto(); }
         }
 
+        private bool _isUpdateAvailable;
+        /// <summary>
+        /// Gets or sets a value indicating whether a new version update is available.
+        /// </summary>
+        public bool IsUpdateAvailable
+        {
+            get { return _isUpdateAvailable; }
+            set { _isUpdateAvailable = value; RaisePropertyChangedAuto(); }
+        }
+
+        private String _latestVersion;
+        /// <summary>
+        /// Gets or sets the latest version.
+        /// </summary>
+        public String LatestVersion
+        {
+            get { return _latestVersion; }
+            set { _latestVersion = value; RaisePropertyChangedAuto(); }
+        }
+
+
+        private bool _disableCheckForUpdates;
+        /// <summary>
+        /// Gets or sets a value indicating whether [disable check for updates].
+        /// </summary>
+        public bool DisableCheckForUpdates
+        {
+            get { return _disableCheckForUpdates; }
+            set { _disableCheckForUpdates = value; RaisePropertyChangedAuto(); }
+        }
+
+
         /// <summary>
         /// Initializes a new instance of the <see cref="MainViewVM"/> class.
         /// </summary>
@@ -181,6 +216,43 @@ namespace Tango.MachineStudio.UI.ViewModels
             OpenModuleInWindowCommand = new RelayCommand<IStudioModule>(OpenModuleInWindow);
             ExitCommand = new RelayCommand(ExitApplication);
             UpdateCenterCommand = new RelayCommand(NavigateToUpdateCenter);
+
+            _updateCheckThread = new Thread(UpdateCheckThreadMethod);
+            _updateCheckThread.IsBackground = true;
+            _updateCheckThread.Start();
+        }
+
+        private void UpdateCheckThreadMethod()
+        {
+            while (!DisableCheckForUpdates)
+            {
+                Thread.Sleep(TimeSpan.FromMinutes(1));
+
+                try
+                {
+                    if (_authenticationProvider.CurrentUser != null)
+                    {
+                        var service = UpdateServiceHelper.GetUpdateServiceChannel();
+                        var client = service.CreateChannel();
+
+                        CheckForUpdatesResponse response = client.CheckForUpdates(new CheckForUpdatesRequest()
+                        {
+                            Email = _authenticationProvider.CurrentUser.Email,
+                            Password = _authenticationProvider.CurrentUser.Password,
+                            Version = _applicationManager.Version,
+                        });
+
+                        IsUpdateAvailable = response.IsUpdateAvailable;
+                        LatestVersion = response.Version;
+                    }
+                }
+                catch (Exception ex)
+                {
+                    LogManager.Log(ex, "Error in version update periodic check...");
+                }
+
+                Thread.Sleep(TimeSpan.FromMinutes(4));
+            }
         }
 
         /// <summary>
diff --git a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/UpdateViewVM.cs b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/UpdateViewVM.cs
index c24204fb6..6be4ba4ca 100644
--- a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/UpdateViewVM.cs
+++ b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/UpdateViewVM.cs
@@ -1,5 +1,6 @@
 using FluentFTP;
 using Ionic.Zip;
+using Microsoft.Practices.ServiceLocation;
 using System;
 using System.Collections.Generic;
 using System.Diagnostics;
@@ -75,6 +76,14 @@ namespace Tango.MachineStudio.UI.ViewModels
             set { _updateProgress = value; RaisePropertyChangedAuto(); }
         }
 
+        private String _currentUpdateFile;
+
+        public String CurrentUpdateFile
+        {
+            get { return _currentUpdateFile; }
+            set { _currentUpdateFile = value; RaisePropertyChanged(nameof(CurrentUpdateFile)); }
+        }
+
         public RelayCommand UpdateCommand { get; set; }
 
         public RelayCommand BackCommand { get; set; }
@@ -95,7 +104,7 @@ namespace Tango.MachineStudio.UI.ViewModels
             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);
+            TryAgainCommand = new RelayCommand(TryAgain, () => Status == UpdateStatus.Error);
         }
 
         public void OnNavigatedInto()
@@ -174,10 +183,10 @@ namespace Tango.MachineStudio.UI.ViewModels
 
             Task.Factory.StartNew(() =>
             {
+                var tempFile = PathHelper.GetTempFilePath() + ".zip";
+
                 try
                 {
-                    var tempFile = PathHelper.GetTempFilePath() + ".zip";
-
                     LogManager.Log("Creating temporary file " + tempFile);
 
                     int fileSize = 0;
@@ -235,6 +244,7 @@ namespace Tango.MachineStudio.UI.ViewModels
                                 }
                                 else
                                 {
+                                    CurrentUpdateFile = Path.GetFileName(entry.FileName);
                                     entry.Extract(_newPackageTempFolder, ExtractExistingFileAction.OverwriteSilently);
                                 }
                             }
@@ -245,6 +255,7 @@ namespace Tango.MachineStudio.UI.ViewModels
                         }
                     }
 
+                    ServiceLocator.Current.GetInstance<MainViewVM>().DisableCheckForUpdates = true;
                     Status = UpdateStatus.UpdateCompleted;
                 }
                 catch (Exception ex)
@@ -252,6 +263,10 @@ namespace Tango.MachineStudio.UI.ViewModels
                     LogManager.Log(ex, "Error while extracting update package.");
                     Status = UpdateStatus.Error;
                 }
+                finally
+                {
+                    PathHelper.TryDeleteFile(tempFile);
+                }
             });
         }
 
@@ -262,11 +277,22 @@ namespace Tango.MachineStudio.UI.ViewModels
 
         private void RestartApplication()
         {
-            Process p = new Process();
-            p.StartInfo.FileName = _appPath + "\\Tango.MachineStudio.Updater.exe";
-            p.StartInfo.UseShellExecute = true;
-            p.StartInfo.Arguments = _newPackageTempFolder;
-            p.Start();
+            try
+            {
+                Process p = new Process();
+                p.StartInfo.FileName = _appPath + "\\Tango.MachineStudio.Updater.exe";
+                p.StartInfo.UseShellExecute = true;
+                p.StartInfo.Arguments = _newPackageTempFolder;
+                p.Start();
+            }
+            catch (Exception ex)
+            {
+                if (ex.Message == "The operation was canceled by the user")
+                {
+                    _notification.ShowWarning("It seems like you refused to invoke our update utility. This prevents Machine Studio from completing the update process!");
+                    return;
+                }
+            }
             Environment.Exit(0);
         }