From 76a2811093232bb11402d650b96bb473f805be01 Mon Sep 17 00:00:00 2001
From: tienifr <christianwen18@gmail.com>
Date: Tue, 6 Feb 2024 23:24:41 +0700
Subject: [PATCH] create util function

---
 src/components/LocalePicker.tsx                           | 3 ++-
 src/libs/FormUtils.ts                                     | 7 ++++++-
 src/pages/signin/ValidateCodeForm/BaseValidateCodeForm.js | 4 ++--
 3 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/src/components/LocalePicker.tsx b/src/components/LocalePicker.tsx
index 8022efd211b0..17520a467d3c 100644
--- a/src/components/LocalePicker.tsx
+++ b/src/components/LocalePicker.tsx
@@ -4,6 +4,7 @@ import {withOnyx} from 'react-native-onyx';
 import useLocalize from '@hooks/useLocalize';
 import useTheme from '@hooks/useTheme';
 import useThemeStyles from '@hooks/useThemeStyles';
+import FormUtils from '@libs/FormUtils';
 import * as App from '@userActions/App';
 import CONST from '@src/CONST';
 import ONYXKEYS from '@src/ONYXKEYS';
@@ -34,7 +35,7 @@ function LocalePicker({account, preferredLocale = CONST.LOCALES.DEFAULT, size =
         keyForList: language,
         isSelected: preferredLocale === language,
     }));
-    const shouldDisablePicker = account?.isLoading && account.loadingForm === (account.requiresTwoFactorAuth ? CONST.FORMS.VALIDATE_TFA_CODE_FORM : CONST.FORMS.VALIDATE_CODE_FORM);
+    const shouldDisablePicker = FormUtils.isValidateCodeFormSubmitting(account ?? {});
 
     return (
         <Picker
diff --git a/src/libs/FormUtils.ts b/src/libs/FormUtils.ts
index 37241df49af7..1628e4178ae8 100644
--- a/src/libs/FormUtils.ts
+++ b/src/libs/FormUtils.ts
@@ -1,7 +1,12 @@
 import type {OnyxFormKeyWithoutDraft} from '@components/Form/types';
+import CONST from '@src/CONST';
+import type {Account} from '@src/types/onyx';
+
+const isValidateCodeFormSubmitting = (account: Account) =>
+    account.isLoading && account.loadingForm === (account.requiresTwoFactorAuth ? CONST.FORMS.VALIDATE_TFA_CODE_FORM : CONST.FORMS.VALIDATE_CODE_FORM);
 
 function getDraftKey(formID: OnyxFormKeyWithoutDraft): `${OnyxFormKeyWithoutDraft}Draft` {
     return `${formID}Draft`;
 }
 
-export default {getDraftKey};
+export default {getDraftKey, isValidateCodeFormSubmitting};
diff --git a/src/pages/signin/ValidateCodeForm/BaseValidateCodeForm.js b/src/pages/signin/ValidateCodeForm/BaseValidateCodeForm.js
index fd5e9b952612..b1423555ee7d 100755
--- a/src/pages/signin/ValidateCodeForm/BaseValidateCodeForm.js
+++ b/src/pages/signin/ValidateCodeForm/BaseValidateCodeForm.js
@@ -22,6 +22,7 @@ import useThemeStyles from '@hooks/useThemeStyles';
 import canFocusInputOnScreenFocus from '@libs/canFocusInputOnScreenFocus';
 import compose from '@libs/compose';
 import * as ErrorUtils from '@libs/ErrorUtils';
+import FormUtils from '@libs/FormUtils';
 import * as ValidationUtils from '@libs/ValidationUtils';
 import ChangeExpensifyLoginLink from '@pages/signin/ChangeExpensifyLoginLink';
 import Terms from '@pages/signin/Terms';
@@ -105,8 +106,7 @@ function BaseValidateCodeForm(props) {
     const hasError = Boolean(props.account) && !_.isEmpty(props.account.errors) && !needToClearError;
     const isLoadingResendValidationForm = props.account.loadingForm === CONST.FORMS.RESEND_VALIDATE_CODE_FORM;
     const shouldDisableResendValidateCode = props.network.isOffline || props.account.isLoading;
-    const isValidateCodeFormSubmitting =
-        props.account.isLoading && props.account.loadingForm === (props.account.requiresTwoFactorAuth ? CONST.FORMS.VALIDATE_TFA_CODE_FORM : CONST.FORMS.VALIDATE_CODE_FORM);
+    const isValidateCodeFormSubmitting = FormUtils.isValidateCodeFormSubmitting(props.account);
 
     useEffect(() => {
         if (!(inputValidateCodeRef.current && hasError && (props.session.autoAuthState === CONST.AUTO_AUTH_STATE.FAILED || props.account.isLoading))) {