import type {
  Account,
  Address,
  Hash,
  SendTransactionParameters,
  Transport,
} from 'viem'
import { sendTransaction } from 'viem/actions'
import type { ChainWithPns, ClientWithAccount } from '../../contracts/consts.js'
import type {
  Prettify,
  SimpleTransactionRequest,
  WriteTransactionParameters,
} from '../../types.js'
import { encodeSetText } from '../../utils/encoders/encodeSetText.js'
import { namehash } from '../../utils/normalise.js'

export type SetTextRecordDataParameters = {
  /** The name to set a text record for */
  name: string
  /** The text record key to set */
  key: string
  /** The text record value to set */
  value: string | null
  /** The resolver address to use */
  resolverAddress: Address
}

export type SetTextRecordDataReturnType = SimpleTransactionRequest

export type SetTextRecordParameters<
  TChain extends ChainWithPns,
  TAccount extends Account | undefined,
  TChainOverride extends ChainWithPns | undefined,
> = Prettify<
  SetTextRecordDataParameters &
    WriteTransactionParameters<TChain, TAccount, TChainOverride>
>

export type SetTextRecordReturnType = Hash

export const makeFunctionData = <
  TChain extends ChainWithPns,
  TAccount extends Account | undefined,
>(
  _wallet: ClientWithAccount<Transport, TChain, TAccount>,
  { name, key, value, resolverAddress }: SetTextRecordDataParameters,
): SetTextRecordDataReturnType => {
  return {
    to: resolverAddress,
    data: encodeSetText({ namehash: namehash(name), key, value }),
  }
}

/**
 * Sets a text record for a name on a resolver.
 * @param wallet - {@link ClientWithAccount}
 * @param parameters - {@link SetTextRecordParameters}
 * @returns Transaction hash. {@link SetTextRecordReturnType}
 *
 * @example
 * import { createWalletClient, custom } from 'viem'
 * import { mainnet } from 'viem/chains'
 * import { addPnsContracts } from '@pnsdomains/pnsjs'
 * import { setTextRecord } from '@pnsdomains/pnsjs/wallet'
 *
 * const wallet = createWalletClient({
 *   chain: addPnsContracts(mainnet),
 *   transport: custom(window.ethereum),
 * })
 * const hash = await setTextRecord(wallet, {
 *   name: 'pns.pls',
 *   key: 'foo',
 *   value: 'bar',
 *   resolverAddress: '0x4976fb03C32e5B8cfe2b6cCB31c09Ba78EBaBa41',
 * })
 * // 0x...
 */
async function setTextRecord<
  TChain extends ChainWithPns,
  TAccount extends Account | undefined,
  TChainOverride extends ChainWithPns | undefined = ChainWithPns,
>(
  wallet: ClientWithAccount<Transport, TChain, TAccount>,
  {
    name,
    key,
    value,
    resolverAddress,
    ...txArgs
  }: SetTextRecordParameters<TChain, TAccount, TChainOverride>,
): Promise<SetTextRecordReturnType> {
  const data = makeFunctionData(wallet, {
    name,
    key,
    value,
    resolverAddress,
  })
  const writeArgs = {
    ...data,
    ...txArgs,
  } as SendTransactionParameters<TChain, TAccount, TChainOverride>
  return sendTransaction(wallet, writeArgs)
}

setTextRecord.makeFunctionData = makeFunctionData

export default setTextRecord