permissionless/actions/smartAccount/getCallsStatus.ts

A utility library for working with ERC-4337

import type {
    Chain,
    Client,
    GetCallsStatusParameters,
    GetCallsStatusReturnType,
    Hex,
    Transport
} from "viem"
import {
    type SmartAccount,
    getUserOperationReceipt
} from "viem/account-abstraction"
import { getChainId } from "viem/actions"
import { getAction } from "viem/utils"

const getStatus = (_status: number) => {
    const [status, statusCode] = (() => {
        const statusCode = _status
        if (statusCode >= 100 && statusCode < 200)
            return ["pending", statusCode] as const
        if (statusCode >= 200 && statusCode < 300)
            return ["success", statusCode] as const
        if (statusCode >= 300 && statusCode < 700)
            return ["failure", statusCode] as const
        // @ts-expect-error: for backwards compatibility
        if (statusCode === "CONFIRMED") return ["success", 200] as const
        // @ts-expect-error: for backwards compatibility
        if (statusCode === "PENDING") return ["pending", 100] as const
        return [undefined, statusCode]
    })()

    return [status, statusCode] as const
}

export async function getCallsStatus<
    account extends SmartAccount | undefined,
    chain extends Chain | undefined
>(
    client: Client<Transport, chain, account>,
    args: GetCallsStatusParameters
): Promise<GetCallsStatusReturnType> {
    const userOperationHash = args.id as Hex

    const chainId =
        client.chain?.id ??
        client.account?.client.chain?.id ??
        getAction(client, getChainId, "getChainId")(client)

    try {
        const receipt = await getAction(
            client,
            getUserOperationReceipt,
            "getUserOperationReceipt"
        )({
            hash: args.id as Hex
        })

        const userOpStatus = receipt.success
        const [status, statusCode] = getStatus(userOpStatus ? 200 : 500)

        return {
            id: userOperationHash,
            version: "1.0",
            chainId: await chainId,
            status,
            statusCode,
            atomic: true,
            receipts: [
                {
                    status: receipt.receipt.status,
                    logs: receipt.receipt.logs,
                    blockHash: receipt.receipt.blockHash,
                    blockNumber: receipt.receipt.blockNumber,
                    gasUsed: receipt.receipt.gasUsed,
                    transactionHash: receipt.receipt.transactionHash
                }
            ]
        }
    } catch {
        const [status, statusCode] = getStatus(100)

        return {
            id: userOperationHash,
            version: "1.0",
            chainId: await chainId,
            atomic: true,
            status,
            statusCode
        }
    }
}