Create separate transition pages to sign the user in / out

package.json

@@ -91,7 +91,7 @@
     "react-native-image-size": "^1.1.3",
     "react-native-keyboard-spacer": "^0.4.1",
     "react-native-modal": "^13.0.0",
-    "react-native-onyx": "git+https://github.com/Expensify/react-native-onyx.git#4cc46d1ad70312d2e10adf7cdd7065ab9b472113",
+    "react-native-onyx": "1.0.5",
     "react-native-pdf": "^6.2.2",
     "react-native-performance": "^2.0.0",
     "react-native-permissions": "^3.0.1",

src/ROUTES.js

@@ -81,7 +81,7 @@ export default {
     getReportDetailsRoute: reportID => `r/${reportID}/details`,
     REPORT_SETTINGS: 'r/:reportID/settings',
     getReportSettingsRoute: reportID => `r/${reportID}/settings`,
-    LOGIN_WITH_SHORT_LIVED_TOKEN: 'transition',
+    TRANSITION_FROM_OLD_DOT: 'transition',
     VALIDATE_LOGIN: 'v/:accountID/:validateCode',
     GET_ASSISTANCE: 'get-assistance/:taskID',
     getGetAssistanceRoute: taskID => `get-assistance/${taskID}`,

src/SCREENS.js

@@ -6,5 +6,5 @@ export default {
     HOME: 'Home',
     LOADING: 'Loading',
     REPORT: 'Report',
-    LOG_IN_WITH_SHORT_LIVED_TOKEN: 'LogInWithShortLivedToken',
+    TRANSITION_FROM_OLD_DOT: 'TransitionFromOldDot',
 };

src/libs/Navigation/AppNavigator/AuthScreens.js

@@ -1,7 +1,5 @@
 import React from 'react';
-import {Linking} from 'react-native';
 import Onyx, {withOnyx} from 'react-native-onyx';
-import Str from 'expensify-common/lib/str';
 import moment from 'moment';
 import _ from 'underscore';
 import lodashGet from 'lodash/get';
@@ -34,11 +32,11 @@ import MainDrawerNavigator from './MainDrawerNavigator';
 import * as ModalStackNavigators from './ModalStackNavigators';
 import SCREENS from '../../../SCREENS';
 import Timers from '../../Timers';
-import LogInWithShortLivedTokenPage from '../../../pages/LogInWithShortLivedTokenPage';
 import ValidateLoginPage from '../../../pages/ValidateLoginPage';
 import defaultScreenOptions from './defaultScreenOptions';
 import * as App from '../../actions/App';
 import * as Session from '../../actions/Session';
+import LogOutPreviousUserPage from '../../../pages/LogOutPreviousUserPage';
 import networkPropTypes from '../../../components/networkPropTypes';
 import {withNetwork} from '../../../components/OnyxProvider';
 
@@ -109,17 +107,7 @@ class AuthScreens extends React.Component {
         App.getAppData(false);
 
         App.fixAccountAndReloadData();
-
-        // Load policies, maybe creating a new policy first.
-        Linking.getInitialURL()
-            .then((url) => {
-                if (this.shouldCreateFreePolicy(url)) {
-                    Policy.createAndGetPolicyList();
-                    return;
-                }
-
-                Policy.getPolicyList();
-            });
+        App.setUpPoliciesAndNavigate(this.props.session);
 
         // Refresh the personal details, timezone and betas every 30 minutes
         // There is no pusher event that sends updated personal details data yet
@@ -165,25 +153,6 @@ class AuthScreens extends React.Component {
         this.interval = null;
     }
 
-    /**
-     * @param {String} [url]
-     * @returns {Boolean}
-     */
-    shouldCreateFreePolicy(url = '') {
-        if (!url) {
-            return false;
-        }
-
-        const path = new URL(url).pathname;
-        const params = new URLSearchParams(url);
-        const exitTo = params.get('exitTo');
-        const email = params.get('email');
-        const isLoggingInAsNewUser = !_.isNull(this.props.session.email) && (email !== this.props.session.email);
-        return !isLoggingInAsNewUser
-            && Str.startsWith(path, Str.normalizeUrl(ROUTES.LOGIN_WITH_SHORT_LIVED_TOKEN))
-            && exitTo === ROUTES.WORKSPACE_NEW;
-    }
-
     render() {
         const commonModalScreenOptions = {
             headerShown: false,
@@ -239,9 +208,9 @@ class AuthScreens extends React.Component {
                     component={ValidateLoginPage}
                 />
                 <RootStack.Screen
-                    name={SCREENS.LOG_IN_WITH_SHORT_LIVED_TOKEN}
+                    name={SCREENS.TRANSITION_FROM_OLD_DOT}
                     options={defaultScreenOptions}
-                    component={LogInWithShortLivedTokenPage}
+                    component={LogOutPreviousUserPage}
                 />
 
                 {/* These are the various modal routes */}

src/libs/Navigation/AppNavigator/PublicScreens.js

@@ -17,7 +17,7 @@ const PublicScreens = () => (
             component={SignInPage}
         />
         <RootStack.Screen
-            name={SCREENS.LOG_IN_WITH_SHORT_LIVED_TOKEN}
+            name={SCREENS.TRANSITION_FROM_OLD_DOT}
             options={defaultScreenOptions}
             component={LogInWithShortLivedTokenPage}
         />

src/libs/Navigation/Navigation.js

@@ -10,6 +10,11 @@ import ONYXKEYS from '../../ONYXKEYS';
 import linkingConfig from './linkingConfig';
 import navigationRef from './navigationRef';
 
+let resolveNavigationIsReadyPromise;
+let navigationIsReadyPromise = new Promise((resolve) => {
+    resolveNavigationIsReadyPromise = resolve;
+});
+
 let isLoggedIn = false;
 Onyx.connect({
     key: ONYXKEYS.SESSION,
@@ -186,6 +191,23 @@ function isActiveRoute(routePath) {
     return getActiveRoute().substring(1) === routePath;
 }
 
+/**
+ * @returns {Promise}
+ */
+function isNavigationReady() {
+    return navigationIsReadyPromise;
+}
+
+function setIsNavigationReady() {
+    resolveNavigationIsReadyPromise();
+}
+
+function resetIsNavigationReady() {
+    navigationIsReadyPromise = new Promise((resolve) => {
+        resolveNavigationIsReadyPromise = resolve;
+    });
+}
+
 export default {
     canNavigate,
     navigate,
@@ -196,6 +218,9 @@ export default {
     closeDrawer,
     getDefaultDrawerState,
     setDidTapNotification,
+    isNavigationReady,
+    setIsNavigationReady,
+    resetIsNavigationReady,
 };
 
 export {

src/libs/Navigation/linkingConfig.js

@@ -27,7 +27,7 @@ export default {
             // Main Routes
             SetPassword: ROUTES.SET_PASSWORD_WITH_VALIDATE_CODE,
             ValidateLogin: ROUTES.VALIDATE_LOGIN,
-            [SCREENS.LOG_IN_WITH_SHORT_LIVED_TOKEN]: ROUTES.LOGIN_WITH_SHORT_LIVED_TOKEN,
+            [SCREENS.TRANSITION_FROM_OLD_DOT]: ROUTES.TRANSITION_FROM_OLD_DOT,
 
             // Modal Screens
             Settings: {

src/libs/actions/App.js

@@ -1,6 +1,7 @@
-import {AppState} from 'react-native';
+import {AppState, Linking} from 'react-native';
 import Onyx from 'react-native-onyx';
 import lodashGet from 'lodash/get';
+import Str from 'expensify-common/lib/str';
 import ONYXKEYS from '../../ONYXKEYS';
 import * as DeprecatedAPI from '../deprecatedAPI';
 import CONST from '../../CONST';
@@ -15,6 +16,8 @@ import * as GeoLocation from './GeoLocation';
 import * as BankAccounts from './BankAccounts';
 import * as Policy from './Policy';
 import NetworkConnection from '../NetworkConnection';
+import Navigation from '../Navigation/Navigation';
+import ROUTES from '../../ROUTES';
 
 let currentUserAccountID;
 Onyx.connect({
@@ -135,6 +138,60 @@ function fixAccountAndReloadData() {
         });
 }
 
+/**
+ * This action runs every time the AuthScreens are mounted. The navigator may
+ * not be ready yet, and therefore we need to wait before navigating within this
+ * action and any actions this method calls.
+ *
+ * getInitialURL allows us to access params from the transition link more easily
+ * than trying to extract them from the navigation state.
+
+ * The transition link contains an exitTo param that contains the route to
+ * navigate to after the user is signed in. A user can transition from OldDot
+ * with a different account than the one they are currently signed in with, so
+ * we only navigate if they are not signing in as a new user. Once they are
+ * signed in as that new user, this action will run again and the navigation
+ * will occur.
+
+ * When the exitTo route is 'workspace/new', we create a new
+ * workspace and navigate to it via Policy.createAndGetPolicyList.
+ *
+ * We subscribe to the session using withOnyx in the AuthScreens and
+ * pass it in as a parameter. withOnyx guarantees that the value has been read
+ * from Onyx because it will not render the AuthScreens until that point.
+ * @param {Object} session
+ */
+function setUpPoliciesAndNavigate(session) {
+    Linking.getInitialURL()
+        .then((url) => {
+            if (!url) {
+                Policy.getPolicyList();
+                return;
+            }
+            const path = new URL(url).pathname;
+            const params = new URLSearchParams(url);
+            const exitTo = params.get('exitTo');
+            const email = params.get('email');
+            const isLoggingInAsNewUser = session.email !== email;
+            const shouldCreateFreePolicy = !isLoggingInAsNewUser
+                        && Str.startsWith(path, Str.normalizeUrl(ROUTES.TRANSITION_FROM_OLD_DOT))
+                        && exitTo === ROUTES.WORKSPACE_NEW;
+            if (shouldCreateFreePolicy) {
+                Policy.createAndGetPolicyList();
+                return;
+            }
+            Policy.getPolicyList();
+            if (!isLoggingInAsNewUser && exitTo) {
+                Navigation.isNavigationReady()
+                    .then(() => {
+                        // We must call dismissModal() to remove the /transition route from history
+                        Navigation.dismissModal();
+                        Navigation.navigate(exitTo);
+                    });
+            }
+        });
+}
+
 // When the app reconnects from being offline, fetch all initialization data
 NetworkConnection.onReconnect(() => getAppData(true, false));
 
@@ -145,4 +202,5 @@ export {
     getLocale,
     getAppData,
     fixAccountAndReloadData,
+    setUpPoliciesAndNavigate,
 };

src/libs/actions/Policy.js

@@ -238,6 +238,7 @@ function createAndGetPolicyList() {
             newPolicyID = policyID;
             return getPolicyList();
         })
+        .then(Navigation.isNavigationReady)
         .then(() => {
             Navigation.dismissModal();
             navigateToPolicy(newPolicyID);

src/libs/actions/Session/index.js

@@ -40,6 +40,7 @@ function setSuccessfulSignInData(data) {
     PushNotification.register(data.accountID);
     Onyx.merge(ONYXKEYS.SESSION, {
         shouldShowComposeInput: true,
+        error: null,
         ..._.pick(data, 'authToken', 'accountID', 'email', 'encryptedAuthToken'),
     });
 }
@@ -270,28 +271,24 @@ function signIn(password, twoFactorAuthCode) {
 /**
  * Uses a short lived authToken to continue a user's session from OldDot
  *
- * @param {String} accountID
  * @param {String} email
  * @param {String} shortLivedToken
  * @param {String} exitTo
  */
-function signInWithShortLivedToken(accountID, email, shortLivedToken) {
+function signInWithShortLivedToken(email, shortLivedToken) {
     Onyx.merge(ONYXKEYS.ACCOUNT, {...CONST.DEFAULT_ACCOUNT_DATA, loading: true});
 
-    createTemporaryLogin(shortLivedToken, email).then((response) => {
-        Onyx.merge(ONYXKEYS.SESSION, {
-            accountID,
-            email,
-        });
-        if (response.jsonCode !== CONST.JSON_CODE.SUCCESS) {
-            return;
-        }
+    createTemporaryLogin(shortLivedToken, email)
+        .then((response) => {
+            if (response.jsonCode !== CONST.JSON_CODE.SUCCESS) {
+                return;
+            }
 
-        User.getUserDetails();
-        Onyx.merge(ONYXKEYS.ACCOUNT, {success: true});
-    }).finally(() => {
-        Onyx.merge(ONYXKEYS.ACCOUNT, {loading: false});
-    });
+            User.getUserDetails();
+            Onyx.merge(ONYXKEYS.ACCOUNT, {success: true});
+        }).finally(() => {
+            Onyx.merge(ONYXKEYS.ACCOUNT, {loading: false});
+        });
 }
 
 /**

src/libs/actions/Welcome.js

@@ -8,6 +8,7 @@ import * as Policy from './Policy';
 import ONYXKEYS from '../../ONYXKEYS';
 import NameValuePair from './NameValuePair';
 import CONST from '../../CONST';
+import SCREENS from '../../SCREENS';
 
 let resolveIsReadyPromise;
 let isReadyPromise = new Promise((resolve) => {
@@ -113,20 +114,20 @@ function show({routes, showCreateMenu}) {
         // Set the NVP back to false so we don't automatically run welcome actions again
         NameValuePair.set(CONST.NVP.IS_FIRST_TIME_NEW_EXPENSIFY_USER, false, ONYXKEYS.NVP_IS_FIRST_TIME_NEW_EXPENSIFY_USER);
 
+        // If we are rendering the SidebarScreen at the same time as a workspace route that means we've already created a workspace via workspace/new and should not open the global
+        // create menu right now. We should also stay on the workspace page if that is our destination.
+        const topRouteName = lodashGet(_.last(routes), 'name', '');
+        const transitionRoute = _.find(routes, route => route.name === SCREENS.TRANSITION_FROM_OLD_DOT);
+        const exitingToWorkspaceRoute = lodashGet(transitionRoute, 'params.exitTo', '') === 'workspace/new';
+        const isDisplayingWorkspaceRoute = topRouteName.toLowerCase().includes('workspace') || exitingToWorkspaceRoute;
+
         // We want to display the Workspace chat first since that means a user is already in a Workspace and doesn't need to create another one
         const workspaceChatReport = _.find(allReports, report => ReportUtils.isPolicyExpenseChat(report) && report.ownerEmail === email);
-        if (workspaceChatReport) {
+        if (workspaceChatReport && !isDisplayingWorkspaceRoute) {
             Navigation.navigate(ROUTES.getReportRoute(workspaceChatReport.reportID));
             return;
         }
 
-        // If we are rendering the SidebarScreen at the same time as a workspace route that means we've already created a workspace via workspace/new and should not open the global
-        // create menu right now.
-        const topRouteName = lodashGet(_.last(routes), 'name', '');
-        const loginWithShortLivedTokenRoute = _.find(routes, route => route.name === 'LogInWithShortLivedToken');
-        const exitingToWorkspaceRoute = lodashGet(loginWithShortLivedTokenRoute, 'params.exitTo', '') === 'workspace/new';
-        const isDisplayingWorkspaceRoute = topRouteName.toLowerCase().includes('workspace') || exitingToWorkspaceRoute;
-
         // If user is not already an admin of a free policy and we are not navigating them to their workspace or creating a new workspace via workspace/new then
         // we will show the create menu.
         if (!Policy.isAdminOfFreePolicy(allPolicies) && !isDisplayingWorkspaceRoute) {