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

3 files changed, 177 insertions, 61 deletions

diff --git a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/ConnectionLostViewVM.cs b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/ConnectionLostViewVM.cs
index f1f4f69c0..e96f0ab62 100644
--- a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/ConnectionLostViewVM.cs
+++ b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/ConnectionLostViewVM.cs
@@ -3,12 +3,76 @@ using System.Collections.Generic;
 using System.Linq;
 using System.Text;
 using System.Threading.Tasks;
+using System.Timers;
+using System.Windows.Threading;
 using Tango.SharedUI;
 
 namespace Tango.MachineStudio.UI.ViewModels
 {
     public class ConnectionLostViewVM : DialogViewVM
     {
+        private Timer _reconnectTimer;
+
         public String Exception { get; set; }
+
+        private int _reconnectinSeconds;
+        public int ReconnectinSeconds
+        {
+            get { return _reconnectinSeconds; }
+            set { _reconnectinSeconds = value; RaisePropertyChangedAuto(); }
+        }
+
+        private bool _autoReconnect;
+        public bool AutoReconnect
+        {
+            get { return _autoReconnect; }
+            set { _autoReconnect = value; RaisePropertyChangedAuto(); }
+        }
+
+
+        public ConnectionLostViewVM() : base()
+        {
+            ReconnectinSeconds = 10;
+            _reconnectTimer = new Timer();
+            _reconnectTimer.Interval = TimeSpan.FromSeconds(1).TotalMilliseconds;
+            _reconnectTimer.Elapsed += _reconnectTimer_Elapsed;
+        }
+
+        private void _reconnectTimer_Elapsed(object sender, ElapsedEventArgs e)
+        {
+            ReconnectinSeconds--;
+
+            if (ReconnectinSeconds == -1)
+            {
+                _reconnectTimer.Stop();
+
+                InvokeUI(() =>
+                {
+                    Accept();
+                });
+            }
+        }
+
+        public override void OnShow()
+        {
+            base.OnShow();
+
+            if (AutoReconnect)
+            {
+                _reconnectTimer.Start();
+            }
+        }
+
+        protected override void Accept()
+        {
+            _reconnectTimer.Stop();
+            base.Accept();
+        }
+
+        protected override void Cancel()
+        {
+            base.Cancel();
+            _reconnectTimer.Stop();
+        }
     }
 }
diff --git a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/LoginViewVM.cs b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/LoginViewVM.cs
index cf34764d9..9c2367f93 100644
--- a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/LoginViewVM.cs
+++ b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/LoginViewVM.cs
@@ -102,6 +102,21 @@ namespace Tango.MachineStudio.UI.ViewModels
             set { _enableSlotSelection = value; RaisePropertyChangedAuto(); }
         }
 
+        private bool _isActiveDirectory;
+        public bool IsActiveDirectory
+        {
+            get { return _isActiveDirectory; }
+            set { _isActiveDirectory = value; RaisePropertyChangedAuto(); if (value) IsStandardUser = false; }
+        }
+
+        private bool _isStandardUser;
+        public bool IsStandardUser
+        {
+            get { return _isStandardUser; }
+            set { _isStandardUser = value; RaisePropertyChangedAuto(); if (value) IsActiveDirectory = false; }
+        }
+
+
 
         /// <summary>
         /// Gets or sets the login command.
@@ -131,6 +146,15 @@ namespace Tango.MachineStudio.UI.ViewModels
             Email = _settings.LastLoginEmail;
             DeploymentSlot = _settings.DeploymentSlot;
             RememberMe = _settings.RememberMe;
+            
+            if (_settings.LastLoginMethod == LoginMethod.ActiveDirectory)
+            {
+                IsActiveDirectory = true;
+            }
+            else
+            {
+                IsStandardUser = true;
+            }
 
             try
             {
@@ -154,11 +178,13 @@ namespace Tango.MachineStudio.UI.ViewModels
                     IsLogging = true;
                     InvalidateRelayCommands();
 
+                    LoginMethod loginMethod = IsActiveDirectory ? LoginMethod.ActiveDirectory : LoginMethod.StandardUser;
+
                     await Task.Factory.StartNew(() =>
                     {
                         _settings.DeploymentSlot = DeploymentSlot;
 
-                        LoginResponse result = _authenticationProvider.Login(Email, Password, _settings.ByPassEnvironmentVersionCheck).Response;
+                        LoginResponse result = _authenticationProvider.Login(Email, Password, loginMethod, _settings.ByPassEnvironmentVersionCheck).Response;
 
                         if (result.VersionChangeRequired && !_settings.ByPassEnvironmentVersionCheck)
                         {
@@ -183,6 +209,7 @@ namespace Tango.MachineStudio.UI.ViewModels
 
                         _settings.LastLoginEmail = Email;
                         _settings.RememberMe = RememberMe;
+                        _settings.LastLoginMethod = loginMethod;
                         _settings.LastLoginPassword = RememberMe ? cryptographer.Encrypt(Password) : null;
                         _settings.Save();
 
diff --git a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/MainViewVM.cs b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/MainViewVM.cs
index a7397835b..ce3b825c8 100644
--- a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/MainViewVM.cs
+++ b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/MainViewVM.cs
@@ -58,6 +58,8 @@ namespace Tango.MachineStudio.UI.ViewModels
         private IEventLogger _eventLogger;
         private MachineStudioSettings _settings;
         private MachineStudioWebClient _machineStudioWebClient;
+        private IExternalBridgeSecureClient _reconnectionMachine;
+        private MachineLoginViewVM _reconnectionMachineConfig;
 
         /// <summary>
         /// Gets or sets the current loaded module.
@@ -344,6 +346,13 @@ namespace Tango.MachineStudio.UI.ViewModels
             AboutCommand = new RelayCommand(ShowAboutDialog);
 
             ChangeAppThemeCommand = new RelayCommand<MachineStudioTheme>(ChangeTheme);
+
+            ApplicationManager.ReconnectionRequired += ApplicationManager_ReconnectionRequired;
+        }
+
+        private void ApplicationManager_ReconnectionRequired(object sender, EventArgs e)
+        {
+            ConnectToMachineSecure(_reconnectionMachine, _reconnectionMachineConfig);
         }
 
         private void MachineEventsStateProvider_EventsResolved(object sender, IEnumerable<MachinesEvent> e)
@@ -465,62 +474,7 @@ namespace Tango.MachineStudio.UI.ViewModels
 
                         if (x.SelectedMachine.RequiresAuthentication)
                         {
-                            //Check machine exist on my database first
-                            if (x.SelectedMachine.Machine == null)
-                            {
-                                _notificationProvider.ShowError($"The specified machine '{x.SelectedMachine.SerialNumber}' could not be found on the database. Aborting connection.");
-                                return;
-                            }
-
-                            _notificationProvider.ShowModalDialog<MachineLoginViewVM>(async (login) =>
-                            {
-                                using (NotificationProvider.PushTaskItem("Connecting to machine " + x.SelectedMachine.ToString() + "..."))
-                                {
-                                    try
-                                    {
-                                        await x.SelectedMachine.As<IExternalBridgeSecureClient>().Connect(new PMR.Integration.ExternalBridgeLoginRequest()
-                                        {
-                                            AppID = "Machine Studio",
-                                            HostName = Environment.MachineName,
-                                            Password = login.Password,
-                                            UserGuid = AuthenticationProvider.CurrentUser.Guid,
-                                            Intent = login.Intent,
-                                        });
-
-                                        ApplicationManager.SetConnectedMachine(x.SelectedMachine);
-                                        (x.SelectedMachine as IExternalBridgeSecureClient).SessionClosed += (_, __) =>
-                                        {
-                                            InvokeUI(() =>
-                                            {
-                                                _notificationProvider.ShowError("The remote machine has closed the current session. Machine disconnected.");
-                                                ApplicationManager.SetConnectedMachine(null);
-                                            });
-                                        };
-                                        PostMessage(new MachineConnectionChangedMessage() { Machine = x.SelectedMachine });
-                                        _eventLogger.Log(String.Format("Successfully connected to machine {0} via TCP", x.SelectedMachine.SerialNumber));
-
-                                        //if (x.UploadHardwareConfiguration)
-                                        //{
-                                        //    UploadHardwareConfiguration(false);
-                                        //}
-
-                                    }
-                                    catch (ResponseErrorException ex)
-                                    {
-                                        LogManager.Log(ex);
-                                        _eventLogger.Log(ex, "Error connecting to machine " + x.SelectedMachine.SerialNumber);
-                                        _notificationProvider.ShowError("Could not connect to the selected machine." + Environment.NewLine + ex.Container.ErrorMessage);
-                                    }
-                                    catch (Exception ex)
-                                    {
-                                        LogManager.Log(ex);
-                                        _eventLogger.Log(ex, "Error connecting to machine " + x.SelectedMachine.SerialNumber);
-                                        _notificationProvider.ShowError("Could not connect to the selected machine." + Environment.NewLine + ex.Message);
-                                    }
-
-                                    InvalidateRelayCommands();
-                                }
-                            });
+                            ConnectToMachineSecure(x.SelectedMachine as IExternalBridgeSecureClient);
                         }
                         else
                         {
@@ -633,6 +587,77 @@ namespace Tango.MachineStudio.UI.ViewModels
             InvalidateRelayCommands();
         }
 
+        private async void ConnectToMachineSecure(IExternalBridgeSecureClient machine, MachineLoginViewVM config)
+        {
+            using (NotificationProvider.PushTaskItem("Connecting to machine " + machine.ToString() + "..."))
+            {
+                try
+                {
+                    await machine.Connect(new PMR.Integration.ExternalBridgeLoginRequest()
+                    {
+                        AppID = "Machine Studio",
+                        HostName = Environment.MachineName,
+                        Password = config.Password,
+                        UserGuid = AuthenticationProvider.CurrentUser.Guid,
+                        Intent = config.Intent,
+                    });
+
+                    _reconnectionMachine = machine;
+                    _reconnectionMachineConfig = config;
+
+                    ApplicationManager.SetConnectedMachine(machine);
+                    machine.SessionClosed -= Machine_SessionClosed;
+                    machine.SessionClosed += Machine_SessionClosed;
+                    PostMessage(new MachineConnectionChangedMessage() { Machine = machine });
+                    _eventLogger.Log(String.Format("Successfully connected to machine {0} via TCP", machine.SerialNumber));
+
+                    //if (x.UploadHardwareConfiguration)
+                    //{
+                    //    UploadHardwareConfiguration(false);
+                    //}
+
+                }
+                catch (ResponseErrorException ex)
+                {
+                    LogManager.Log(ex);
+                    _eventLogger.Log(ex, "Error connecting to machine " + machine.SerialNumber);
+                    _notificationProvider.ShowError("Could not connect to the selected machine." + Environment.NewLine + ex.Container.ErrorMessage);
+                }
+                catch (Exception ex)
+                {
+                    LogManager.Log(ex);
+                    _eventLogger.Log(ex, "Error connecting to machine " + machine.SerialNumber);
+                    _notificationProvider.ShowError("Could not connect to the selected machine." + Environment.NewLine + ex.Message);
+                }
+
+                InvalidateRelayCommands();
+            }
+        }
+
+        private void ConnectToMachineSecure(IExternalBridgeSecureClient machine)
+        {
+            //Check machine exist on my database first
+            if (machine.Machine == null)
+            {
+                _notificationProvider.ShowError($"The specified machine '{machine.SerialNumber}' could not be found on the database. Aborting connection.");
+                return;
+            }
+
+            _notificationProvider.ShowModalDialog<MachineLoginViewVM>((login) =>
+            {
+                ConnectToMachineSecure(machine, login);
+            });
+        }
+
+        private void Machine_SessionClosed(object sender, EventArgs e)
+        {
+            InvokeUI(() =>
+            {
+                _notificationProvider.ShowError("The remote machine has closed the current session. Machine disconnected.");
+                ApplicationManager.SetConnectedMachine(null);
+            });
+        }
+
         private async void UploadHardwareConfiguration(bool showSuccessMessage = true)
         {
             try
@@ -706,10 +731,10 @@ namespace Tango.MachineStudio.UI.ViewModels
             {
                 IsModuleLoaded = false;
                 LogManager.Log(String.Format("Navigating to Home..."));
-                (MainView.Self as MainView).NavigationControl.NavigateTo("Home",() => 
-                {
-                    CurrentModule = module;
-                });
+                (MainView.Self as MainView).NavigationControl.NavigateTo("Home", () =>
+                 {
+                     CurrentModule = module;
+                 });
             }
         }