/* eslint-disable @typescript-eslint/no-explicit-any */ "use client"; import { useQuery, useQueryClient } from "@tanstack/react-query"; import { useCookie } from "react-use"; import { useRouter } from "next/navigation"; import { ProjectType, User } from "@/types"; import { api } from "@/lib/api"; import { toast } from "sonner"; import MY_TOKEN_KEY from "@/lib/get-cookie-name"; export const useUser = (initialData?: { user: User | null; errCode: number | null; projects: ProjectType[]; }) => { const client = useQueryClient(); const router = useRouter(); const [, setCurrentRoute, removeCurrentRoute] = useCookie("deepsite-currentRoute"); const [, setToken, removeToken] = useCookie(MY_TOKEN_KEY()); const { data: { user, errCode } = { user: null, errCode: null }, isLoading } = useQuery({ queryKey: ["user.me"], queryFn: async () => { return { user: initialData?.user || null, errCode: initialData?.errCode || null }; }, refetchOnWindowFocus: false, refetchOnReconnect: false, refetchOnMount: false, retry: false, initialData: initialData ? { user: initialData?.user, errCode: initialData?.errCode } : undefined, enabled: false, }); const { data: loadingAuth } = useQuery({ queryKey: ["loadingAuth"], queryFn: async () => false, refetchOnWindowFocus: false, refetchOnReconnect: false, refetchOnMount: false, }); const setLoadingAuth = (value: boolean) => { client.setQueryData(["setLoadingAuth"], value); }; const { data: projects } = useQuery({ queryKey: ["me.projects"], queryFn: async () => [], refetchOnWindowFocus: false, refetchOnReconnect: false, refetchOnMount: false, initialData: initialData?.projects || [], }); const setProjects = (projects: ProjectType[]) => { client.setQueryData(["me.projects"], projects); }; const openLoginWindow = async () => { setCurrentRoute(window.location.pathname); return router.push("/auth"); }; const loginFromCode = async (code: string) => { setLoadingAuth(true); if (loadingAuth) return; await api .post("/auth", { code }) .then(async (res: any) => { if (res.data && res.data.access_token) { const expiresIn = res.data.expires_in || 3600; const expiresDate = new Date(); expiresDate.setTime(expiresDate.getTime() + expiresIn * 1000); const cookieOptions: any = { expires: expiresDate, path: '/', sameSite: 'lax', }; if (window.location.protocol === 'https:') { cookieOptions.secure = true; } setToken(res.data.access_token, cookieOptions); const cookieString = `${MY_TOKEN_KEY()}=${res.data.access_token}; path=/; max-age=${expiresIn}; samesite=lax${cookieOptions.secure ? '; secure' : ''}`; document.cookie = cookieString; const meResponse = await api.get("/me"); if (meResponse.data) { client.setQueryData(["user.me"], { user: meResponse.data.user, errCode: null, }); if (meResponse.data.projects) { setProjects(meResponse.data.projects); } } setTimeout(() => { window.location.href = "/"; }, 100); toast.success("Login successful"); } }) .catch((err: any) => { toast.error(err?.data?.message ?? err.message ?? "An error occurred"); }) .finally(() => { setLoadingAuth(false); }); }; const logout = async () => { removeToken(); removeCurrentRoute(); toast.success("Logout successful"); client.clear(); window.location.reload(); }; return { user, projects, setProjects, errCode, loading: isLoading || loadingAuth, openLoginWindow, loginFromCode, logout, }; };