1 files changed, 20 insertions, 2 deletions
diff --git a/Software/Visual_Studio/Azure/Tango.AzureUtils.UI/ViewModels/EnvironmentFirmwareUpgradeViewVM.cs b/Software/Visual_Studio/Azure/Tango.AzureUtils.UI/ViewModels/EnvironmentFirmwareUpgradeViewVM.cs
index 931182c90..743561e8b 100644
--- a/Software/Visual_Studio/Azure/Tango.AzureUtils.UI/ViewModels/EnvironmentFirmwareUpgradeViewVM.cs
+++ b/Software/Visual_Studio/Azure/Tango.AzureUtils.UI/ViewModels/EnvironmentFirmwareUpgradeViewVM.cs
@@ -10,6 +10,7 @@ using System.Threading.Tasks;
 using System.Windows;
 using Tango.AzureUtils.Environment;
 using Tango.AzureUtils.Firmware;
+using Tango.BL.Enumerations;
 using Tango.Core.Commands;
 
 namespace Tango.AzureUtils.UI.ViewModels
@@ -32,6 +33,21 @@ namespace Tango.AzureUtils.UI.ViewModels
             set { _selectedDeploymentSlot = value; RaisePropertyChangedAuto(); }
         }
 
+        private List<MachineTypes> _machineTypes;
+        public List<MachineTypes> MachineTypes
+        {
+            get { return _machineTypes; }
+            set { _machineTypes = value; RaisePropertyChangedAuto(); }
+        }
+
+        private MachineTypes _selectedMachineType;
+        public MachineTypes SelectedMachineType
+        {
+            get { return _selectedMachineType; }
+            set { _selectedMachineType = value; RaisePropertyChangedAuto(); }
+        }
+
+
         private String _filePath;
         public String FilePath
         {
@@ -53,6 +69,8 @@ namespace Tango.AzureUtils.UI.ViewModels
         {
             UpgradeFirmwareCommand = new RelayCommand(UpgradeFirmware, () => FilePath != null);
             BrowseFileCommand = new RelayCommand(BrowseFile);
+
+            MachineTypes = Enum.GetValues(typeof(MachineTypes)).Cast<MachineTypes>().ToList();
         }
 
         public override void OnAuthenticated(IAzure azure, List<IWebAppBase> apps)
@@ -68,7 +86,7 @@ namespace Tango.AzureUtils.UI.ViewModels
         private async void BrowseFile()
         {
             OpenFileDialog dlg = new OpenFileDialog();
-            dlg.Title = "Select Tango Firmware Package";
+            dlg.Title = "Select Firmware Package";
             dlg.Filter = "Tango Firmware Package Files|*.tfp";
             if (dlg.ShowDialog().Value)
             {
@@ -93,7 +111,7 @@ namespace Tango.AzureUtils.UI.ViewModels
 
                 IsFree = false;
 
-                await _firmwareManager.InjectFirmwarePackage(SelectedDeploymentSlot, FilePath, null);
+                await _firmwareManager.InjectFirmwarePackage(SelectedDeploymentSlot, SelectedMachineType, FilePath, null);
             }
             catch (Exception ex)
             {