import { getTronWeb, isTronWalletAccount } from "@dynamic-labs-sdk/tron";
import { getPrimaryWalletAccount } from "@dynamic-labs-sdk/client";

const walletAccount = getPrimaryWalletAccount();

if (walletAccount && isTronWalletAccount(walletAccount)) {
  const tronWeb = await getTronWeb({ walletAccount });

  // Use TronWeb to interact with the Tron blockchain
  const balance = await tronWeb.trx.getBalance(walletAccount.address);
  console.log("Balance in SUN:", balance);

  // Convert to TRX
  const balanceInTrx = tronWeb.fromSun(balance);
  console.log("Balance in TRX:", balanceInTrx);
}

import { getTronWeb, isTronWalletAccount } from '@dynamic-labs-sdk/tron';
import { useGetWalletAccounts } from '@dynamic-labs-sdk/react-hooks';
import { useEffect, useState } from 'react';

function TronBalance() {
  const { data: walletAccounts = [] } = useGetWalletAccounts();
  const walletAccount = walletAccounts.find(isTronWalletAccount);
  const [balance, setBalance] = useState('');

  useEffect(() => {
    if (!walletAccount) return;
    getTronWeb({ walletAccount }).then(async (tronWeb) => {
      const bal = await tronWeb.trx.getBalance(walletAccount.address);
      setBalance(tronWeb.fromSun(bal));
    });
  }, [walletAccount]);

  return <p>TRX balance: {balance}</p>;
}