import React, { createContext, useCallback, useReducer } from "react"; // uuid import { v4 as uuid } from "uuid"; // components import ToastAlert from "@/components/ui/toast-alert"; export const toastContext = createContext({} as ContextType); // types type ToastAlert = { id: string; title: string; message?: string; type: "success" | "error" | "warning" | "info"; }; type ReducerActionType = { type: "SET_TOAST_ALERT" | "REMOVE_TOAST_ALERT"; payload: ToastAlert; }; type ContextType = { alerts?: ToastAlert[]; removeAlert: (id: string) => void; setToastAlert: (data: { title: string; type?: "success" | "error" | "warning" | "info" | undefined; message?: string | undefined; }) => void; }; type StateType = { toastAlerts?: ToastAlert[]; }; type ReducerFunctionType = (state: StateType, action: ReducerActionType) => StateType; export const initialState: StateType = { toastAlerts: [], }; export const reducer: ReducerFunctionType = (state, action) => { const { type, payload } = action; switch (type) { case "SET_TOAST_ALERT": return { ...state, toastAlerts: [...(state.toastAlerts ?? []), payload], }; case "REMOVE_TOAST_ALERT": return { ...state, toastAlerts: state.toastAlerts?.filter((toastAlert) => toastAlert.id !== payload.id), }; default: { return state; } } }; export const ToastContextProvider: React.FC<{ children: React.ReactNode }> = ({ children }) => { const [state, dispatch] = useReducer(reducer, initialState); const removeAlert = useCallback((id: string) => { dispatch({ type: "REMOVE_TOAST_ALERT", payload: { id, title: "", message: "", type: "success" }, }); }, []); const setToastAlert = useCallback( (data: { title: string; type?: "success" | "error" | "warning" | "info"; message?: string }) => { const id = uuid(); const { title, type, message } = data; dispatch({ type: "SET_TOAST_ALERT", payload: { id, title, message, type: type ?? "success" }, }); const timer = setTimeout(() => { removeAlert(id); clearTimeout(timer); }, 3000); }, [removeAlert] ); return ( {children} ); };