1 files changed, 222 insertions, 3 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 3942a1b84..965584767 100644
--- a/Software/Visual_Studio/PPC/Tango.PPC.UI/ViewModels/MachineUpdateViewVM.cs
+++ b/Software/Visual_Studio/PPC/Tango.PPC.UI/ViewModels/MachineUpdateViewVM.cs
@@ -5,24 +5,31 @@ using System.Diagnostics;
 using System.IO;
 using System.Linq;
 using System.Text;
+using System.Threading;
 using System.Threading.Tasks;
 using Tango.BL;
+using Tango.Core;
 using Tango.Core.Commands;
 using Tango.Core.ExtensionMethods;
 using Tango.Core.Helpers;
+using Tango.Core.Threading;
 using Tango.Explorer;
+using Tango.Integration.ExternalBridge;
 using Tango.PMR.FirmwareUpgrade;
 using Tango.PPC.Common;
+using Tango.PPC.Common.ExternalBridge;
 using Tango.PPC.Common.MachineUpdate;
 using Tango.PPC.Common.Publish;
 using Tango.PPC.Common.Web;
+using Tango.PPC.Shared.RemoteUpgrade;
 using Tango.PPC.UI.Dialogs;
 using Tango.PPC.UI.Notifications.NotificationItems;
 using Tango.PPC.UI.ViewsContracts;
+using Tango.Transport;
 
 namespace Tango.PPC.UI.ViewModels
 {
-    public class MachineUpdateViewVM : PPCViewModel<IMachineUpdateView>
+    public class MachineUpdateViewVM : PPCViewModel<IMachineUpdateView>, IExternalBridgeRequestHandler
     {
         public enum MachineUpdateView
         {
@@ -114,9 +121,10 @@ namespace Tango.PPC.UI.ViewModels
 
         #region Constructors
 
-        public MachineUpdateViewVM(IMachineUpdateManager machineUpdateManager)
+        public MachineUpdateViewVM(IMachineUpdateManager machineUpdateManager, IPPCExternalBridgeService externalBridge)
         {
             MachineUpdateManager = machineUpdateManager;
+            externalBridge.RegisterRequestHandler(this);
 
             CompleteCommand = new RelayCommand(CompleteUpdate);
             UpdateCommand = new RelayCommand(Update);
@@ -491,7 +499,7 @@ namespace Tango.PPC.UI.ViewModels
                     LogManager.Log("Firmware upgrade from package completed.");
                     _update_result = new MachineUpdateResult()
                     {
-                         RequiresBinariesUpdate = false,
+                        RequiresBinariesUpdate = false,
                     };
                     await NavigateTo(MachineUpdateView.UpdateCompletedView);
                 }
@@ -540,5 +548,216 @@ namespace Tango.PPC.UI.ViewModels
         }
 
         #endregion
+
+        #region External Bridge Handler
+
+        [ExternalBridgeRequestHandlerMethod(typeof(StartRemoteApplicationUpgradeRequest), RequestHandlerLoggingMode.LogRequestName)]
+        public async Task OnStartRemoteApplicationUpgradeRequest(StartRemoteApplicationUpgradeRequest request, String token, ExternalBridgeReceiver receiver)
+        {
+            await receiver.SendGenericResponse(new StartRemoteApplicationUpgradeResponse(), token);
+
+            bool stopReporting = false;
+
+            try
+            {
+                ThreadFactory.StartNew(async () =>
+                {
+                    while (!stopReporting)
+                    {
+                        if (MachineUpdateManager.Status != null)
+                        {
+                            try
+                            {
+                                await receiver.SendGenericResponse(new StartRemoteApplicationUpgradeResponse()
+                                {
+                                    Progress = new TangoProgress<double>()
+                                    {
+                                        Message = MachineUpdateManager.Status.Message,
+                                        IsIndeterminate = MachineUpdateManager.Status.IsIntermediate,
+                                        Maximum = MachineUpdateManager.Status.Total,
+                                        Value = MachineUpdateManager.Status.Progress
+                                    },
+                                }, token, new TransportResponseConfig() { Priority = QueuePriority.Low });
+                            }
+                            catch (Exception ex)
+                            {
+                                LogManager.Log(ex, "Error sending remote upgrade progress.");
+                            }
+                        }
+
+                        Thread.Sleep(500);
+                    }
+                });
+
+                InvokeUI(() =>
+                {
+                    NavigationManager.NavigateTo(Common.Navigation.NavigationView.MachineUpdateView);
+                    NavigateTo(MachineUpdateView.UpdateProgressView);
+                });
+
+                LogManager.Log("Starting machine update from package...");
+
+                try
+                {
+                    _update_result = await MachineUpdateManager.UpdateFromTUP(request.RemoteTupFilePath, request.SetupFirmware, request.SetupFPGA);
+                    LogManager.Log("Machine update from package completed.");
+
+                    InvokeUI(() =>
+                    {
+                        NavigateTo(MachineUpdateView.UpdateCompletedView);
+                    });
+                }
+                catch (Exception ex)
+                {
+                    LogManager.Log(ex, "Machine update from package failed.");
+                    FailedError = ex.FlattenMessage();
+
+                    InvokeUI(() =>
+                    {
+                        NavigateTo(MachineUpdateView.UpdateFailedFromPackageView);
+                    });
+                }
+
+                await receiver.SendGenericResponse(new StartRemoteApplicationUpgradeResponse()
+                {
+                    Progress = new TangoProgress<double>("Completed", false, 100, 100),
+                }, token, new TransportResponseConfig()
+                {
+                    Completed = true
+                });
+
+                try
+                {
+                    File.Delete(request.RemoteTupFilePath);
+                }
+                catch { }
+
+                await Task.Delay(2000);
+
+                CompleteUpdate();
+            }
+            catch (Exception ex)
+            {
+                await receiver.SendErrorResponse(ex, token);
+            }
+            finally
+            {
+                try
+                {
+                    File.Delete(request.RemoteTupFilePath);
+                }
+                catch { }
+            }
+        }
+
+        [ExternalBridgeRequestHandlerMethod(typeof(StartRemoteFirmwareUpgradeRequest), RequestHandlerLoggingMode.LogRequestName)]
+        public async Task OnStartRemoteFirmwareUpgradeRequest(StartRemoteFirmwareUpgradeRequest request, String token, ExternalBridgeReceiver receiver)
+        {
+            await receiver.SendGenericResponse(new StartRemoteFirmwareUpgradeResponse(), token);
+
+            bool stopReporting = false;
+
+            try
+            {
+                ThreadFactory.StartNew(async () =>
+                {
+                    while (!stopReporting)
+                    {
+                        if (MachineUpdateManager.Status != null)
+                        {
+                            try
+                            {
+                                await receiver.SendGenericResponse(new StartRemoteFirmwareUpgradeResponse()
+                                {
+                                    Progress = new TangoProgress<double>()
+                                    {
+                                        Message = MachineUpdateManager.Status.Message,
+                                        IsIndeterminate = MachineUpdateManager.Status.IsIntermediate,
+                                        Maximum = MachineUpdateManager.Status.Total,
+                                        Value = MachineUpdateManager.Status.Progress
+                                    }
+                                }, token, new TransportResponseConfig() { Priority = QueuePriority.Low });
+                            }
+                            catch (Exception ex)
+                            {
+                                LogManager.Log(ex, "Error sending remote upgrade progress.");
+                            }
+                        }
+
+                        Thread.Sleep(500);
+                    }
+                });
+
+                InvokeUI(() =>
+                {
+                    NavigationManager.NavigateTo(Common.Navigation.NavigationView.MachineUpdateView);
+                    NavigateTo(MachineUpdateView.UpdateProgressView);
+                });
+
+                LogManager.Log("Starting firmware upgrade from package...");
+
+                try
+                {
+                    await MachineUpdateManager.UpdateFromTFP(request.RemoteTfpFilePath);
+                    LogManager.Log("Firmware upgrade from package completed.");
+                    _update_result = new MachineUpdateResult()
+                    {
+                        RequiresBinariesUpdate = false,
+                    };
+
+                    InvokeUI(() =>
+                    {
+                        NavigateTo(MachineUpdateView.UpdateCompletedView);
+                    });
+                }
+                catch (Exception ex)
+                {
+                    LogManager.Log(ex, "Firmware upgrade from package failed.");
+                    FailedError = ex.FlattenMessage();
+
+                    InvokeUI(() =>
+                    {
+                        NavigateTo(MachineUpdateView.UpdateFailedFromPackageView);
+                    });
+                }
+
+                await receiver.SendGenericResponse(new StartRemoteFirmwareUpgradeResponse()
+                {
+                    Progress = new TangoProgress<double>("Completed", false, 100, 100),
+                }, token, new TransportResponseConfig()
+                {
+                    Completed = true
+                });
+
+                try
+                {
+                    File.Delete(request.RemoteTfpFilePath);
+                }
+                catch { }
+
+                await Task.Delay(2000);
+
+                CompleteUpdate();
+            }
+            catch (Exception ex)
+            {
+                await receiver.SendErrorResponse(ex, token);
+            }
+            finally
+            {
+                try
+                {
+                    File.Delete(request.RemoteTfpFilePath);
+                }
+                catch { }
+            }
+        }
+
+        public void OnReceiverDisconnected(ExternalBridgeReceiver receiver)
+        {
+            //Do Nothing.
+        }
+
+        #endregion
     }
 }