mirror of https://github.com/kremalicious/blog.git
34 lines
950 B
TypeScript
34 lines
950 B
TypeScript
import { useEffect, useState } from 'react'
|
|
import useSWR, { type SWRResponse } from 'swr'
|
|
import { useChainId, useAccount } from 'wagmi'
|
|
import type { GetToken } from './types'
|
|
|
|
const fetcher = (url: string) => fetch(url).then((res) => res.json())
|
|
const apiUrl = import.meta.env.PUBLIC_WEB3_API_URL
|
|
|
|
//
|
|
// Wrapper for fetching user tokens with swr.
|
|
//
|
|
export function useFetchTokens(): SWRResponse<GetToken[] | undefined, Error> {
|
|
const chainId = useChainId()
|
|
const { address } = useAccount()
|
|
|
|
const [url, setUrl] = useState<string | undefined>()
|
|
|
|
const fetchResults = useSWR<GetToken[] | undefined>(url, fetcher)
|
|
|
|
// Set url only after we have all data loaded on client,
|
|
// preventing initial fetch.
|
|
useEffect(() => {
|
|
if (!address || !chainId) {
|
|
setUrl(undefined)
|
|
return
|
|
}
|
|
|
|
const url = `${apiUrl}/balance?address=${address}&chainId=${chainId}`
|
|
setUrl(url)
|
|
}, [address, chainId])
|
|
|
|
return fetchResults
|
|
}
|