import {
    EstimateUserOperationGasReturnType,
    GetSupportedEntryPointsReturnType,
    GetUserOperationReceiptReturnType,
    SendUserOperationReturnType,
    UserOperation,
} from "viem/account-abstraction";
import { Transport, Chain, Account, Client, Address, GetChainIdReturnType, Hash } from "viem";

import { getChainId } from "viem/actions";
import {
    estimateUserOperationGas,
    EstimateUserOperationGasParameters07,
} from "../../actions/bundler/estimateUserOperationGas.js";
import { getSupportedEntryPoints } from "../../actions/bundler/getSupportedEntryPoints.js";
import { sendUserOperation } from "../../actions/bundler/sendUserOperation.js";
import { getUserOperationReceipt } from "../../actions/index.js";

export type BackendBundlerActions = {
    getChainId: () => Promise<GetChainIdReturnType>;
    estimateUserOperationGas: (
        parameters: EstimateUserOperationGasParameters07,
    ) => Promise<EstimateUserOperationGasReturnType<undefined, undefined, undefined, "0.7">>;
    getSupportedEntryPoints: () => Promise<GetSupportedEntryPointsReturnType>;
    getUserOperationReceipt: (parameters: { hash: Hash }) => Promise<GetUserOperationReceiptReturnType>;
    sendUserOperation: (parameters: UserOperation<"0.7">) => Promise<SendUserOperationReturnType>;
};

export function backendBundlerActions<
    transport extends Transport = Transport,
    chain extends Chain | undefined = Chain | undefined,
    account extends Account = Account,
>(
    client: Client<transport, chain, account> & {
        entryPointSimulationsAddress: Address;
    },
): BackendBundlerActions {
    return {
        estimateUserOperationGas: (parameters) => estimateUserOperationGas(client, parameters),
        getChainId: () => getChainId(client),
        getSupportedEntryPoints: () => getSupportedEntryPoints(client),
        getUserOperationReceipt: (parameters) => getUserOperationReceipt(client, parameters),
        sendUserOperation: (parameters) => sendUserOperation(client, parameters),
    };
}