d0zingcat/solana-agent-kit
1
0
Fork 0
mirror of https://github.com/d0zingcat/solana-agent-kit.git synced 2026-05-24 15:10:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: parse provided wallet address to token balances

Browse Source
This commit is contained in:
michaelessiet
2025-01-10 17:38:20 +01:00
parent 51cf665c74
commit 6ef55a04da
2 changed files with 55 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

86
src/tools/get_token_balances.ts
View File

@@ -11,55 +11,49 @@ import { getTokenMetadata } from "../utils/tokenMetadata";
*/
export async function get_token_balance(
agent: SolanaAgentKit,
token_address?: PublicKey,
): Promise<
| number
| {
sol: number;
tokens: Array<{
tokenAddress: string;
name: string;
symbol: string;
balance: number;
decimals: number;
}>;
}
> {
if (!token_address) {
const [lamportsBalance, tokenAccountData] = await Promise.all([
agent.connection.getBalance(agent.wallet_address),
agent.connection.getParsedTokenAccountsByOwner(agent.wallet_address, {
walletAddress?: PublicKey,
): Promise<{
sol: number;
tokens: Array<{
tokenAddress: string;
name: string;
symbol: string;
balance: number;
decimals: number;
}>;
}> {
const [lamportsBalance, tokenAccountData] = await Promise.all([
agent.connection.getBalance(walletAddress ?? agent.wallet_address),
agent.connection.getParsedTokenAccountsByOwner(
walletAddress ?? agent.wallet_address,
{
programId: TOKEN_PROGRAM_ID,
}),
]);
},
),
]);
const removedZeroBalance = tokenAccountData.value.filter(
(v) => v.account.data.parsed.info.tokenAmount.uiAmount !== 0,
);
const removedZeroBalance = tokenAccountData.value.filter(
(v) => v.account.data.parsed.info.tokenAmount.uiAmount !== 0,
);
const tokenBalances = await Promise.all(
removedZeroBalance.map(async (v) => {
const mint = v.account.data.parsed.info.mint;
const mintInfo = await getTokenMetadata(agent.connection, mint);
return {
tokenAddress: mint,
name: mintInfo.name ?? "",
symbol: mintInfo.symbol ?? "",
balance: v.account.data.parsed.info.tokenAmount.uiAmount as number,
decimals: v.account.data.parsed.info.tokenAmount.decimals as number,
};
}),
);
const tokenBalances = await Promise.all(
removedZeroBalance.map(async (v) => {
const mint = v.account.data.parsed.info.mint;
const mintInfo = await getTokenMetadata(agent.connection, mint);
return {
tokenAddress: mint,
name: mintInfo.name ?? "",
symbol: mintInfo.symbol ?? "",
balance: v.account.data.parsed.info.tokenAmount.uiAmount as number,
decimals: v.account.data.parsed.info.tokenAmount.decimals as number,
};
}),
);
const solBalance = lamportsBalance / LAMPORTS_PER_SOL;
const solBalance = lamportsBalance / LAMPORTS_PER_SOL;
return {
sol: solBalance,
tokens: tokenBalances,
};
}
const token_account =
await agent.connection.getTokenAccountBalance(token_address);
return token_account.value.uiAmount || 0;
return {
sol: solBalance,
tokens: tokenBalances,
};
}