import type { MutateOptions, MutationOptions } from '@tanstack/query-core'
import type { Abi, ContractConstructorArgs } from 'viem'
import {
  type DeployContractErrorType,
  type DeployContractParameters,
  type DeployContractReturnType,
  deployContract,
} from '../actions/deployContract.js'
import type { Config } from '../createConfig.js'
import type { MutationParameter } from '../types/query.js'
import type { Compute } from '../types/utils.js'

export type DeployContractOptions<
  config extends Config,
  context = unknown,
> = MutationParameter<
  DeployContractData,
  DeployContractErrorType,
  DeployContractVariables<Abi, config, config['chains'][number]['id']>,
  context
>

export function deployContractMutationOptions<config extends Config, context>(
  config: config,
  options: DeployContractOptions<config, context> = {},
): DeployContractMutationOptions<config> {
  return {
    ...(options.mutation as any),
    mutationFn(variables) {
      return deployContract(config, variables)
    },
    mutationKey: ['deployContract'],
  }
}

export type DeployContractMutationOptions<config extends Config> =
  MutationOptions<
    DeployContractData,
    DeployContractErrorType,
    DeployContractVariables<Abi, config, config['chains'][number]['id']>
  >

export type DeployContractData = Compute<DeployContractReturnType>

export type DeployContractVariables<
  abi extends Abi | readonly unknown[],
  config extends Config,
  chainId extends config['chains'][number]['id'],
  ///
  allArgs = ContractConstructorArgs<abi>,
> = DeployContractParameters<abi, config, chainId, allArgs>

export type DeployContractMutate<config extends Config, context = unknown> = <
  abi extends Abi | readonly unknown[],
  chainId extends config['chains'][number]['id'],
>(
  variables: DeployContractVariables<abi, config, chainId>,
  options?:
    | Compute<
        MutateOptions<
          DeployContractData,
          DeployContractErrorType,
          Compute<DeployContractVariables<abi, config, chainId>>,
          context
        >
      >
    | undefined,
) => void

export type DeployContractMutateAsync<
  config extends Config,
  context = unknown,
> = <
  abi extends Abi | readonly unknown[],
  chainId extends config['chains'][number]['id'],
>(
  variables: DeployContractVariables<abi, config, chainId>,
  options?:
    | Compute<
        MutateOptions<
          DeployContractData,
          DeployContractErrorType,
          Compute<DeployContractVariables<abi, config, chainId>>,
          context
        >
      >
    | undefined,
) => Promise<DeployContractData>