mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2024-12-31 16:21:21 +00:00
75 lines
2.5 KiB
Go
75 lines
2.5 KiB
Go
package sentinelcli
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/ledgerwatch/erigon/cl/clparams"
|
|
"github.com/ledgerwatch/erigon/cmd/sentinel/sentinelflags"
|
|
"github.com/ledgerwatch/erigon/cmd/utils"
|
|
|
|
"github.com/ledgerwatch/erigon-lib/common"
|
|
|
|
"github.com/urfave/cli/v2"
|
|
|
|
"github.com/ledgerwatch/erigon/turbo/logging"
|
|
|
|
"github.com/ledgerwatch/log/v3"
|
|
)
|
|
|
|
type SentinelCliCfg struct {
|
|
GenesisCfg *clparams.GenesisConfig
|
|
BeaconCfg *clparams.BeaconChainConfig
|
|
NetworkCfg *clparams.NetworkConfig
|
|
NetworkType clparams.NetworkType
|
|
Port uint `json:"port"`
|
|
Addr string `json:"address"`
|
|
ServerAddr string `json:"server_addr"`
|
|
ServerProtocol string `json:"server_protocol"`
|
|
ServerTcpPort uint `json:"server_tcp_port"`
|
|
LogLvl uint `json:"log_level"`
|
|
NoDiscovery bool `json:"no_discovery"`
|
|
LocalDiscovery bool `json:"local_discovery"`
|
|
}
|
|
|
|
func SetupSentinelCli(ctx *cli.Context) (*SentinelCliCfg, error) {
|
|
cfg := &SentinelCliCfg{}
|
|
chainName := ctx.String(utils.ChainFlag.Name)
|
|
var err error
|
|
cfg.GenesisCfg, cfg.NetworkCfg, cfg.BeaconCfg, cfg.NetworkType, err = clparams.GetConfigsByNetworkName(chainName)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if ctx.String(sentinelflags.BeaconConfigFlag.Name) != "" {
|
|
cfg.BeaconCfg = new(clparams.BeaconChainConfig)
|
|
if *cfg.BeaconCfg, err = clparams.CustomConfig(ctx.String(sentinelflags.BeaconConfigFlag.Name)); err != nil {
|
|
return nil, err
|
|
}
|
|
if ctx.String(sentinelflags.GenesisSSZFlag.Name) == "" {
|
|
return nil, fmt.Errorf("no genesis file provided")
|
|
}
|
|
cfg.GenesisCfg = new(clparams.GenesisConfig)
|
|
|
|
}
|
|
cfg.ServerAddr = fmt.Sprintf("%s:%d", ctx.String(sentinelflags.SentinelServerAddr.Name), ctx.Int(sentinelflags.SentinelServerPort.Name))
|
|
cfg.ServerProtocol = "tcp"
|
|
|
|
cfg.Port = uint(ctx.Int(sentinelflags.SentinelDiscoveryPort.Name))
|
|
cfg.Addr = ctx.String(sentinelflags.SentinelDiscoveryAddr.Name)
|
|
|
|
cfg.LogLvl = ctx.Uint(logging.LogVerbosityFlag.Name)
|
|
if cfg.LogLvl == uint(log.LvlInfo) || cfg.LogLvl == 0 {
|
|
cfg.LogLvl = uint(log.LvlDebug)
|
|
}
|
|
cfg.NoDiscovery = ctx.Bool(sentinelflags.NoDiscovery.Name)
|
|
cfg.LocalDiscovery = ctx.Bool(sentinelflags.LocalDiscovery.Name)
|
|
|
|
// Process bootnodes
|
|
if ctx.String(sentinelflags.BootnodesFlag.Name) != "" {
|
|
cfg.NetworkCfg.BootNodes = common.CliString2Array(ctx.String(sentinelflags.BootnodesFlag.Name))
|
|
}
|
|
if ctx.String(sentinelflags.SentinelStaticPeersFlag.Name) != "" {
|
|
cfg.NetworkCfg.StaticPeers = common.CliString2Array(ctx.String(sentinelflags.SentinelStaticPeersFlag.Name))
|
|
}
|
|
return cfg, nil
|
|
}
|