Fix: Push notification onyx updates not applied when notification is selected

src/libs/Notification/PushNotification/getPushNotificationData.ts

@@ -0,0 +1,15 @@
+import type {PushPayload} from '@ua/react-native-airship';
+import type {PushNotificationData} from './NotificationType';
+
+function getPushNotificationData(notification: PushPayload): PushNotificationData {
+    let payload = notification.extras.payload;
+
+    // On Android, some notification payloads are sent as a JSON string rather than an object
+    if (typeof payload === 'string') {
+        payload = JSON.parse(payload);
+    }
+
+    return payload as PushNotificationData;
+}
+
+export default getPushNotificationData;

src/libs/Notification/PushNotification/index.native.ts

@@ -5,6 +5,7 @@ import Log from '@libs/Log';
 import * as PushNotificationActions from '@userActions/PushNotification';
 import ONYXKEYS from '@src/ONYXKEYS';
 import ForegroundNotifications from './ForegroundNotifications';
+import getPushNotificationData from './getPushNotificationData';
 import type {PushNotificationData} from './NotificationType';
 import NotificationType from './NotificationType';
 import type {ClearNotifications, Deregister, Init, OnReceived, OnSelected, Register} from './types';
@@ -27,14 +28,7 @@ const notificationEventActionMap: NotificationEventActionMap = {};
  */
 function pushNotificationEventCallback(eventType: EventType, notification: PushPayload) {
     const actionMap = notificationEventActionMap[eventType] ?? {};
-    let payload = notification.extras.payload;
-
-    // On Android, some notification payloads are sent as a JSON string rather than an object
-    if (typeof payload === 'string') {
-        payload = JSON.parse(payload);
-    }
-
-    const data = payload as PushNotificationData;
+    const data = getPushNotificationData(notification);
 
     Log.info(`[PushNotification] Callback triggered for ${eventType}`);
 

src/libs/Notification/PushNotification/subscribePushNotification/index.ts

@@ -1,8 +1,12 @@
+import Airship from '@ua/react-native-airship';
 import Onyx from 'react-native-onyx';
 import applyOnyxUpdatesReliably from '@libs/actions/applyOnyxUpdatesReliably';
+import type {DeferredUpdatesDictionary} from '@libs/actions/OnyxUpdateManager/types';
+import * as DeferredOnyxUpdates from '@libs/actions/OnyxUpdateManager/utils/DeferredOnyxUpdates';
 import * as ActiveClientManager from '@libs/ActiveClientManager';
 import Log from '@libs/Log';
 import Navigation from '@libs/Navigation/Navigation';
+import getPushNotificationData from '@libs/Notification/PushNotification/getPushNotificationData';
 import type {ReportActionPushNotificationData} from '@libs/Notification/PushNotification/NotificationType';
 import getPolicyEmployeeAccountIDs from '@libs/PolicyEmployeeListUtils';
 import {extractPolicyIDFromPath} from '@libs/PolicyUtils';
@@ -13,6 +17,7 @@ import CONST from '@src/CONST';
 import ONYXKEYS from '@src/ONYXKEYS';
 import ROUTES from '@src/ROUTES';
 import type {OnyxUpdatesFromServer} from '@src/types/onyx';
+import type {OnyxServerUpdate} from '@src/types/onyx/OnyxUpdatesFromServer';
 import {isEmptyObject} from '@src/types/utils/EmptyObject';
 import PushNotification from '..';
 
@@ -27,6 +32,20 @@ Onyx.connect({
     },
 });
 
+function buildOnyxUpdatesFromServer({onyxData, lastUpdateID, previousUpdateID}: {onyxData: OnyxServerUpdate[]; lastUpdateID: number; previousUpdateID: number}) {
+    return {
+        type: CONST.ONYX_UPDATE_TYPES.AIRSHIP,
+        lastUpdateID,
+        previousUpdateID,
+        updates: [
+            {
+                eventType: 'eventType',
+                data: onyxData,
+            },
+        ],
+    } as OnyxUpdatesFromServer;
+}
+
 function getLastUpdateIDAppliedToClient(): Promise<number> {
     return new Promise((resolve) => {
         Onyx.connect({
@@ -50,17 +69,8 @@ function applyOnyxData({reportID, reportActionID, onyxData, lastUpdateID, previo
     }
 
     Log.info('[PushNotification] reliable onyx update received', false, {lastUpdateID, previousUpdateID, onyxDataCount: onyxData?.length ?? 0});
-    const updates: OnyxUpdatesFromServer = {
-        type: CONST.ONYX_UPDATE_TYPES.AIRSHIP,
-        lastUpdateID,
-        previousUpdateID,
-        updates: [
-            {
-                eventType: 'eventType',
-                data: onyxData,
-            },
-        ],
-    };
+
+    const updates = buildOnyxUpdatesFromServer({onyxData, lastUpdateID, previousUpdateID});
 
     /**
      * When this callback runs in the background on Android (via Headless JS), no other Onyx.connect callbacks will run. This means that
@@ -71,6 +81,31 @@ function applyOnyxData({reportID, reportActionID, onyxData, lastUpdateID, previo
 }
 
 function navigateToReport({reportID, reportActionID}: ReportActionPushNotificationData): Promise<void> {
+    Log.info('[PushNotification] Adding push notification updates to deferred updates queue', false, {reportID, reportActionID});
+
+    // Onyx data from push notifications might not have been applied when they were received in the background
+    // due to OS limitations. So we'll also attempt to apply them here so they can display immediately. Reliable
+    // updates will prevent any old updates from being duplicated and any gaps in them will be handled
+    Airship.push.getActiveNotifications().then((notifications) => {
+        const onyxUpdates = notifications.reduce<DeferredUpdatesDictionary>((updates, notification) => {
+            const pushNotificationData = getPushNotificationData(notification);
+            const lastUpdateID = pushNotificationData.lastUpdateID;
+            const previousUpdateID = pushNotificationData.previousUpdateID;
+
+            if (pushNotificationData.onyxData == null || lastUpdateID == null || previousUpdateID == null) {
+                return updates;
+            }
+
+            const newUpdates = buildOnyxUpdatesFromServer({onyxData: pushNotificationData.onyxData, lastUpdateID, previousUpdateID});
+
+            // eslint-disable-next-line no-param-reassign
+            updates[lastUpdateID] = newUpdates;
+            return updates;
+        }, {});
+
+        return DeferredOnyxUpdates.enqueueAndProcess(onyxUpdates);
+    });
+
     Log.info('[PushNotification] Navigating to report', false, {reportID, reportActionID});
 
     const policyID = lastVisitedPath && extractPolicyIDFromPath(lastVisitedPath);

src/libs/actions/OnyxUpdateManager/utils/DeferredOnyxUpdates.ts

@@ -65,6 +65,8 @@ function process() {
     }
 
     missingOnyxUpdatesQueryPromise = OnyxUpdateManagerUtils.validateAndApplyDeferredUpdates();
+
+    return missingOnyxUpdatesQueryPromise;
 }
 
 type EnqueueDeferredOnyxUpdatesOptions = {
@@ -105,7 +107,7 @@ function enqueue(updates: OnyxUpdatesFromServer | DeferredUpdatesDictionary, opt
  */
 function enqueueAndProcess(updates: OnyxUpdatesFromServer | DeferredUpdatesDictionary, options?: EnqueueDeferredOnyxUpdatesOptions) {
     enqueue(updates, options);
-    process();
+    return process();
 }
 
 type ClearDeferredOnyxUpdatesOptions = {