@metamask-previews/account-tree-controller 7.5.3-preview-ea5e7f6d0 → 7.5.3-preview-34f2bcbf7

package/dist/AccountTreeController.mjs

@@ -9,18 +9,20 @@ 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 _AccountTreeController_instances, _AccountTreeController_accountIdToContext, _AccountTreeController_groupIdToWalletId, _AccountTreeController_backupAndSyncService, _AccountTreeController_rules, _AccountTreeController_trace, _AccountTreeController_backupAndSyncConfig, _AccountTreeController_accountOrderCallbacks, _AccountTreeController_initialized, _AccountTreeController_initTreeContext, _AccountTreeController_getEntropyRule, _AccountTreeController_getSnapRule, _AccountTreeController_getKeyringRule, _AccountTreeController_applyAccountWalletMetadata, _AccountTreeController_getRuleForWallet, _AccountTreeController_getComputedAccountGroupName, _AccountTreeController_getDefaultAccountGroupName, _AccountTreeController_applyAccountGroupMetadata, _AccountTreeController_handleAccountsAdded, _AccountTreeController_handleAccountsRemoved, _AccountTreeController_pruneEmptyGroupAndWallet, _AccountTreeController_publishAccountGroupCreated, _AccountTreeController_publishAccountGroupUpdated, _AccountTreeController_publishAccountGroupRemoved, _AccountTreeController_insert, _AccountTreeController_listAccounts, _AccountTreeController_assertAccountGroupExists, _AccountTreeController_assertAccountWalletExists, _AccountTreeController_assertAccountGroupNameIsUnique, _AccountTreeController_setSelectedAccountGroup, _AccountTreeController_getDefaultSelectedAccountGroup, _AccountTreeController_handleSelectedAccountChange, _AccountTreeController_handleMultichainAccountWalletStatusChange, _AccountTreeController_getAccountGroup, _AccountTreeController_getDefaultAccountFromAccountGroupId, _AccountTreeController_getDefaultAccountGroupId, _AccountTreeController_hasEvmAccount, _AccountTreeController_resolveNameConflict, _AccountTreeController_createBackupAndSyncContext;
-import { AccountWalletType, select } from "@metamask/account-api";
+var _AccountTreeController_instances, _AccountTreeController_accountIdToContext, _AccountTreeController_groupIdToWalletId, _AccountTreeController_backupAndSyncService, _AccountTreeController_rules, _AccountTreeController_trace, _AccountTreeController_backupAndSyncConfig, _AccountTreeController_accountOrderCallbacks, _AccountTreeController_initialized, _AccountTreeController_initTreeContext, _AccountTreeController_getEntropyRule, _AccountTreeController_getSnapRule, _AccountTreeController_getKeyringRule, _AccountTreeController_applyAccountWalletMetadata, _AccountTreeController_getRuleForWallet, _AccountTreeController_getComputedAccountGroupName, _AccountTreeController_getDefaultAccountGroupName, _AccountTreeController_applyAccountGroupMetadata, _AccountTreeController_handleAccountsAdded, _AccountTreeController_handleAccountsRemoved, _AccountTreeController_pruneEmptyGroupAndWallet, _AccountTreeController_publishAccountGroupCreated, _AccountTreeController_publishAccountGroupUpdated, _AccountTreeController_publishAccountGroupRemoved, _AccountTreeController_insert, _AccountTreeController_listAccounts, _AccountTreeController_assertAccountGroupExists, _AccountTreeController_assertAccountWalletExists, _AccountTreeController_assertAccountGroupNameIsUnique, _AccountTreeController_setSelectedAccountGroup, _AccountTreeController_getDefaultSelectedAccountGroup, _AccountTreeController_handleSelectedAccountChange, _AccountTreeController_handleMultichainAccountWalletStatusChange, _AccountTreeController_handleMultichainAccountGroupStatusChange, _AccountTreeController_getMultichainAccountWalletStatus, _AccountTreeController_getMultichainAccountGroupStatus, _AccountTreeController_getAccountGroup, _AccountTreeController_getDefaultAccountFromAccountGroupId, _AccountTreeController_getDefaultAccountGroupId, _AccountTreeController_hasEvmAccount, _AccountTreeController_resolveNameConflict, _AccountTreeController_createBackupAndSyncContext;
+import { AccountGroupType, AccountWalletType, select } from "@metamask/account-api";
 import { BaseController } from "@metamask/base-controller";
 import { isEvmAccountType } from "@metamask/keyring-api";
 import { assert } from "@metamask/utils";
 import { formatAnalyticsEvent, traceFallback } from "./backup-and-sync/analytics/index.mjs";
 import { BackupAndSyncService } from "./backup-and-sync/service/index.mjs";
-import { ACCOUNT_TYPE_TO_SORT_ORDER, isAccountGroupNameUnique, isAccountGroupNameUniqueFromWallet, MAX_SORT_ORDER } from "./group.mjs";
+import { createSyncMutationTracker } from "./backup-and-sync/utils/index.mjs";
+import { ACCOUNT_TYPE_TO_SORT_ORDER, isAccountGroupNameUnique, isAccountGroupNameUniqueFromWallet, isMultichainAccountGroup, MAX_SORT_ORDER } from "./group.mjs";
 import { projectLogger as log } from "./logger.mjs";
 import { EntropyRule } from "./rules/entropy.mjs";
 import { KeyringRule } from "./rules/keyring.mjs";
 import { SnapRule } from "./rules/snap.mjs";
+import { isMultichainAccountWallet } from "./wallet.mjs";
 export const controllerName = 'AccountTreeController';
 const MESSENGER_EXPOSED_METHODS = [
     'getSelectedAccountGroup',
@@ -176,6 +178,9 @@ export class AccountTreeController extends BaseController {
         this.messenger.subscribe('MultichainAccountService:walletStatusChange', (walletId, status) => {
             __classPrivateFieldGet(this, _AccountTreeController_instances, "m", _AccountTreeController_handleMultichainAccountWalletStatusChange).call(this, walletId, status);
         });
+        this.messenger.subscribe('MultichainAccountService:groupStatusChange', (groupId, status) => {
+            __classPrivateFieldGet(this, _AccountTreeController_instances, "m", _AccountTreeController_handleMultichainAccountGroupStatusChange).call(this, groupId, status);
+        });
         this.messenger.registerMethodActionHandlers(this, MESSENGER_EXPOSED_METHODS);
     }
     /**
@@ -914,7 +919,9 @@ _AccountTreeController_accountIdToContext = new WeakMap(), _AccountTreeControlle
         log(`[${walletId}] Added as new wallet`);
         wallets[walletId] = {
             ...result.wallet,
-            status: 'ready',
+            status: isMultichainAccountWallet(result.wallet)
+                ? __classPrivateFieldGet(this, _AccountTreeController_instances, "m", _AccountTreeController_getMultichainAccountWalletStatus).call(this, result.wallet.metadata.entropy.id)
+                : 'ready',
             groups: {},
             metadata: {
                 name: '', // Will get updated later.
@@ -925,7 +932,7 @@ _AccountTreeController_accountIdToContext = new WeakMap(), _AccountTreeControlle
         };
         wallet = wallets[walletId];
         // Trigger atomic sync for new wallet (only for entropy wallets)
-        if (wallet.type === AccountWalletType.Entropy) {
+        if (isMultichainAccountWallet(wallet)) {
             __classPrivateFieldGet(this, _AccountTreeController_backupAndSyncService, "f").enqueueSingleWalletSync(walletId);
         }
     }
@@ -940,6 +947,10 @@ _AccountTreeController_accountIdToContext = new WeakMap(), _AccountTreeControlle
             ...result.group,
             // Type-wise, we are guaranteed to always have at least 1 account.
             accounts: [id],
+            ...(isMultichainAccountGroup(result.group) &&
+                isMultichainAccountWallet(result.wallet) && {
+                status: __classPrivateFieldGet(this, _AccountTreeController_instances, "m", _AccountTreeController_getMultichainAccountGroupStatus).call(this, result.wallet.metadata.entropy.id, result.group.metadata.entropy.groupIndex),
+            }),
             metadata: {
                 name: '',
                 ...{ pinned: false, hidden: false, lastSelected: 0 }, // Default UI states
@@ -952,7 +963,7 @@ _AccountTreeController_accountIdToContext = new WeakMap(), _AccountTreeControlle
         // Map group ID to its containing wallet ID for efficient direct access
         __classPrivateFieldGet(this, _AccountTreeController_groupIdToWalletId, "f").set(groupId, walletId);
         // Trigger atomic sync for new group (only for entropy wallets)
-        if (wallet.type === AccountWalletType.Entropy) {
+        if (isMultichainAccountWallet(wallet)) {
             __classPrivateFieldGet(this, _AccountTreeController_backupAndSyncService, "f").enqueueSingleGroupSync(groupId);
         }
     }
@@ -1009,9 +1020,6 @@ _AccountTreeController_accountIdToContext = new WeakMap(), _AccountTreeControlle
     if (!accountToSelect) {
         throw new Error('No accounts found in group');
     }
-    const gutoSetGroupStart = Date.now();
-    console.log('[GUTO PERFORMANCE LOG] AccountTreeController.#setSelectedAccountGroup START');
-    const gutoUpdateStart = Date.now();
     this.update((state) => {
         // Update our selected account group first.
         state.selectedAccountGroup = groupId;
@@ -1028,20 +1036,14 @@ _AccountTreeController_accountIdToContext = new WeakMap(), _AccountTreeControlle
             state.accountTree.wallets[walletId].groups[groupId].metadata.lastSelected = now;
         }
     });
-    console.log(`[GUTO PERFORMANCE LOG] AccountTreeController state.update (sync) took ${Date.now() - gutoUpdateStart}ms`);
     log(`Selected group is now: [${this.state.selectedAccountGroup}]`);
-    const gutoPublishStart = Date.now();
     this.messenger.publish(`${controllerName}:selectedAccountGroupChange`, groupId, previousSelectedAccountGroup);
-    console.log(`[GUTO PERFORMANCE LOG] AccountTreeController publish selectedAccountGroupChange (sync subscriber fan-out) took ${Date.now() - gutoPublishStart}ms`);
     if (forwardSelectedAccount) {
         // Update AccountsController - this will trigger selectedAccountChange event,
         // but our handler is idempotent so it won't cause infinite loop
-        const gutoSetAccountStart = Date.now();
         this.messenger.call('AccountsController:setSelectedAccount', accountToSelect);
-        console.log(`[GUTO PERFORMANCE LOG] AccountTreeController AccountsController:setSelectedAccount (sync selectedAccountChange fan-out) took ${Date.now() - gutoSetAccountStart}ms`);
         log(`Selected account is now: ${accountToSelect}`);
     }
-    console.log(`[GUTO PERFORMANCE LOG] AccountTreeController.#setSelectedAccountGroup END (total, sync) took ${Date.now() - gutoSetGroupStart}ms`);
 }, _AccountTreeController_getDefaultSelectedAccountGroup = function _AccountTreeController_getDefaultSelectedAccountGroup(wallets) {
     const selectedAccount = this.messenger.call('AccountsController:getSelectedMultichainAccount');
     if (selectedAccount?.id) {
@@ -1070,6 +1072,30 @@ _AccountTreeController_accountIdToContext = new WeakMap(), _AccountTreeControlle
             wallet.status = walletStatus;
         }
     });
+}, _AccountTreeController_handleMultichainAccountGroupStatusChange = function _AccountTreeController_handleMultichainAccountGroupStatusChange(groupId, groupStatus) {
+    this.update((state) => {
+        const walletId = __classPrivateFieldGet(this, _AccountTreeController_groupIdToWalletId, "f").get(groupId);
+        if (walletId) {
+            const group = state.accountTree.wallets[walletId]?.groups[groupId];
+            if (group && isMultichainAccountGroup(group)) {
+                group.status = groupStatus;
+            }
+        }
+    });
+}, _AccountTreeController_getMultichainAccountWalletStatus = function _AccountTreeController_getMultichainAccountWalletStatus(entropySource) {
+    try {
+        return this.messenger.call('MultichainAccountService:getMultichainAccountWallet', { entropySource }).status;
+    }
+    catch {
+        return 'uninitialized';
+    }
+}, _AccountTreeController_getMultichainAccountGroupStatus = function _AccountTreeController_getMultichainAccountGroupStatus(entropySource, groupIndex) {
+    try {
+        return this.messenger.call('MultichainAccountService:getMultichainAccountGroup', { entropySource, groupIndex }).status;
+    }
+    catch {
+        return 'uninitialized';
+    }
 }, _AccountTreeController_getAccountGroup = function _AccountTreeController_getAccountGroup(groupId) {
     const found = Object.values(this.state.accountTree.wallets).find((wallet) => wallet.groups[groupId] !== undefined);
     return found?.groups[groupId];
@@ -1138,6 +1164,7 @@ _AccountTreeController_accountIdToContext = new WeakMap(), _AccountTreeControlle
         controllerStateUpdateFn: this.update.bind(this),
         traceFn: __classPrivateFieldGet(this, _AccountTreeController_trace, "f").bind(this),
         groupIdToWalletId: __classPrivateFieldGet(this, _AccountTreeController_groupIdToWalletId, "f"),
+        mutationTracker: createSyncMutationTracker(),
     };
 };
 //# sourceMappingURL=AccountTreeController.mjs.map