import type { Address } from "viem"
import type { CAB_VERSION } from "../types"
import {
    CAB_V0_1,
    CAB_V0_2,
    CAB_V0_2_1,
    cabPaymasterAddressV1,
    cabPaymasterAddressV2,
    cabPaymasterAddressV2_1
} from "./constants"

export const getCabPaymasterAddress = (cabVersion?: CAB_VERSION): Address => {
    if (cabVersion === CAB_V0_1) {
        return cabPaymasterAddressV1
    }
    if (cabVersion === CAB_V0_2) {
        return cabPaymasterAddressV2
    }
    if (cabVersion === CAB_V0_2_1) {
        return cabPaymasterAddressV2_1
    }
    return cabPaymasterAddressV2_1
}