diff --git a/packages/assets-controller/src/data-sources/SnapDataSource.ts b/packages/assets-controller/src/data-sources/SnapDataSource.ts index 17da7b0b5e3..1615bb2838b 100644 --- a/packages/assets-controller/src/data-sources/SnapDataSource.ts +++ b/packages/assets-controller/src/data-sources/SnapDataSource.ts @@ -47,6 +47,7 @@ export type AccountBalancesUpdatedEventPayload = { [assetId: string]: { amount: string; unit: string; + metadata?: Record; }; }; }; @@ -292,6 +293,7 @@ export class SnapDataSource extends AbstractDataSource< accountAssets ??= {}; accountAssets[assetId as Caip19AssetId] = { amount: balance.amount, + ...(balance.metadata ? { metadata: balance.metadata } : {}), }; } } @@ -483,7 +485,7 @@ export class SnapDataSource extends AbstractDataSource< } // Step 2: Get balances for those specific assets - const balances: Record = + const balances: Record = await client.getAccountBalances( accountId, accountAssets as CaipAssetType[], @@ -497,6 +499,7 @@ export class SnapDataSource extends AbstractDataSource< if (accountBalances) { (accountBalances as Record)[assetId] = { amount: balance.amount, + ...(balance.metadata ? { metadata: balance.metadata } : {}), }; } }