import type { Address } from "viem"
import type { CAB_VERSION } from "../types"
import {
    CAB_V0_1,
    CAB_V0_2,
    CAB_V0_2_1,
    invoiceManagerAddressV1,
    invoiceManagerAddressV2,
    invoiceManagerAddressV2_1
} from "./constants"

export const getInvoiceManagerAddress = (cabVersion?: CAB_VERSION): Address => {
    if (cabVersion === CAB_V0_1) {
        return invoiceManagerAddressV1
    }
    if (cabVersion === CAB_V0_2) {
        return invoiceManagerAddressV2
    }
    if (cabVersion === CAB_V0_2_1) {
        return invoiceManagerAddressV2_1
    }
    return invoiceManagerAddressV2_1
}