fix: host connections failing with "Authentication failed" (#791)

Co-authored-by: LukeGus <bugattiguy527@gmail.com>

src/backend/ssh/docker-console.ts

@@ -257,6 +257,12 @@ wss.on("connection", async (ws: WebSocket, req) => {
     }
   }
 
+  if (!token) {
+    const urlObj = new URL(req.url || "", "http://localhost");
+    const qp = urlObj.searchParams.get("token");
+    if (qp) token = qp;
+  }
+
   if (!token) {
     ws.close(1008, "Authentication required");
     return;

src/backend/ssh/terminal.ts

@@ -505,6 +505,12 @@ wss.on("connection", async (ws: WebSocket, req) => {
       }
     }
 
+    if (!token) {
+      const urlObj = new URL(req.url || "", "http://localhost");
+      const qp = urlObj.searchParams.get("token");
+      if (qp) token = qp;
+    }
+
     if (!token) {
       ws.close(1008, "Authentication required");
       return;

src/ui/features/terminal/Terminal.tsx

@@ -884,6 +884,10 @@ const TerminalInner = forwardRef<TerminalHandle, SSHTerminalProps>(
             .replace(/^https?:\/\//, "")
             .replace(/\/$/, "");
           baseWsUrl = `${wsProtocol}${wsHost}/ssh/websocket/`;
+          const storedJwt = localStorage.getItem("jwt");
+          if (storedJwt) {
+            baseWsUrl += `?token=${encodeURIComponent(storedJwt)}`;
+          }
         }
       } else {
         baseWsUrl = `${getBasePath()}/ssh/websocket/`;