From 7fff53b4e158212572dd0d83b5628e799e05ba01 Mon Sep 17 00:00:00 2001
From: Roy Ben-Shabat <Roy@Twine-s.com>
Date: Thu, 24 Jan 2019 13:28:53 +0200
Subject: Implemented forced environment version in machine studio.

---
 .../ViewModels/UpdateViewVM.cs                     | 23 ++++++++++++++++++++++
 1 file changed, 23 insertions(+)

(limited to 'Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/UpdateViewVM.cs')

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 60a023071..f671251f7 100644
--- a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/UpdateViewVM.cs
+++ b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/UpdateViewVM.cs
@@ -158,6 +158,29 @@ namespace Tango.MachineStudio.UI.ViewModels
             IsRollbackAvailable = File.Exists(GetRollbackFile());
 
             TangoMessenger.Default.Register<Messages.ForcedUpdateMessage>(HandleForcedUpdateMessage);
+            TangoMessenger.Default.Register<Messages.ChangeVersionMessage>(HandleChangeVersionMessage);
+        }
+
+        private async void HandleChangeVersionMessage(ChangeVersionMessage msg)
+        {
+            ForcedUpdate = true;
+            InvalidateRelayCommands();
+
+            Status = UpdateStatus.CheckingForUpdate;
+
+            var client = new MachineStudioUpdateService();
+
+            DownloadLatestVersionResponse response = await client.DownloadLatestVersion(new DownloadLatestVersionRequest()
+            {
+                AccessToken = _authentication.AccessToken,
+            });
+
+            _updateInfo = new CheckForUpdatesResponse();
+            _updateInfo.BlobAddress = response.BlobAddress;
+            _updateInfo.Version = response.Version;
+            LatestVersion = _updateInfo.Version;
+
+            StartUpdate();
         }
 
         private void HandleForcedUpdateMessage(ForcedUpdateMessage msg)
-- 
cgit v1.3.1