fix(core): ensure passing onSuccess wouldn't override Refine behavior

refinedev · Apr 24, 2024 · 167e100 · 167e100
1 parent 9868454
commit 167e100
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/packages/core/src/hooks/auth/useLogin/index.ts b/packages/core/src/hooks/auth/useLogin/index.ts
@@ -144,7 +144,10 @@ export function useLogin<TVariables = {}>({
   >({
     mutationKey: keys().auth().action("login").get(preferLegacyKeys),
     mutationFn: loginFromContext,
-    onSuccess: async ({ success, redirectTo, error, successNotification }) => {
+    ...(v3LegacyAuthProviderCompatible === true ? {} : mutationOptions),
+    onSuccess: async (data, variables, context) => {
+      const { success, redirectTo, error, successNotification } = data;
+
       if (success) {
         close?.("login-error");
 
@@ -176,11 +179,14 @@ export function useLogin<TVariables = {}>({
       }
 
       await invalidateAuthStore();
+
+      if (!v3LegacyAuthProviderCompatible) {
+        mutationOptions?.onSuccess?.(data, variables, context);
+      }
     },
     onError: (error: any) => {
       open?.(buildNotification(error));
     },
-    ...(v3LegacyAuthProviderCompatible === true ? {} : mutationOptions),
     meta: {
       ...(v3LegacyAuthProviderCompatible === true ? {} : mutationOptions?.meta),
       ...getXRay("useLogin", preferLegacyKeys),