@greatapps/common 1.1.478 → 1.1.479

package/src/providers/auth.provider.tsx

@@ -1,160 +1,162 @@
-"use client";
-
-import {
-  createContext,
-  ReactNode,
-  useCallback,
-  useContext,
-  useMemo,
-} from "react";
-import { useQueryClient } from "@tanstack/react-query";
-import {
-  useUserQuery,
-  useSetUserData,
-  useUserValidateSession,
-} from "../modules/auth/hooks/useUserQuery";
-import { useLogin } from "../modules/auth/hooks/login.hook";
-import { useRegister } from "../modules/auth/hooks/register.hook";
-import { useLogout } from "../modules/auth/hooks/logout.hook";
-import {
-  AuthState,
-  LoginRequest,
-  RegisterRequest,
-} from "../modules/auth/schema";
-
-export type LoginResult =
-  | { result: "success" }
-  | { result: "two_factor_required"; cookie: string; twoFactorMode: "verify" | "setup" }
-  | { result: "error"; message: string };
-import {
-  useCurrentAccount,
-  ACCOUNT_QUERY_KEY,
-} from "../modules/accounts/hooks/current-account.hook";
-import { useWhitelabelUrls } from "./whitelabel.provider";
-
-interface AuthContextProps extends AuthState {
-  login: (credentials: LoginRequest) => Promise<LoginResult>;
-  register: (data: RegisterRequest) => Promise<void>;
-  logout: () => Promise<void>;
-}
-
-export const AuthContext = createContext<AuthContextProps | undefined>(
-  undefined,
-);
-
-interface AuthProviderProps {
-  children: ReactNode;
-}
-
-export function AuthProvider({ children }: AuthProviderProps) {
-  const { data: user, isLoading: isQueryLoading } = useUserQuery();
-  const { data: account, isLoading: isAccountLoading } = useCurrentAccount();
-  const { isLoading: isSessionLoading } = useUserValidateSession();
-
-  const queryClient = useQueryClient();
-  const setUserData = useSetUserData();
-  const { accountsUrl, pagesUrl } = useWhitelabelUrls();
-
-  const { mutateAsync: loginMutate, isPending: isLogging } = useLogin();
-  const { mutateAsync: registerMutate, isPending: isRegistering } =
-    useRegister();
-  const { mutateAsync: logoutMutate, isPending: isLoggingOut } = useLogout();
-
-  const isAuthenticated = !!user;
-  const isLoading =
-    isLogging ||
-    isRegistering ||
-    isLoggingOut ||
-    isQueryLoading ||
-    isAccountLoading ||
-    isSessionLoading;
-
-  const login = useCallback(
-    async (credentials: LoginRequest): Promise<LoginResult> => {
-      try {
-        const data = await loginMutate(credentials);
-
-        if (data.result === "two_factor_required") {
-          return { result: "two_factor_required", cookie: data.cookie, twoFactorMode: data.twoFactorMode };
-        }
-
-        queryClient.clear();
-        setUserData(data.user);
-        queryClient.setQueryData(ACCOUNT_QUERY_KEY, data.account);
-
-        return { result: "success" };
-      } catch (error) {
-        const message =
-          error instanceof Error
-            ? error.message
-            : "Ocorreu um erro ao fazer login. Tente novamente.";
-        return { result: "error", message };
-      }
-    },
-    [queryClient, setUserData],
-  );
-
-  const register = useCallback(
-    async (data: RegisterRequest): Promise<void> => {
-      await registerMutate(data, {
-        onSuccess: (data) => {
-          queryClient.clear();
-          setUserData(data.user);
-          queryClient.setQueryData(ACCOUNT_QUERY_KEY, data.account);
-        },
-      });
-    },
-    [queryClient, setUserData],
-  );
-
-  const resolveLogoutRedirect = useCallback(() => {
-    try {
-      const currentOrigin = window.location.origin;
-      const pagesOrigin = pagesUrl ? new URL(pagesUrl).origin : "";
-
-      if (pagesOrigin && currentOrigin === pagesOrigin) {
-        return `${accountsUrl}/login?redirect=${encodeURIComponent(pagesUrl)}`;
-      }
-    } catch {
-      // fallback below
-    }
-
-    return `${accountsUrl}/login?redirect=${encodeURIComponent(pagesUrl || "/")}`;
-  }, [accountsUrl, pagesUrl]);
-
-  const logout = useCallback(async (): Promise<void> => {
-    try {
-      await logoutMutate(undefined);
-    } catch {
-      // cookie já removido no servidor mesmo se a API falhou
-    } finally {
-      queryClient.clear();
-      window.location.assign(resolveLogoutRedirect());
-    }
-  }, [queryClient, logoutMutate, resolveLogoutRedirect]);
-
-  const value: AuthContextProps = useMemo(
-    () => ({
-      user: user ?? null,
-      account: account ?? null,
-      isAuthenticated,
-      isLoading,
-      login,
-      register,
-      logout,
-    }),
-    [user, account, isAuthenticated, isLoading, login, register, logout],
-  );
-
-  return <AuthContext.Provider value={value}>{children}</AuthContext.Provider>;
-}
-
-export function useAuth(): AuthContextProps {
-  const context = useContext(AuthContext);
-
-  if (context === undefined) {
-    throw new Error("useAuth deve ser usado dentro de um AuthProvider");
-  }
-
-  return context;
-}
+"use client";
+
+import {
+  createContext,
+  ReactNode,
+  useCallback,
+  useContext,
+  useMemo,
+} from "react";
+import { useQueryClient } from "@tanstack/react-query";
+import {
+  useUserQuery,
+  useSetUserData,
+  useUserValidateSession,
+} from "../modules/auth/hooks/useUserQuery";
+import { useLogin } from "../modules/auth/hooks/login.hook";
+import { useRegister } from "../modules/auth/hooks/register.hook";
+import { useLogout } from "../modules/auth/hooks/logout.hook";
+import {
+  AuthState,
+  LoginRequest,
+  RegisterRequest,
+} from "../modules/auth/schema";
+
+export type LoginResult =
+  | { result: "success" }
+  | { result: "two_factor_required"; cookie: string; twoFactorMode: "verify" | "setup" }
+  | { result: "error"; message: string };
+import {
+  useCurrentAccount,
+  ACCOUNT_QUERY_KEY,
+} from "../modules/accounts/hooks/current-account.hook";
+import { useWhitelabelUrls } from "./whitelabel.provider";
+import { useWlQueryKey } from "../hooks/useAuthQueryKey";
+
+interface AuthContextProps extends AuthState {
+  login: (credentials: LoginRequest) => Promise<LoginResult>;
+  register: (data: RegisterRequest) => Promise<void>;
+  logout: () => Promise<void>;
+}
+
+export const AuthContext = createContext<AuthContextProps | undefined>(
+  undefined,
+);
+
+interface AuthProviderProps {
+  children: ReactNode;
+}
+
+export function AuthProvider({ children }: AuthProviderProps) {
+  const { data: user, isLoading: isQueryLoading } = useUserQuery();
+  const { data: account, isLoading: isAccountLoading } = useCurrentAccount();
+  const { isLoading: isSessionLoading } = useUserValidateSession();
+
+  const queryClient = useQueryClient();
+  const setUserData = useSetUserData();
+  const { accountsUrl, pagesUrl } = useWhitelabelUrls();
+  const accountKey = useWlQueryKey([...ACCOUNT_QUERY_KEY]);
+
+  const { mutateAsync: loginMutate, isPending: isLogging } = useLogin();
+  const { mutateAsync: registerMutate, isPending: isRegistering } =
+    useRegister();
+  const { mutateAsync: logoutMutate, isPending: isLoggingOut } = useLogout();
+
+  const isAuthenticated = !!user;
+  const isLoading =
+    isLogging ||
+    isRegistering ||
+    isLoggingOut ||
+    isQueryLoading ||
+    isAccountLoading ||
+    isSessionLoading;
+
+  const login = useCallback(
+    async (credentials: LoginRequest): Promise<LoginResult> => {
+      try {
+        const data = await loginMutate(credentials);
+
+        if (data.result === "two_factor_required") {
+          return { result: "two_factor_required", cookie: data.cookie, twoFactorMode: data.twoFactorMode };
+        }
+
+        queryClient.clear();
+        setUserData(data.user);
+        queryClient.setQueryData(accountKey, data.account);
+
+        return { result: "success" };
+      } catch (error) {
+        const message =
+          error instanceof Error
+            ? error.message
+            : "Ocorreu um erro ao fazer login. Tente novamente.";
+        return { result: "error", message };
+      }
+    },
+    [queryClient, setUserData, accountKey],
+  );
+
+  const register = useCallback(
+    async (data: RegisterRequest): Promise<void> => {
+      await registerMutate(data, {
+        onSuccess: (data) => {
+          queryClient.clear();
+          setUserData(data.user);
+          queryClient.setQueryData(accountKey, data.account);
+        },
+      });
+    },
+    [queryClient, setUserData, accountKey],
+  );
+
+  const resolveLogoutRedirect = useCallback(() => {
+    try {
+      const currentOrigin = window.location.origin;
+      const pagesOrigin = pagesUrl ? new URL(pagesUrl).origin : "";
+
+      if (pagesOrigin && currentOrigin === pagesOrigin) {
+        return `${accountsUrl}/login?redirect=${encodeURIComponent(pagesUrl)}`;
+      }
+    } catch {
+      // fallback below
+    }
+
+    return `${accountsUrl}/login?redirect=${encodeURIComponent(pagesUrl || "/")}`;
+  }, [accountsUrl, pagesUrl]);
+
+  const logout = useCallback(async (): Promise<void> => {
+    try {
+      await logoutMutate(undefined);
+    } catch {
+      // cookie já removido no servidor mesmo se a API falhou
+    } finally {
+      queryClient.clear();
+      window.location.assign(resolveLogoutRedirect());
+    }
+  }, [queryClient, logoutMutate, resolveLogoutRedirect]);
+
+  const value: AuthContextProps = useMemo(
+    () => ({
+      user: user ?? null,
+      account: account ?? null,
+      isAuthenticated,
+      isLoading,
+      login,
+      register,
+      logout,
+    }),
+    [user, account, isAuthenticated, isLoading, login, register, logout],
+  );
+
+  return <AuthContext.Provider value={value}>{children}</AuthContext.Provider>;
+}
+
+export function useAuth(): AuthContextProps {
+  const context = useContext(AuthContext);
+
+  if (context === undefined) {
+    throw new Error("useAuth deve ser usado dentro de um AuthProvider");
+  }
+
+  return context;
+}