Show error message when can't connect

1 files changed, 26 insertions, 2 deletions

@@ -2,6 +2,7 @@ import asyncio
 import os
 import pkgutil
 import subprocess
+import sys
 from typing import Any
 
 import websockets
@@ -106,6 +107,17 @@ class Lingo2GameContext:
 
         async_start(self.send_msgs([msg]), name="game Connected")
 
+    def send_connection_refused(self, text):
+        if self.server is None:
+            return
+
+        msg = {
+            "cmd": "ConnectionRefused",
+            "text": text,
+        }
+
+        async_start(self.send_msgs([msg]), name="game ConnectionRefused")
+
     def send_item_sent_notification(self, item_name, receiver_name, item_flags):
         if self.server is None:
             return
@@ -242,8 +254,20 @@ class Lingo2ClientContext(CommonContext):
         return ui
 
     async def server_auth(self, password_requested: bool = False):
-        self.auth = self.username
-        await self.send_connect()
+        if password_requested:
+            if self.password is None:
+                self.manager.game_ctx.send_connection_refused("Slot requires a password.")
+            else:
+                self.manager.game_ctx.send_connection_refused("Invalid password.")
+        else:
+            self.auth = self.username
+            await self.send_connect()
+
+    def handle_connection_loss(self, msg: str):
+        super().handle_connection_loss(msg)
+
+        exc_info = sys.exc_info()
+        self.manager.game_ctx.send_connection_refused(str(exc_info[1]))
 
     def on_package(self, cmd: str, args: dict):
         if cmd == "RoomInfo":