import {useQuery, useQueryClient} from "react-query";
import firestore from "@react-native-firebase/firestore";
import {useAuthContext} from "@/contexts/AuthContext";
import {useAtomValue} from "jotai";
import {isFamilyViewAtom} from "@/components/pages/calendar/atoms";
import {colorMap} from "@/constants/colorMap";
import {uuidv4} from "@firebase/util";
import {useEffect} from "react";

export const useGetEvents = () => {
    const {user, profileData} = useAuthContext();
    const isFamilyView = useAtomValue(isFamilyViewAtom);
    const queryClient = useQueryClient();

    useEffect(() => {
        if (!profileData?.familyId) return;

        console.log(`[SYNC] Setting up sync listener for family: ${profileData.familyId}`);

        const unsubscribe = firestore()
            .collection('Households')
            .where("familyId", "==", profileData.familyId)
            .onSnapshot((snapshot) => {
                snapshot.docChanges().forEach((change) => {
                    if (change.type === 'modified') {
                        const data = change.doc.data();
                        if (data?.lastSyncTimestamp) {
                            console.log(`[SYNC] Change detected at ${data.lastSyncTimestamp.toDate()}`);
                            console.log(`[SYNC] Household ${change.doc.id} triggered refresh`);
                            queryClient.invalidateQueries(["events", user?.uid, isFamilyView]);
                        }
                    }
                });
            }, (error) => {
                console.error('[SYNC] Listener error:', error);
            });

        return () => {
            console.log('[SYNC] Cleaning up sync listener');
            unsubscribe();
        };
    }, [profileData?.familyId, user?.uid, isFamilyView, queryClient]);


    return useQuery({
        queryKey: ["events", user?.uid, isFamilyView],
        queryFn: async () => {
            console.log(`Fetching events - Family View: ${isFamilyView}, User: ${user?.uid}`);

            const db = firestore();
            const userId = user?.uid;
            const familyId = profileData?.familyId;
            let allEvents = [];

            if (isFamilyView) {
                const publicFamilyEvents = await db.collection("Events")
                    .where("familyId", "==", familyId)
                    .where("private", "==", false)
                    .get();

                const privateCreatorEvents = await db.collection("Events")
                    .where("familyId", "==", familyId)
                    .where("private", "==", true)
                    .where("creatorId", "==", userId)
                    .get();

                const privateAttendeeEvents = await db.collection("Events")
                    .where("private", "==", true)
                    .where("attendees", "array-contains", userId)
                    .get();

                console.log(`Found ${publicFamilyEvents.size} public events, ${privateCreatorEvents.size} private creator events, ${privateAttendeeEvents.size} private attendee events`);

                allEvents = [
                    ...publicFamilyEvents.docs.map(doc => ({...doc.data(), id: doc.id})),
                    ...privateCreatorEvents.docs.map(doc => ({...doc.data(), id: doc.id})),
                    ...privateAttendeeEvents.docs.map(doc => ({...doc.data(), id: doc.id}))
                ];
            } else {
                const [creatorEvents, attendeeEvents] = await Promise.all([
                    db.collection("Events")
                        .where("creatorId", "==", userId)
                        .get(),
                    db.collection("Events")
                        .where("attendees", "array-contains", userId)
                        .get()
                ]);

                console.log(`Found ${creatorEvents.size} creator events, ${attendeeEvents.size} attendee events`);

                allEvents = [
                    ...creatorEvents.docs.map(doc => ({...doc.data(), id: doc.id})),
                    ...attendeeEvents.docs.map(doc => ({...doc.data(), id: doc.id}))
                ];
            }

            const uniqueEventsMap = new Map();
            allEvents.forEach(event => {
                if (event.id) {
                    uniqueEventsMap.set(event.id, event);
                } else {
                    const newId = uuidv4();
                    console.log(`Generated new ID for event without ID: ${newId}`);
                    uniqueEventsMap.set(newId, {...event, id: newId});
                }
            });

            console.log(`Processing ${uniqueEventsMap.size} unique events`);

            const processedEvents = await Promise.all(
                Array.from(uniqueEventsMap.values()).map(async (event) => {
                    const profileSnapshot = await db
                        .collection("Profiles")
                        .doc(event.creatorId)
                        .get();

                    const profileData = profileSnapshot.data();
                    const eventColor = profileData?.eventColor || colorMap.pink;

                    return {
                        ...event,
                        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,
                        notes: event.notes,
                    };
                })
            );

            console.log(`Events processing completed, returning ${processedEvents.length} events`);
            return processedEvents;
        },
        staleTime: 5 * 60 * 1000,
        cacheTime: 30 * 60 * 1000,
        keepPreviousData: true,
        onError: (error) => {
            console.error('Error fetching events:', error);
        }
    });
};