Function useInAppNotification

packages/shared/src/hooks/useInAppNotification.ts:37–109

()

const queue: NewNotification[] = [];
export const useInAppNotification = (): UseInAppNotification => {
  const client = useQueryClient();
  const { incrementUnreadCount } = useNotificationContext();
  const { data: notification } = useQuery<InAppNotification>({
    queryKey: IN_APP_NOTIFICATION_KEY,

    queryFn: () =>
      client.getQueryData<InAppNotification>(IN_APP_NOTIFICATION_KEY) || null,
  });
  const hasNotification = (): boolean =>
    !!client.getQueryData(IN_APP_NOTIFICATION_KEY);
  const setInAppNotification = (data: InAppNotification) =>
    client.setQueryData(IN_APP_NOTIFICATION_KEY, data);

  const displayNotification = (
    payload: NewNotification,
    { timer = 5000 }: NotifyOptionalProps = {},
  ) => {
    setInAppNotification({ notification: payload, timer });
  };

  const addToQueue = (newNotification: NewNotification) => {
    if (!newNotification || registeredNotification[newNotification.id]) {
      return;
    }

    registeredNotification[newNotification.id] = true;
    incrementUnreadCount();
    queue.push(newNotification);
    if (queue.length >= MAX_QUEUE_LENGTH) {
      queue.shift();
    }
    if (!hasNotification()) {
      displayNotification(queue.shift());
    }
  };

  const dismissNotification = () => {
    if (queue.length) {
      displayNotification(queue.shift());
      return;
    }
    setInAppNotification(null);
  };

  const clearNotifications = () => {
    queue.length = 0;
    dismissNotification();
  };

  useSubscription(
    () => ({
      query: NEW_NOTIFICATIONS_SUBSCRIPTION,
    }),
    {
      next: ({ newNotification }: PushNotificationSubscription) => {
        addToQueue(newNotification);
      },