Working on Firmware upgrade !

1 files changed, 12 insertions, 1 deletions

diff --git a/Software/Visual_Studio/MachineStudio/Modules/Tango.MachineStudio.Storage/ViewModels/MainViewVM.cs b/Software/Visual_Studio/MachineStudio/Modules/Tango.MachineStudio.Storage/ViewModels/MainViewVM.cs
index 12bb8242a..a9cfed937 100644
--- a/Software/Visual_Studio/MachineStudio/Modules/Tango.MachineStudio.Storage/ViewModels/MainViewVM.cs
+++ b/Software/Visual_Studio/MachineStudio/Modules/Tango.MachineStudio.Storage/ViewModels/MainViewVM.cs
@@ -12,6 +12,7 @@ using Tango.Core.IO;
 using Tango.Integration.ExternalBridge;
 using Tango.Integration.Storage;
 using Tango.MachineStudio.Common;
+using Tango.MachineStudio.Common.FirmwareUpgrade;
 using Tango.MachineStudio.Common.Notifications;
 using Tango.MachineStudio.Common.StudioApplication;
 using Tango.MachineStudio.Storage.Models;
@@ -24,6 +25,7 @@ namespace Tango.MachineStudio.Storage.ViewModels
 
         private IStudioApplicationManager _applicationManager;
         private INotificationProvider _notification;
+        private IFirmwareUpgrader _firmwareUpgrader;
         private bool _machine_operator_changed = true;
 
         private StorageManager _storageManager;
@@ -71,10 +73,13 @@ namespace Tango.MachineStudio.Storage.ViewModels
 
         public RelayCommand UploadFileCommand { get; set; }
 
-        public MainViewVM(IStudioApplicationManager applicationManager, INotificationProvider notificationProvider)
+        public RelayCommand UploadVersionCommand { get; set; }
+
+        public MainViewVM(IStudioApplicationManager applicationManager, INotificationProvider notificationProvider, IFirmwareUpgrader firmwareUpgrader)
         {
             _applicationManager = applicationManager;
             _notification = notificationProvider;
+            _firmwareUpgrader = firmwareUpgrader;
             _applicationManager.ConnectedMachineChanged += _applicationManager_ConnectedMachineChanged;
 
             FileHandlers = new ObservableCollection<StorageFileHandlerModel>();
@@ -90,6 +95,7 @@ namespace Tango.MachineStudio.Storage.ViewModels
             CreateFolderCommand = new RelayCommand(CreateFolder, () => StorageManager != null && StorageManager.CurrentFolder != null);
             DeleteFolderCommand = new RelayCommand(DeleteFolder, () => StorageManager != null && SelectedStorageItem != null && SelectedStorageItem is StorageFolder);
             UploadFileCommand = new RelayCommand(UploadFile, () => StorageManager != null && StorageManager.CurrentFolder != null);
+            UploadVersionCommand = new RelayCommand(UploadVersion, () => StorageManager != null && StorageManager.CurrentFolder != null);
         }
 
         private void UploadFile()
@@ -351,5 +357,10 @@ namespace Tango.MachineStudio.Storage.ViewModels
                 FileHandlers.Remove(handler);
             }
         }
+
+        private void UploadVersion()
+        {
+            _firmwareUpgrader.InvokeUpgradeUI();
+        }
     }
 }