0g-chain/tests/e2e/runner/chain.go
2024-08-02 12:59:25 +08:00

88 lines
2.4 KiB
Go

package runner
import (
"errors"
"fmt"
"github.com/0glabs/0g-chain/chaincfg"
rpchttpclient "github.com/cometbft/cometbft/rpc/client/http"
"github.com/ethereum/go-ethereum/ethclient"
)
var (
ErrChainAlreadyExists = errors.New("chain already exists")
)
// ChainDetails wraps information about the properties & endpoints of a chain.
type ChainDetails struct {
RpcUrl string
GrpcUrl string
EvmRpcUrl string
ChainId string
StakingDenom string
}
// EvmClient dials the underlying EVM RPC url and returns an ethclient.
func (c ChainDetails) EvmClient() (*ethclient.Client, error) {
return ethclient.Dial(c.EvmRpcUrl)
}
// RpcConn creates a new connection to the underlying Rpc url.
func (c ChainDetails) RpcConn() (*rpchttpclient.HTTP, error) {
return rpchttpclient.New(c.RpcUrl, "/websocket")
}
// Chains wraps a map of name -> details about how to connect to a chain.
// It prevents registering multiple chains with the same name & encapsulates
// panicking if attempting to access a chain that does not exist.
type Chains struct {
byName map[string]*ChainDetails
}
// NewChains creates an empty Chains map.
func NewChains() Chains {
return Chains{byName: make(map[string]*ChainDetails, 0)}
}
// MustGetChain returns the chain of a given name,
// or panics if a chain with that name has not been registered.
func (c Chains) MustGetChain(name string) *ChainDetails {
chain, found := c.byName[name]
if !found {
panic(fmt.Sprintf("no chain with name %s found", name))
}
return chain
}
// Register adds a chain to the map.
// It returns an error if a ChainDetails with that name has already been registered.
func (c *Chains) Register(name string, chain *ChainDetails) error {
if _, found := c.byName[name]; found {
return ErrChainAlreadyExists
}
c.byName[name] = chain
return nil
}
// the Chain details are all hardcoded because they are currently fixed by kvtool.
// someday they may be accepted as configurable parameters.
var (
kvtoolZgChainChain = ChainDetails{
RpcUrl: "http://localhost:26657",
GrpcUrl: "http://localhost:9090",
EvmRpcUrl: "http://localhost:8545",
ChainId: "0gchainlocalnet_8888-1",
StakingDenom: chaincfg.GasDenom,
}
kvtoolIbcChain = ChainDetails{
RpcUrl: "http://localhost:26658",
GrpcUrl: "http://localhost:9092",
EvmRpcUrl: "http://localhost:8547",
ChainId: "0gchainlocalnet_8889-2",
StakingDenom: "uatom",
}
)