Reconnect functionallity.

Program.cs

@@ -8,17 +8,30 @@ using System.Diagnostics;
 
 class Program
 {
+    static async void notify(string title, string message)
+    {
+        string command = $"notify-send '{title}' '{message}'";
+        await ExecuteCommand(command);
+    }
+
     static async Task Main(string[] args)
     {
         string username = Environment.GetEnvironmentVariable("SNEK_USERNAME") ?? "anonymous";
         string password = Environment.GetEnvironmentVariable("SNEK_PASSWORD") ?? "anonymous";
 
+        while (true) // Loop for auto-reconnection
+        {
             using (var client = new ClientWebSocket())
             {
                 Uri serverUri = new Uri("wss://snek.molodetz.nl/rpc.ws");
-            await client.ConnectAsync(serverUri, CancellationToken.None);
 
-            var jsonMessage = JsonConvert.SerializeObject(new {
+                try
+                {
+                    await client.ConnectAsync(serverUri, CancellationToken.None);
+                    Console.WriteLine("Connected to the server.");
+
+                    var jsonMessage = JsonConvert.SerializeObject(new
+                    {
                         method = "login",
                         args = new[] { username, password }
                     });
@@ -26,10 +39,11 @@ class Program
                     var bytes = Encoding.UTF8.GetBytes(jsonMessage);
                     var buffer = new ArraySegment<byte>(bytes);
                     await client.SendAsync(buffer, WebSocketMessageType.Text, true, CancellationToken.None);
-
                     Console.WriteLine("Message sent: " + jsonMessage);
+
                     Boolean isLoggedIn = false;
-            var receiveBuffer = new byte[1024000]; // Increased buffer size
+                    var receiveBuffer = new byte[1024 * 1024]; // Optimized buffer size
+
                     while (client.State == WebSocketState.Open)
                     {
                         try
@@ -41,6 +55,13 @@ class Program
                             dynamic jsonResponse = JsonConvert.DeserializeObject(responseMessage);
                             if (!isLoggedIn)
                             {
+                                Boolean isSuccess = jsonResponse.success != null ? (bool)jsonResponse.success : false;
+                                if (!isSuccess)
+                                {
+                                    notify("Snek", $"Login as {username} failed");
+                                    break;
+                                }
+                                notify("Snek", "Logged in as " + username);
                                 isLoggedIn = true;
                                 continue;
                             }
@@ -49,16 +70,30 @@ class Program
 
                             if (receivedUsername != username)
                             {
-                        string command = $"notify-send '{receivedUsername}' '{message}'";
-                        await ExecuteCommand(command);
+                                notify(receivedUsername, message);
                             }
                         }
                         catch (Exception ex)
                         {
                             Console.WriteLine("Error processing message: " + ex.Message);
+                            break; // Exit the inner loop to reconnect
                         }
                     }
                 }
+                catch (System.InvalidOperationException ex)
+                {
+                    Console.WriteLine("Error connecting to server: " + ex.Message);
+                }
+                catch (Exception ex)
+                {
+                    Console.WriteLine("An error occurred: " + ex.Message);
+                }
+
+                // Wait before attempting to reconnect
+                Console.WriteLine("Reconnecting in 5 seconds...");
+                await Task.Delay(5000);
+            }
+        }
     }
 
     private static async Task ExecuteCommand(string command)