web3-tools-mcp 1.6.0 → 2.0.0

package/src/tools/balance.ts

@@ -1,16 +1,13 @@
 import { type Address, formatEther, formatUnits, isAddress } from 'viem'
 import { z } from 'zod'
-import type { ChainName } from '../types.js'
 import { getClientManager, SUPPORTED_CHAINS } from '../client.js'
+import type { ChainName } from '../types.js'
 import { createTool, formatResponse } from '../utils.js'
 
 const BalanceQuerySchema = z.object({
   chain: z.enum(SUPPORTED_CHAINS).describe('The blockchain network to use'),
   address: z.string().describe('The address to check balance for'),
-  tokenAddress: z
-    .string()
-    .optional()
-    .describe('Token contract address (optional, if not provided returns native token balance)'),
+  tokenAddress: z.string().optional().describe('Token contract address (optional, if not provided returns native token balance)'),
   blockNumber: z.string().optional().describe('Block number to query (optional, defaults to latest)'),
   label: z.string().optional().describe('Optional label for batch results')
 })
@@ -20,7 +17,9 @@ export default {
     'Get Token Balances',
     'Retrieve native and ERC20 token balances for multiple addresses. BATCH OPTIMIZED - use for portfolio tracking, wallet monitoring.',
     z.object({
-      queries: z.array(BalanceQuerySchema).describe('Array of balance queries. Omit tokenAddress for native balance, include for ERC20.')
+      queries: z
+        .array(BalanceQuerySchema)
+        .describe('Array of balance queries. Omit tokenAddress for native balance, include for ERC20.')
     }),
     async (args) => {
       // Group by chain and blockNumber for efficient processing
@@ -47,130 +46,126 @@ export default {
         error?: string
       }> = []
 
-      try {
-        // Process each group
-        for (const [key, queries] of groupedQueries) {
-          const [chain, blockNum] = key.split(':')
-          const clientManager = getClientManager()
-          const client = clientManager.getClient(chain as ChainName)
-          const blockTag = blockNum === 'latest' ? 'latest' : BigInt(blockNum)
-
-          // Process all queries in this group
-          const results = await Promise.allSettled(
-            queries.map(async (query) => {
-              if (!isAddress(query.address)) {
-                throw new Error(`Invalid address format: ${query.address}`)
-              }
-
-              const originalIndex = args.queries.indexOf(query)
-
-              // Native token balance
-              if (!query.tokenAddress) {
-                const balance = await client.getBalance({
-                  address: query.address as Address,
-                  blockNumber: blockTag === 'latest' ? undefined : blockTag
-                })
+      // Process each group
+      for (const [key, queries] of groupedQueries) {
+        const [chain, blockNum] = key.split(':')
+        const clientManager = getClientManager()
+        const client = clientManager.getClient(chain as ChainName)
+        const blockTag = blockNum === 'latest' ? 'latest' : BigInt(blockNum)
 
-                return {
-                  index: originalIndex,
-                  label: query.label,
-                  address: query.address,
-                  chain: query.chain,
-                  blockNumber: query.blockNumber || 'latest',
-                  success: true,
-                  balance,
-                  balanceFormatted: formatEther(balance)
-                }
-              }
+        // Process all queries in this group
+        const results = await Promise.allSettled(
+          queries.map(async (query) => {
+            if (!isAddress(query.address)) {
+              throw new Error(`Invalid address format: ${query.address}`)
+            }
 
-              // ERC20 token balance
-              if (!isAddress(query.tokenAddress)) {
-                throw new Error(`Invalid token address format: ${query.tokenAddress}`)
-              }
+            const originalIndex = args.queries.indexOf(query)
 
-              const balance = await client.readContract({
-                address: query.tokenAddress as Address,
-                abi: [
-                  {
-                    name: 'balanceOf',
-                    type: 'function',
-                    stateMutability: 'view',
-                    inputs: [{ name: 'account', type: 'address' }],
-                    outputs: [{ name: '', type: 'uint256' }]
-                  }
-                ],
-                functionName: 'balanceOf',
-                args: [query.address as Address],
+            // Native token balance
+            if (!query.tokenAddress) {
+              const balance = await client.getBalance({
+                address: query.address as Address,
                 blockNumber: blockTag === 'latest' ? undefined : blockTag
               })
 
-              // Try to get decimals
-              let decimals = 18
-              try {
-                decimals = (await client.readContract({
-                  address: query.tokenAddress as Address,
-                  abi: [
-                    {
-                      name: 'decimals',
-                      type: 'function',
-                      stateMutability: 'view',
-                      inputs: [],
-                      outputs: [{ name: '', type: 'uint8' }]
-                    }
-                  ],
-                  functionName: 'decimals',
-                  blockNumber: blockTag === 'latest' ? undefined : blockTag
-                })) as number
-              } catch {
-                // Use default decimals if call fails
-              }
-
               return {
                 index: originalIndex,
                 label: query.label,
                 address: query.address,
-                tokenAddress: query.tokenAddress,
                 chain: query.chain,
                 blockNumber: query.blockNumber || 'latest',
                 success: true,
                 balance,
-                balanceFormatted: formatUnits(balance as bigint, decimals),
-                decimals
+                balanceFormatted: formatEther(balance)
               }
+            }
+
+            // ERC20 token balance
+            if (!isAddress(query.tokenAddress)) {
+              throw new Error(`Invalid token address format: ${query.tokenAddress}`)
+            }
+
+            const balance = await client.readContract({
+              address: query.tokenAddress as Address,
+              abi: [
+                {
+                  name: 'balanceOf',
+                  type: 'function',
+                  stateMutability: 'view',
+                  inputs: [{ name: 'account', type: 'address' }],
+                  outputs: [{ name: '', type: 'uint256' }]
+                }
+              ],
+              functionName: 'balanceOf',
+              args: [query.address as Address],
+              blockNumber: blockTag === 'latest' ? undefined : blockTag
             })
-          )
 
-          // Process results
-          results.forEach((result, idx) => {
-            if (result.status === 'fulfilled') {
-              allResults[result.value.index] = result.value
-            } else {
-              const query = queries[idx]!
-              const originalIndex = args.queries.indexOf(query)
-              allResults[originalIndex] = {
-                index: originalIndex,
-                label: query.label,
-                address: query.address,
-                tokenAddress: query.tokenAddress,
-                chain: query.chain,
-                blockNumber: query.blockNumber || 'latest',
-                success: false,
-                error: result.reason instanceof Error ? result.reason.message : String(result.reason)
-              }
+            // Try to get decimals
+            let decimals = 18
+            try {
+              decimals = (await client.readContract({
+                address: query.tokenAddress as Address,
+                abi: [
+                  {
+                    name: 'decimals',
+                    type: 'function',
+                    stateMutability: 'view',
+                    inputs: [],
+                    outputs: [{ name: '', type: 'uint8' }]
+                  }
+                ],
+                functionName: 'decimals',
+                blockNumber: blockTag === 'latest' ? undefined : blockTag
+              })) as number
+            } catch {
+              // Use default decimals if call fails
+            }
+
+            return {
+              index: originalIndex,
+              label: query.label,
+              address: query.address,
+              tokenAddress: query.tokenAddress,
+              chain: query.chain,
+              blockNumber: query.blockNumber || 'latest',
+              success: true,
+              balance,
+              balanceFormatted: formatUnits(balance as bigint, decimals),
+              decimals
             }
           })
-        }
+        )
 
-        return formatResponse({
-          success: true,
-          totalQueries: args.queries.length,
-          successfulQueries: allResults.filter((r) => r.success).length,
-          failedQueries: allResults.filter((r) => !r.success).length,
-          results: allResults
+        // Process results
+        results.forEach((result, idx) => {
+          if (result.status === 'fulfilled') {
+            allResults[result.value.index] = result.value
+          } else {
+            const query = queries[idx]!
+            const originalIndex = args.queries.indexOf(query)
+            allResults[originalIndex] = {
+              index: originalIndex,
+              label: query.label,
+              address: query.address,
+              tokenAddress: query.tokenAddress,
+              chain: query.chain,
+              blockNumber: query.blockNumber || 'latest',
+              success: false,
+              error: result.reason instanceof Error ? result.reason.message : String(result.reason)
+            }
+          }
         })
-      } catch (error) {
-        throw new Error(`Balance query failed: ${error}`)
       }
+
+      return formatResponse({
+        success: true,
+        totalQueries: args.queries.length,
+        successfulQueries: allResults.filter((r) => r.success).length,
+        failedQueries: allResults.filter((r) => !r.success).length,
+        results: allResults
+      })
     }
   )
 }