Add helper functions in getMatchingCentralPaneRouteForState

src/libs/Navigation/linkTo.ts

@@ -215,7 +215,7 @@ export default function linkTo(navigation: NavigationContainerRef<RootStackParam
             if (!getIsNarrowLayout()) {
                 // stateFromPath should always include bottom tab navigator state, so getMatchingCentralPaneRouteForState will be always defined.
                 // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
-                const matchingCentralPaneRoute = getMatchingCentralPaneRouteForState(stateFromPath)!;
+                const matchingCentralPaneRoute = getMatchingCentralPaneRouteForState(stateFromPath, rootState)!;
                 if (matchingCentralPaneRoute && 'name' in matchingCentralPaneRoute) {
                     root.dispatch({
                         type: CONST.NAVIGATION.ACTION_TYPE.PUSH,

src/libs/Navigation/linkingConfig/getMatchingCentralPaneRouteForState.ts

@@ -1,9 +1,11 @@
 import getTopmostBottomTabRoute from '@libs/Navigation/getTopmostBottomTabRoute';
-import type {CentralPaneName, NavigationPartialRoute, RootStackParamList, State} from '@libs/Navigation/types';
+import type {CentralPaneName, CentralPaneNavigatorParamList, NavigationPartialRoute, RootStackParamList, State} from '@libs/Navigation/types';
 import NAVIGATORS from '@src/NAVIGATORS';
 import SCREENS from '@src/SCREENS';
 import TAB_TO_CENTRAL_PANE_MAPPING from './TAB_TO_CENTRAL_PANE_MAPPING';
 
+const WORKSPACES_SCREENS = TAB_TO_CENTRAL_PANE_MAPPING[SCREENS.WORKSPACE.INITIAL].concat(TAB_TO_CENTRAL_PANE_MAPPING[SCREENS.ALL_SETTINGS]);
+
 /**
  * @param state - react-navigation state
  */
@@ -31,8 +33,47 @@ const getTopMostReportIDFromRHP = (state: State): string => {
     return '';
 };
 
+// Check if the given route has a policyID equal to the id provided in the function params
+function hasRouteMatchingPolicyID(route: NavigationPartialRoute<CentralPaneName>, policyID?: string) {
+    if (!route.params) {
+        return false;
+    }
+
+    const params = `params` in route?.params ? (route.params.params as Record<string, string | undefined>) : undefined;
+
+    // If params are not defined, then we need to check if the policyID exists
+    if (!params) {
+        return !policyID;
+    }
+
+    return 'policyID' in params && params.policyID === policyID;
+}
+
+// Get already opened settings screen within the policy
+function getAlreadyOpenedSettingsScreen(rootState?: State, policyID?: string): keyof CentralPaneNavigatorParamList | undefined {
+    if (!rootState) {
+        return undefined;
+    }
+
+    // If one of the screen from WORKSPACES_SCREENS is now in the navigation state, we can decide which screen we should display.
+    // A screen from the navigation state can be pushed to the navigation state again only if it has a matching policyID with the currently selected workspace.
+    // Otherwise, when we switch the workspace, we want to display the initial screen in the settings tab.
+    const alreadyOpenedSettingsTab = rootState.routes
+        .filter((item) => item.params && 'screen' in item.params && WORKSPACES_SCREENS.includes(item.params.screen as keyof CentralPaneNavigatorParamList))
+        .at(-1);
+
+    if (!hasRouteMatchingPolicyID(alreadyOpenedSettingsTab as NavigationPartialRoute<CentralPaneName>, policyID)) {
+        return undefined;
+    }
+
+    const settingsScreen =
+        alreadyOpenedSettingsTab?.params && 'screen' in alreadyOpenedSettingsTab?.params ? (alreadyOpenedSettingsTab?.params?.screen as keyof CentralPaneNavigatorParamList) : undefined;
+
+    return settingsScreen;
+}
+
 // Get matching central pane route for bottom tab navigator. e.g HOME -> REPORT
-function getMatchingCentralPaneRouteForState(state: State<RootStackParamList>): NavigationPartialRoute<CentralPaneName> | undefined {
+function getMatchingCentralPaneRouteForState(state: State<RootStackParamList>, rootState?: State): NavigationPartialRoute<CentralPaneName> | undefined {
     const topmostBottomTabRoute = getTopmostBottomTabRoute(state);
 
     if (!topmostBottomTabRoute) {
@@ -42,7 +83,10 @@ function getMatchingCentralPaneRouteForState(state: State<RootStackParamList>):
     const centralPaneName = TAB_TO_CENTRAL_PANE_MAPPING[topmostBottomTabRoute.name][0];
 
     if (topmostBottomTabRoute.name === SCREENS.WORKSPACE.INITIAL) {
-        return {name: centralPaneName, params: topmostBottomTabRoute.params};
+        // When we go back to the settings tab without switching the workspace id, we want to return to the previously opened screen
+        const policyID = topmostBottomTabRoute?.params && 'policyID' in topmostBottomTabRoute?.params ? (topmostBottomTabRoute.params.policyID as string) : undefined;
+        const screen = getAlreadyOpenedSettingsScreen(rootState, policyID) ?? centralPaneName;
+        return {name: screen, params: topmostBottomTabRoute.params};
     }
 
     if (topmostBottomTabRoute.name === SCREENS.HOME) {