semaphore/packages/contracts/scripts/utils.ts

import { readFileSync, writeFileSync } from "fs"

export type NetworkDeployedContracts = {
    name: "Semaphore" | "SemaphoreVerifier" | "PoseidonT3"
    address: string
}[]

export type DeployedContracts = {
    network: string
    contracts: NetworkDeployedContracts
}[]

const supportedNetworks = ["sepolia", "arbitrum", "mumbai", "optimism-sepolia", "arbitrum-sepolia"]

export function getDeployedContracts(): DeployedContracts {
    return JSON.parse(readFileSync(`./deployed-contracts.json`, "utf8"))
}

export function getDeployedContractsByNetwork(network: string): NetworkDeployedContracts {
    const deployedContracts = getDeployedContracts()
    const networkDeployedContracts = deployedContracts.find((n) => n.network === network)

    if (!networkDeployedContracts) {
        throw Error(`Network '${network}' is not supported`)
    }

    return networkDeployedContracts.contracts
}

export function getDeployedContractAddress(network: string, contractName: string): string {
    const contracts = getDeployedContractsByNetwork(network)
    const semaphoreAddress = contracts.find((contract) => contract.name === contractName)

    if (!semaphoreAddress) {
        throw Error(`Contract with name '${contractName}' does not exist`)
    }

    return semaphoreAddress.address
}

export function saveDeployedContracts(contracts: NetworkDeployedContracts, network?: string) {
    if (network && supportedNetworks.includes(network)) {
        const deployedContracts = getDeployedContracts() as DeployedContracts

        for (let i = 0; i < deployedContracts.length; i += 1) {
            if (deployedContracts[i].network === network) {
                deployedContracts[i].contracts = contracts

                writeFileSync(`./deployed-contracts.json`, JSON.stringify(deployedContracts, null, 4))

                return
            }
        }

        deployedContracts.push({
            network,
            contracts
        })

        writeFileSync(`./deployed-contracts.json`, JSON.stringify(deployedContracts, null, 4))
    }
}