Add logic for other page

Author: nkdengineer

src/pages/iou/request/step/IOURequestEditReportCommon.tsx

@@ -48,9 +48,19 @@ type Props = {
     removeFromReport?: () => void;
     isEditing?: boolean;
     isUnreported?: boolean;
+    shouldShowNotFoundPage?: boolean;
 };
 
-function IOURequestEditReportCommon({backTo, transactionsReports, selectReport, policyID: policyIDFromProps, removeFromReport, isEditing = false, isUnreported}: Props) {
+function IOURequestEditReportCommon({
+    backTo,
+    transactionsReports,
+    selectReport,
+    policyID: policyIDFromProps,
+    removeFromReport,
+    isEditing = false,
+    isUnreported,
+    shouldShowNotFoundPage = false,
+}: Props) {
     const {translate, localeCompare} = useLocalize();
     const {options} = useOptionsList();
     const [allReports] = useOnyx(ONYXKEYS.COLLECTION.REPORT, {selector: (reports) => mapOnyxCollectionItems(reports, reportSelector), canBeMissing: true});
@@ -119,7 +129,7 @@ function IOURequestEditReportCommon({backTo, transactionsReports, selectReport,
             shouldShowWrapper
             testID="IOURequestEditReportCommon"
             includeSafeAreaPaddingBottom
-            shouldShowNotFoundPage={expenseReports.length === 0}
+            shouldShowNotFoundPage={expenseReports.length === 0 || shouldShowNotFoundPage}
         >
             <SelectionList
                 sections={[{data: reportOptions}]}

src/pages/iou/request/step/IOURequestStepDistance.tsx

@@ -13,6 +13,7 @@ import DotIndicatorMessage from '@components/DotIndicatorMessage';
 import DraggableList from '@components/DraggableList';
 import withCurrentUserPersonalDetails from '@components/withCurrentUserPersonalDetails';
 import type {WithCurrentUserPersonalDetailsProps} from '@components/withCurrentUserPersonalDetails';
+import useCanEditStep from '@hooks/useCanEditStep';
 import useFetchRoute from '@hooks/useFetchRoute';
 import useLocalize from '@hooks/useLocalize';
 import useNetwork from '@hooks/useNetwork';
@@ -139,6 +140,7 @@ function IOURequestStepDistance({
     const [recentWaypoints, {status: recentWaypointsStatus}] = useOnyx(ONYXKEYS.NVP_RECENT_WAYPOINTS, {canBeMissing: true});
     const iouRequestType = getRequestType(transaction);
     const customUnitRateID = getRateID(transaction);
+    const [canEditDistance] = useCanEditStep(action, iouType, report, CONST.EDIT_REQUEST_FIELD.DISTANCE);
 
     // Sets `amount` and `split` share data before moving to the next step to avoid briefly showing `0.00` as the split share for participants
     const setDistanceRequestData = useCallback(
@@ -529,7 +531,7 @@ function IOURequestStepDistance({
             headerTitle={translate('common.distance')}
             onBackButtonPress={navigateBack}
             testID={IOURequestStepDistance.displayName}
-            shouldShowNotFoundPage={isEditing && !transaction?.comment?.waypoints}
+            shouldShowNotFoundPage={(isEditing && !transaction?.comment?.waypoints) || !canEditDistance}
             shouldShowWrapper={!isCreatingNewRequest}
         >
             <>

src/pages/iou/request/step/IOURequestStepReport.tsx

@@ -1,6 +1,7 @@
 import React from 'react';
 import {InteractionManager} from 'react-native';
 import type {ListItem} from '@components/SelectionList/types';
+import useCanEditStep from '@hooks/useCanEditStep';
 import useOnyx from '@hooks/useOnyx';
 import {changeTransactionsReport, setTransactionReport} from '@libs/actions/Transaction';
 import getNonEmptyStringOnyxID from '@libs/getNonEmptyStringOnyxID';
@@ -126,6 +127,8 @@ function IOURequestStepReport({route, transaction}: IOURequestStepReportProps) {
         Navigation.dismissModal();
     };
 
+    const [canEditReport] = useCanEditStep(action, iouType, reportOrDraftReport, CONST.EDIT_REQUEST_FIELD.REPORT);
+
     return (
         <IOURequestEditReportCommon
             backTo={backTo}
@@ -135,6 +138,7 @@ function IOURequestStepReport({route, transaction}: IOURequestStepReportProps) {
             removeFromReport={removeFromReport}
             isEditing={isEditing}
             isUnreported={isUnreported}
+            shouldShowNotFoundPage={!canEditReport}
         />
     );
 }