fix(notification): refactors and fixes user notifications feature (#446)

Author: aimedivin

package-lock.json

@@ -8,6 +8,7 @@ export const getAllNotification = gql`
       receiver
       createdAt
       read
+      type
       sender {
         profile {
           firstName
@@ -38,6 +39,27 @@ export const NotificationSubscription = gql`
       createdAt
       message
       read
+      type
+      receiver
+      id
+    }
+  }
+`;
+export const PUSH_NOTIFICATION_SUB = gql`
+  subscription PushNotificationSub($receiverId: String!) {
+    pushNotification(receiverId: $receiverId) {
+      sender {
+        profile {
+          firstName
+          lastName
+          name
+          avatar
+        }
+      }
+      createdAt
+      message
+      read
+      type
       receiver
       id
     }
@@ -82,4 +104,4 @@ export const updatedPushNotifications = gql`
   query Query($getUpdatedPushNotificationsId: ID!) {
     getUpdatedPushNotifications(id: $getUpdatedPushNotificationsId)
   }
-`;
+`;

src/Mutations/notificationMutation.tsx

@@ -12,6 +12,7 @@ import { UserContext } from '../hook/useAuth';
 import {
   getAllNotification,
   NotificationSubscription,
+  PUSH_NOTIFICATION_SUB,
 } from '../Mutations/notificationMutation';
 import { MenuContext } from '../hook/menuProvider';
 import ToggleThemeButton from './TogglethemeIcon';
@@ -74,6 +75,14 @@ function DashHeader() {
       receiver: user?.userId,
     },
   });
+  useSubscription(PUSH_NOTIFICATION_SUB, {
+    onData: (data) => {
+      setNotificationData([data.data.data.pushNotification, ...notifications]);
+    },
+    variables: {
+      receiverId: user?.userId,
+    },
+  });
 
   /* istanbul ignore next */
   const handleShowProfileDropdown = () =>