import {
  type Account,
  type Client,
  type Hex,
  type SendTransactionRequest,
  type SignMessageParameters,
  type SignTypedDataParameters,
  type Transport,
  toHex,
} from "viem";
import type {
  ChainEIP712,
  SendEip712TransactionParameters,
  SignEip712TransactionParameters,
  SignEip712TransactionReturnType,
} from "viem/zksync";

import { replaceBigInts } from "../replaceBigInts.js";
import type { SendTransactionBatchParameters } from "./sendTransactionBatch.js";
import type { SignTransactionBatchParameters } from "./signTransactionBatch.js";

export async function sendPrivyTransaction<
  chain extends ChainEIP712 | undefined = ChainEIP712 | undefined,
  account extends Account | undefined = Account | undefined,
  chainOverride extends ChainEIP712 | undefined = ChainEIP712 | undefined,
  const request extends SendTransactionRequest<
    chain,
    chainOverride
  > = SendTransactionRequest<chain, chainOverride>,
  const calls extends readonly unknown[] = readonly unknown[],
>(
  client: Client<Transport, ChainEIP712, Account>,
  parameters:
    | SendEip712TransactionParameters<chain, account, chainOverride, request>
    | SendTransactionBatchParameters<calls>,
): Promise<SignEip712TransactionReturnType> {
  const result = (await client.request(
    {
      method: "privy_sendSmartWalletTx",
      params: [replaceBigInts(parameters, toHex)],
      // eslint-disable-next-line @typescript-eslint/no-explicit-any
    } as any,
    { retryCount: 0 },
  )) as SignEip712TransactionReturnType;
  return result;
}

export async function sendPrivySignMessage(
  client: Client<Transport, ChainEIP712, Account>,
  parameters: Omit<SignMessageParameters, "account">,
): Promise<Hex> {
  const result = (await client.request(
    {
      method: "privy_signSmartWalletMessage",
      params: [parameters.message],
      // eslint-disable-next-line @typescript-eslint/no-explicit-any
    } as any,
    { retryCount: 0 },
  )) as Hex;
  return result;
}

export async function sendPrivySignTypedData(
  client: Client<Transport, ChainEIP712, Account>,
  parameters: Omit<SignTypedDataParameters, "account" | "privateKey">,
): Promise<Hex> {
  const result = (await client.request(
    {
      method: "privy_signSmartWalletTypedData",
      params: [client.account.address, parameters],
      // eslint-disable-next-line @typescript-eslint/no-explicit-any
    } as any,
    { retryCount: 0 },
  )) as Hex;
  return result;
}

export async function signPrivyTransaction<
  chain extends ChainEIP712 | undefined = ChainEIP712 | undefined,
  account extends Account | undefined = Account | undefined,
  chainOverride extends ChainEIP712 | undefined = ChainEIP712 | undefined,
  const calls extends readonly unknown[] = readonly unknown[],
>(
  client: Client<Transport, ChainEIP712, Account>,
  parameters:
    | SignEip712TransactionParameters<chain, account, chainOverride>
    | SignTransactionBatchParameters<chain, account, chainOverride, calls>,
): Promise<SignEip712TransactionReturnType> {
  const { chain: _chain, account: _account, ...request } = parameters;

  const result = (await client.request(
    {
      method: "privy_signSmartWalletTx",
      params: [replaceBigInts(request, toHex)],
      // eslint-disable-next-line @typescript-eslint/no-explicit-any
    } as any,
    { retryCount: 0 },
  )) as SignEip712TransactionReturnType;
  return result;
}