import { erc165SupportsInterfaceSnippet } from '@ensdomains/ensjs-abi/erc165'
import type { Address, Chain, Hex, MulticallErrorType } from 'viem'
import { multicall } from 'viem/actions'
import { getAction } from 'viem/utils'
import type { RequireClientContracts } from '../../../clients/chain.js'
import { ASSERT_NO_TYPE_ERROR } from '../../../types/internal.js'

export type GetSupportedInterfacesParameters<
  interfaces extends readonly Hex[],
> = {
  address: Address
  interfaces: interfaces
  /** Block number to read at */
  blockNumber?: bigint
}

export type GetSupportedInterfacesReturnType<
  interfaces extends readonly Hex[],
> = {
  -readonly [K in keyof interfaces]: boolean
}

export type GetSupportedInterfacesErrorType = MulticallErrorType

/**
 * Gets the supported interfaces for any contract address.
 * @param client - {@link Client}
 * @param parameters - {@link GetSupportedInterfacesParameters}
 * @returns Array of booleans matching the input array {@link GetSupportedInterfacesReturnType}
 *
 * @example
 * import { createPublicClient, http } from 'viem'
 * import { mainnet } from 'viem/chains'
 * import { addEnsContracts } from '@ensdomains/ensjs'
 * import { getSupportedInterfaces } from '@ensdomains/ensjs/public'
 *
 * const client = createPublicClient({
 *   chain: addEnsContracts(mainnet),
 *   transport: http(),
 * })
 * const result = await getSupportedInterfaces(client, {
 *   address: '0x58774Bb8acD458A640aF0B88238369A167546ef2',
 *   interfaces: ['0x2f435428', '0x23b872dd'],
 * })
 * // [true, false]
 */
export async function getSupportedInterfaces<
  chain extends Chain,
  const interfaces extends readonly Hex[],
>(
  client: RequireClientContracts<chain, 'multicall3'>,
  {
    address,
    interfaces,
    blockNumber,
  }: GetSupportedInterfacesParameters<interfaces>,
): Promise<GetSupportedInterfacesReturnType<interfaces>> {
  ASSERT_NO_TYPE_ERROR(client)

  const multicallAction = getAction(client, multicall, 'multicall')

  const result = await multicallAction({
    contracts: interfaces.map((interfaceId) => ({
      address,
      abi: erc165SupportsInterfaceSnippet,
      functionName: 'supportsInterface',
      args: [interfaceId],
    })),
    blockNumber,
  })

  return result.map(
    (r) => r.status === 'success' && r.result === true,
  ) as GetSupportedInterfacesReturnType<interfaces>
}