diff --git a/hooks/firebase/useGetEvents.ts b/hooks/firebase/useGetEvents.ts index db4a658..b7114e0 100644 --- a/hooks/firebase/useGetEvents.ts +++ b/hooks/firebase/useGetEvents.ts @@ -20,14 +20,97 @@ const createEventHash = (event: any): string => { return hash.toString(36); }; +const fetchEvents = async (userId: string, familyId: string | undefined, isFamilyView: boolean) => { + const db = firestore(); + const eventsQuery = db.collection("Events"); + let constraints = []; + + if (isFamilyView) { + constraints = [ + eventsQuery.where("familyId", "==", familyId).where("private", "==", false), + eventsQuery.where("creatorId", "==", userId), + eventsQuery.where("attendees", "array-contains", userId) + ]; + } else { + constraints = [ + eventsQuery.where("creatorId", "==", userId), + eventsQuery.where("attendees", "array-contains", userId) + ]; + } + + const snapshots = await Promise.all(constraints.map(query => query.get())); + + const uniqueEvents = new Map(); + const processedHashes = new Set(); + const creatorIds = new Set(); + + snapshots.forEach(snapshot => { + snapshot.docs.forEach(doc => { + const event = doc.data(); + const hash = createEventHash(event); + + if (!processedHashes.has(hash)) { + processedHashes.add(hash); + creatorIds.add(event.creatorId); + uniqueEvents.set(doc.id, event); + } + }); + }); + + const creatorIdsArray = Array.from(creatorIds); + const creatorProfiles = new Map(); + + for (let i = 0; i < creatorIdsArray.length; i += 10) { + const chunk = creatorIdsArray.slice(i, i + 10); + const profilesSnapshot = await db + .collection("Profiles") + .where(firestore.FieldPath.documentId(), "in", chunk) + .get(); + + profilesSnapshot.docs.forEach(doc => { + creatorProfiles.set(doc.id, doc.data()?.eventColor || colorMap.pink); + }); + } + + return Array.from(uniqueEvents.entries()).map(([id, event]) => ({ + ...event, + id, + start: event.allDay + ? new Date(new Date(event.startDate.seconds * 1000).setHours(0, 0, 0, 0)) + : new Date(event.startDate.seconds * 1000), + end: event.allDay + ? new Date(new Date(event.endDate.seconds * 1000).setHours(0, 0, 0, 0)) + : new Date(event.endDate.seconds * 1000), + hideHours: event.allDay, + eventColor: creatorProfiles.get(event.creatorId) || colorMap.pink, + notes: event.notes + })); +}; + export const useGetEvents = () => { - const {user, profileData} = useAuthContext(); + const { user, profileData } = useAuthContext(); const isFamilyView = useAtomValue(isFamilyViewAtom); const queryClient = useQueryClient(); const lastSyncTimestamp = useRef(0); useEffect(() => { - if (!profileData?.familyId) return; + if (!user?.uid || !profileData?.familyId) return; + + const prefetchEvents = async () => { + await queryClient.prefetchQuery({ + queryKey: ["events", user.uid, false], // Personal events + queryFn: () => fetchEvents(user.uid, profileData.familyId, false), + staleTime: 5 * 60 * 1000, + }); + + await queryClient.prefetchQuery({ + queryKey: ["events", user.uid, true], // Family events + queryFn: () => fetchEvents(user.uid, profileData.familyId, true), + staleTime: 5 * 60 * 1000, + }); + }; + + prefetchEvents(); const unsubscribe = firestore() .collection('Households') @@ -40,7 +123,10 @@ export const useGetEvents = () => { const newTimestamp = data.lastSyncTimestamp.seconds; if (newTimestamp > lastSyncTimestamp.current) { lastSyncTimestamp.current = newTimestamp; - queryClient.invalidateQueries({queryKey: ["events", user?.uid, isFamilyView]}); + // Invalidate both queries + queryClient.invalidateQueries({ + queryKey: ["events", user.uid] + }); } } } @@ -48,92 +134,14 @@ export const useGetEvents = () => { }, console.error); return unsubscribe; - }, [profileData?.familyId, user?.uid, isFamilyView, queryClient]); + }, [profileData?.familyId, user?.uid, queryClient]); return useQuery({ queryKey: ["events", user?.uid, isFamilyView], - queryFn: async () => { - const db = firestore(); - const userId = user?.uid; - const familyId = profileData?.familyId; - - const eventsQuery = db.collection("Events"); - let constraints = []; - - if (isFamilyView) { - constraints = [ - eventsQuery - .where("familyId", "==", familyId) - .where("private", "==", false), - eventsQuery - .where("creatorId", "==", userId), - eventsQuery - .where("attendees", "array-contains", userId) - ]; - } else { - constraints = [ - eventsQuery.where("creatorId", "==", userId), - eventsQuery.where("attendees", "array-contains", userId) - ]; - } - - const snapshots = await Promise.all( - constraints.map(query => query.get()) - ); - - const uniqueEvents = new Map(); - const processedHashes = new Set(); - const creatorIds = new Set(); - - snapshots.forEach(snapshot => { - snapshot.docs.forEach(doc => { - const event = doc.data(); - const hash = createEventHash(event); - - if (!processedHashes.has(hash)) { - processedHashes.add(hash); - creatorIds.add(event.creatorId); - uniqueEvents.set(doc.id, event); - } - }); - }); - - const creatorIdsArray = Array.from(creatorIds); - const creatorProfiles = new Map(); - - for (let i = 0; i < creatorIdsArray.length; i += 10) { - const chunk = creatorIdsArray.slice(i, i + 10); - const profilesSnapshot = await db - .collection("Profiles") - .where(firestore.FieldPath.documentId(), "in", chunk) - .get(); - - profilesSnapshot.docs.forEach(doc => { - creatorProfiles.set(doc.id, doc.data()?.eventColor || colorMap.pink); - }); - } - - return Array.from(uniqueEvents.entries()).map(([id, event]) => { - const startSeconds = event.startDate.seconds; - const endSeconds = event.endDate.seconds; - - return { - ...event, - id, - start: event.allDay - ? new Date(new Date(startSeconds * 1000).setHours(0, 0, 0, 0)) - : new Date(startSeconds * 1000), - end: event.allDay - ? new Date(new Date(endSeconds * 1000).setHours(0, 0, 0, 0)) - : new Date(endSeconds * 1000), - hideHours: event.allDay, - eventColor: creatorProfiles.get(event.creatorId) || colorMap.pink, - notes: event.notes - }; - }); - }, + queryFn: () => fetchEvents(user?.uid!, profileData?.familyId, isFamilyView), staleTime: 5 * 60 * 1000, gcTime: Infinity, placeholderData: (previousData) => previousData, + enabled: Boolean(user?.uid), }); }; \ No newline at end of file