-
Notifications
You must be signed in to change notification settings - Fork 49
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
fix: added token balance fix bug with duplicate baseUrl (#183)
* fix: added balance fix bug wuth url path * fix: removed logs
- Loading branch information
1 parent
c668050
commit 40d44ac
Showing
4 changed files
with
221 additions
and
12 deletions.
There are no files selected for viewing
This file contains 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 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 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 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,203 @@ | ||
| import type {Address, Chain} from '@starknet-react/chains'; | ||
| import {useContract, useNetwork} from '@starknet-react/core'; | ||
| import {QueryKey, useQuery, UseQueryOptions, UseQueryResult} from '@tanstack/react-query'; | ||
| import {useMemo} from 'react'; | ||
| import {type BlockNumber, type CallOptions, BlockTag, num, shortString} from 'starknet'; | ||
| import {formatUnits} from 'viem'; | ||
|
|
||
| export type Balance = { | ||
| decimals: number; | ||
| symbol: string; | ||
| formatted: string; | ||
| value: bigint; | ||
| }; | ||
| export type UseQueryProps< | ||
| TQueryFnData = unknown, | ||
| TError = unknown, | ||
| TData = TQueryFnData, | ||
| TQueryKey extends QueryKey = QueryKey, | ||
| > = Pick< | ||
| UseQueryOptions<TQueryFnData, TError, TData, TQueryKey>, | ||
| 'enabled' | 'refetchInterval' | 'retry' | 'retryDelay' | ||
| >; | ||
|
|
||
| export type UseBalanceProps = UseQueryProps< | ||
| Balance, | ||
| Error, | ||
| Balance, | ||
| ReturnType<typeof queryKey> | ||
| > & { | ||
| /** The contract's address. Defaults to the native currency. */ | ||
| token?: Address | string; | ||
| /** The address to fetch balance for. */ | ||
| address?: Address | string; | ||
| /** Whether to watch for changes. */ | ||
| watch?: boolean; | ||
| /** Block identifier used when performing call. */ | ||
| blockIdentifier?: BlockNumber; | ||
| }; | ||
|
|
||
| export type UseBalanceResult = UseQueryResult<Balance, Error>; | ||
|
|
||
| /** | ||
| * Fetch the balance for the provided address and token. | ||
| * | ||
| * If no token is provided, the native currency is used. | ||
| */ | ||
| export function useBalanceUtil({ | ||
| token: token_, | ||
| address, | ||
| enabled: enabled_ = true, | ||
| blockIdentifier = BlockTag.latest, | ||
| ...props | ||
| }: UseBalanceProps) { | ||
| const {chain} = useNetwork(); | ||
| const token = token_ ?? chain.nativeCurrency.address; | ||
|
|
||
| const {contract} = useContract({ | ||
| abi: balanceABIFragment, | ||
| address: token, | ||
| }); | ||
|
|
||
| const queryKey_ = useMemo( | ||
| () => queryKey({chain, token, address, blockIdentifier}), | ||
| [chain, token, address, blockIdentifier], | ||
| ); | ||
|
|
||
| const enabled = useMemo( | ||
| () => Boolean(enabled_ && contract && address), | ||
| [enabled_, contract, address], | ||
| ); | ||
|
|
||
| return useQuery({ | ||
| enabled, | ||
| queryKey: queryKey_, | ||
| queryFn: queryFn({chain, contract, token, address, blockIdentifier}), | ||
| ...props, | ||
| }); | ||
| } | ||
|
|
||
| function queryKey({ | ||
| chain, | ||
| token, | ||
| address, | ||
| blockIdentifier, | ||
| }: { | ||
| chain: Chain; | ||
| token?: string; | ||
| address?: string; | ||
| blockIdentifier?: BlockNumber; | ||
| }) { | ||
| return [ | ||
| { | ||
| entity: 'balance', | ||
| chainId: chain?.name, | ||
| token, | ||
| address, | ||
| blockIdentifier, | ||
| }, | ||
| ] as const; | ||
| } | ||
|
|
||
| function queryFn({ | ||
| chain, | ||
| token, | ||
| address, | ||
| contract, | ||
| blockIdentifier, | ||
| }: { | ||
| chain: Chain; | ||
| token?: string; | ||
| address?: string; | ||
| contract?: any; | ||
| blockIdentifier?: BlockNumber; | ||
| }) { | ||
| return async () => { | ||
| if (!address) throw new Error('address is required'); | ||
| if (!contract) throw new Error('contract is required'); | ||
|
|
||
| const options: CallOptions = { | ||
| blockIdentifier, | ||
| }; | ||
|
|
||
| const isNativeCurrency = token === chain.nativeCurrency.address; | ||
|
|
||
| let symbol = chain.nativeCurrency.symbol; | ||
| if (!isNativeCurrency) { | ||
| const symbol_ = await contract.symbol(options); | ||
| symbol = shortString.decodeShortString(num.toHex(symbol_)); | ||
| } | ||
|
|
||
| let decimals = chain.nativeCurrency.decimals; | ||
| if (!isNativeCurrency) { | ||
| const decimals_ = await contract.decimals(options); | ||
| decimals = Number(decimals_); | ||
| } | ||
|
|
||
| const balanceOf = (await contract.balanceOf(address, options)) as bigint; | ||
|
|
||
| const formatted = formatUnits(balanceOf, decimals); | ||
|
|
||
| return { | ||
| value: balanceOf, | ||
| decimals, | ||
| symbol, | ||
| formatted, | ||
| }; | ||
| }; | ||
| } | ||
|
|
||
| const balanceABIFragment = [ | ||
| { | ||
| name: 'core::integer::u256', | ||
| type: 'struct', | ||
| members: [ | ||
| { | ||
| name: 'low', | ||
| type: 'core::integer::u128', | ||
| }, | ||
| { | ||
| name: 'high', | ||
| type: 'core::integer::u128', | ||
| }, | ||
| ], | ||
| }, | ||
| { | ||
| name: 'balanceOf', | ||
| type: 'function', | ||
| inputs: [ | ||
| { | ||
| name: 'account', | ||
| type: 'core::starknet::contract_address::ContractAddress', | ||
| }, | ||
| ], | ||
| outputs: [ | ||
| { | ||
| type: 'core::integer::u256', | ||
| }, | ||
| ], | ||
| state_mutability: 'view', | ||
| }, | ||
| { | ||
| name: 'symbol', | ||
| type: 'function', | ||
| inputs: [], | ||
| outputs: [ | ||
| { | ||
| type: 'core::felt252', | ||
| }, | ||
| ], | ||
| state_mutability: 'view', | ||
| }, | ||
| { | ||
| name: 'decimals', | ||
| type: 'function', | ||
| inputs: [], | ||
| outputs: [ | ||
| { | ||
| type: 'core::integer::u8', | ||
| }, | ||
| ], | ||
| state_mutability: 'view', | ||
| }, | ||
| ] as const; |