Uh oh!
There was an error while loading. Please reload this page.
feat: Add comprehensive BONK token support with multi-token swap func…#12502
Uh oh!
There was an error while loading. Please reload this page.
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,47 +1,40 @@ | ||
| import { PublicKey } from '@solana/web3.js' | ||
| import { Env } from '~/services/env' | ||
| import { getOrInitializeRegistry, type TokenConfig } from '~/services/tokens' | ||
| import { UserBankManagedTokenInfo } from './types' | ||
| const CLAIMABLE_TOKEN_MINTS = ['wAUDIO', 'USDC', 'BONK'] as const | ||
| type ClaimableTokenMint = (typeof CLAIMABLE_TOKEN_MINTS)[number] | ||
| const isClaimableTokenMint = (symbol: string): symbol is ClaimableTokenMint => { | ||
| return CLAIMABLE_TOKEN_MINTS.includes(symbol as ClaimableTokenMint) | ||
| } | ||
| export const JUPITER_PROGRAM_ID = new PublicKey( | ||
| 'JUP6LkbZbjS1jKKwapdHNy74zcZ3tLUZoi5QNyVTaV4' | ||
| ) | ||
| export const createUserBankManagedTokens = ( | ||
| env: Env | ||
| ): Record<string, UserBankManagedTokenInfo> => { | ||
| const registry = getOrInitializeRegistry(env.ENVIRONMENT) | ||
| // Get all userbank-enabled tokens from registry | ||
| const userbankTokens: TokenConfig[] = registry.getUserbankTokens() | ||
| const managedTokens: Record<string, UserBankManagedTokenInfo> = {} | ||
| // Convert registry tokens to UserBankManagedTokenInfo format | ||
| userbankTokens.forEach((token: TokenConfig) => { | ||
| if (isClaimableTokenMint(token.symbol)) { | ||
| managedTokens[token.address.toUpperCase()] = { | ||
Contributor There was a problem hiding this comment. i think solana addresses are case sensitive? | ||
| mintAddress: token.address, | ||
| claimableTokenMint: token.symbol, | ||
| decimals: token.decimals | ||
| } | ||
| } | ||
| }) | ||
| return managedTokens | ||
| } | ||
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,145 @@ | ||
| import { useCallback } from 'react' | ||
| import { FixedDecimal } from '@audius/fixed-decimal' | ||
| import { TokenAccountNotFoundError } from '@solana/spl-token' | ||
| import { Commitment } from '@solana/web3.js' | ||
| import { useQuery, useQueryClient } from '@tanstack/react-query' | ||
| import { useCurrentAccountUser } from '~/api' | ||
| import { useQueryContext } from '~/api/tan-query/utils' | ||
| import { Status } from '~/models/Status' | ||
| import { MintName } from '~/services/audius-backend/solana' | ||
| import { getUserbankAccountInfo, getTokenBySymbol } from '~/services/index' | ||
| import { QUERY_KEYS } from '../queryKeys' | ||
| import { QueryOptions, type QueryKey } from '../types' | ||
| const createTokenBalance = ( | ||
Contributor There was a problem hiding this comment. nit: is this rly the cleanest way? can we not do something like:
Contributor There was a problem hiding this comment. idk i guess that's kind of annoying too.. | ||
| amount: string | number | bigint | null | undefined, | ||
| decimals: number | ||
| ): FixedDecimal | null => { | ||
| if (amount === null || amount === undefined) { | ||
| return null | ||
| } | ||
| return new FixedDecimal(BigInt(amount.toString()), decimals) | ||
| } | ||
| export const getTokenBalanceQueryKey = ( | ||
| ethAddress: string | null, | ||
| token: MintName, | ||
| commitment: Commitment | ||
| ) => | ||
| [ | ||
| QUERY_KEYS.tokenBalance, | ||
| ethAddress, | ||
| token, | ||
| commitment | ||
| ] as unknown as QueryKey<FixedDecimal | null> | ||
| /** | ||
| * Hook to get the balance for any supported token for the current user. | ||
| * Uses TanStack Query for data fetching and caching. | ||
| * | ||
| * @param token The token to fetch balance for | ||
| * @param options Options for the query and polling | ||
| * @returns Object with status, data, refresh, and cancelPolling | ||
| */ | ||
| export const useTokenBalance = ({ | ||
| token, | ||
| isPolling, | ||
| pollingInterval = 1000, | ||
| commitment = 'processed', | ||
| ...queryOptions | ||
| }: { | ||
| token: MintName | ||
| isPolling?: boolean | ||
| pollingInterval?: number | ||
| commitment?: Commitment | ||
| } & QueryOptions) => { | ||
| const { audiusSdk, env } = useQueryContext() | ||
| const { data: user } = useCurrentAccountUser() | ||
| const ethAddress = user?.wallet ?? null | ||
| const queryClient = useQueryClient() | ||
| const result = useQuery({ | ||
| queryKey: getTokenBalanceQueryKey(ethAddress, token, commitment), | ||
| queryFn: async () => { | ||
| const sdk = await audiusSdk() | ||
| if (!ethAddress || !token) { | ||
| return null | ||
| } | ||
| try { | ||
| const account = await getUserbankAccountInfo( | ||
| sdk, | ||
| { | ||
| ethAddress, | ||
| mint: token | ||
| }, | ||
| commitment | ||
| ) | ||
| // Get token configuration from registry to get decimal places | ||
| const tokenConfig = getTokenBySymbol(env, token) | ||
| if (!tokenConfig) { | ||
| console.warn( | ||
| `Token not found in registry: ${token}, returning null balance` | ||
| ) | ||
| return null | ||
| } | ||
| return createTokenBalance(account?.amount, tokenConfig.decimals) | ||
| } catch (e) { | ||
| // If user doesn't have a token account yet, return 0 balance | ||
| if (e instanceof TokenAccountNotFoundError) { | ||
| const tokenConfig = getTokenBySymbol(env, token) | ||
| if (!tokenConfig) { | ||
| console.warn( | ||
| `Token not found in registry: ${token}, returning null balance` | ||
| ) | ||
| return null | ||
| } | ||
| return createTokenBalance(BigInt(0), tokenConfig.decimals) | ||
| } | ||
| console.error(`Error fetching ${token} balance:`, e) | ||
| // Return null instead of throwing to prevent infinite loading | ||
| return null | ||
| } | ||
| }, | ||
| enabled: !!ethAddress && !!token, | ||
| // TanStack Query's built-in polling - only poll when isPolling is true | ||
| refetchInterval: isPolling ? pollingInterval : false, | ||
| // Prevent refetching when window regains focus during polling to avoid conflicts | ||
| refetchOnWindowFocus: !isPolling, | ||
| ...queryOptions | ||
| }) | ||
| // Map TanStack Query states to the Status enum for API compatibility | ||
Contributor There was a problem hiding this comment. damn is this another migration we need to do | ||
| let status = Status.IDLE | ||
| if (result.isPending) { | ||
| status = Status.LOADING | ||
| } else if (result.isError) { | ||
| status = Status.ERROR | ||
| } else if (result.isSuccess) { | ||
| status = Status.SUCCESS | ||
| } | ||
| // For compatibility with existing code | ||
| const data = result.data ?? null | ||
| // Function to cancel polling by invalidating and refetching the query | ||
| // This effectively stops the current polling cycle | ||
| const cancelPolling = useCallback(() => { | ||
| queryClient.cancelQueries({ | ||
| queryKey: getTokenBalanceQueryKey(ethAddress, token, commitment) | ||
| }) | ||
| }, [queryClient, ethAddress, token, commitment]) | ||
| return { | ||
| status, | ||
| data, | ||
| error: result.error, | ||
| refresh: result.refetch, | ||
| cancelPolling | ||
| } | ||
| } | ||
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.
Uh oh!
There was an error while loading. Please reload this page.
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
whoa cool