@metamask-previews/notification-services-controller 10.0.0-preview-7dcef379 → 10.0.0-preview-f9611039

package/dist/NotificationServicesController/NotificationServicesController.mjs

@@ -9,7 +9,7 @@ var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (
     if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
     return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
 };
-var _NotificationServicesController_instances, _NotificationServicesController_keyringController, _NotificationServicesController_auth, _NotificationServicesController_pushNotifications, _NotificationServicesController_accounts, _NotificationServicesController_featureAnnouncementEnv, _NotificationServicesController_registerMessageHandlers, _NotificationServicesController_clearLoadingStates, _NotificationServicesController_assertAuthEnabled, _NotificationServicesController_enableAuth, _NotificationServicesController_getBearerToken, _NotificationServicesController_setIsUpdatingMetamaskNotifications, _NotificationServicesController_setIsFetchingMetamaskNotifications, _NotificationServicesController_setIsCheckingAccountsPresence, _NotificationServicesController_updateUpdatingAccountsState, _NotificationServicesController_clearUpdatingAccountsState;
+var _NotificationServicesController_instances, _NotificationServicesController_keyringController, _NotificationServicesController_auth, _NotificationServicesController_storage, _NotificationServicesController_pushNotifications, _NotificationServicesController_accounts, _NotificationServicesController_featureAnnouncementEnv, _NotificationServicesController_registerMessageHandlers, _NotificationServicesController_clearLoadingStates, _NotificationServicesController_assertAuthEnabled, _NotificationServicesController_enableAuth, _NotificationServicesController_getValidStorageKeyAndBearerToken, _NotificationServicesController_assertUserStorage, _NotificationServicesController_getUserStorage, _NotificationServicesController_setIsUpdatingMetamaskNotifications, _NotificationServicesController_setIsFetchingMetamaskNotifications, _NotificationServicesController_setIsCheckingAccountsPresence, _NotificationServicesController_updateUpdatingAccountsState, _NotificationServicesController_clearUpdatingAccountsState;
 function $importDefault(module) {
     if (module?.__esModule) {
         return module.default;
@@ -22,10 +22,12 @@ import { KeyringTypes } from "@metamask/keyring-controller";
 import { assert } from "@metamask/utils";
 import $log from "loglevel";
 const log = $importDefault($log);
+import { USER_STORAGE_VERSION_KEY } from "./constants/constants.mjs";
 import { TRIGGER_TYPES } from "./constants/notification-schema.mjs";
-import { processAndFilterNotifications, safeProcessNotification } from "./processors/process-notifications.mjs";
+import { safeProcessNotification } from "./processors/process-notifications.mjs";
 import * as FeatureNotifications from "./services/feature-announcements.mjs";
 import * as OnChainNotifications from "./services/onchain-notifications.mjs";
+import * as Utils from "./utils/utils.mjs";
 // Unique name for the controller
 const controllerName = 'NotificationServicesController';
 const metadata = {
@@ -135,6 +137,17 @@ class NotificationServicesController extends BaseController {
                 return await this.messagingSystem.call('AuthenticationController:performSignIn');
             },
         });
+        _NotificationServicesController_storage.set(this, {
+            getStorageKey: () => {
+                return this.messagingSystem.call('UserStorageController:getStorageKey');
+            },
+            getNotificationStorage: async () => {
+                return await this.messagingSystem.call('UserStorageController:performGetStorage', 'notifications.notification_settings');
+            },
+            setNotificationStorage: async (state) => {
+                return await this.messagingSystem.call('UserStorageController:performSetStorage', 'notifications.notification_settings', state);
+            },
+        });
         _NotificationServicesController_pushNotifications.set(this, {
             // Flag to check is notifications have been setup when the browser/extension is initialized.
             // We want to re-initialize push notifications when the browser/extension is refreshed
@@ -143,9 +156,9 @@ class NotificationServicesController extends BaseController {
             subscribeToPushNotifications: async () => {
                 await this.messagingSystem.call('NotificationServicesPushController:subscribeToPushNotifications');
             },
-            enablePushNotifications: async (addresses) => {
+            enablePushNotifications: async (UUIDs) => {
                 try {
-                    await this.messagingSystem.call('NotificationServicesPushController:enablePushNotifications', addresses);
+                    await this.messagingSystem.call('NotificationServicesPushController:enablePushNotifications', UUIDs);
                 }
                 catch (e) {
                     log.error('Silently failed to enable push notifications', e);
@@ -159,6 +172,14 @@ class NotificationServicesController extends BaseController {
                     log.error('Silently failed to disable push notifications', e);
                 }
             },
+            updatePushNotifications: async (UUIDs) => {
+                try {
+                    await this.messagingSystem.call('NotificationServicesPushController:updateTriggerPushNotifications', UUIDs);
+                }
+                catch (e) {
+                    log.error('Silently failed to update push notifications', e);
+                }
+            },
             subscribe: () => {
                 this.messagingSystem.subscribe('NotificationServicesPushController:onNewNotifications', (notification) => {
                     // eslint-disable-next-line @typescript-eslint/no-floating-promises
@@ -193,7 +214,7 @@ class NotificationServicesController extends BaseController {
         _NotificationServicesController_accounts.set(this, {
             // Flag to ensure we only setup once
             isNotificationAccountsSetup: false,
-            getNotificationAccounts: () => {
+            getNotificationAccounts: async () => {
                 const { keyrings } = this.messagingSystem.call('KeyringController:getState');
                 const firstHDKeyring = keyrings.find((k) => k.type === KeyringTypes.hd.toString());
                 const keyringAccounts = firstHDKeyring?.accounts ?? null;
@@ -204,9 +225,9 @@ class NotificationServicesController extends BaseController {
              *
              * @returns addresses removed, added, and latest list of addresses
              */
-            listAccounts: () => {
+            listAccounts: async () => {
                 // Get previous and current account sets
-                const nonChecksumAccounts = __classPrivateFieldGet(this, _NotificationServicesController_accounts, "f").getNotificationAccounts();
+                const nonChecksumAccounts = await __classPrivateFieldGet(this, _NotificationServicesController_accounts, "f").getNotificationAccounts();
                 if (!nonChecksumAccounts) {
                     return {
                         accountsAdded: [],
@@ -243,11 +264,13 @@ class NotificationServicesController extends BaseController {
             },
             /**
              * Initializes the cache/previous list. This is handy so we have an accurate in-mem state of the previous list of accounts.
+             *
+             * @returns result from list accounts
              */
-            initialize: () => {
+            initialize: async () => {
                 if (__classPrivateFieldGet(this, _NotificationServicesController_keyringController, "f").isUnlocked &&
                     !__classPrivateFieldGet(this, _NotificationServicesController_accounts, "f").isNotificationAccountsSetup) {
-                    __classPrivateFieldGet(this, _NotificationServicesController_accounts, "f").listAccounts();
+                    await __classPrivateFieldGet(this, _NotificationServicesController_accounts, "f").listAccounts();
                     __classPrivateFieldGet(this, _NotificationServicesController_accounts, "f").isNotificationAccountsSetup = true;
                 }
             },
@@ -263,15 +286,15 @@ class NotificationServicesController extends BaseController {
                         !hasTotalAccountsChanged) {
                         return;
                     }
-                    const { accountsAdded, accountsRemoved } = __classPrivateFieldGet(this, _NotificationServicesController_accounts, "f").listAccounts();
+                    const { accountsAdded, accountsRemoved } = await __classPrivateFieldGet(this, _NotificationServicesController_accounts, "f").listAccounts();
                     const promises = [];
                     if (accountsAdded.length > 0) {
-                        promises.push(this.enableAccounts(accountsAdded));
+                        promises.push(this.updateOnChainTriggersByAccount(accountsAdded));
                     }
                     if (accountsRemoved.length > 0) {
-                        promises.push(this.disableAccounts(accountsRemoved));
+                        promises.push(this.deleteOnChainTriggersByAccount(accountsRemoved));
                     }
-                    await Promise.allSettled(promises);
+                    await Promise.all(promises);
                 }, (state) => {
                     return (state?.keyrings?.flatMap?.((keyring) => keyring.accounts)?.length ??
                         0);
@@ -285,9 +308,10 @@ class NotificationServicesController extends BaseController {
     }
     init() {
         __classPrivateFieldGet(this, _NotificationServicesController_keyringController, "f").setupLockedStateSubscriptions(async () => {
-            __classPrivateFieldGet(this, _NotificationServicesController_accounts, "f").initialize();
+            await __classPrivateFieldGet(this, _NotificationServicesController_accounts, "f").initialize();
             await __classPrivateFieldGet(this, _NotificationServicesController_pushNotifications, "f").initializePushNotifications();
         });
+        // eslint-disable-next-line @typescript-eslint/no-floating-promises
         __classPrivateFieldGet(this, _NotificationServicesController_accounts, "f").initialize();
         // eslint-disable-next-line @typescript-eslint/no-floating-promises
         __classPrivateFieldGet(this, _NotificationServicesController_pushNotifications, "f").initializePushNotifications();
@@ -298,20 +322,13 @@ class NotificationServicesController extends BaseController {
      * Public method to expose enabling push notifications
      */
     async enablePushNotifications() {
-        try {
-            const { bearerToken } = await __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_getBearerToken).call(this);
-            const { accounts } = __classPrivateFieldGet(this, _NotificationServicesController_accounts, "f").listAccounts();
-            const addressesWithNotifications = await OnChainNotifications.getOnChainNotificationsConfigCached(bearerToken, accounts);
-            const addresses = addressesWithNotifications
-                .filter((a) => Boolean(a.enabled))
-                .map((a) => a.address);
-            if (addresses.length > 0) {
-                await __classPrivateFieldGet(this, _NotificationServicesController_pushNotifications, "f").enablePushNotifications(addresses);
-            }
-        }
-        catch (e) {
-            log.error('Failed to enable push notifications', e);
+        await __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_enableAuth).call(this);
+        const storage = await __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_getUserStorage).call(this);
+        if (!storage) {
+            throw new Error('Unable to get triggers');
         }
+        const uuids = Utils.getAllUUIDs(storage);
+        await __classPrivateFieldGet(this, _NotificationServicesController_pushNotifications, "f").enablePushNotifications(uuids);
     }
     /**
      * Public method to expose disabling push notifications
@@ -323,13 +340,10 @@ class NotificationServicesController extends BaseController {
         try {
             __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_setIsCheckingAccountsPresence).call(this, true);
             // Retrieve user storage
-            const { bearerToken } = await __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_getBearerToken).call(this);
-            const addressesWithNotifications = await OnChainNotifications.getOnChainNotificationsConfigCached(bearerToken, accounts);
-            const result = {};
-            addressesWithNotifications.forEach((a) => {
-                result[a.address] = a.enabled;
-            });
-            return result;
+            const userStorage = await __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_getUserStorage).call(this);
+            __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_assertUserStorage).call(this, userStorage);
+            const presence = Utils.checkAccountsPresence(userStorage, accounts);
+            return presence;
         }
         catch (error) {
             log.error('Failed to check accounts presence', error);
@@ -373,32 +387,40 @@ class NotificationServicesController extends BaseController {
     async createOnChainTriggers(opts) {
         try {
             __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_setIsUpdatingMetamaskNotifications).call(this, true);
-            const { bearerToken } = await __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_getBearerToken).call(this);
-            const { accounts } = __classPrivateFieldGet(this, _NotificationServicesController_accounts, "f").listAccounts();
-            // 1. See if has enabled notifications before
-            const addressesWithNotifications = await OnChainNotifications.getOnChainNotificationsConfigCached(bearerToken, accounts);
-            // Notifications API can return array with addresses set to false
-            // So assert that at least one address is enabled
-            let accountsWithNotifications = addressesWithNotifications
-                .filter((a) => Boolean(a.enabled))
-                .map((a) => a.address);
-            // 2. Enable Notifications (if no accounts subscribed or we are resetting)
-            if (accountsWithNotifications.length === 0 || opts?.resetNotifications) {
-                await OnChainNotifications.updateOnChainNotifications(bearerToken, accounts.map((address) => ({ address, enabled: true })));
-                accountsWithNotifications = accounts;
+            const { bearerToken, storageKey } = await __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_getValidStorageKeyAndBearerToken).call(this);
+            const { accounts } = await __classPrivateFieldGet(this, _NotificationServicesController_accounts, "f").listAccounts();
+            // Attempt Get User Storage
+            // Will be null if entry does not exist, or a user is resetting their notifications
+            // Will be defined if entry exists
+            // Will throw if fails to get the user storage entry
+            let userStorage = opts?.resetNotifications
+                ? null
+                : await __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_getUserStorage).call(this);
+            // If userStorage does not exist, create a new one
+            // All the triggers created are set as: "disabled"
+            if (userStorage?.[USER_STORAGE_VERSION_KEY] === undefined) {
+                userStorage = Utils.initializeUserStorage(accounts.map((account) => ({ address: account })), false);
+                // Write the userStorage
+                await __classPrivateFieldGet(this, _NotificationServicesController_storage, "f").setNotificationStorage(JSON.stringify(userStorage));
             }
-            // 3. Lazily enable push notifications (FCM may take some time, so keeps UI unblocked)
-            __classPrivateFieldGet(this, _NotificationServicesController_pushNotifications, "f")
-                .enablePushNotifications(accountsWithNotifications)
-                .catch(() => {
+            // Create the triggers
+            const triggers = Utils.traverseUserStorageTriggers(userStorage);
+            await OnChainNotifications.createOnChainTriggers(userStorage, storageKey, bearerToken, triggers);
+            // Create push notifications triggers in background
+            const allUUIDS = Utils.getAllUUIDs(userStorage);
+            // We do not want to wait for this request as it may take a while (e.g. for Firebase to setup)
+            __classPrivateFieldGet(this, _NotificationServicesController_pushNotifications, "f").enablePushNotifications(allUUIDS).catch(() => {
                 // Do Nothing
             });
+            // Write the new userStorage (triggers are now "enabled")
+            await __classPrivateFieldGet(this, _NotificationServicesController_storage, "f").setNotificationStorage(JSON.stringify(userStorage));
             // Update the state of the controller
             this.update((state) => {
                 state.isNotificationServicesEnabled = true;
                 state.isFeatureAnnouncementsEnabled = true;
                 state.isMetamaskNotificationsFeatureSeen = true;
             });
+            return userStorage;
         }
         catch (err) {
             log.error('Failed to create On Chain triggers', err);
@@ -411,6 +433,10 @@ class NotificationServicesController extends BaseController {
     /**
      * Enables all MetaMask notifications for the user.
      * This is identical flow when initializing notifications for the first time.
+     * 1. Enable Profile Syncing
+     * 2. Get or Create Notification User Storage
+     * 3. Upsert Triggers
+     * 4. Update Push notifications
      *
      * @throws {Error} If there is an error during the process of enabling notifications.
      */
@@ -458,11 +484,12 @@ class NotificationServicesController extends BaseController {
         __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_setIsUpdatingMetamaskNotifications).call(this, false);
     }
     /**
-     * Deletes on-chain triggers associated with a specific account/s.
+     * Deletes on-chain triggers associated with a specific account.
      * This method performs several key operations:
-     * 1. Validates Auth
-     * 2. Deletes accounts
-     * (note) We do not need to look through push notifications as we've deleted triggers
+     * 1. Validates Auth & Storage
+     * 2. Finds and deletes all triggers associated with the account
+     * 3. Disables any related push notifications
+     * 4. Updates Storage to reflect new state.
      *
      * **Action** - When a user disables notifications for a given account in settings.
      *
@@ -470,13 +497,29 @@ class NotificationServicesController extends BaseController {
      * @returns A promise that resolves to void or an object containing a success message.
      * @throws {Error} Throws an error if unauthenticated or from other operations.
      */
-    async disableAccounts(accounts) {
+    async deleteOnChainTriggersByAccount(accounts) {
         try {
             __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_updateUpdatingAccountsState).call(this, accounts);
             // Get and Validate BearerToken and User Storage Key
-            const { bearerToken } = await __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_getBearerToken).call(this);
+            const { bearerToken, storageKey } = await __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_getValidStorageKeyAndBearerToken).call(this);
+            // Get & Validate User Storage
+            const userStorage = await __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_getUserStorage).call(this);
+            __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_assertUserStorage).call(this, userStorage);
+            // Get the UUIDs to delete
+            const UUIDs = accounts
+                .map((a) => Utils.getUUIDsForAccount(userStorage, a.toLowerCase()))
+                .flat();
+            if (UUIDs.length === 0) {
+                return userStorage;
+            }
             // Delete these UUIDs (Mutates User Storage)
-            await OnChainNotifications.updateOnChainNotifications(bearerToken, accounts.map((address) => ({ address, enabled: false })));
+            await OnChainNotifications.deleteOnChainTriggers(userStorage, storageKey, bearerToken, UUIDs);
+            // Update Push Notifications with new list of IDs
+            const remainingTriggerIds = Utils.getAllUUIDs(userStorage);
+            await __classPrivateFieldGet(this, _NotificationServicesController_pushNotifications, "f").updatePushNotifications(remainingTriggerIds);
+            // Update User Storage
+            await __classPrivateFieldGet(this, _NotificationServicesController_storage, "f").setNotificationStorage(JSON.stringify(userStorage));
+            return userStorage;
         }
         catch (err) {
             log.error('Failed to delete OnChain triggers', err);
@@ -501,11 +544,45 @@ class NotificationServicesController extends BaseController {
      * @returns A promise that resolves to the updated user storage.
      * @throws {Error} Throws an error if unauthenticated or from other operations.
      */
-    async enableAccounts(accounts) {
+    async updateOnChainTriggersByAccount(accounts) {
         try {
             __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_updateUpdatingAccountsState).call(this, accounts);
-            const { bearerToken } = await __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_getBearerToken).call(this);
-            await OnChainNotifications.updateOnChainNotifications(bearerToken, accounts.map((address) => ({ address, enabled: true })));
+            // Get and Validate BearerToken and User Storage Key
+            const { bearerToken, storageKey } = await __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_getValidStorageKeyAndBearerToken).call(this);
+            // Get & Validate User Storage
+            const userStorage = await __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_getUserStorage).call(this);
+            __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_assertUserStorage).call(this, userStorage);
+            // Add any missing triggers
+            accounts.forEach((a) => Utils.upsertAddressTriggers(a, userStorage));
+            const newTriggers = Utils.traverseUserStorageTriggers(userStorage, {
+                mapTrigger: (t) => {
+                    if (!t.enabled) {
+                        return t;
+                    }
+                    return undefined;
+                },
+            });
+            // Create any missing triggers.
+            if (newTriggers.length > 0) {
+                // Write te updated userStorage (where triggers are disabled)
+                await __classPrivateFieldGet(this, _NotificationServicesController_storage, "f").setNotificationStorage(JSON.stringify(userStorage));
+                // Create the triggers
+                const triggers = Utils.traverseUserStorageTriggers(userStorage, {
+                    mapTrigger: (t) => {
+                        if (accounts.some((a) => a.toLowerCase() === t.address.toLowerCase())) {
+                            return t;
+                        }
+                        return undefined;
+                    },
+                });
+                await OnChainNotifications.createOnChainTriggers(userStorage, storageKey, bearerToken, triggers);
+            }
+            // Update Push Notifications Triggers
+            const UUIDs = Utils.getAllUUIDs(userStorage);
+            await __classPrivateFieldGet(this, _NotificationServicesController_pushNotifications, "f").updatePushNotifications(UUIDs);
+            // Update the userStorage (where triggers are enabled)
+            await __classPrivateFieldGet(this, _NotificationServicesController_storage, "f").setNotificationStorage(JSON.stringify(userStorage));
+            return userStorage;
         }
         catch (err) {
             log.error('Failed to update OnChain triggers', err);
@@ -532,31 +609,37 @@ class NotificationServicesController extends BaseController {
             // Not used by Snaps
             const isGlobalNotifsEnabled = this.state.isNotificationServicesEnabled;
             // Raw Feature Notifications
-            const rawAnnouncements = isGlobalNotifsEnabled && this.state.isFeatureAnnouncementsEnabled
+            const rawFeatureAnnouncementNotifications = isGlobalNotifsEnabled && this.state.isFeatureAnnouncementsEnabled
                 ? await FeatureNotifications.getFeatureAnnouncementNotifications(__classPrivateFieldGet(this, _NotificationServicesController_featureAnnouncementEnv, "f"), previewToken).catch(() => [])
                 : [];
             // Raw On Chain Notifications
             const rawOnChainNotifications = [];
             if (isGlobalNotifsEnabled) {
-                try {
-                    const { bearerToken } = await __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_getBearerToken).call(this);
-                    const { accounts } = __classPrivateFieldGet(this, _NotificationServicesController_accounts, "f").listAccounts();
-                    const addressesWithNotifications = (await OnChainNotifications.getOnChainNotificationsConfigCached(bearerToken, accounts)).map((a) => a.address);
-                    const notifications = await OnChainNotifications.getOnChainNotifications(bearerToken, addressesWithNotifications).catch(() => []);
+                const userStorage = await __classPrivateFieldGet(this, _NotificationServicesController_storage, "f")
+                    .getNotificationStorage()
+                    .then((s) => s && JSON.parse(s))
+                    .catch(() => null);
+                const bearerToken = await __classPrivateFieldGet(this, _NotificationServicesController_auth, "f").getBearerToken().catch(() => null);
+                if (userStorage && bearerToken) {
+                    const notifications = await OnChainNotifications.getOnChainNotifications(userStorage, bearerToken).catch(() => []);
                     rawOnChainNotifications.push(...notifications);
                 }
-                catch {
-                    // Do nothing
-                }
             }
             // Snap Notifications (original)
             // We do not want to remove them
             const snapNotifications = this.state.metamaskNotificationsList.filter((notification) => notification.type === TRIGGER_TYPES.SNAP);
+            // Process Notifications
             const readIds = this.state.metamaskNotificationsReadList;
+            const isNotUndefined = (t) => Boolean(t);
+            const processAndFilter = (ns) => ns
+                .map((n) => safeProcessNotification(n, readIds))
+                .filter(isNotUndefined);
+            const featureAnnouncementNotifications = processAndFilter(rawFeatureAnnouncementNotifications);
+            const onChainNotifications = processAndFilter(rawOnChainNotifications);
             // Combine Notifications
             const metamaskNotifications = [
-                ...processAndFilterNotifications(rawAnnouncements, readIds),
-                ...processAndFilterNotifications(rawOnChainNotifications, readIds),
+                ...featureAnnouncementNotifications,
+                ...onChainNotifications,
                 ...snapNotifications,
             ];
             // Sort Notifications
@@ -724,7 +807,7 @@ class NotificationServicesController extends BaseController {
         }
     }
 }
-_NotificationServicesController_keyringController = new WeakMap(), _NotificationServicesController_auth = new WeakMap(), _NotificationServicesController_pushNotifications = new WeakMap(), _NotificationServicesController_accounts = new WeakMap(), _NotificationServicesController_featureAnnouncementEnv = new WeakMap(), _NotificationServicesController_instances = new WeakSet(), _NotificationServicesController_registerMessageHandlers = function _NotificationServicesController_registerMessageHandlers() {
+_NotificationServicesController_keyringController = new WeakMap(), _NotificationServicesController_auth = new WeakMap(), _NotificationServicesController_storage = new WeakMap(), _NotificationServicesController_pushNotifications = new WeakMap(), _NotificationServicesController_accounts = new WeakMap(), _NotificationServicesController_featureAnnouncementEnv = new WeakMap(), _NotificationServicesController_instances = new WeakSet(), _NotificationServicesController_registerMessageHandlers = function _NotificationServicesController_registerMessageHandlers() {
     this.messagingSystem.registerActionHandler(`${controllerName}:updateMetamaskNotificationsList`, this.updateMetamaskNotificationsList.bind(this));
     this.messagingSystem.registerActionHandler(`${controllerName}:disableNotificationServices`, this.disableNotificationServices.bind(this));
     this.messagingSystem.registerActionHandler(`${controllerName}:getNotificationsByType`, this.getNotificationsByType.bind(this));
@@ -748,13 +831,42 @@ _NotificationServicesController_keyringController = new WeakMap(), _Notification
     if (!isSignedIn) {
         await __classPrivateFieldGet(this, _NotificationServicesController_auth, "f").signIn();
     }
-}, _NotificationServicesController_getBearerToken = async function _NotificationServicesController_getBearerToken() {
+}, _NotificationServicesController_getValidStorageKeyAndBearerToken = async function _NotificationServicesController_getValidStorageKeyAndBearerToken() {
     __classPrivateFieldGet(this, _NotificationServicesController_instances, "m", _NotificationServicesController_assertAuthEnabled).call(this);
     const bearerToken = await __classPrivateFieldGet(this, _NotificationServicesController_auth, "f").getBearerToken();
-    if (!bearerToken) {
-        throw new Error('Missing BearerToken');
+    const storageKey = await __classPrivateFieldGet(this, _NotificationServicesController_storage, "f").getStorageKey();
+    if (!bearerToken || !storageKey) {
+        throw new Error('Missing BearerToken or storage key');
+    }
+    return { bearerToken, storageKey };
+}, _NotificationServicesController_assertUserStorage = function _NotificationServicesController_assertUserStorage(storage) {
+    if (!storage) {
+        throw new Error('User Storage does not exist');
+    }
+}, _NotificationServicesController_getUserStorage = 
+/**
+ * Retrieves and parses the user storage from the storage key.
+ *
+ * This method attempts to retrieve the user storage using the specified storage key,
+ * then parses the JSON string to an object. If the storage is not found or cannot be parsed,
+ * it throws an error.
+ *
+ * @returns The parsed user storage object or null
+ */
+async function _NotificationServicesController_getUserStorage() {
+    const userStorageString = await __classPrivateFieldGet(this, _NotificationServicesController_storage, "f").getNotificationStorage();
+    if (!userStorageString) {
+        return null;
+    }
+    try {
+        const userStorage = JSON.parse(userStorageString);
+        Utils.cleanUserStorage(userStorage);
+        return userStorage;
+    }
+    catch {
+        log.error('Unable to parse User Storage');
+        return null;
     }
-    return { bearerToken };
 }, _NotificationServicesController_setIsUpdatingMetamaskNotifications = function _NotificationServicesController_setIsUpdatingMetamaskNotifications(isUpdatingMetamaskNotifications) {
     this.update((state) => {
         state.isUpdatingMetamaskNotifications = isUpdatingMetamaskNotifications;