Fix reconnecting

Author: PiotrMachowski

Home Assistant Taskbar Menu/Connection/HaClientContext.cs

@@ -1,6 +1,7 @@
 using System;
 using System.Collections.Generic;
 using System.Threading.Tasks;
+using System.Windows;
 using System.Windows.Threading;
 using Home_Assistant_Taskbar_Menu.Entities;
 using Home_Assistant_Taskbar_Menu.Utils;
@@ -18,11 +19,10 @@ public static void Initialize(Configuration configuration)
             HomeAssistantWebsocketClient = new HomeAssistantWebsocketsClient(Configuration);
         }
 
-        public static async void Recreate()
+        public static void Recreate()
         {
-            HomeAssistantWebsocketClient?.Disconnect();
-            HomeAssistantWebsocketClient = new HomeAssistantWebsocketsClient(Configuration);
-            await Start();
+            System.Diagnostics.Process.Start(Application.ResourceAssembly.Location);
+            Application.Current.Shutdown();
         }
 
         public static async Task Start()

Home Assistant Taskbar Menu/Connection/HomeAssistantWebsocketsClient.cs

@@ -95,9 +95,14 @@ public HomeAssistantWebsocketsClient(Configuration configuration)
             _websocketClient.ReconnectionHappened.Subscribe(recInfo =>
             {
                 {
+                    ConsoleWriter.WriteLine($"RECONNECTION HAPPENED: {recInfo.Type}", ConsoleColor.Yellow);
+                    if (recInfo.Type == ReconnectionType.NoMessageReceived)
+                    {
+                        HaClientContext.Recreate();
+                    }
+
                     if (recInfo.Type != ReconnectionType.Initial)
                     {
-                        ConsoleWriter.WriteLine($"RECONNECTION HAPPENED: {recInfo.Type}", ConsoleColor.Yellow);
                         Authenticated = false;
                         Authenticate();
                     }