synapsecns/sanguine

package relayer

import "fmt"

// TODO: this file can be moved to somewhere common

// Mainnet chain IDs.
const ethereumChainID = 1
const avalancheChainID = 43114
const optimismChainID = 10
const arbitrumChainID = 42161
const baseChainID = 8453
const polygonChainID = 137

// Testnet chain IDs.
const sepoliaChainID = 11155111
const avalancheFujiChainID = 43113
const opSepoliaChainID = 11155420
const arbitrumSepoliaChainID = 421614
const baseSepoliaChainID = 84532
const polygonAmoyChainID = 80002

// see https://developers.circle.com/stablecoins/docs/evm-smart-contracts#mainnet-contract-addresses
var mainnetDomainMap = map[uint32]uint32{
    0: ethereumChainID,
    1: avalancheChainID,
    2: optimismChainID,
    3: arbitrumChainID,
    6: baseChainID,
    7: polygonChainID,
}

// see https://developers.circle.com/stablecoins/docs/evm-smart-contracts#testnet-contract-addresses
var testnetDomainMap = map[uint32]uint32{
    0: sepoliaChainID,
    1: avalancheFujiChainID,
    2: opSepoliaChainID,
    3: arbitrumSepoliaChainID,
    6: baseSepoliaChainID,
    7: polygonAmoyChainID,
}

var testnetChainIDMap map[uint32]uint32 // Maps chainID to domain for testnet
var mainnetChainIDMap map[uint32]uint32 // Maps chainID to domain for mainnet

// IsTestnetChainID returns true if the chain ID is a testnet chain ID.
func IsTestnetChainID(chainID uint32) bool {
    return chainID == sepoliaChainID || chainID == opSepoliaChainID || chainID == arbitrumSepoliaChainID
}

// CircleDomainToChainID converts a Circle domain to a chain ID.
func CircleDomainToChainID(domain uint32, isTestnet bool) (chainID uint32, err error) {
    if isTestnet {
        chainID, ok := testnetDomainMap[domain]
        if !ok {
            return 0, fmt.Errorf("domain %d not found in testnet domain map", domain)
        }
        return chainID, nil
    }

    chainID, ok := mainnetDomainMap[domain]
    if !ok {
        return 0, fmt.Errorf("domain %d not found in mainnet domain map", domain)
    }
    return chainID, nil
}

// ChainIDToCircleDomain converts a chain ID to a Circle domain.
func ChainIDToCircleDomain(chainID uint32, isTestnet bool) (domain uint32, err error) {
    if testnetChainIDMap == nil || mainnetChainIDMap == nil {
        initReverseMaps()
    }

    if isTestnet {
        domain, ok := testnetChainIDMap[chainID]
        if !ok {
            return 0, fmt.Errorf("chain ID %d not found in testnet chain ID map", chainID)
        }
        return domain, nil
    }

    domain, ok := mainnetChainIDMap[chainID]
    if !ok {
        return 0, fmt.Errorf("chain ID %d not found in mainnet chain ID map", chainID)
    }
    return domain, nil
}

func initReverseMaps() {
    testnetChainIDMap = make(map[uint32]uint32)
    mainnetChainIDMap = make(map[uint32]uint32)

    for domain, chainID := range testnetDomainMap {
        testnetChainIDMap[chainID] = domain
    }

    for domain, chainID := range mainnetDomainMap {
        mainnetChainIDMap[chainID] = domain
    }
}