Merge branch 'main' into feature/remove-canBeMissing

Author: fabioh8010

lib/Onyx.ts

@@ -137,6 +137,21 @@ function getDeferredInitTask(): DeferredTask {
     return deferredInitTask;
 }
 
+/**
+ * Executes an action after Onyx has been initialized.
+ * If Onyx is already initialized, the action is executed immediately.
+ * Otherwise, it waits for initialization to complete before executing.
+ *
+ * @param action The action to execute after initialization
+ * @returns The result of the action
+ */
+function afterInit<T>(action: () => Promise<T>): Promise<T> {
+    if (deferredInitTask.isResolved) {
+        return action();
+    }
+    return deferredInitTask.promise.then(action);
+}
+
 /**
  * Getter - returns the skippable collection member IDs.
  */
@@ -1749,6 +1764,7 @@ const OnyxUtils = {
     getMergeQueuePromise,
     getDefaultKeyStates,
     getDeferredInitTask,
+    afterInit,
     initStoreValues,
     sendActionToDevTools,
     get,

lib/OnyxUtils.ts

@@ -1,6 +1,7 @@
 type DeferredTask = {
     promise: Promise<void>;
-    resolve?: () => void;
+    resolve: () => void;
+    isResolved: boolean;
 };
 
 /**
@@ -9,11 +10,16 @@ type DeferredTask = {
  * Useful when we want to wait for a tasks that is resolved from an external action
  */
 export default function createDeferredTask(): DeferredTask {
-    const deferred = {} as DeferredTask;
+    const {promise, resolve: originalResolve} = Promise.withResolvers<void>();
 
-    deferred.promise = new Promise((res) => {
-        deferred.resolve = res;
-    });
+    const deferred: DeferredTask = {
+        promise,
+        resolve: () => {
+            deferred.isResolved = true;
+            originalResolve();
+        },
+        isResolved: false,
+    };
 
     return deferred;
 }

lib/createDeferredTask.ts

@@ -334,7 +334,7 @@ type ExpandOnyxKeys<TKey extends OnyxKey> = TKey extends CollectionKeyBase ? NoI
  * If a new method is added to OnyxUtils.METHOD constant, it must be added to OnyxMethodValueMap type.
  * Otherwise it will show static type errors.
  */
-type OnyxUpdate<TKey extends OnyxKey = OnyxKey> = {
+type OnyxUpdate<TKey extends OnyxKey> = {
     // ⚠️ DO NOT CHANGE THIS TYPE, UNLESS YOU KNOW WHAT YOU ARE DOING. ⚠️
     [K in TKey]:
         | {onyxMethod: typeof OnyxUtils.METHOD.SET; key: ExpandOnyxKeys<K>; value: OnyxSetInput<K>}

lib/types.ts

@@ -1,6 +1,6 @@
 {
   "name": "react-native-onyx",
-  "version": "3.0.34",
+  "version": "3.0.36",
   "author": "Expensify, Inc.",
   "homepage": "https://expensify.com",
   "description": "State management for React Native",