import type { SmartAccount } from "permissionless/accounts"
import type {
    EntryPoint,
    GetEntryPointVersion,
    UserOperation
} from "permissionless/types"
import { deepHexlify } from "permissionless/utils"
import type { Address, Chain, Client, Transport } from "viem"
import type {
    CabPaymasterRpcScheme,
    RepayTokenInfo,
    SponsorTokenInfo,
    SwapTokenInfo,
    UserOperationWithBigIntAsHex
} from "../../types"
import { cabPaymasterAddress } from "../../utils"

export type GetCabPaymasterSponsorTokensParameters<
    TEntryPoint extends EntryPoint
> = {
    userOperation: UserOperation<GetEntryPointVersion<TEntryPoint>>
    entryPoint: TEntryPoint
    sponsorChain: Chain
    cabPaymasterAddress?: Address
}

export type GetCabPaymasterSponsorTokensReturnType = {
    paymaster: Address
    sponsorTokensInfo: SponsorTokenInfo[]
    swapTokensInfo: SwapTokenInfo[]
    repayTokensInfo: RepayTokenInfo[]
}

export async function getCabPaymasterSponsorTokens<
    TEntryPoint extends EntryPoint,
    TChain extends Chain | undefined,
    TTransport extends Transport = Transport,
    TSmartAccount extends
        | SmartAccount<TEntryPoint, string, TTransport, TChain>
        | undefined =
        | SmartAccount<TEntryPoint, string, TTransport, TChain>
        | undefined
>(
    client: Client<
        TTransport,
        TChain,
        TSmartAccount,
        CabPaymasterRpcScheme<TEntryPoint>
    >,
    {
        userOperation,
        entryPoint,
        sponsorChain,
        cabPaymasterAddress: cabPaymasterAddress_
    }: GetCabPaymasterSponsorTokensParameters<TEntryPoint>
): Promise<GetCabPaymasterSponsorTokensReturnType> {
    const paymasterAddress = cabPaymasterAddress_ ?? cabPaymasterAddress
    const params: [
        UserOperationWithBigIntAsHex<GetEntryPointVersion<TEntryPoint>>,
        TEntryPoint,
        number,
        Address
    ] = [
        deepHexlify(userOperation) as UserOperationWithBigIntAsHex<
            GetEntryPointVersion<TEntryPoint>
        >,
        entryPoint,
        sponsorChain.id,
        paymasterAddress
    ]
    const response = await client.request({
        method: "pm_getCabPaymasterSponsorTokens",
        params
    })
    return {
        paymaster: response.paymaster,
        sponsorTokensInfo: response.sponsorTokensInfo,
        swapTokensInfo: response.swapTokensInfo,
        repayTokensInfo: response.repayTokensInfo
    }
}