import { permissionedRegistryGetExpirySnippet } from '@ensdomains/ensjs-abi/v2/permissionedRegistry'
import {
  type Address,
  type Chain,
  labelhash,
  type ReadContractErrorType,
} from 'viem'
import { readContract } from 'viem/actions'
import { type GetChainContractAddressErrorType, getAction } from 'viem/utils'
import type { RequireClientContracts } from '../../../../clients/chain.js'
import { getChainContractAddress } from '../../../../clients/chain.js'
import { UnsupportedNameTypeError } from '../../../../errors/general.js'
import { ASSERT_NO_TYPE_ERROR } from '../../../../types/internal.js'
import { getNameType } from '../../../../utils/name/getNameType.js'
import { checkIsDotEth } from '../../../../utils/name/validation.js'

export type GetExpiryParameters = {
  /** Name to get expiry for */
  name: string
  /** Optional custom registry address (defaults to chain's ensRegistry) */
  registryAddress?: Address
}

export type GetExpiryReturnType = bigint

export type GetExpiryErrorType =
  | UnsupportedNameTypeError
  | GetChainContractAddressErrorType
  | ReadContractErrorType

export async function getExpiry<chain extends Chain>(
  client: RequireClientContracts<chain, 'ensRegistry'>,
  { name, registryAddress }: GetExpiryParameters,
): Promise<GetExpiryReturnType> {
  ASSERT_NO_TYPE_ERROR(client)
  const labels = name.split('.')
  const nameType = getNameType(name)
  if (!checkIsDotEth(labels) && nameType !== 'eth-subname')
    throw new UnsupportedNameTypeError({
      nameType,
      supportedNameTypes: ['eth-2ld', 'eth-subname'],
      details: 'Only .eth names can be looked up via the registry',
    })

  const readContractAction = getAction(client, readContract, 'readContract')

  const currentRegistry =
    registryAddress ??
    getChainContractAddress({
      chain: client.chain,
      contract: 'ensRegistry',
    })

  const labelHash = BigInt(labelhash(labels[0]))

  return readContractAction({
    address: currentRegistry,
    abi: permissionedRegistryGetExpirySnippet,
    functionName: 'getExpiry',
    args: [labelHash],
  })
}