added refresh token

src/hooks/useChatSocket.js

@@ -2,35 +2,56 @@ import { useEffect, useRef, useState, useCallback } from "react";
 import { Client } from "@stomp/stompjs";
 import SockJS from "sockjs-client";
 import { API_URL } from "../config/authConstants";
+import { getAccessToken, refreshAccessToken } from "../api/apiClient";
 
 export function useChatSocket(chatId, onMessage) {
   const clientRef = useRef(null);
+  const onMessageRef = useRef(onMessage);
   const [connected, setConnected] = useState(false);
 
+  // Держим актуальный колбэк в ref, чтобы не пересоздавать соединение
+  // при каждом ре-рендере родителя из-за нового onMessage
+  useEffect(() => {
+    onMessageRef.current = onMessage;
+  }, [onMessage]);
+
   useEffect(() => {
     if (!chatId) return;
 
-    const token = localStorage.getItem("token");
-    const socket = new SockJS(`${API_URL}/ws?token=${token}`);
-
     const client = new Client({
-      webSocketFactory: () => socket,
+      webSocketFactory: () => new SockJS(`${API_URL}/ws`),
       reconnectDelay: 3000,
+      beforeConnect: async () => {
+        // На каждую попытку коннекта — свежий токен из localStorage
+        client.connectHeaders = {
+          Authorization: `Bearer ${getAccessToken()}`,
+        };
+      },
       onConnect: () => {
         setConnected(true);
         client.subscribe(`/topic/chat/${chatId}`, (msg) => {
           const body = JSON.parse(msg.body);
-          onMessage(body);
+          onMessageRef.current(body);
         });
       },
       onDisconnect: () => setConnected(false),
+      onStompError: async (frame) => {
+        // Если сервер отклонил CONNECT из-за протухшего токена — пробуем обновить и переподключиться
+        if (frame.headers?.message?.includes("Unauthorized")) {
+          try {
+            await refreshAccessToken();
+          } catch {
+            // refresh тоже не удался — дальше решает общий логин-флоу (redirect на /)
+          }
+        }
+      },
     });
 
     client.activate();
     clientRef.current = client;
 
     return () => client.deactivate();
-  }, [chatId, onMessage]);
+  }, [chatId]);
 
   const sendMessage = useCallback((content) => {
     if (!clientRef.current?.connected) return;
@@ -41,4 +62,4 @@ export function useChatSocket(chatId, onMessage) {
   }, [chatId]);
 
   return { connected, sendMessage };
-}
+}