From 82ec837569914a7e4566a52c1b0a393ee396b4b8 Mon Sep 17 00:00:00 2001
From: Cyberarm <matthewlikesrobots@gmail.com>
Date: Tue, 13 Oct 2026 23:33:32 -0500
Subject: [PATCH] Move WebSocketClient error handler up a level to catch
 initial connection errors

---
 lib/websocket_client.rb | 17 +++++++++--------
 1 file changed, 9 insertions(+), 8 deletions(-)

diff --git a/lib/websocket_client.rb b/lib/websocket_client.rb
index f12b746..d4d22c9 100644
--- a/lib/websocket_client.rb
+++ b/lib/websocket_client.rb
@@ -34,15 +34,16 @@ module W3DHubLauncher
           while message = connection.read
             @events[:message].call(message)
           end
-        # FIXME: Don't rescue for all ta errors?
-        rescue => error
-          @errored = true
-          @events[:error]&.call(error)
-        ensure
-          @events[:close]&.call unless @errored
-          @connection = nil
-          @errored = false
         end
+
+      # FIXME: Don't rescue for all ta errors?
+      rescue => error
+        @errored = true
+        @events[:error]&.call(error)
+      ensure
+        @events[:close]&.call unless @errored
+        @connection = nil
+        @errored = false
       end
 
       self