import { permissionedRegistryRoleCountSnippet } from '@ensdomains/ensjs-abi/v2/permissionedRegistry'
import { type Client, labelhash } from 'viem'
import type { Address } from 'viem/accounts'
import { type ReadContractErrorType, readContract } from 'viem/actions'
import { getAction } from 'viem/utils'
import { ASSERT_NO_TYPE_ERROR } from '../../../../types/internal.js'
import {
  type DecodeRoleCountsReturnType,
  decodeRoleCounts,
  registryRoles,
} from '../../../../utils/v2/index.js'

export type GetRoleCountsParameters = {
  registryAddress: Address
  label: string
}

export type GetRoleCountsReturnType = {
  decoded: DecodeRoleCountsReturnType<typeof registryRoles>
  raw: bigint
}

export type GetRoleCountsErrorType = ReadContractErrorType

/**
 * Gets the registry's role counts for a name's label.
 * @param client - {@link Client}
 * @param parameters - {@link GetRoleCountsParameters}
 * @returns Decoded and raw role counts. {@link GetRoleCountsReturnType}
 *
 * @example
 * import { createPublicClient, http } from 'viem'
 * import { mainnet } from 'viem/chains'
 * import { addEnsContracts } from '@ensdomains/ensjs'
 * import { getRoleCounts } from '@ensdomains/ensjs/public/v2'
 *
 * const client = createPublicClient({
 *   chain: addEnsContracts(mainnet),
 *   transport: http(),
 * })
 * const result = await getRoleCounts(client, { registryAddress: '0x0f3eb298470639a96bd548cea4a648bc80b2cee2', label: 'raffy' })
 */
export async function getRoleCounts(
  client: Client,
  { registryAddress, label }: GetRoleCountsParameters,
): Promise<GetRoleCountsReturnType> {
  ASSERT_NO_TYPE_ERROR(client)

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

  const roleBitmap = await readContractAction({
    address: registryAddress,
    abi: permissionedRegistryRoleCountSnippet,
    functionName: 'roleCount',
    args: [BigInt(labelhash(label))],
  })

  return {
    decoded: decodeRoleCounts(roleBitmap, registryRoles),
    raw: roleBitmap,
  }
}