import type { SmartAccount } from "permissionless/accounts"
import type { EntryPoint } from "permissionless/types"
import type { Chain, Client, Transport } from "viem"
import {
    type PrepareUserOperationRequestCABParameters,
    type PrepareUserOperationRequestCABReturnType,
    type SendUserOperationCABParameters,
    type SendUserOperationCABReturnType,
    prepareUserOperationRequestCAB,
    sendUserOperationCAB
} from "../../actions/cab"
import type { CAB_VERSION } from "../../types"
import { getCabPaymasterAddress } from "../../utils"

export type CABActions<
    TEntryPoint extends EntryPoint,
    TTransport extends Transport,
    TChain extends Chain | undefined,
    TSmartAccount extends
        | SmartAccount<TEntryPoint, string, TTransport, TChain>
        | undefined =
        | SmartAccount<TEntryPoint, string, TTransport, TChain>
        | undefined
> = {
    prepareUserOperationRequestCAB: (
        args: PrepareUserOperationRequestCABParameters<
            TEntryPoint,
            TTransport,
            TChain,
            TSmartAccount
        >
    ) => Promise<PrepareUserOperationRequestCABReturnType<TEntryPoint>>
    sendUserOperationCAB: (
        args: SendUserOperationCABParameters<TEntryPoint>
    ) => Promise<SendUserOperationCABReturnType>
}

export const cabActions =
    <
        TEntryPoint extends EntryPoint,
        TTransport extends Transport,
        TChain extends Chain | undefined
    >(
        transport: Transport,
        cabVersion?: CAB_VERSION
    ) =>
    (
        client: Client<
            TTransport,
            TChain,
            SmartAccount<TEntryPoint, string, TTransport, TChain>
        >
    ): CABActions<TEntryPoint, TTransport, TChain> => ({
        prepareUserOperationRequestCAB: async (args) =>
            prepareUserOperationRequestCAB<
                TEntryPoint,
                TChain,
                TTransport,
                SmartAccount<TEntryPoint, string, TTransport, TChain>
            >(
                client,
                {
                    ...args,
                    cabPaymasterAddress: getCabPaymasterAddress(cabVersion)
                },
                transport
            ),
        sendUserOperationCAB: async (args) =>
            sendUserOperationCAB<
                TEntryPoint,
                TChain,
                TTransport,
                SmartAccount<TEntryPoint, string, TTransport, TChain>
            >(
                client,
                {
                    ...args,
                    cabPaymasterAddress: getCabPaymasterAddress(cabVersion)
                },
                transport
            )
    })