From 2dd26947afaff36b5c0cb7f2deafd55120e12d16 Mon Sep 17 00:00:00 2001
From: Stanley Yuen <102275989+stanleyyconsensys@users.noreply.github.com>
Date: Tue, 21 Jul 2026 12:33:43 +0800
Subject: [PATCH] feat: add balance metadata for snap datasource

---
 .../assets-controller/src/data-sources/SnapDataSource.ts     | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

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<string, Json>;
       };
     };
   };
@@ -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<CaipAssetType, Balance> =
+        const balances: Record<CaipAssetType, Balance & { metadata?: Json }> =
           await client.getAccountBalances(
             accountId,
             accountAssets as CaipAssetType[],
@@ -497,6 +499,7 @@ export class SnapDataSource extends AbstractDataSource<
             if (accountBalances) {
               (accountBalances as Record<string, unknown>)[assetId] = {
                 amount: balance.amount,
+                ...(balance.metadata ? { metadata: balance.metadata } : {}),
               };
             }
           }