forbidals

Back Back Up Advanced Editor

'use client';

import React from 'react';
import { useForegroundNotifications } from '@/lib/firebase/useForegroundNotifications';
import { useSyncFCMToken } from '@/lib/firebase/useSyncFCMToken';
import ServiceWorkerNavigationListener from '@/lib/firebase/ServiceWorkerNavigationListener';
import {
  NotificationProvider,
  useNotification,
} from '@/lib/firebase/NotificationContext';
import { useQueryClient } from '@tanstack/react-query';
import { useGetSchoolSettings } from '@/lib/api/student/queryHooks';
import { showNotificationIfNeeded } from './notificationHelper';

/**
 * Inner component to use the hook inside the provider
 */

const handleNotification = React.useCallback(
    (payload) => {
      // Check if this is a navigation message (from clicked notification)
      // Navigation messages should NOT trigger a new notification
      const isNavigationMessage =
        payload.action === 'navigate' || payload.type === 'NAVIGATE';

// If it's a navigation message, skip notification display
      if (isNavigationMessage) {
        // console.log('[PushNotification] Navigation message received, skipping notification display');
        return;
      }

// Update the context
      setLastNotification(payload);

// Show notification if needed (e.g. not on chat page)
      showNotificationIfNeeded(payload, logoUrl);

// Invalidate queries based on notification type to ensure fresh data
      const type = payload.data?.type || payload.notification?.type;
      const lowerType = type?.toLowerCase();

// Always invalidate notifications list
      queryClient.invalidateQueries({ queryKey: ['student', 'notifications'] });

if (lowerType) {
        if (lowerType === 'assignment') {
          queryClient.invalidateQueries({
            queryKey: ['student', 'assignments'],
          });
        } else if (lowerType === 'message') {
          // ChatWindow handles this, but good to have backup
          queryClient.invalidateQueries({ queryKey: ['chat', 'messages'] });
        } else if (
          lowerType === 'exam' ||
          lowerType === 'exam result' ||
          lowerType === 'result'
        ) {
          queryClient.invalidateQueries({ queryKey: ['online-exam-list'] });
          queryClient.invalidateQueries({ queryKey: ['offline-exam-list'] });
          queryClient.invalidateQueries({
            queryKey: ['online-exam-result-list'],
          });
        } else if (lowerType === 'lesson') {
          queryClient.invalidateQueries({ queryKey: ['lessons'] });
          queryClient.invalidateQueries({ queryKey: ['subject-details'] });
        } else if (lowerType === 'attendance') {
          queryClient.invalidateQueries({ queryKey: ['attendance'] });
        } else if (lowerType === 'class section') {
          queryClient.invalidateQueries({ queryKey: ['subject-details'] });
        }
      }

// Call the original prop if provided
      if (onNotificationReceived) {
        onNotificationReceived(payload);
      }
    },
    [setLastNotification, queryClient, onNotificationReceived, logoUrl]
  );

useForegroundNotifications(handleNotification);
  useSyncFCMToken();

return (
    <>
      <ServiceWorkerNavigationListener onMessage={handleNotification} />
      {children}
    </>
  );
};

/**
 * PushNotificationLayout centralizes all notification-related hooks.
 * It keeps hooks alive and renders the service worker listener once.
 */
const PushNotificationLayout = ({
  children,
  onNotificationReceived = () => {},
}) => {
  return (
    <NotificationProvider>
      <PushNotificationContent onNotificationReceived={onNotificationReceived}>
        {children}
      </PushNotificationContent>
    </NotificationProvider>
  );
};

export default PushNotificationLayout;