import {
  type Account,
  type Address,
  type Client,
  type ClientConfig,
  type ParseAccount,
  type Transport,
  type WalletActions,
  type WalletRpcSchema,
  createWalletClient,
} from 'viem'
import { addEnsContracts } from '../contracts/addEnsContracts.js'
import type {
  ChainWithBaseContracts,
  ChainWithEns,
  CheckedChainWithEns,
} from '../contracts/consts.js'
import type { Assign, Prettify } from '../types.js'
import { type EnsWalletActions, ensWalletActions } from './decorators/wallet.js'

export type EnsWalletClientConfig<
  TTransport extends Transport,
  TChain extends ChainWithBaseContracts,
  TAccountOrAddress extends Account | Address | undefined =
    | Account
    | Address
    | undefined,
> = Assign<
  Pick<
    ClientConfig<TTransport, TChain, TAccountOrAddress>,
    'account' | 'chain' | 'key' | 'name' | 'pollingInterval' | 'transport'
  >,
  {
    chain: TChain
  }
>

export type EnsWalletClient<
  TTransport extends Transport = Transport,
  TChain extends ChainWithEns = ChainWithEns,
  TAccount extends Account | undefined = Account | undefined,
> = Prettify<
  Client<
    TTransport,
    TChain,
    TAccount,
    WalletRpcSchema,
    WalletActions<TChain, TAccount> & EnsWalletActions<TChain, TAccount>
  >
>

/**
 * Creates an ENS Wallet Client with a given [Transport](https://viem.sh/docs/clients/intro.html) configured for a [Chain](https://viem.sh/docs/clients/chains.html).
 *
 * @param config - {@link EnsWalletClientConfig}
 * @returns An ENS Wallet Client. {@link EnsWalletClient}
 *
 * @example
 * import { custom } from 'viem'
 * import { mainnet } from 'viem/chains'
 * import { createEnsWalletClient } from '@etn-sc/ensjs'
 *
 * const client = createEnsWalletClient({
 *   chain: mainnet,
 *   transport: custom(window.ethereum),
 * })
 */
export const createEnsWalletClient = <
  TTransport extends Transport,
  TChain extends ChainWithBaseContracts,
  TAccountOrAddress extends Account | Address | undefined = undefined,
>({
  account,
  chain,
  key = 'ensWallet',
  name = 'ENS Wallet Client',
  transport,
  pollingInterval,
}: EnsWalletClientConfig<
  TTransport,
  TChain,
  TAccountOrAddress
>): EnsWalletClient<
  TTransport,
  CheckedChainWithEns<TChain>,
  ParseAccount<TAccountOrAddress>
> => {
  return createWalletClient({
    account,
    chain: addEnsContracts(chain),
    key,
    name,
    pollingInterval,
    transport,
  }).extend(ensWalletActions)
}