const balance = await sdk.getBalance('0x...token-address');
console.log('Shielded:', balance.shieldedBalance);
console.log('Wallet:', balance.walletBalance);

interface TokenBalance {
  tokenAddress: string;
  shieldedBalance: string;  // Amount in wei
  walletBalance: string;    // Amount in wei
  symbol?: string;
  decimals: number;
}

const allBalances = await sdk.getAllBalances();
for (const balance of allBalances) {
  const formatted = sdk.formatAmount(balance.shieldedBalance, balance.decimals);
  console.log(`${balance.symbol}: ${formatted} (shielded)`);
}

const balance = await sdk.getBalance(tokenAddress);

const shielded = sdk.formatAmount(balance.shieldedBalance, balance.decimals);
const wallet = sdk.formatAmount(balance.walletBalance, balance.decimals);

console.log(`Shielded: ${shielded} ${balance.symbol}`);
console.log(`Wallet: ${wallet} ${balance.symbol}`);

// Set tracked tokens
sdk.setTrackedTokens(['0x...token1', '0x...token2']);

// Add more tokens
sdk.addTrackedTokens(['0x...token3']);

// Get tracked list
const tracked = sdk.getTrackedTokens();

try {
  const balance = await sdk.getBalance(tokenAddress);
} catch (error) {
  if (error.message.includes('Not authenticated')) {
    console.log('Please log in first');
  } else if (error.message.includes('Network error')) {
    console.log('Network error, please retry');
  } else {
    console.log('Balance error:', error.message);
  }
}