import { useEffect, useState } from 'react'
import { getENS, ResolvedENS } from 'get-ens'
import type { BaseProvider as Provider } from '@ethersproject/providers'

export type UseENSOptions = {
  provider: Provider
  chainId: number
  domain: string
  fetchOptions?: RequestInit
  contractAddress?: string
}

/**
 * A React hook to fetch ENS records from a domain.
 * @returns
 */
export const useENS = ({ provider, domain, fetchOptions, contractAddress, chainId }: UseENSOptions): ResolvedENS => {
  const [data, set] = useState<ResolvedENS>({ address: null, owner: null, records: {}, domain: '' })

  useEffect(() => {
    if (provider && domain && data.owner == null) {
      if (chainId) {
        if (contractAddress || chainId === 1) {
          getENS(provider, contractAddress)(domain, fetchOptions).then((data) => {
            set(data)
          })
        }
      } else {
        provider.getNetwork().then(({ chainId }) => {
          if (contractAddress || chainId === 1) {
            getENS(provider, contractAddress)(domain, fetchOptions).then((data) => {
              set(data)
            })
          }
        })
      }
    }
  }, [contractAddress, domain, fetchOptions])

  return data
}