synapsecns/sanguine

package cctp

import (
    "github.com/ethereum/go-ethereum/accounts/abi/bind"
    "github.com/ethereum/go-ethereum/common"
    "github.com/ethereum/go-ethereum/core/vm"
)

// CCTPRef is a reference to a deployed CCTP contract.
//
//nolint:golint
type CCTPRef struct {
    *SynapseCCTP
    address common.Address
}

// MessageTransmitterRef is a reference to a deployed message transmitter contract.
//
//nolint:golint
type MessageTransmitterRef struct {
    *MessageTransmitter
    address common.Address
}

// Address is the contract address.
func (s CCTPRef) Address() common.Address {
    return s.address
}

// Address is the contract address.
func (s MessageTransmitterRef) Address() common.Address {
    return s.address
}

// NewCCTPRef creates a new CCTPRef instance.
//
//nolint:golint
func NewCCTPRef(address common.Address, backend bind.ContractBackend) (*CCTPRef, error) {
    synapseCCTP, err := NewSynapseCCTP(address, backend)
    if err != nil {
        return nil, err
    }
    return &CCTPRef{
        SynapseCCTP: synapseCCTP,
        address:     address,
    }, nil
}

// NewMessageTransmitterRef creates a new MessageTransmitterRef instance.
func NewMessageTransmitterRef(address common.Address, backend bind.ContractBackend) (*MessageTransmitterRef, error) {
    messageTransmitter, err := NewMessageTransmitter(address, backend)
    if err != nil {
        return nil, err
    }
    return &MessageTransmitterRef{
        MessageTransmitter: messageTransmitter,
        address:            address,
    }, nil
}

var _ vm.ContractRef = &CCTPRef{}
var _ vm.ContractRef = &MessageTransmitterRef{}