chore: all review comments

src/features/authorization/components/AuthenticationSteps/AuthenticationSteps.tsx

@@ -1,6 +1,6 @@
 import { useState, type JSX } from 'react';
 import { LoginRegisterForm } from './LoginRegiserForm';
-import type { AuthMode, AuthType } from '../../types/authTypes';
+import type { AuthMode, AuthStep, AuthType } from '../../types/authTypes';
 import { OtpVerifyForm } from './OtpVerifyForm';
 import { isNumeric } from '@/utils/regexes/isNumeric';
 import { CompleteSignUp } from './CompleteSignUp';
@@ -8,22 +8,17 @@ import { EnterPasswordForm } from './EnterPasswordForm';
 import { UserStatus } from '../../types/userTypes';
 import type { CountryCode, GUID } from '@/types/commonTypes';
 import { VerifyPhoneNumber } from './VerifyPhoneNumber';
-import { useSearchParams } from 'react-router';
+import { useNavigate, useSearchParams } from 'react-router-dom';
 
 export const AuthenticationSteps = (): JSX.Element => {
-  const DEFAULT_RETURN_URL = 'https://account.business-harmony.com/';
+  const navigate = useNavigate();
+  const DEFAULT_RETURN_URL = '/profile';
   const [searchParams] = useSearchParams();
   const authReturnUrl: string =
     searchParams.get('returnUrl') ?? DEFAULT_RETURN_URL;
   const [authMode, setAuthMode] = useState<AuthMode>('register');
   const [authType, setAuthType] = useState<AuthType>('phone');
-  const [currentStep, setCurrentStep] = useState<
-    | 'emailOrPhone'
-    | 'verify'
-    | 'enterPassword'
-    | 'addPhoneNumber'
-    | 'addedPhoneNumberVerify'
-  >('emailOrPhone');
+  const [currentStep, setCurrentStep] = useState<AuthStep>('emailOrPhone');
   const [loginRegisterValue, setLoginRegisterValue] = useState<string>('');
   const [countryCode, setCountryCode] = useState<CountryCode>('+98');
   const [addPhoneCountryCode, setAddPhoneCountryCode] =
@@ -54,24 +49,28 @@ export const AuthenticationSteps = (): JSX.Element => {
     }
   };
 
-  const handleOTPVerfied = (userId: GUID) => {
-    handleUserLoggedIn(userId);
-  };
-
   const handleUserLoggedIn = (userId: GUID) => {
     localStorage.setItem('userID', userId);
 
-    location.href = authReturnUrl;
+    redirectToReturnUrl();
   };
 
   const handleConfrimPhoneNumber = (userId: GUID) => {
-    handleUserLoggedIn(userId);
+    localStorage.setItem('userID', userId);
 
     setCurrentStep('addPhoneNumber');
   };
 
   const handlePhoneNumberVerified = () => {
-    location.href = authReturnUrl;
+    redirectToReturnUrl();
+  };
+
+  const redirectToReturnUrl = () => {
+    if (authReturnUrl === DEFAULT_RETURN_URL) {
+      navigate(DEFAULT_RETURN_URL);
+    } else {
+      location.href = authReturnUrl;
+    }
   };
 
   return (
@@ -95,7 +94,7 @@ export const AuthenticationSteps = (): JSX.Element => {
           onVerifyPhoneNumber={handleConfrimPhoneNumber}
           authReturnUrl={authReturnUrl}
           countryCode={countryCode}
-          onOTPVerified={handleOTPVerfied}
+          onOTPVerified={handleUserLoggedIn}
           onEditValue={() => setCurrentStep('emailOrPhone')}
           authMode={authMode}
           authType={authType}