@deserialize/multi-vm-wallet 1.6.0 → 1.6.1

package/dist/savings/multi-chain-savings.d.ts

@@ -17,6 +17,7 @@ export interface PocketBalance {
     balances: {
         token: string;
         balance: Balance;
+        balanceInUsd: number;
     }[];
 }
 export declare class MultiChainSavingsManager {
@@ -25,6 +26,9 @@ export declare class MultiChainSavingsManager {
     private evmManagers;
     private svmManagers;
     private chainConfigs;
+    private getTokenKey;
+    private normalizePriceMap;
+    private getPriceMap;
     constructor(mnemonic: string, chains: ChainConfig[], walletIndex?: number);
     addChain(chain: ChainConfig): void;
     removeChain(chainId: string): void;

package/dist/savings/multi-chain-savings.js

@@ -3,12 +3,41 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.MultiChainSavingsManager = void 0;
 const evm_savings_1 = require("./evm-savings");
 const svm_savings_1 = require("./svm-savings");
+const price_1 = require("../price");
 class MultiChainSavingsManager {
     mnemonic;
     walletIndex;
     evmManagers = new Map();
     svmManagers = new Map();
     chainConfigs = new Map();
+    getTokenKey(tokenAddress, chainType) {
+        if (chainType === 'EVM') {
+            return tokenAddress.toLowerCase();
+        }
+        return tokenAddress;
+    }
+    normalizePriceMap(prices, chainType) {
+        const priceMap = new Map();
+        for (const item of prices) {
+            if (!item?.tokenAddress || typeof item.price !== 'number')
+                continue;
+            priceMap.set(this.getTokenKey(item.tokenAddress, chainType), item.price);
+        }
+        return priceMap;
+    }
+    async getPriceMap(chain, tokenAddresses) {
+        const chainWithId = chain.config;
+        if (typeof chainWithId.chainId !== 'number') {
+            return new Map();
+        }
+        const uniqueTokenAddresses = Array.from(new Set(tokenAddresses));
+        const priceResult = await (0, price_1.fetchPrices)({
+            vm: chain.type,
+            chainId: chainWithId.chainId,
+            tokenAddresses: uniqueTokenAddresses
+        });
+        return this.normalizePriceMap(priceResult.data?.prices ?? [], chain.type);
+    }
     constructor(mnemonic, chains, walletIndex = 0) {
         if (!mnemonic || typeof mnemonic !== 'string') {
             throw new Error('Mnemonic must be a non-empty string');
@@ -96,14 +125,21 @@ class MultiChainSavingsManager {
             const manager = this.evmManagers.get(chainId);
             const balances = await manager.getPocketBalance(pocketIndex, tokens);
             const pocket = manager.getPocket(pocketIndex);
+            const balanceRows = balances.map(b => ({
+                token: b.address === 'native' ? 'native' : b.address,
+                balance: b.balance
+            }));
+            const priceMap = await this.getPriceMap(chain, balanceRows.map(row => row.token));
             return {
                 chainId,
                 chainType: 'EVM',
                 pocketIndex,
                 address: pocket.address,
-                balances: balances.map(b => ({
-                    token: b.address === 'native' ? 'native' : b.address,
-                    balance: b.balance
+                balances: balanceRows.map(row => ({
+                    token: row.token,
+                    balance: row.balance,
+                    balanceInUsd: row.balance.formatted *
+                        (priceMap.get(this.getTokenKey(row.token, 'EVM')) ?? 0)
                 }))
             };
         }
@@ -111,14 +147,21 @@ class MultiChainSavingsManager {
             const manager = this.svmManagers.get(chainId);
             const balances = await manager.getPocketBalance(pocketIndex, tokens);
             const pocket = manager.getPocket(pocketIndex);
+            const balanceRows = balances.map(b => ({
+                token: b.address === 'native' ? 'native' : b.address.toBase58(),
+                balance: b.balance
+            }));
+            const priceMap = await this.getPriceMap(chain, balanceRows.map(row => row.token));
             return {
                 chainId,
                 chainType: 'SVM',
                 pocketIndex,
                 address: pocket.address.toBase58(),
-                balances: balances.map(b => ({
-                    token: b.address === 'native' ? 'native' : b.address.toBase58(),
-                    balance: b.balance
+                balances: balanceRows.map(row => ({
+                    token: row.token,
+                    balance: row.balance,
+                    balanceInUsd: row.balance.formatted *
+                        (priceMap.get(this.getTokenKey(row.token, 'SVM')) ?? 0)
                 }))
             };
         }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@deserialize/multi-vm-wallet",
-  "version": "1.6.0",
+  "version": "1.6.1",
   "devDependencies": {
     "@types/bn.js": "^5.2.0",
     "@types/crypto-js": "^4.2.2",

package/utils/savings/multi-chain-savings.ts

@@ -10,6 +10,7 @@ import { SVMSavingsManager } from "./svm-savings";
 import { ChainWalletConfig, Balance } from "../types";
 import { Hex } from "viem";
 import { PublicKey } from "@solana/web3.js";
+import { fetchPrices } from "../price";
 
 /**
  * Chain type identifier
@@ -46,6 +47,8 @@ export interface PocketBalance {
         token: string;
         /** Balance information */
         balance: Balance;
+        /** Balance in USD (if unavailable, defaults to 0) */
+        balanceInUsd: number;
     }[];
 }
 
@@ -90,6 +93,44 @@ export class MultiChainSavingsManager {
     // Track chain configs
     private chainConfigs: Map<string, ChainConfig> = new Map();
 
+    private getTokenKey(tokenAddress: string, chainType: ChainType): string {
+        if (chainType === 'EVM') {
+            return tokenAddress.toLowerCase();
+        }
+        return tokenAddress;
+    }
+
+    private normalizePriceMap(
+        prices: Array<{ tokenAddress: string; price: number }>,
+        chainType: ChainType
+    ): Map<string, number> {
+        const priceMap = new Map<string, number>();
+        for (const item of prices) {
+            if (!item?.tokenAddress || typeof item.price !== 'number') continue;
+            priceMap.set(this.getTokenKey(item.tokenAddress, chainType), item.price);
+        }
+        return priceMap;
+    }
+
+    private async getPriceMap(
+        chain: ChainConfig,
+        tokenAddresses: string[]
+    ): Promise<Map<string, number>> {
+        const chainWithId = chain.config as Partial<ChainWalletConfig>;
+        if (typeof chainWithId.chainId !== 'number') {
+            return new Map();
+        }
+
+        const uniqueTokenAddresses = Array.from(new Set(tokenAddresses));
+        const priceResult = await fetchPrices({
+            vm: chain.type,
+            chainId: chainWithId.chainId,
+            tokenAddresses: uniqueTokenAddresses
+        });
+
+        return this.normalizePriceMap(priceResult.data?.prices ?? [], chain.type);
+    }
+
     /**
      * Create a new MultiChainSavingsManager
      *
@@ -253,30 +294,52 @@ export class MultiChainSavingsManager {
             const manager = this.evmManagers.get(chainId)!;
             const balances = await manager.getPocketBalance(pocketIndex, tokens);
             const pocket = manager.getPocket(pocketIndex);
+            const balanceRows = balances.map(b => ({
+                token: b.address === 'native' ? 'native' : b.address,
+                balance: b.balance
+            }));
+            const priceMap = await this.getPriceMap(
+                chain,
+                balanceRows.map(row => row.token)
+            );
 
             return {
                 chainId,
                 chainType: 'EVM',
                 pocketIndex,
                 address: pocket.address,
-                balances: balances.map(b => ({
-                    token: b.address === 'native' ? 'native' : b.address,
-                    balance: b.balance
+                balances: balanceRows.map(row => ({
+                    token: row.token,
+                    balance: row.balance,
+                    balanceInUsd:
+                        row.balance.formatted *
+                        (priceMap.get(this.getTokenKey(row.token, 'EVM')) ?? 0)
                 }))
             };
         } else {
             const manager = this.svmManagers.get(chainId)!;
             const balances = await manager.getPocketBalance(pocketIndex, tokens);
             const pocket = manager.getPocket(pocketIndex);
+            const balanceRows = balances.map(b => ({
+                token: b.address === 'native' ? 'native' : b.address.toBase58(),
+                balance: b.balance
+            }));
+            const priceMap = await this.getPriceMap(
+                chain,
+                balanceRows.map(row => row.token)
+            );
 
             return {
                 chainId,
                 chainType: 'SVM',
                 pocketIndex,
                 address: pocket.address.toBase58(),
-                balances: balances.map(b => ({
-                    token: b.address === 'native' ? 'native' : b.address.toBase58(),
-                    balance: b.balance
+                balances: balanceRows.map(row => ({
+                    token: row.token,
+                    balance: row.balance,
+                    balanceInUsd:
+                        row.balance.formatted *
+                        (priceMap.get(this.getTokenKey(row.token, 'SVM')) ?? 0)
                 }))
             };
         }