fix: WebSocket cookie auth + connect on layout mount

2026-06-29 14:30:29 -04:00
parent 90289fede1
commit 369737565e
4 changed files with 36 additions and 21 deletions
@@ -1,6 +1,7 @@
 import { useEffect } from 'react';
 import { Link, Outlet, useLocation, useNavigate } from 'react-router-dom';
 import { useAuthStore } from '../stores/auth.ts';
+import { useWebSocketStore } from '../stores/ws.ts';
 import { ServerBar } from './ServerBar.tsx';
 import { ChannelList } from './ChannelList.tsx';
 import { MemberList } from './MemberList.tsx';
@@ -12,12 +13,16 @@ export function Layout() {
  const user = useAuthStore((state) => state.user);
  const fetchMe = useAuthStore((state) => state.fetchMe);
  const logout = useAuthStore((state) => state.logout);
+  const wsConnect = useWebSocketStore((s) => s.connect);
+  const wsDisconnect = useWebSocketStore((s) => s.disconnect);
  const navigate = useNavigate();
  const location = useLocation();

  useEffect(() => {
    fetchMe();
-  }, [fetchMe]);
+    wsConnect();
+    return () => { wsDisconnect(); };
+  }, [fetchMe, wsConnect, wsDisconnect]);

  useEffect(() => {
    if (!isLoading && !isAuthenticated && location.pathname !== '/login') {