- updates;

src/components/NotificationsSidebar/index.js

@@ -1,9 +1,11 @@
-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';
+import { useStore } from '../../store';
 
 // api
 import NotificationService from '../../service/notification-service';
@@ -15,6 +17,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();
@@ -24,6 +28,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) => {
@@ -60,21 +68,94 @@ const NotificationsSidebar = () => {
 
     }
 
+    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) => {
+        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(() => {
         const role = pathOr('', ['role', 'roleType'], userData);
 
+        if (currentSubscription) {
+            currentSubscription.unsubscribe();
+            setCurrentSubscription(null);
+        }
+
         console.log('chosenCompanyId', chosenCompanyId, role)
-        if (userData.id && chosenCompanyId !== 0 && role === 'ROLE_BENEFICIARY') {
+        if (isConnected && userData.id && chosenCompanyId !== 0 && role === 'ROLE_BENEFICIARY') {
             NotificationService.getNotifications(userData.id, getNotifications, errGetNotifications, [
                 ['status', 'UNREAD'],
                 ['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') {
             NotificationService.getNotifications(userData.id, getNotifications, errGetNotifications, [
                 ['status', 'UNREAD']
             ]);
+            if (socket.current) {
+                subscribeTo(`/topic/notifications_user_${userData.id}`)
+            }
         }
-    }, [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 (
         <>