Merge branch 'master' of https://twinetfs.visualstudio.com/Tango/_git/Tango

1 files changed, 231 insertions, 17 deletions

diff --git a/Software/Visual_Studio/PPC/Tango.PPC.UI/ViewModels/MachineUpdateViewVM.cs b/Software/Visual_Studio/PPC/Tango.PPC.UI/ViewModels/MachineUpdateViewVM.cs
index f265c5dbf..3942a1b84 100644
--- a/Software/Visual_Studio/PPC/Tango.PPC.UI/ViewModels/MachineUpdateViewVM.cs
+++ b/Software/Visual_Studio/PPC/Tango.PPC.UI/ViewModels/MachineUpdateViewVM.cs
@@ -1,16 +1,23 @@
 using System;
 using System.Collections.Generic;
+using System.Data.Entity;
 using System.Diagnostics;
 using System.IO;
 using System.Linq;
 using System.Text;
 using System.Threading.Tasks;
+using Tango.BL;
 using Tango.Core.Commands;
+using Tango.Core.ExtensionMethods;
 using Tango.Core.Helpers;
 using Tango.Explorer;
+using Tango.PMR.FirmwareUpgrade;
 using Tango.PPC.Common;
 using Tango.PPC.Common.MachineUpdate;
+using Tango.PPC.Common.Publish;
+using Tango.PPC.Common.Web;
 using Tango.PPC.UI.Dialogs;
+using Tango.PPC.UI.Notifications.NotificationItems;
 using Tango.PPC.UI.ViewsContracts;
 
 namespace Tango.PPC.UI.ViewModels
@@ -34,6 +41,8 @@ namespace Tango.PPC.UI.ViewModels
         private MachineUpdateResult _update_result;
         private DbCompareResult _db_compare_result;
         private bool _isChecking;
+        private CheckForUpdateResponse _checkUpdateResponse;
+        private UpdateAvailableNotificationItem _updateNotificationItem;
 
         #region Properties
 
@@ -123,6 +132,8 @@ namespace Tango.PPC.UI.ViewModels
                 NavigationManager.NavigateTo(Common.Navigation.NavigationView.HomeModule);
                 NavigateTo(MachineUpdateView.UpdateCheckView);
             });
+
+            machineUpdateManager.UpdateAvailable += MachineUpdateManager_UpdateAvailable;
         }
 
         #endregion
@@ -150,11 +161,43 @@ namespace Tango.PPC.UI.ViewModels
 
                 var response = await MachineUpdateManager.CheckForUpdate(MachineProvider.Machine.SerialNumber);
 
+                try
+                {
+                    if (response.UsedNotExistingRmlsGuids.Count > 0)
+                    {
+                        using (ObservablesContext db = ObservablesContext.CreateDefault())
+                        {
+                            var arr = response.UsedNotExistingRmlsGuids.ToArray();
+                            var jobs = await db.Jobs.Where(x => arr.Contains(x.RmlGuid)).ToListAsync();
+                            FailedError = $"The following jobs must be removed or change thread type before the system can be updated:\n{String.Join("\n", jobs.Select(x => x.Name))}";
+                            _isChecking = false;
+                            await NavigateTo(MachineUpdateView.UpdateFailedView);
+                            return;
+                        }
+                    }
+                }
+                catch (Exception ex)
+                {
+                    LogManager.Log(ex, "Error on used RML check procedure.");
+                }
+
+                _checkUpdateResponse = response;
+
                 if (response.IsUpdateAvailable)
                 {
                     LatestVersion = response.Version;
                     await NavigateTo(MachineUpdateView.UpdateAvailableView);
                 }
+                else if (response.IsDatabaseUpdateAvailable)
+                {
+                    IsDbUpdate = true;
+                    _db_compare_result = new DbCompareResult()
+                    {
+                        RequiresUpdate = true,
+                        UpdateDBResponse = response.UpdateDBResponse
+                    };
+                    await NavigateTo(MachineUpdateView.UpdateAvailableView);
+                }
                 else
                 {
                     _db_compare_result = await MachineUpdateManager.UpdateDBCheck(MachineProvider.Machine.SerialNumber);
@@ -192,7 +235,7 @@ namespace Tango.PPC.UI.ViewModels
 
                 try
                 {
-                    _update_result = await MachineUpdateManager.Update(MachineProvider.Machine.SerialNumber);
+                    _update_result = await MachineUpdateManager.Update(_checkUpdateResponse.SetupFirmware, _checkUpdateResponse.SetupFPGA);
                     LogManager.Log("Machine update completed.");
                     await NavigateTo(MachineUpdateView.UpdateCompletedView);
                 }
@@ -231,15 +274,15 @@ namespace Tango.PPC.UI.ViewModels
         {
             LogManager.Log("Completing machine update...");
 
-            if (!IsDbUpdate)
+            if (IsDbUpdate || !_update_result.RequiresBinariesUpdate)
             {
-                String updater_exe = Path.Combine(_update_result.UpdatePackagePath, "Tango.PPC.Updater.exe");
-                ApplicationManager.UpdateApplication(updater_exe, PathHelper.GetStartupPath());
+                LogManager.Log("Restarting Application...");
+                ApplicationManager.Restart();
             }
             else
             {
-                LogManager.Log("Restarting Application...");
-                ApplicationManager.Restart();
+                String updater_exe = Path.Combine(_update_result.UpdatePackagePath, "Tango.PPC.Updater.exe");
+                ApplicationManager.UpdateApplication(updater_exe, PathHelper.GetStartupPath());
             }
         }
 
@@ -272,6 +315,85 @@ namespace Tango.PPC.UI.ViewModels
             base.OnApplicationReady();
 
             StorageProvider.RegisterFileHandler(ExplorerFileDefinition.Update.Extension, HandleSoftwareUpdatePackageLoaded);
+            StorageProvider.RegisterFileHandler(ExplorerFileDefinition.Firmware.Extension, HandleFirmwareUpgradeLoaded);
+
+            if (ApplicationManager.IsAfterUpdate)
+            {
+                RunPostUpdatePackages();
+            }
+            else
+            {
+                MachineUpdateManager.EnableAutoCheckForUpdates = true;
+            }
+        }
+
+        /// <summary>
+        /// Called when the navigation system has navigated to this VM view.
+        /// </summary>
+        public override void OnNavigatedTo()
+        {
+            base.OnNavigatedTo();
+
+            if (_updateNotificationItem != null)
+            {
+                _updateNotificationItem.Close();
+                _updateNotificationItem = null;
+            }
+        }
+
+        #endregion
+
+        #region Post Update Packages
+
+        private async void RunPostUpdatePackages()
+        {
+            await Task.Delay(1000);
+
+            LogManager.Log("Application was loaded after an update. Checking for required post-update packages...");
+
+            bool required = false;
+
+            try
+            {
+                required = await MachineUpdateManager.PostUpdatePackagesRequired();
+            }
+            catch (Exception ex)
+            {
+                LogManager.Log(ex, "Error checking for post-update packages.");
+            }
+
+            if (required)
+            {
+                LogManager.Log("Post-update packages found and needs to be installed. Navigating to machine update and running post-update packages...");
+                await NavigationManager.NavigateTo(Common.Navigation.NavigationView.MachineUpdateView);
+                await NavigateTo(MachineUpdateView.UpdateProgressView);
+                try
+                {
+                    var result = await MachineUpdateManager.RunPostUpdatePackages();
+
+                    LogManager.Log("Post-update packages installed successfully.");
+
+                    await Task.Delay(2000);
+
+                    if (result.RestartRequired)
+                    {
+                        LogManager.Log("Restart required. Restarting...");
+                        ApplicationManager.Restart();
+                    }
+                    else
+                    {
+                        await NavigationManager.NavigateTo(Common.Navigation.NavigationView.LayoutView);
+                    }
+                }
+                catch (Exception ex)
+                {
+                    LogManager.Log(ex, "Error occurred while running post-update packages.");
+                }
+            }
+            else
+            {
+                LogManager.Log("No post-update packages installation required.");
+            }
         }
 
         #endregion
@@ -280,7 +402,9 @@ namespace Tango.PPC.UI.ViewModels
 
         private async void HandleSoftwareUpdatePackageLoaded(ExplorerFileItem fileItem)
         {
-            UpdatePackageFile packageFile = null;
+            PublishInfo packageFile = null;
+
+            LogManager.Log("TUP file loaded from storage...");
 
             try
             {
@@ -288,43 +412,133 @@ namespace Tango.PPC.UI.ViewModels
             }
             catch (Exception ex)
             {
-                LogManager.Log(ex, $"Error loading update package file from {fileItem.Path}.");
+                LogManager.Log(ex, $"Error loading publish info from {fileItem.Path}.");
                 await NotificationProvider.ShowError("An error occurred while trying to load the selected software update package. Please make sure the package is valid.");
                 return;
             }
 
-            if (ApplicationManager.Version <= packageFile.Version)
-            {
-                await NotificationProvider.ShowError($"The selected update package (v{packageFile.Version.ToString()}) contains an older software version.");
-                return;
-            }
-
             UpdateFromFileViewVM vm = new UpdateFromFileViewVM();
-            vm.Version = packageFile.Version.ToString();
+            vm.PublishInfo = packageFile;
+
+            LogManager.Log($"TUP publish info:\n{packageFile.ToJson()}");
+
+            LogManager.Log("Displaying TUP update dialog...");
 
             await NotificationProvider.ShowDialog(vm);
 
             if (vm.DialogResult)
             {
                 await NavigationManager.NavigateTo(Common.Navigation.NavigationView.MachineUpdateView);
-                await NavigateTo(MachineUpdateView.UpdateFromPackageView);
+                await NavigateTo(MachineUpdateView.UpdateProgressView);
 
                 LogManager.Log("Starting machine update from package...");
 
                 try
                 {
-                    _update_result = await MachineUpdateManager.UpdateFromTUP(fileItem.Path);
+                    _update_result = await MachineUpdateManager.UpdateFromTUP(fileItem.Path, MachineProvider.Machine.SetupFirmware, MachineProvider.Machine.SetupFpga);
                     LogManager.Log("Machine update from package completed.");
                     await NavigateTo(MachineUpdateView.UpdateCompletedView);
                 }
                 catch (Exception ex)
                 {
                     LogManager.Log(ex, "Machine update from package failed.");
+                    FailedError = ex.FlattenMessage();
                     await NavigateTo(MachineUpdateView.UpdateFailedFromPackageView);
                 }
             }
         }
 
+        private async void HandleFirmwareUpgradeLoaded(ExplorerFileItem fileItem)
+        {
+            LogManager.Log("TFP file loaded from storage...");
+
+            VersionPackageDescriptor packageInfo;
+            FirmwareUpgradeFromFileViewVM vm = new FirmwareUpgradeFromFileViewVM();
+
+            try
+            {
+                using (FileStream st = File.OpenRead(fileItem.Path))
+                {
+                    packageInfo = await MachineProvider.MachineOperator.GetFirmwarePackageInfo(st);
+                }
+
+                vm.Version = packageInfo.FileDescriptors.SingleOrDefault(x => x.Destination == VersionFileDestination.Mcu).Version;
+            }
+            catch (Exception ex)
+            {
+                LogManager.Log(ex, $"Error loading package info from {fileItem.Path}.");
+                await NotificationProvider.ShowError("An error occurred while trying to load the selected firmware upgrade package. Please make sure the package is valid.");
+                return;
+            }
+
+
+            LogManager.Log($"TFP publish info:\n{packageInfo.ToJsonString()}");
+
+            LogManager.Log("Displaying TFP update dialog...");
+
+            await NotificationProvider.ShowDialog(vm);
+
+            if (vm.DialogResult)
+            {
+                await NavigationManager.NavigateTo(Common.Navigation.NavigationView.MachineUpdateView);
+                await NavigateTo(MachineUpdateView.UpdateProgressView);
+
+                LogManager.Log("Starting firmware upgrade from package...");
+
+                try
+                {
+                    await MachineUpdateManager.UpdateFromTFP(fileItem.Path);
+                    LogManager.Log("Firmware upgrade from package completed.");
+                    _update_result = new MachineUpdateResult()
+                    {
+                         RequiresBinariesUpdate = false,
+                    };
+                    await NavigateTo(MachineUpdateView.UpdateCompletedView);
+                }
+                catch (Exception ex)
+                {
+                    LogManager.Log(ex, "Firmware upgrade from package failed.");
+                    FailedError = ex.FlattenMessage();
+                    await NavigateTo(MachineUpdateView.UpdateFailedFromPackageView);
+                }
+            }
+        }
+
+        #endregion
+
+        #region Auto Check For Update
+
+        private void MachineUpdateManager_UpdateAvailable(object sender, CheckForUpdateResponse e)
+        {
+            if (!IsVisible && _updateNotificationItem == null)
+            {
+                LogManager.Log($"New {(e.IsDatabaseUpdateAvailable ? "database updates" : "application version")} detected ({e.Version}). Pushing notification...");
+
+                InvokeUI(() =>
+                {
+                    _updateNotificationItem = new UpdateAvailableNotificationItem();
+                    _updateNotificationItem.Version = Version.Parse(e.Version).ToString(3);
+                    _updateNotificationItem.IsDatabaseUpdate = e.IsDatabaseUpdateAvailable && !e.IsUpdateAvailable;
+                    _updateNotificationItem.Pressed += (_, __) =>
+                    {
+                        _updateNotificationItem = null;
+
+                        if (!IsVisible)
+                        {
+                            LogManager.Log("Update available notification pressed. Navigating to update view...");
+                            NavigationManager.NavigateTo(Common.Navigation.NavigationView.MachineUpdateView);
+                            CheckForUpdates();
+                        }
+                    };
+                    _updateNotificationItem.Closed += (_, __) =>
+                    {
+                        _updateNotificationItem = null;
+                    };
+                    NotificationProvider.PushNotification(_updateNotificationItem);
+                });
+            }
+        }
+
         #endregion
     }
 }