Check result of Connect in MainWindowViewModel.

Author: bitbound

Desktop.Win/ViewModels/MainWindowViewModel.cs

@@ -206,48 +206,51 @@ public async Task Init()
 
             try
             {
-                await _casterSocket.Connect(_conductor.Host);
+                var result = await _casterSocket.Connect(_conductor.Host);
 
-                if (_casterSocket.Connection is null)
+                if (result)
                 {
-                    return;
-                }
+                    _casterSocket.Connection.Closed += (ex) =>
+                    {
+                        App.Current?.Dispatcher?.Invoke(() =>
+                        {
+                            Viewers.Clear();
+                            SessionID = "Disconnected";
+                        });
+                        return Task.CompletedTask;
+                    };
 
-                _casterSocket.Connection.Closed += async (ex) =>
-                {
-                    await App.Current?.Dispatcher?.InvokeAsync(() =>
+                    _casterSocket.Connection.Reconnecting += (ex) =>
                     {
-                        Viewers.Clear();
-                        SessionID = "Disconnected";
-                    });
-                };
+                        App.Current?.Dispatcher?.Invoke(() =>
+                        {
+                            Viewers.Clear();
+                            SessionID = "Reconnecting";
+                        });
+                        return Task.CompletedTask;
+                    };
 
-                _casterSocket.Connection.Reconnecting += async (ex) =>
-                {
-                    await App.Current?.Dispatcher?.InvokeAsync(() =>
+                    _casterSocket.Connection.Reconnected += async (arg) =>
                     {
-                        Viewers.Clear();
-                        SessionID = "Reconnecting";
-                    });
-                };
+                        await GetSessionID();
+                    };
 
-                _casterSocket.Connection.Reconnected += async (arg) =>
-                {
-                    await GetSessionID();
-                };
+                    await DeviceInitService.GetInitParams();
+                    ApplyBranding();
 
-                await DeviceInitService.GetInitParams();
-                ApplyBranding();
+                    await GetSessionID();
 
-                await GetSessionID();
+                    return;
+                }
             }
             catch (Exception ex)
             {
                 Logger.Write(ex);
-                SessionID = "Failed";
-                MessageBox.Show(Application.Current.MainWindow, "Failed to connect to server.", "Connection Failed", MessageBoxButton.OK, MessageBoxImage.Warning);
-                return;
             }
+
+            // If we got here, something went wrong.
+            SessionID = "Failed";
+            MessageBox.Show(Application.Current.MainWindow, "Failed to connect to server.", "Connection Failed", MessageBoxButton.OK, MessageBoxImage.Warning);
         }
 
         public void PromptForHostName()

Desktop.XPlat/ViewModels/MainWindowViewModel.cs

@@ -163,48 +163,55 @@ public async Task Init()
 
                 _conductor.ProcessArgs(new string[] { "-mode", "Normal", "-host", Host });
 
-                await _casterSocket.Connect(_conductor.Host);
+                var result = await _casterSocket.Connect(_conductor.Host);
 
-                if (_casterSocket.Connection is null)
+                if (result)
                 {
-                    return;
-                }
+                    if (_casterSocket.Connection is null)
+                    {
+                        return;
+                    }
 
-                _casterSocket.Connection.Closed += async (ex) =>
-                {
-                    await Dispatcher.UIThread.InvokeAsync(() =>
+                    _casterSocket.Connection.Closed += async (ex) =>
                     {
-                        SessionID = "Disconnected";
-                    });
-                };
+                        await Dispatcher.UIThread.InvokeAsync(() =>
+                        {
+                            SessionID = "Disconnected";
+                        });
+                    };
 
-                _casterSocket.Connection.Reconnecting += async (ex) =>
-                {
-                    await Dispatcher.UIThread.InvokeAsync(() =>
+                    _casterSocket.Connection.Reconnecting += async (ex) =>
                     {
-                        SessionID = "Reconnecting";
-                    });
-                };
+                        await Dispatcher.UIThread.InvokeAsync(() =>
+                        {
+                            SessionID = "Reconnecting";
+                        });
+                    };
 
-                _casterSocket.Connection.Reconnected += async (arg) =>
-                {
-                    await GetSessionID();
-                };
+                    _casterSocket.Connection.Reconnected += async (arg) =>
+                    {
+                        await GetSessionID();
+                    };
 
-                await DeviceInitService.GetInitParams();
+                    await DeviceInitService.GetInitParams();
 
-                ApplyBranding();
+                    ApplyBranding();
+
+                    await _casterSocket.SendDeviceInfo(_conductor.ServiceID, Environment.MachineName, _conductor.DeviceID);
+                    await _casterSocket.GetSessionID();
+
+                    return;
+                }
 
-                await _casterSocket.SendDeviceInfo(_conductor.ServiceID, Environment.MachineName, _conductor.DeviceID);
-                await _casterSocket.GetSessionID();
             }
             catch (Exception ex)
             {
                 Logger.Write(ex);
-                _sessionID = "Failed";
-                await MessageBox.Show("Failed to connect to server.", "Connection Failed", MessageBoxType.OK);
-                return;
             }
+
+            // If we got here, something went wrong.
+            _sessionID = "Failed";
+            await MessageBox.Show("Failed to connect to server.", "Connection Failed", MessageBoxType.OK);
         }
 
         public async Task PromptForHostName()