- added websocket;

src/components/NotificationsSidebar/index.js

@@ -1,6 +1,8 @@
-import React, { useEffect, useState } from 'react';
+import React, { useEffect, useRef, useState } from 'react';
 import { __ } from '@wordpress/i18n';
 import { head, isEmpty, pathOr } from 'ramda';
+import SockJS from 'sockjs-client';
+import { Stomp } from '@stomp/stompjs';
 
 // store
 import { storeGet, useStore } from '../../store';
@@ -18,6 +20,8 @@ import { TabPanel, TabView } from 'primereact/tabview';
 import NotificationItem from './components/NotificationItem';
 import NotificationItemChosen from './components/NotificationItemChosen';
 
+const socketUrl = process.env.REACT_APP_API_ADDRESS_WS;
+
 const NotificationsSidebar = () => {
     const chosenCompanyId = useStore().main.chosenCompanyId();
     const userData = useStore().main.userData();
@@ -27,6 +31,10 @@ const NotificationsSidebar = () => {
     const [notifications, setNotifications] = useState([]);
     const [notificationsRead, setNotificationsRead] = useState([]);
     const [chosenMsg, setChosenMsg] = useState({});
+    const socket = useRef(null);
+    const stomp = useRef(null);
+    const [currentSubscription, setCurrentSubscription] = useState(null);
+    const [isConnected, setIsConnected] = useState(false);
 
     // Handle tab change
     const handleTabChange = (e) => {
@@ -61,7 +69,13 @@ const NotificationsSidebar = () => {
         const userData = storeGet.main.userData();
         const role = pathOr('', ['role', 'roleType'], userData);
 
-        if (userData.id && chosenCompanyId !== 0 && role === 'ROLE_BENEFICIARY') {
+        if (currentSubscription) {
+            console.log('UNsubscribed')
+            currentSubscription.unsubscribe();
+            setCurrentSubscription(null);
+        }
+
+        if (isConnected && userData.id && chosenCompanyId !== 0 && role === 'ROLE_BENEFICIARY') {
             setLoading(true);
             NotificationService.getNotifications(
                 userData.id,
@@ -72,7 +86,10 @@ const NotificationsSidebar = () => {
                     ['companyId', chosenCompanyId]
                 ]
             );
-        } else if (userData.id && role !== 'ROLE_BENEFICIARY') {
+            if (socket.current) {
+                subscribeTo(`/topic/notifications_user_${userData.id}_company_${chosenCompanyId}`)
+            }
+        } else if (isConnected && userData.id && role !== 'ROLE_BENEFICIARY') {
             setLoading(true);
             NotificationService.getNotifications(
                 userData.id,
@@ -82,6 +99,9 @@ const NotificationsSidebar = () => {
                     ['status', status]
                 ]
             );
+            if (socket.current) {
+                subscribeTo(`/topic/notifications_user_${userData.id}`)
+            }
         }
     }
 
@@ -127,9 +147,72 @@ const NotificationsSidebar = () => {
         set404FromErrorResponse(resp);
     }
 
+    const connectWebSocket = () => {
+        socket.current = new SockJS(socketUrl);
+        stomp.current = Stomp.over(socket.current);
+
+        stomp.current.configure({
+            debug: function(str) {
+                //console.log(str);
+            },
+            reconnectDelay: 5000,
+            heartbeatIncoming: 20000,
+            heartbeatOutgoing: 20000
+        });
+
+        stomp.current.connect(
+            {},
+            () => {
+                // connected
+                console.log('Websocket connected');
+                setIsConnected(true);
+            },
+            (error) => {
+                console.error('WebSocket Connection Error:', error);
+                setIsConnected(false);
+                setTimeout(connectWebSocket, 5000);
+            }
+        );
+    };
+
+    const subscribeTo = (topic) => {
+        console.log('subscribeTo', topic)
+        const subscription = stomp.current.subscribe(
+            topic,
+            (message) => {
+                try {
+                    const notification = JSON.parse(message.body);
+                    console.log('notification', notification)
+                    //setNotifications(prev => [notification, ...prev]);
+                } catch (error) {
+                    console.error('Error parsing notification:', error);
+                }
+            }
+        );
+
+        setCurrentSubscription(subscription);
+    }
+
     useEffect(() => {
         fetchMessages();
-    }, [chosenCompanyId, userData.id]);
+    }, [chosenCompanyId, userData.id, isConnected]);
+
+    useEffect(() => {
+        connectWebSocket();
+
+        return () => {
+            if (currentSubscription) {
+                currentSubscription.unsubscribe();
+                setCurrentSubscription(null);
+            }
+
+            if (stomp.current) {
+                stomp.current.disconnect(() => {
+                    console.log('WebSocket Disconnected');
+                });
+            }
+        };
+    }, []);
 
     return (
         <>