import {useCallback} from 'react';
import {ProfileDocument, useUpdateMeMutation} from '../generated/graphql';
import {create} from 'zustand';
import {persist} from 'zustand/middleware';
import {useSession} from 'next-auth/react';

type Store = {
  eventsToBeAdded: string[];
  addEvent: (eventId: string) => void;
  clear: () => void;
};

const store = create<Store>(
  persist(
    (set, get) => ({
      eventsToBeAdded: [],
      addEvent: eventId =>
        set({eventsToBeAdded: [...get().eventsToBeAdded, eventId]}),
      clear: () => set({eventsToBeAdded: []}),
    }),
    {
      name: 'add-events',
    }
  )
);

const useAddToEvents = () => {
  const [updateProfile] = useUpdateMeMutation();
  const session = useSession();
  const isAuth = session.status === 'authenticated';
  const eventsToBeAdded = store(s => s.eventsToBeAdded);
  const addEvent = store(s => s.addEvent);
  const clearStore = store(s => s.clear);

  const saveStoredEvents = useCallback(() => {
    if (eventsToBeAdded.length > 0) {
      updateProfile({
        variables: {
          userUpdate: {
            events: eventsToBeAdded,
          },
        },
        refetchQueries: [ProfileDocument],
      });
      clearStore();
    }
  }, [eventsToBeAdded]);

  const addToEvent = (eventId: string) => {
    if (isAuth) {
      updateProfile({
        variables: {
          userUpdate: {
            events: [eventId],
          },
        },
        refetchQueries: [ProfileDocument],
      });
    } else addEvent(eventId);
  };

  return {eventsToBeAdded, saveStoredEvents, addToEvent};
};

export default useAddToEvents;