@tuturuuu/ui 0.32.0 → 0.33.0

package/src/hooks/use-notifications.ts

@@ -1,16 +1,14 @@
 import {
-  type QueryClient,
   useInfiniteQuery,
   useMutation,
   useQuery,
   useQueryClient,
 } from '@tanstack/react-query';
-import {
-  createRealtimeClient,
-  type RealtimeChannel,
-  type SupabaseClient,
-} from '@tuturuuu/supabase/next/realtime-browser';
-import { useEffect } from 'react';
+
+export {
+  __resetNotificationSubscriptionRegistryForTests,
+  useNotificationSubscription,
+} from './notification-subscription';
 export type NotificationType =
   | 'task_assigned'
   | 'task_updated'
@@ -78,98 +76,8 @@ interface UseNotificationsOptions {
   readOnly?: boolean;
   type?: NotificationType;
 }
-interface NotificationSubscriptionEntry {
-  channel: RealtimeChannel;
-  queryClientRefs: Map<QueryClient, number>;
-  subscriberCount: number;
-  supabase: SupabaseClient;
-}
 export const UNREAD_COUNT_STALE_TIME_MS = 5 * 60 * 1000;
 export const UNREAD_COUNT_FALLBACK_INTERVAL_MS = 15 * 60 * 1000;
-const notificationSubscriptionRegistry = new Map<
-  string,
-  NotificationSubscriptionEntry
->();
-function invalidateNotificationQueries(
-  queryClientRefs: Map<QueryClient, number>
-) {
-  for (const queryClient of queryClientRefs.keys()) {
-    queryClient.invalidateQueries({ queryKey: ['notifications'] });
-    queryClient.invalidateQueries({
-      queryKey: ['notifications', 'unread-count'],
-    });
-  }
-}
-function addNotificationQueryClientRef(
-  entry: NotificationSubscriptionEntry,
-  queryClient: QueryClient
-) {
-  entry.subscriberCount += 1;
-  entry.queryClientRefs.set(
-    queryClient,
-    (entry.queryClientRefs.get(queryClient) ?? 0) + 1
-  );
-}
-function releaseNotificationQueryClientRef(
-  entry: NotificationSubscriptionEntry,
-  queryClient: QueryClient
-) {
-  entry.subscriberCount -= 1;
-  const currentRefCount = entry.queryClientRefs.get(queryClient) ?? 0;
-  if (currentRefCount <= 1) {
-    entry.queryClientRefs.delete(queryClient);
-  } else {
-    entry.queryClientRefs.set(queryClient, currentRefCount - 1);
-  }
-}
-function createNotificationSubscriptionEntry(
-  userId: string,
-  queryClient: QueryClient
-) {
-  const queryClientRefs = new Map<QueryClient, number>([[queryClient, 1]]);
-  const supabase = createRealtimeClient();
-  const invalidateQueries = () =>
-    invalidateNotificationQueries(queryClientRefs);
-  const channel = supabase
-    .channel(`notifications-${userId}`)
-    .on(
-      'postgres_changes',
-      {
-        event: 'INSERT',
-        schema: 'public',
-        table: 'notifications',
-        filter: `user_id=eq.${userId}`,
-      },
-      invalidateQueries
-    )
-    .on(
-      'postgres_changes',
-      {
-        event: 'UPDATE',
-        schema: 'public',
-        table: 'notifications',
-        filter: `user_id=eq.${userId}`,
-      },
-      invalidateQueries
-    )
-    .on(
-      'postgres_changes',
-      {
-        event: 'DELETE',
-        schema: 'public',
-        table: 'notifications',
-        filter: `user_id=eq.${userId}`,
-      },
-      invalidateQueries
-    )
-    .subscribe();
-  return {
-    channel,
-    queryClientRefs,
-    subscriberCount: 1,
-    supabase,
-  } satisfies NotificationSubscriptionEntry;
-}
 export function dedupeNotifications(
   notifications: Notification[]
 ): Notification[] {
@@ -665,47 +573,3 @@ export function useDeleteNotification() {
     },
   });
 }
-
-/**
- * Hook to subscribe to realtime notification updates
- * Subscribes to all notifications for the user, regardless of workspace
- */
-export function useNotificationSubscription(
-  _wsId: string | null,
-  userId: string
-) {
-  const queryClient = useQueryClient();
-
-  useEffect(() => {
-    if (!userId) return;
-
-    const registryKey = userId;
-    let entry = notificationSubscriptionRegistry.get(registryKey);
-
-    if (entry) {
-      addNotificationQueryClientRef(entry, queryClient);
-    } else {
-      entry = createNotificationSubscriptionEntry(userId, queryClient);
-      notificationSubscriptionRegistry.set(registryKey, entry);
-    }
-
-    return () => {
-      if (!entry) return;
-
-      releaseNotificationQueryClientRef(entry, queryClient);
-
-      if (entry.subscriberCount <= 0) {
-        notificationSubscriptionRegistry.delete(registryKey);
-        entry.supabase.removeChannel(entry.channel);
-      }
-    };
-  }, [userId, queryClient]);
-}
-
-export function __resetNotificationSubscriptionRegistryForTests() {
-  for (const entry of notificationSubscriptionRegistry.values()) {
-    entry.supabase.removeChannel(entry.channel);
-  }
-
-  notificationSubscriptionRegistry.clear();
-}