refactor: extract common selection mode logic into shared hooks

src/components/MoneyRequestReportView/SelectionToolbar/SelectionDropdown.tsx

@@ -11,7 +11,7 @@ import useLocalize from '@hooks/useLocalize';
 import useThemeStyles from '@hooks/useThemeStyles';
 import type {KYCFlowEvent, TriggerKYCFlow} from '@libs/PaymentUtils';
 import type CONST from '@src/CONST';
-import type {Report} from '@src/types/onyx';
+import type {Policy, Report} from '@src/types/onyx';
 import type {PaymentMethodType} from '@src/types/onyx/OriginalMessage';
 
 type SelectionDropdownProps = {
@@ -30,6 +30,9 @@ type SelectionDropdownProps = {
     /** Callback for the end of the onContinue trigger on option selection */
     selectionModeKYCSuccess: (type?: PaymentMethodType) => void;
 
+    /** Callback when a workspace policy payment option is selected */
+    onWorkspacePolicySelect: (policy: Policy, triggerKYCFlow: TriggerKYCFlow) => void;
+
     /** Reference to the KYC wall */
     kycWallRef: React.RefObject<KYCWallRef | null>;
 
@@ -43,6 +46,7 @@ function SelectionDropdown({
     report,
     onSelectionModePaymentSelect,
     selectionModeKYCSuccess,
+    onWorkspacePolicySelect,
     primaryAction,
     selectedTransactionsOptions,
     selectedTransactionIDs,
@@ -60,6 +64,7 @@ function SelectionDropdown({
                     chatReportID={chatReport?.reportID}
                     iouReport={report}
                     onPaymentSelect={onSelectionModePaymentSelect}
+                    onWorkspacePolicySelect={onWorkspacePolicySelect}
                     onSuccessfulKYC={selectionModeKYCSuccess}
                     primaryAction={primaryAction}
                     applicableSecondaryActions={selectedTransactionsOptions}

src/components/MoneyRequestReportView/SelectionToolbar/index.tsx

@@ -157,6 +157,7 @@ function SelectionToolbar({reportID, transactions, reportActions}: SelectionTool
         hasPayInSelectionMode,
         onSelectionModePaymentSelect,
         selectionModeKYCSuccess,
+        shouldBlockAction,
         primaryAction,
         kycWallRef,
         isHoldMenuVisible,
@@ -251,6 +252,12 @@ function SelectionToolbar({reportID, transactions, reportActions}: SelectionTool
                             report={report}
                             onSelectionModePaymentSelect={onSelectionModePaymentSelect}
                             selectionModeKYCSuccess={selectionModeKYCSuccess}
+                            onWorkspacePolicySelect={(selectedPolicy, triggerKYCFlow) => {
+                                if (shouldBlockAction()) {
+                                    return;
+                                }
+                                triggerKYCFlow({policy: selectedPolicy});
+                            }}
                             primaryAction={primaryAction}
                             selectedTransactionsOptions={selectedTransactionsOptions}
                             selectedTransactionIDs={selectedTransactionIDs}

src/hooks/useLifecycleActions.tsx

@@ -7,7 +7,6 @@ import {ModalActions} from '@components/Modal/Global/ModalContext';
 import type {SecondaryActionEntry} from '@components/MoneyReportHeaderActions/types';
 import {useSearchQueryContext, useSearchResultsContext, useSearchSelectionActions} from '@components/Search/SearchContext';
 import Text from '@components/Text';
-import {search} from '@libs/actions/Search';
 import getNonEmptyStringOnyxID from '@libs/getNonEmptyStringOnyxID';
 import {getValidConnectedIntegration, isSubmitPolicy} from '@libs/PolicyUtils';
 import {getFilteredReportActionsForReportView} from '@libs/ReportActionsUtils';
@@ -21,6 +20,7 @@ import {
     shouldBlockSubmitDueToStrictPolicyRules,
     shouldShowMarkAsDone,
 } from '@libs/ReportUtils';
+import refreshSearchAfterReportAction from '@libs/SearchRefreshUtils';
 import {hasAnyPendingRTERViolation as hasAnyPendingRTERViolationTransactionUtils, hasOnlyPendingCardTransactions, showPendingCardTransactionsBlockModal} from '@libs/TransactionUtils';
 import {cancelPayment, markReportPaymentReceived} from '@userActions/IOU/PayMoneyRequest';
 import {approveMoneyRequest, reopenReport, retractReport, submitReport, unapproveExpenseReport} from '@userActions/IOU/ReportWorkflow';
@@ -49,6 +49,7 @@ type UseLifecycleActionsParams = {
     startAnimation: () => void;
     startSubmittingAnimation: () => void;
     onHoldMenuOpen: (requestType: ActionHandledType, onConfirm?: () => void, paymentType?: PaymentMethodType) => void;
+    onCleanup?: () => void;
 };
 
 type UseLifecycleActionsResult = {
@@ -63,7 +64,7 @@ type UseLifecycleActionsResult = {
  * Provides report lifecycle transition actions (submit, approve, unapprove, cancel payment, retract, reopen)
  * and their associated guards (delegate access, hold, pending RTER, strict policy rules).
  */
-function useLifecycleActions({reportID, startApprovedAnimation, startAnimation, startSubmittingAnimation, onHoldMenuOpen}: UseLifecycleActionsParams): UseLifecycleActionsResult {
+function useLifecycleActions({reportID, startApprovedAnimation, startAnimation, startSubmittingAnimation, onHoldMenuOpen, onCleanup}: UseLifecycleActionsParams): UseLifecycleActionsResult {
     const [moneyRequestReport] = useOnyx(`${ONYXKEYS.COLLECTION.REPORT}${reportID}`);
     const [policy] = useOnyx(`${ONYXKEYS.COLLECTION.POLICY}${getNonEmptyStringOnyxID(moneyRequestReport?.policyID)}`);
     const [chatReport] = useOnyx(`${ONYXKEYS.COLLECTION.REPORT}${getNonEmptyStringOnyxID(moneyRequestReport?.chatReportID)}`);
@@ -179,6 +180,7 @@ function useLifecycleActions({reportID, startApprovedAnimation, startAnimation,
         });
         if (skipAnimation) {
             clearSelectedTransactions(true);
+            onCleanup?.();
         }
     };
 
@@ -213,18 +215,16 @@ function useLifecycleActions({reportID, startApprovedAnimation, startAnimation,
                 delegateEmail,
                 submitterLogin,
             });
-            if (currentSearchQueryJSON && !isOffline) {
-                search({
-                    searchKey: currentSearchKey,
-                    shouldCalculateTotals,
-                    offset: 0,
-                    queryJSON: currentSearchQueryJSON,
-                    isOffline,
-                    isLoading: !!currentSearchResults?.search?.isLoading,
-                });
-            }
+            refreshSearchAfterReportAction({
+                currentSearchQueryJSON,
+                currentSearchKey,
+                shouldCalculateTotals,
+                isOffline,
+                isLoading: !!currentSearchResults?.search?.isLoading,
+            });
             if (skipAnimation) {
                 clearSelectedTransactions(true);
+                onCleanup?.();
             }
         };