From b57f99763d0f53e4bda9afd47841e9245784c224 Mon Sep 17 00:00:00 2001
From: Roy <roy.mail.net@gmail.com>
Date: Sat, 16 Dec 2017 14:25:47 +0200
Subject: Implemented login remember me. Implemented database connection while
 loading screen + error handling.

---
 .../ViewModels/LoadingViewVM.cs                    | 50 ++++++++++++++++++----
 .../ViewModels/LoginViewVM.cs                      | 17 +++++++-
 2 files changed, 57 insertions(+), 10 deletions(-)

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

diff --git a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/LoadingViewVM.cs b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/LoadingViewVM.cs
index 72ab5aca9..f50b734f4 100644
--- a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/LoadingViewVM.cs
+++ b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/LoadingViewVM.cs
@@ -4,26 +4,58 @@ using System.Linq;
 using System.Text;
 using System.Threading;
 using System.Threading.Tasks;
+using Tango.Core.Threading;
+using Tango.DAL.Observables;
 using Tango.MachineStudio.Common.Modules;
 using Tango.MachineStudio.Common.Navigation;
+using Tango.MachineStudio.Common.Notifications;
 using Tango.SharedUI;
 
 namespace Tango.MachineStudio.UI.ViewModels
 {
     public class LoadingViewVM : ViewModel
     {
-        public LoadingViewVM(INavigationManager navigationManager, IStudioModuleLoader studioModuleLoader)
+        private INotificationProvider _notificationProvider;
+        private INavigationManager _navigationManager;
+        private IStudioModuleLoader _studioModuleLoader;
+
+        public LoadingViewVM(INavigationManager navigationManager, IStudioModuleLoader studioModuleLoader, INotificationProvider notificationProvider)
+        {
+            _navigationManager = navigationManager;
+            _studioModuleLoader = studioModuleLoader;
+            _notificationProvider = notificationProvider;
+            Load();
+        }
+
+        private void Load()
         {
-            Task.Factory.StartNew(() =>
+            StaThreadHelper.StartStaThread(() =>
             {
                 Thread.Sleep(3000);
-            }).ContinueWith((x) => 
-            {
-
-                studioModuleLoader.LoadModules();
-                navigationManager.NavigateTo(NavigationView.LoginView);
-
-            }, TaskScheduler.FromCurrentSynchronizationContext());
+                try
+                {
+                    ObservablesEntitiesAdapter.Instance.Initialize();
+                    InvokeUI(() =>
+                    {
+                        _studioModuleLoader.LoadModules();
+                        _navigationManager.NavigateTo(NavigationView.LoginView);
+                    });
+                }
+                catch (Exception ex)
+                {
+                    InvokeUINow(() =>
+                    {
+                        if (_notificationProvider.ShowQuestion("An error occurred while trying to connect to Twine database." + Environment.NewLine + "Would you like to try again?"))
+                        {
+                            Load();
+                        }
+                        else
+                        {
+                            Environment.Exit(0);
+                        }
+                    });
+                }
+            });
         }
     }
 }
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 7d75d3805..6fe90fa99 100644
--- a/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/LoginViewVM.cs
+++ b/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/ViewModels/LoginViewVM.cs
@@ -7,6 +7,7 @@ using System.Linq;
 using System.Text;
 using System.Threading.Tasks;
 using Tango.Core.Commands;
+using Tango.Core.Cryptography;
 using Tango.MachineStudio.Common.Authentication;
 using Tango.MachineStudio.Common.Navigation;
 using Tango.MachineStudio.Common.Notifications;
@@ -20,6 +21,7 @@ namespace Tango.MachineStudio.UI.ViewModels
         private IAuthenticationProvider _authenticationProvider;
         private INavigationManager _navigationManager;
         private INotificationProvider _notificationProvider;
+        private Rfc2898Cryptographer cryptographer;
 
         private String _email;
         [Required(ErrorMessage = "Email is required")]
@@ -30,6 +32,15 @@ namespace Tango.MachineStudio.UI.ViewModels
             set { _email = value; RaisePropertyChangedAuto(); }
         }
 
+        private bool _rememberMe;
+
+        public bool RememberMe
+        {
+            get { return _rememberMe; }
+            set { _rememberMe = value; RaisePropertyChangedAuto(); }
+        }
+
+
         public RelayCommand<String> LoginCommand { get; set; }
 
         public LoginViewVM(IAuthenticationProvider authenticationProvider, INavigationManager navigationManager, INotificationProvider notificationProvider)
@@ -39,7 +50,9 @@ namespace Tango.MachineStudio.UI.ViewModels
             _authenticationProvider = authenticationProvider;
             LoginCommand = new RelayCommand<String>(Login);
 
+            cryptographer = new Rfc2898Cryptographer();
             Email = SettingsManager.Default.MachineStudio.LastLoginEmail;
+            RememberMe = SettingsManager.Default.MachineStudio.RememberMe;
         }
 
         private void Login(String password)
@@ -51,7 +64,9 @@ namespace Tango.MachineStudio.UI.ViewModels
                     _authenticationProvider.Login(Email, password);
                     _navigationManager.NavigateTo(NavigationView.MainView);
                     SettingsManager.Default.MachineStudio.LastLoginEmail = Email;
-                    SettingsManager.Default.MachineStudio.LastLoginPassword = password;
+                    SettingsManager.Default.MachineStudio.RememberMe = RememberMe;
+
+                    SettingsManager.Default.MachineStudio.LastLoginPassword = RememberMe ? cryptographer.Encrypt(password) : null;
                     SettingsManager.SaveDefaultSettings();
                 }
                 catch
-- 
cgit v1.3.1