import type { BaseError, Hex } from 'viem'
import type { ClientWithPns } from '../../contracts/consts.js'
import type {
  GenericPassthrough,
  Prettify,
  SimpleTransactionRequest,
} from '../../types.js'
import {
  generateFunction,
  type GeneratedFunction,
} from '../../utils/generateFunction.js'
import _getText, {
  type InternalGetTextParameters,
  type InternalGetTextReturnType,
} from './_getText.js'
import universalWrapper from './universalWrapper.js'

export type GetTextRecordParameters = Prettify<
  InternalGetTextParameters & {
    /** Batch gateway URLs to use for resolving CCIP-read requests. */
    gatewayUrls?: string[]
  }
>

export type GetTextRecordReturnType = Prettify<InternalGetTextReturnType>

const encode = (
  client: ClientWithPns,
  { name, key, gatewayUrls }: Omit<GetTextRecordParameters, 'strict'>,
): SimpleTransactionRequest => {
  const prData = _getText.encode(client, { name, key })
  return universalWrapper.encode(client, {
    name,
    data: prData.data,
    gatewayUrls,
  })
}

const decode = async (
  client: ClientWithPns,
  data: Hex | BaseError,
  passthrough: GenericPassthrough,
  {
    strict,
    gatewayUrls,
  }: Pick<GetTextRecordParameters, 'strict' | 'gatewayUrls'>,
): Promise<GetTextRecordReturnType> => {
  const urData = await universalWrapper.decode(client, data, passthrough, {
    strict,
    gatewayUrls,
  })
  if (!urData) return null
  return _getText.decode(client, urData.data, { strict })
}

type BatchableFunctionObject = GeneratedFunction<typeof encode, typeof decode>

/**
 * Gets a text record for a name.
 * @param client - {@link ClientWithPns}
 * @param parameters - {@link GetTextRecordParameters}
 * @returns Text record string, or null if none is found. {@link GetTextRecordReturnType}
 *
 * @example
 * import { createPublicClient, http } from 'viem'
 * import { mainnet } from 'viem/chains'
 * import { addPnsContracts } from '@pnsdomains/pnsjs'
 * import { getTextRecord } from '@pnsdomains/pnsjs/public'
 *
 * const client = createPublicClient({
 *   chain: addPnsContracts(mainnet),
 *   transport: http(),
 * })
 * const result = await getTextRecord(client, { name: 'pns.pls', key: 'com.twitter' })
 * // pnsdomains
 */
const getTextRecord = generateFunction({ encode, decode }) as ((
  client: ClientWithPns,
  { name, key, strict, gatewayUrls }: GetTextRecordParameters,
) => Promise<GetTextRecordReturnType>) &
  BatchableFunctionObject

export default getTextRecord