import { getAptosClient, isAptosWalletAccount } from "@dynamic-labs-sdk/aptos";
import { getPrimaryWalletAccount } from "@dynamic-labs-sdk/client";

const walletAccount = getPrimaryWalletAccount();

if (walletAccount && isAptosWalletAccount(walletAccount)) {
  const aptosClient = await getAptosClient({ walletAccount });

  // Use the client to interact with the Aptos blockchain
  const balance = await aptosClient.getAccountAPTAmount({
    accountAddress: walletAccount.address,
  });
}

import { getAptosClient, isAptosWalletAccount } from '@dynamic-labs-sdk/aptos';
import { useGetWalletAccounts } from '@dynamic-labs-sdk/react-hooks';
import { useEffect, useState } from 'react';

function AptosBalance() {
  const { data: walletAccounts = [] } = useGetWalletAccounts();
  const walletAccount = walletAccounts.find(isAptosWalletAccount);
  const [balance, setBalance] = useState(null);

  useEffect(() => {
    if (!walletAccount) return;
    getAptosClient({ walletAccount }).then(async (client) => {
      const bal = await client.getAccountAPTAmount({ accountAddress: walletAccount.address });
      setBalance(bal);
    });
  }, [walletAccount]);

  return <p>APT Balance: {balance?.toString() ?? '...'}</p>;
}