import type { Prettify } from '../../../types/index.js'
import type { EncodeAbiParameters } from '../../resolver/encodeAbi.js'
import {
  type SetAbiParametersErrorType,
  type SetAbiParametersReturnType,
  setAbiParameters,
} from '../../resolver/setAbi.js'
import {
  type SetAddrParametersErrorType,
  type SetAddrParametersParameters,
  type SetAddrParametersReturnType,
  setAddrParameters,
} from '../../resolver/setAddr.js'
import {
  type SetContentHashParametersErrorType,
  type SetContentHashParametersReturnType,
  setContentHashParameters,
} from '../../resolver/setContentHash.js'
import {
  type SetTextParameters,
  type SetTextParametersErrorType,
  type SetTextParametersReturnType,
  setTextParameters,
} from '../../resolver/setText.js'
import {
  type ClearRecordsParametersErrorType,
  type ClearRecordsParametersReturnType,
  clearRecordsParameters,
} from './clearRecords.js'

export type RecordOptions = Prettify<{
  /** Clears all current records */
  clearRecords?: boolean
  /** ContentHash value */
  contentHash?: string | null
  /** Array of text records */
  texts?: Omit<SetTextParameters, 'name'>[]
  /** Array of coin records */
  coins?: Omit<SetAddrParametersParameters, 'name'>[]
  /** ABI value */
  abi?: EncodeAbiParameters | EncodeAbiParameters[]
}>

export type ResolverMulticallItem =
  | ClearRecordsParametersReturnType
  | SetContentHashParametersReturnType
  | SetAbiParametersReturnType
  | SetTextParametersReturnType
  | SetAddrParametersReturnType

// ================================
// Resolver multicall item
// ================================

export type ResolverMulticallParametersReturnType = ResolverMulticallItem[]

export type ResolverMulticallItemErrorType =
  | ClearRecordsParametersErrorType
  | SetContentHashParametersErrorType
  | SetAbiParametersErrorType
  | SetTextParametersErrorType
  | SetAddrParametersErrorType

/**
 * Generates multicall parameters for setting records.
 * All individual calls compute the namehash internally.
 */
export const resolverMulticallParameters = async ({
  name,
  clearRecords,
  contentHash,
  texts,
  coins,
  abi,
}: {
  name: string
} & RecordOptions): Promise<ResolverMulticallParametersReturnType> => {
  const calls: ResolverMulticallParametersReturnType = []

  if (clearRecords) {
    calls.push(clearRecordsParameters(name))
  }

  if (contentHash !== undefined) {
    const data = setContentHashParameters({ name, contentHash })
    if (data) calls.push(data)
  }

  if (abi !== undefined) {
    const abis = Array.isArray(abi) ? abi : [abi]
    const data = await Promise.all(
      abis.map(async (abiItem) => setAbiParameters({ name, ...abiItem })),
    )
    if (data) calls.push(...data)
  }

  if (texts && texts.length > 0) {
    const data = texts.map((textItem) =>
      setTextParameters({ name, ...textItem }),
    )
    if (data) calls.push(...data)
  }

  if (coins && coins.length > 0) {
    const data = coins.map((coinItem) =>
      setAddrParameters({ name, ...coinItem }),
    )
    if (data) calls.push(...data)
  }

  return calls
}